fix: provider generation issues

2025-01-09 11:05:03 +08:00
parent 1c5b908269
commit bd6d6bc6e8
4 changed files with 37 additions and 13 deletions
--- a/pkg/ast/provider/provider.go.tpl
+++ b/pkg/ast/provider/provider.go.tpl
@@ -31,7 +31,7 @@ func Provide(opts ...opt.Option) error {
 		{{- end }}

 		{{- if eq .Mode "grpc"}}
-		userv1.RegisterUserServiceServer(__grpc.Server, obj)
+		{{ .GrpcRegisterFunc }}(__grpc.Server, obj)
 		{{- end }}

 		{{- if eq .Mode "event"}}