rogee/atomctl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add grpc

Browse Source
This commit is contained in:
Rogee
2024-12-31 17:57:06 +08:00
parent daf39f7255
commit 64c08249df
8 changed files with 221 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/ast/provider/provider.go.tpl
View File

@@ -19,7 +19,7 @@ func Provide(opts ...opt.Option) error {
) ({{.ReturnType}}, error) {
obj := &{{.StructName}}{
{{- range $key, $param := .InjectParams }}
{{- if and (ne $key "__job") (ne $key "__event")}}
{{- if and (ne $key "__job") (ne $key "__event") (ne $key "__grpc")}}
{{$key}}: {{$key}},
{{- end}}
{{- end }}
@@ -30,6 +30,10 @@ func Provide(opts ...opt.Option) error {
}
{{- end }}
{{- if eq .Mode "grpc"}}
userv1.RegisterUserServiceServer(__grpc.Server, obj)
{{- end }}
{{- if eq .Mode "event"}}
__event.Handle("handler:{{.StructName}}", obj.Topic(), obj.PublishToTopic(), obj.Handler)
{{- end }}
@@ -39,6 +43,7 @@ func Provide(opts ...opt.Option) error {
return nil, err
}
{{- end }}
return obj, nil
}{{if .ProviderGroup}}, {{.ProviderGroup}}{{end}}); err != nil {
return err