mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
69 lines
2.1 KiB
Go
69 lines
2.1 KiB
Go
package addsearchattributes
|
|
|
|
import (
|
|
"context"
|
|
|
|
sdkworker "go.temporal.io/sdk/worker"
|
|
"go.temporal.io/sdk/workflow"
|
|
"github.com/hanzoai/tasks/common/headers"
|
|
"github.com/hanzoai/tasks/common/log"
|
|
"github.com/hanzoai/tasks/common/metrics"
|
|
esclient "github.com/hanzoai/tasks/common/persistence/visibility/store/elasticsearch/client"
|
|
"github.com/hanzoai/tasks/common/primitives"
|
|
"github.com/hanzoai/tasks/common/searchattribute"
|
|
workercommon "github.com/hanzoai/tasks/service/worker/common"
|
|
"go.uber.org/fx"
|
|
)
|
|
|
|
type (
|
|
// addSearchAttributes represent background work needed for adding search attributes
|
|
addSearchAttributes struct {
|
|
initParams
|
|
}
|
|
|
|
initParams struct {
|
|
fx.In
|
|
EsClient esclient.Client
|
|
Manager searchattribute.Manager
|
|
MetricsHandler metrics.Handler
|
|
Logger log.Logger
|
|
}
|
|
)
|
|
|
|
var Module = workercommon.AnnotateWorkerComponentProvider(newComponent)
|
|
|
|
func newComponent(params initParams) workercommon.WorkerComponent {
|
|
return &addSearchAttributes{initParams: params}
|
|
}
|
|
|
|
func (wc *addSearchAttributes) RegisterWorkflow(registry sdkworker.Registry) {
|
|
registry.RegisterWorkflowWithOptions(AddSearchAttributesWorkflow, workflow.RegisterOptions{Name: WorkflowName})
|
|
}
|
|
|
|
func (wc *addSearchAttributes) DedicatedWorkflowWorkerOptions() *workercommon.DedicatedWorkerOptions {
|
|
// use default worker
|
|
return nil
|
|
}
|
|
|
|
func (wc *addSearchAttributes) RegisterActivities(registry sdkworker.Registry) {
|
|
registry.RegisterActivity(wc.activities())
|
|
}
|
|
|
|
func (wc *addSearchAttributes) DedicatedActivityWorkerOptions() *workercommon.DedicatedWorkerOptions {
|
|
return &workercommon.DedicatedWorkerOptions{
|
|
TaskQueue: primitives.AddSearchAttributesActivityTQ,
|
|
Options: sdkworker.Options{
|
|
BackgroundActivityContext: headers.SetCallerType(context.Background(), headers.CallerTypeAPI),
|
|
},
|
|
}
|
|
}
|
|
|
|
func (wc *addSearchAttributes) activities() *activities {
|
|
return &activities{
|
|
esClient: wc.EsClient,
|
|
saManager: wc.Manager,
|
|
metricsHandler: wc.MetricsHandler.WithTags(metrics.OperationTag(metrics.AddSearchAttributesWorkflowScope)),
|
|
logger: wc.Logger,
|
|
}
|
|
}
|