MUhahahaha

kyleconroy · kyleconroy · commit 400a305f59db · 2022-02-04T23:17:21.000-08:00
diff --git a/internal/cmd/generate.go b/internal/cmd/generate.go
@@ -202,7 +202,7 @@ func Generate(ctx context.Context, e Env, dir, filename string, stderr io.Writer
 			out = combo.Python.Out
 			req := codeGenRequest(result, combo)
 			if os.Getenv("WASM") != "" {
-				resp, err = pythonGenerate(req)
+				resp, err = pythonGenerate(ctx, req)
 			} else {
 				blob, err := req.MarshalVT()
 				if err != nil {
diff --git a/internal/cmd/sqlc-codegen-python.module b/internal/cmd/sqlc-codegen-python.module
diff --git a/internal/cmd/wasm.go b/internal/cmd/wasm.go
@@ -17,14 +17,43 @@ import (
 //go:embed sqlc-codegen-python.wasm
 var pythonCodeGen []byte
 
-func pythonGenerate(req *plugin.CodeGenRequest) (*plugin.CodeGenResponse, error) {
+//go:embed sqlc-codegen-python.module
+var pythonModule []byte
+
+var engine *wasmtime.Engine
+var linker *wasmtime.Linker
+var module *wasmtime.Module
+
+func init() {
+	var err error
+	engine = wasmtime.NewEngine()
+	module, err = wasmtime.NewModuleDeserialize(engine, pythonModule)
+	if err != nil {
+		panic(err)
+	}
+
+	// out, err := module.Serialize()
+	// if err != nil {
+	// 	panic(err)
+	// }
+
+	// err = os.WriteFile("sqlc-codegen-python.module", out, 0644)
+	// if err != nil {
+	// 	panic(err)
+	// }
+
+	linker = wasmtime.NewLinker(engine)
+	if err := linker.DefineWasi(); err != nil {
+		panic(err)
+	}
+}
+
+func pythonGenerate(cctx context.Context, req *plugin.CodeGenRequest) (*plugin.CodeGenResponse, error) {
 	stdinBlob, err := req.MarshalVT()
 	if err != nil {
 		return nil, err
 	}
 
-	cctx := context.Background()
-
 	ctx, task := trace.NewTask(cctx, "pythonGenerate")
 	defer task.End()
 
@@ -42,14 +71,6 @@ func pythonGenerate(req *plugin.CodeGenRequest) (*plugin.CodeGenResponse, error)
 		return nil, fmt.Errorf("write file: %w", err)
 	}
 
-	engine := wasmtime.NewEngine()
-	linker := wasmtime.NewLinker(engine)
-
-	// Link WASI
-	if err := linker.DefineWasi(); err != nil {
-		return nil, fmt.Errorf("define wasi: %w", err)
-	}
-
 	// Configure WASI imports to write stdout into a file.
 	wasiConfig := wasmtime.NewWasiConfig()
 	wasiConfig.SetStdinFile(stdinPath)
@@ -75,12 +96,12 @@ func pythonGenerate(req *plugin.CodeGenRequest) (*plugin.CodeGenResponse, error)
 	// 	return fmt.Errorf("read file: %w", err)
 	// }
 
-	moduRegion := trace.StartRegion(ctx, "wasmtime.NewModule")
-	module, err := wasmtime.NewModule(store.Engine, pythonCodeGen)
-	moduRegion.End()
-	if err != nil {
-		return nil, fmt.Errorf("define wasi: %w", err)
-	}
+	// moduRegion := trace.StartRegion(ctx, "wasmtime.NewModule")
+	// module, err := wasmtime.NewModule(store.Engine, pythonCodeGen)
+	// moduRegion.End()
+	// if err != nil {
+	// 	return nil, fmt.Errorf("define wasi: %w", err)
+	// }
 
 	linkRegion := trace.StartRegion(ctx, "linker.Instantiate")
 	instance, err := linker.Instantiate(store, module)
