Close queue before waiting for in-progress tasks

Author: cschleiden

internal/worker/activity.go

@@ -58,6 +58,8 @@ func (aw *activityWorker) Start(ctx context.Context) error {
 }
 
 func (aw *activityWorker) WaitForCompletion() error {
+	close(aw.activityTaskQueue)
+
 	aw.wg.Wait()
 
 	return nil

internal/worker/workflow.go

@@ -72,6 +72,8 @@ func (ww *workflowWorker) Start(ctx context.Context) error {
 }
 
 func (ww *workflowWorker) WaitForCompletion() error {
+	close(ww.workflowTaskQueue)
+
 	ww.wg.Wait()
 
 	return nil