mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
Stephan BehnkeandGitHub
4ea9105f9c
## What changed? Extend genrpcserverinterceptors to support `ActivityID` and `OperationID` and `ChasmRunID` ## Why? Fix and improve logging. ## How did you test it? - [ ] built - [ ] run locally and tested manually - [x] covered by existing tests - [ ] added new unit test(s) - [ ] added new functional test(s)
42 lines
1.1 KiB
Cheetah
42 lines
1.1 KiB
Cheetah
{{- /*gotype: go.temporal.io/server/cmd/tools/genrpcserverinterceptors.grpcServerData*/ -}}
|
|
// Code generated by cmd/tools/genrpcserverinterceptors. DO NOT EDIT.
|
|
|
|
package logtags
|
|
|
|
import (
|
|
{{- range .Imports}}
|
|
{{printf "%q" .}}
|
|
{{- end}}
|
|
"go.temporal.io/server/common/log/tag"
|
|
)
|
|
|
|
func (wt *WorkflowTags) extractFrom{{.Server}}Message(message any) []tag.Tag {
|
|
switch r := message.(type) {
|
|
{{- range .Messages}}
|
|
case {{.Type}}:
|
|
{{- if or .TaskTokenGetter .WorkflowIDGetter .RunIDGetter .ActivityIDGetter .OperationIDGetter .ChasmRunIDGetter}}
|
|
{{- if .TaskTokenGetter}}
|
|
return wt.fromTaskToken(r.{{ .TaskTokenGetter}})
|
|
{{- else}}
|
|
return []tag.Tag{
|
|
{{if .WorkflowIDGetter}} tag.WorkflowID(r.{{.WorkflowIDGetter}}),
|
|
{{end -}}
|
|
{{if .ActivityIDGetter}} tag.ActivityID(r.{{.ActivityIDGetter}}),
|
|
{{end -}}
|
|
{{if .OperationIDGetter}} tag.OperationID(r.{{.OperationIDGetter}}),
|
|
{{end -}}
|
|
{{if .RunIDGetter}} tag.WorkflowRunID(r.{{.RunIDGetter}}),
|
|
{{end -}}
|
|
{{if .ChasmRunIDGetter}} tag.ChasmRunID(r.{{.ChasmRunIDGetter}}),
|
|
{{end -}}
|
|
}
|
|
{{- end}}
|
|
{{- else}}
|
|
return nil
|
|
{{- end -}}
|
|
{{- end}}
|
|
default:
|
|
return nil
|
|
}
|
|
}
|