fix data race by replace clone

Signed-off-by: bufferflies <[email protected]>

pkg/core/storelimit/limit.go

@@ -40,4 +40,6 @@ type StoreLimit interface {
 	Take(count int64, typ Type, level constant.PriorityLevel) bool
 	// Reset resets the store limit
 	Reset(rate float64, typ Type)
+	// Cap returns the capacity of the store limit
+	Cap(typ Type) float64
 }

pkg/core/storelimit/sliding_window.go

@@ -44,6 +44,11 @@ func NewSlidingWindows(cap float64) *SlidingWindows {
 	}
 }
 
+// Cap returns the capacity of the sliding windows.
+func (s *SlidingWindows) Cap(typ Type) float64 {
+	return float64(s.windows[typ].capacity)
+}
+
 // Reset resets the capacity of the sliding windows.
 // It doesn't clear all the used, only set the capacity.
 func (s *SlidingWindows) Reset(cap float64, typ Type) {

pkg/core/storelimit/store_limit.go

@@ -90,6 +90,14 @@ func (l *StoreRateLimit) Available(cost int64, typ Type, _ constant.PriorityLeve
 	return l.limits[typ].Available(cost)
 }
 
+// Cap returns the capacity of the store limit.
+func (l *StoreRateLimit) Cap(typ Type) float64 {
+	if l.limits[typ] == nil {
+		return 0.0
+	}
+	return l.limits[typ].ratePerSec
+}
+
 // Take takes count tokens from the bucket without blocking.
 // notice that the priority level is not used.
 func (l *StoreRateLimit) Take(cost int64, typ Type, _ constant.PriorityLevel) bool {

pkg/schedule/operator_controller.go

@@ -866,7 +866,8 @@ func (oc *OperatorController) getOrCreateStoreLimit(storeID uint64, limitType st
 		return nil
 	}
 
-	limit := s.GetStoreLimit()
-	limit.Reset(ratePerSec, limitType)
-	return limit
+	if limit := s.GetStoreLimit(); limit.Cap(limitType) != ratePerSec {
+		oc.cluster.GetBasicCluster().ResetStoreLimit(storeID, limitType)
+	}
+	return s.GetStoreLimit()
 }