Count operation type A & B

internal/bucket/reporter.go

@@ -7,9 +7,9 @@ import (
 
 // MetricsReporter 实现 health.HealthReporter 和 health.StatsReporter 接口
 type MetricsReporter struct {
-	metrics  *metrics.Metrics
-	buckets  map[string]*BucketInfo
-	manager  *Manager
+	metrics *metrics.Metrics
+	buckets map[string]*BucketInfo
+	manager *Manager
 }
 
 // NewMetricsReporter 创建指标报告器
@@ -25,18 +25,20 @@ func (r *MetricsReporter) ReportHealth(targetID string, status health.Status) {
 	if r.metrics == nil {
 		return
 	}
-	
+
 	// 更新存储桶可用性状态
 	r.manager.mu.RLock()
 	bucket, exists := r.manager.buckets[targetID]
 	r.manager.mu.RUnlock()
-	
+
 	if exists {
 		bucket.mu.Lock()
-		bucket.Available = status.Healthy
+		if !bucket.operationLimitReached {
+			bucket.Available = status.Healthy
+		}
 		bucket.LastChecked = status.LastChecked
 		bucket.mu.Unlock()
-		
+
 		// 更新 Prometheus 指标
 		r.metrics.SetBucketHealthy(targetID, bucket.Config.Endpoint, status.Healthy)
 	}
@@ -47,18 +49,18 @@ func (r *MetricsReporter) ReportStats(stats *health.Stats) {
 	if r.metrics == nil {
 		return
 	}
-	
+
 	// 更新存储桶使用统计
 	r.manager.mu.RLock()
 	bucket, exists := r.manager.buckets[stats.TargetID]
 	r.manager.mu.RUnlock()
-	
+
 	if exists {
 		bucket.mu.Lock()
 		bucket.UsedSize = stats.UsedSize
 		bucket.mu.Unlock()
-		
+
 		// 更新 Prometheus 指标
 		r.metrics.SetBucketUsage(stats.TargetID, stats.UsedSize, bucket.Config.MaxSizeBytes)
 	}
-}
+}