fix: do done before wait

PeerDB-io · Dec 8, 2023 · c1ee5b7 · c1ee5b7
1 parent 6834d31
commit c1ee5b7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/flow/activities/flowable.go b/flow/activities/flowable.go
@@ -338,7 +338,7 @@ func (a *FlowableActivity) StartFlow(ctx context.Context,
 		log.Warnf("failed to push records: %v", err)
 		return nil, fmt.Errorf("failed to push records: %w", err)
 	}
-
+	done <- struct{}{}
 	err = errGroup.Wait()
 	if err != nil {
 		return nil, fmt.Errorf("failed to pull records: %w", err)
@@ -386,7 +386,6 @@ func (a *FlowableActivity) StartFlow(ctx context.Context,
 
 	pushedRecordsWithCount := fmt.Sprintf("pushed %d records", numRecords)
 	activity.RecordHeartbeat(ctx, pushedRecordsWithCount)
-	done <- struct{}{}
 
 	return res, nil
 }