forgot handle func and change registerjob channel

lyft · May 3, 2024 · fcd0304 · fcd0304
1 parent b59d996
commit fcd0304
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/server/neptune/temporalworker/job/noop_stream_handler.go b/server/neptune/temporalworker/job/noop_stream_handler.go
@@ -30,6 +30,7 @@ func (n *NoopStreamHandler) CloseJob(ctx context.Context, jobID string) error {
 }
 
 func (n *NoopStreamHandler) CleanUp(ctx context.Context) error {
+	n.wg.Wait()
 	return nil
 }