Merge pull request #125 from thorpelawrence-yoti/wazero-runtime-config

add option to set wazero RuntimeConfig

webassembly/webassembly.go

@@ -38,15 +38,16 @@ type worker struct {
 }
 
 type Config struct {
-	MinIdle      int
-	MaxIdle      int
-	MaxTotal     int
-	WASM         []byte
-	FSConfig     wazero.FSConfig
-	Stdout       io.Writer
-	Stderr       io.Writer
-	RandomSource io.Reader
-	ReuseWorkers bool // WebAssembly runtime by default doesn't use workers because creating new instances is cheap.
+	MinIdle       int
+	MaxIdle       int
+	MaxTotal      int
+	WASM          []byte
+	FSConfig      wazero.FSConfig
+	RuntimeConfig wazero.RuntimeConfig
+	Stdout        io.Writer
+	Stderr        io.Writer
+	RandomSource  io.Reader
+	ReuseWorkers  bool // WebAssembly runtime by default doesn't use workers because creating new instances is cheap.
 }
 
 type pdfiumPool struct {
@@ -106,9 +107,14 @@ func Init(config Config) (pdfium.Pool, error) {
 		config.RandomSource = rand.Reader
 	}
 
+	if config.RuntimeConfig == nil {
+		config.RuntimeConfig = wazero.NewRuntimeConfig()
+	}
+
 	poolContext := context.WithValue(context.Background(), experimental.FunctionListenerFactoryKey{}, logging.NewLoggingListenerFactory(os.Stdout))
 	poolContext = context.Background()
-	runtime := wazero.NewRuntimeWithConfig(poolContext, wazero.NewRuntimeConfig())
+
+	runtime := wazero.NewRuntimeWithConfig(poolContext, config.RuntimeConfig)
 
 	// Import WASI features.
 	if _, err := wasi_snapshot_preview1.Instantiate(poolContext, runtime); err != nil {