lib/storage: update nextRotationTimestamp relative to the timestamp o…

…f the indexdb rotation Updates #1401 Updates #4563
VictoriaMetrics · Jul 29, 2023 · 89ccf19 · 89ccf19
1 parent dfe4816
commit 89ccf19
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/lib/storage/index_db_test.go b/lib/storage/index_db_test.go
@@ -1585,7 +1585,7 @@ func TestIndexDBRepopulateAfterRotation(t *testing.T) {
 	prevGeneration := db.generation
 
 	// force index rotation
-	s.mustRotateIndexDB()
+	s.mustRotateIndexDB(time.Now())
 
 	// check tsidCache wasn't reset after the rotation
 	var cs2 fastcache.Stats

diff --git a/lib/storage/storage.go b/lib/storage/storage.go
@@ -707,8 +707,8 @@ func (s *Storage) retentionWatcher() {
 		select {
 		case <-s.stop:
 			return
-		case <-time.After(time.Second * time.Duration(d)):
-			s.mustRotateIndexDB()
+		case currentTime := <-time.After(time.Second * time.Duration(d)):
+			s.mustRotateIndexDB(currentTime)
 		}
 	}
 }
@@ -765,14 +765,15 @@ func (s *Storage) nextDayMetricIDsUpdater() {
 	}
 }
 
-func (s *Storage) mustRotateIndexDB() {
+func (s *Storage) mustRotateIndexDB(currentTime time.Time) {
 	// Create new indexdb table, which will be used as idbNext
 	newTableName := nextIndexDBTableName()
 	idbNewPath := filepath.Join(s.path, indexdbDirname, newTableName)
 	idbNew := mustOpenIndexDB(idbNewPath, s, &s.isReadOnly)
 
 	// Update nextRotationTimestamp
-	atomic.AddInt64(&s.nextRotationTimestamp, s.retentionMsecs/1000)
+	nextRotationTimestamp := currentTime.UnixMilli() + s.retentionMsecs/1000
+	atomic.StoreInt64(&s.nextRotationTimestamp, nextRotationTimestamp)
 
 	// Set idbNext to idbNew
 	idbNext := s.idbNext.Load()

diff --git a/lib/storage/storage_test.go b/lib/storage/storage_test.go
@@ -1238,7 +1238,7 @@ func TestStorageRotateIndexDB(t *testing.T) {
 				return
 			default:
 				time.Sleep(time.Millisecond)
-				s.mustRotateIndexDB()
+				s.mustRotateIndexDB(time.Now())
 			}
 		}
 	}()