cleanup

PeerDB-io · Jun 23, 2024 · 7030665 · 7030665
1 parent 59de47b
commit 7030665
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flow/cmd/worker.go b/flow/cmd/worker.go
@@ -118,7 +118,6 @@ func WorkerSetup(opts *WorkerSetupOptions) (*workerSetupResponse, error) {
 		clientOptions.Logger.Error("Failed to setup datadog profiling", "error", err)
 		return nil, err
 	}
-	defer shutDownDDProfiler()
 
 	if opts.TemporalCert != "" && opts.TemporalKey != "" {
 		slog.Info("Using temporal certificate/key for authentication")
@@ -195,6 +194,7 @@ func WorkerSetup(opts *WorkerSetupOptions) (*workerSetupResponse, error) {
 		Client: c,
 		Worker: w,
 		Cleanup: func() {
+			shutDownDDProfiler()
 			cleanupOtelManagerFunc()
 			c.Close()
 		},