Update end time for batches where end time is not set (#2260)

PeerDB-io · Nov 15, 2024 · 27996db · 27996db
1 parent 593093e
commit 27996db
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/flow/connectors/utils/monitoring/monitoring.go b/flow/connectors/utils/monitoring/monitoring.go
@@ -96,8 +96,10 @@ func UpdateEndTimeForCDCBatch(
 	batchID int64,
 ) error {
 	_, err := pool.Exec(ctx,
-		"UPDATE peerdb_stats.cdc_batches SET end_time=$1 WHERE flow_name=$2 AND batch_id=$3",
-		time.Now(), flowJobName, batchID)
+		`UPDATE peerdb_stats.cdc_batches
+		SET end_time = COALESCE(end_time, NOW())
+		WHERE flow_name = $1 AND batch_id <= $2`,
+		flowJobName, batchID)
 	if err != nil {
 		return fmt.Errorf("error while updating batch in cdc_batch: %w", err)
 	}