From 214166412d998ef36621d9e9a23f8a8c6b573bf2 Mon Sep 17 00:00:00 2001 From: Johnny Luo Date: Wed, 4 Jun 2025 16:51:04 +1000 Subject: [PATCH] feat: add database configuration and initialize scheduler service --- cmd/payroll/worker/config.go | 3 +++ cmd/payroll/worker/main.go | 12 ++++++++++++ 2 files changed, 15 insertions(+) diff --git a/cmd/payroll/worker/config.go b/cmd/payroll/worker/config.go index 95add63..b2e2d3d 100644 --- a/cmd/payroll/worker/config.go +++ b/cmd/payroll/worker/config.go @@ -18,6 +18,9 @@ type PayrollWorkerConfig struct { Host string `mapstructure:"host" json:"host,omitempty"` Port string `mapstructure:"port" json:"port,omitempty"` } `mapstructure:"datadog" json:"datadog"` + Database struct { + DSN string `mapstructure:"dsn" json:"dsn,omitempty"` + } `mapstructure:"database" json:"database,omitempty"` } func GetConfigure() (*PayrollWorkerConfig, error) { diff --git a/cmd/payroll/worker/main.go b/cmd/payroll/worker/main.go index bcd6a53..61eed6a 100644 --- a/cmd/payroll/worker/main.go +++ b/cmd/payroll/worker/main.go @@ -8,7 +8,9 @@ import ( "github.com/sirupsen/logrus" "github.com/vultisig/verifier/vault" + "github.com/vultisig/plugin/internal/scheduler" "github.com/vultisig/plugin/internal/tasks" + "github.com/vultisig/plugin/storage/postgres" ) func main() { @@ -50,6 +52,16 @@ func main() { if err != nil { panic(fmt.Errorf("failed to create vault service: %w", err)) } + postgressDB, err := postgres.NewPostgresBackend(cfg.Database.DSN, nil) + if err != nil { + panic(fmt.Errorf("failed to create postgres backend: %w", err)) + } + schedulerSvc, err := scheduler.NewSchedulerService(postgressDB, client, redisOptions) + if err != nil { + panic(fmt.Errorf("failed to create scheduler service: %w", err)) + } + schedulerSvc.Start() + defer schedulerSvc.Stop() mux := asynq.NewServeMux() // mux.HandleFunc(tasks.TypePluginTransaction, vaultService.HandlePluginTransaction) mux.HandleFunc(tasks.TypeKeySignDKLS, vaultService.HandleKeySignDKLS)