fix generator

Raezil · Feb 26, 2025 · ae471d9 · ae471d9
1 parent d9bdf10
commit ae471d9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cmd/protoc-gen-rpc-impl.go b/cmd/protoc-gen-rpc-impl.go
@@ -34,6 +34,8 @@ func generateFile(gen *protogen.Plugin, file *protogen.File) {
 	// Write imports
 	g.P("import (")
 	g.P(`    "context"`)
+	g.P(`    "db"`)
+	g.P(`    "go.uber.org/zap"`)
 	g.P(")")
 	g.P()
 
@@ -49,6 +51,8 @@ func generateServiceImplementation(g *protogen.GeneratedFile, service *protogen.
 	// Generate struct
 	g.P("type ", structName, " struct {")
 	g.P("    Unimplemented", service.GoName, "Server")
+	g.P("	 PrismaClient *db.PrismaClient")
+	g.P("	 Logger       *zap.SugaredLogger")
 	g.P("}")
 	g.P()