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

feat: add event provider

Browse Source
This commit is contained in:
Rogee
2024-12-31 15:39:33 +08:00
parent 9e377c917c
commit 81863068cc
13 changed files with 456 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
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 ne $key "__job"}}
{{- if and (ne $key "__job") (ne $key "__event")}}
{{$key}}: {{$key}},
{{- end}}
{{- end }}
@@ -30,6 +30,10 @@ func Provide(opts ...opt.Option) error {
}
{{- end }}
{{- if eq .Mode "event"}}
__event.Handle("handler:{{.StructName}}", obj.Topic(), obj.PublishToTopic(), obj.Handler)
{{- end }}
{{- if eq .Mode "job"}}
if err := river.AddWorkerSafely(__job.Workers, obj); err != nil {
return nil, err