protect stats hashes

dolthub · Sep 10, 2024 · ff4fe7b · ff4fe7b
1 parent c4bf019
commit ff4fe7b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/go/libraries/doltcore/sqle/statspro/dolt_stats.go b/go/libraries/doltcore/sqle/statspro/dolt_stats.go
@@ -44,6 +44,12 @@ func (s *DoltStats) Clone(_ context.Context) sql.JSONWrapper {
 
 var _ sql.Statistic = (*DoltStats)(nil)
 
+func (s *DoltStats) SetChunks(h []hash.Hash) sql.Statistic {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	s.Chunks = h
+}
+
 func (s *DoltStats) WithColSet(set sql.ColSet) sql.Statistic {
 	ret := *s
 	ret.Statistic = ret.Statistic.WithColSet(set).(*stats.Statistic)