heartbeat routine needs to be test aware, prevent panics

PeerDB-io · Oct 25, 2023 · c61bcdd · c61bcdd
1 parent c7b1f06
commit c61bcdd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flow/connectors/utils/heartbeat.go b/flow/connectors/utils/heartbeat.go
@@ -19,7 +19,7 @@ func HeartbeatRoutine(
 	go func() {
 		for {
 			msg := fmt.Sprintf("heartbeat #%d: %s", counter, message())
-			activity.RecordHeartbeat(ctx, msg)
+			RecordHeartbeatWithRecover(ctx, msg)
 			counter += 1
 			to := time.After(interval)
 			select {