jsrouter: parallelize startup of VMs

Author: Snawoot

dialer/jsrouter.go

@@ -8,6 +8,7 @@ import (
 	"os"
 
 	"github.com/dop251/goja"
+	"golang.org/x/sync/errgroup"
 
 	"github.com/SenseUnit/dumbproxy/dialer/dto"
 	"github.com/SenseUnit/dumbproxy/jsext"
@@ -30,39 +31,48 @@ func NewJSRouter(filename string, instances int, factory func(string) (Dialer, e
 
 	instances = max(1, instances)
 	pool := make(chan JSRouterFunc, instances)
+	initGroup, _ := errgroup.WithContext(context.Background())
 
 	for i := 0; i < instances; i++ {
-		vm := goja.New()
-		err := jsext.AddPrinter(vm, logger)
-		if err != nil {
-			return nil, fmt.Errorf("can't add print function to runtime: %w", err)
-		}
-		err = jsext.ConfigureRuntime(vm)
-		if err != nil {
-			return nil, fmt.Errorf("can't configure runtime runtime: %w", err)
-		}
-		vm.SetFieldNameMapper(goja.TagFieldNameMapper("json", true))
-		_, err = vm.RunString(string(script))
-		if err != nil {
-			return nil, fmt.Errorf("script run failed: %w", err)
-		}
-
-		var f JSRouterFunc
-		var routerFnJSVal goja.Value
-		if ex := vm.Try(func() {
-			routerFnJSVal = vm.Get("getProxy")
-		}); ex != nil {
-			return nil, fmt.Errorf("\"getProxy\" function cannot be located in VM context: %w", err)
-		}
-		if routerFnJSVal == nil {
-			return nil, errors.New("\"getProxy\" function is not defined")
-		}
-		err = vm.ExportTo(routerFnJSVal, &f)
-		if err != nil {
-			return nil, fmt.Errorf("can't export \"getProxy\" function from JS VM: %w", err)
-		}
-
-		pool <- f
+		initGroup.Go(func() error {
+			vm := goja.New()
+			err := jsext.AddPrinter(vm, logger)
+			if err != nil {
+				return fmt.Errorf("can't add print function to runtime: %w", err)
+			}
+			err = jsext.ConfigureRuntime(vm)
+			if err != nil {
+				return fmt.Errorf("can't configure runtime runtime: %w", err)
+			}
+			vm.SetFieldNameMapper(goja.TagFieldNameMapper("json", true))
+			_, err = vm.RunString(string(script))
+			if err != nil {
+				return fmt.Errorf("script run failed: %w", err)
+			}
+
+			var f JSRouterFunc
+			var routerFnJSVal goja.Value
+			if ex := vm.Try(func() {
+				routerFnJSVal = vm.Get("getProxy")
+			}); ex != nil {
+				return fmt.Errorf("\"getProxy\" function cannot be located in VM context: %w", err)
+			}
+			if routerFnJSVal == nil {
+				return errors.New("\"getProxy\" function is not defined")
+			}
+			err = vm.ExportTo(routerFnJSVal, &f)
+			if err != nil {
+				return fmt.Errorf("can't export \"getProxy\" function from JS VM: %w", err)
+			}
+
+			pool <- f
+			return nil
+		})
+	}
+
+	err = initGroup.Wait()
+	if err != nil {
+		return nil, err
 	}
 
 	return &JSRouter{