Merge branch '309-reload-connection-options' into 'master'

fix: reload dump/restore connection options (#309)

See merge request postgres-ai/database-lab!401

Author: agneum

Diff for: internal/retrieval/retrieval.go

@@ -80,16 +80,7 @@ func New(cfg *dblabCfg.Config, engineProps global.EngineProps, docker *client.Cl
 		},
 	}
 
-	for _, jobName := range r.cfg.Jobs {
-		jobSpec, ok := r.cfg.JobsSpec[jobName]
-		if !ok {
-			continue
-		}
-
-		jobSpec.Name = jobName
-		r.jobSpecs[jobName] = jobSpec
-	}
-
+	r.formatJobsSpec()
 	r.defineRetrievalMode()
 
 	return r
@@ -99,6 +90,8 @@ func New(cfg *dblabCfg.Config, engineProps global.EngineProps, docker *client.Cl
 func (r *Retrieval) Reload(ctx context.Context, cfg *dblabCfg.Config) {
 	*r.cfg = cfg.Retrieval
 
+	r.formatJobsSpec()
+
 	for _, job := range r.jobs {
 		cfg, ok := r.cfg.JobsSpec[job.Name()]
 		if !ok {
@@ -111,9 +104,22 @@ func (r *Retrieval) Reload(ctx context.Context, cfg *dblabCfg.Config) {
 		}
 	}
 
+	r.stopScheduler()
 	r.setupScheduler(ctx)
 }
 
+func (r *Retrieval) formatJobsSpec() {
+	for _, jobName := range r.cfg.Jobs {
+		jobSpec, ok := r.cfg.JobsSpec[jobName]
+		if !ok {
+			continue
+		}
+
+		jobSpec.Name = jobName
+		r.jobSpecs[jobName] = jobSpec
+	}
+}
+
 // Run start retrieving process.
 func (r *Retrieval) Run(ctx context.Context) error {
 	runCtx, cancel := context.WithCancel(ctx)