fix bug

2025-10-05 20:29:32 +08:00
parent 45079048af
commit ac88ca13eb
4 changed files with 67 additions and 63 deletions
--- a/main.go
+++ b/main.go
@@ -94,7 +94,8 @@ func generateNewServerFile(services []string) error {
 	// register grpc gw
 	var gw []string
 	for _, service := range services {
-		gw = append(gw, "pb.Register"+service+"HandlerFromEndpoint(srv.Ctx, srv.Mux, addr, opts)")
+		handler := strings.ReplaceAll(tpl.Handler, "{service}", service)
+		gw = append(gw, handler)
 	}
 	code = strings.ReplaceAll(code, "{gw}", strings.Join(gw, "\n"))