zdravko/pkg/server/worker.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()
}