flip pages early

gravitational · May 14, 2024 · c78062c · c78062c
1 parent b043444
commit c78062c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/event-handler/teleport_events_watcher.go b/event-handler/teleport_events_watcher.go
@@ -236,6 +236,10 @@ func (t *TeleportEventsWatcher) getEvents(ctx context.Context) (string, error) {
 }
 
 func (t *TeleportEventsWatcher) canSkipToNextWindow(i int, rangeSplitByDay []time.Time, cursor string) bool {
+	if cursor != "" {
+		return false
+
+	}
 	if len(t.batch) == 0 && i < len(rangeSplitByDay)-1 {
 		log.Infof("No events found for the range %v to %v", rangeSplitByDay[i-1], rangeSplitByDay[i])
 		return true
@@ -251,7 +255,7 @@ func (t *TeleportEventsWatcher) canSkipToNextWindow(i int, rangeSplitByDay []tim
 		}
 	}
 
-	if cursor == "" && i < len(rangeSplitByDay)-1 && pos >= len(t.batch) {
+	if i < len(rangeSplitByDay)-1 && pos >= len(t.batch) {
 		log.WithField("pos", pos).WithField("len", len(t.batch)).Infof("No new events found for the range %v to %v", rangeSplitByDay[i-1], rangeSplitByDay[i])
 		return true
 	}
@@ -315,7 +319,7 @@ func (t *TeleportEventsWatcher) Events(ctx context.Context) (chan *TeleportEvent
 				}
 
 				// If there is still nothing, sleep
-				if len(t.batch) == 0 {
+				if len(t.batch) == 0 && t.nextCursor == "" {
 					if t.config.ExitOnLastEvent {
 						log.Info("All events are processed, exiting...")
 						break