From ff4fe7bfe48c349f2df8a119dabfb9aa71644ca8 Mon Sep 17 00:00:00 2001 From: Max Hoffman Date: Tue, 10 Sep 2024 12:28:51 -0700 Subject: [PATCH] protect stats hashes --- go/libraries/doltcore/sqle/statspro/dolt_stats.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/go/libraries/doltcore/sqle/statspro/dolt_stats.go b/go/libraries/doltcore/sqle/statspro/dolt_stats.go index 142e5a063e1..b0223e1c43c 100644 --- 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)