Fix shard epoch race.

2019-02-11 12:15:46 -07:00 · 2019-02-11 12:15:46 -07:00 · b87605f521
parent 198f6fde38
commit b87605f521
1 changed files with 5 additions and 2 deletions
--- a/tsdb/store.go
+++ b/tsdb/store.go
@ -1373,11 +1373,14 @@ func (s *Store) WriteToShard(shardID uint64, points []models.Point) error {
 		s.mu.RUnlock()
 		return ErrShardNotFound
 	}
+
+	epoch := s.epochs[shardID]
+
 	s.mu.RUnlock()

 	// enter the epoch tracker
-	guards, gen := s.epochs[shardID].StartWrite()
-	defer s.epochs[shardID].EndWrite(gen)
+	guards, gen := epoch.StartWrite()
+	defer epoch.EndWrite(gen)

 	// wait for any guards before writing the points.
 	for _, guard := range guards {