mirror of
https://github.com/mentos1386/zdravko.git
synced 2024-11-22 15:53:45 +00:00
47 lines
1.4 KiB
Go
47 lines
1.4 KiB
Go
package server
|
|
|
|
import (
|
|
"log/slog"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/mentos1386/zdravko/database"
|
|
"github.com/mentos1386/zdravko/internal/config"
|
|
"github.com/mentos1386/zdravko/internal/server/activities"
|
|
"github.com/mentos1386/zdravko/internal/server/workflows"
|
|
"github.com/mentos1386/zdravko/internal/temporal"
|
|
"go.temporal.io/sdk/activity"
|
|
"go.temporal.io/sdk/client"
|
|
temporalWorker "go.temporal.io/sdk/worker"
|
|
"go.temporal.io/sdk/workflow"
|
|
)
|
|
|
|
type Worker struct {
|
|
worker temporalWorker.Worker
|
|
}
|
|
|
|
func NewWorker(temporalClient client.Client, cfg *config.ServerConfig, logger *slog.Logger, db *sqlx.DB, kvStore database.KeyValueStore) *Worker {
|
|
worker := temporalWorker.New(temporalClient, temporal.TEMPORAL_SERVER_QUEUE, temporalWorker.Options{})
|
|
|
|
a := activities.NewActivities(cfg, logger, db, kvStore)
|
|
|
|
w := workflows.NewWorkflows()
|
|
|
|
// Register Workflows
|
|
worker.RegisterWorkflowWithOptions(w.CheckWorkflowDefinition, workflow.RegisterOptions{Name: temporal.WorkflowCheckName})
|
|
|
|
// Register Activities
|
|
worker.RegisterActivityWithOptions(a.TargetsFilter, activity.RegisterOptions{Name: temporal.ActivityTargetsFilterName})
|
|
worker.RegisterActivityWithOptions(a.AddTargetHistory, activity.RegisterOptions{Name: temporal.ActivityAddTargetHistoryName})
|
|
|
|
return &Worker{
|
|
worker: worker,
|
|
}
|
|
}
|
|
|
|
func (w *Worker) Start() error {
|
|
return w.worker.Run(temporalWorker.InterruptCh())
|
|
}
|
|
|
|
func (w *Worker) Stop() {
|
|
w.worker.Stop()
|
|
}
|