feat: configurable sync worker job timeout

goto · Jan 5, 2024 · b194794 · b194794
1 parent 4cbcaf2
commit b194794
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/compass.yaml.example b/compass.yaml.example
@@ -65,6 +65,7 @@ worker:
         username: compass
         password: compass_password
     job_manager_port: 8085
+    sync_job_timeout: 15m
 
 client:
     host: localhost:8081

diff --git a/internal/workermanager/discovery_worker.go b/internal/workermanager/discovery_worker.go
@@ -52,7 +52,7 @@ func (m *Manager) syncAssetHandler() worker.JobHandler {
 		Handle: m.SyncAssets,
 		JobOpts: worker.JobOptions{
 			MaxAttempts:     1,
-			Timeout:         5 * time.Minute,
+			Timeout:         m.syncTimeout,
 			BackoffStrategy: worker.DefaultExponentialBackoff,
 		},
 	}

diff --git a/internal/workermanager/worker_manager.go b/internal/workermanager/worker_manager.go
@@ -26,6 +26,7 @@ type Manager struct {
 	discoveryRepo  DiscoveryRepository
 	assetRepo      asset.Repository
 	logger         log.Logger
+	syncTimeout    time.Duration
 }
 
 //go:generate mockery --name=Worker -r --case underscore --with-expecter --structname Worker --filename worker_mock.go --output=./mocks
@@ -44,6 +45,7 @@ type Config struct {
 	ActivePollPercent float64       `mapstructure:"active_poll_percent" default:"20"`
 	PGQ               pgq.Config    `mapstructure:"pgq"`
 	JobManagerPort    int           `mapstructure:"job_manager_port"`
+	SyncJobTimeout    time.Duration `mapstructure:"sync_job_timeout" default:"15m"`
 }
 
 type Deps struct {
@@ -77,6 +79,7 @@ func New(ctx context.Context, deps Deps) (*Manager, error) {
 		discoveryRepo:  deps.DiscoveryRepo,
 		assetRepo:      deps.AssetRepo,
 		logger:         deps.Logger,
+		syncTimeout:    cfg.SyncJobTimeout,
 	}, nil
 }