Add management API endpoints

internal/api/admin_handler.go

@@ -0,0 +1,175 @@
+package api
+
+import (
+	"encoding/json"
+	"net/http"
+	"time"
+
+	"github.com/DullJZ/s3-balance/internal/balancer"
+	"github.com/DullJZ/s3-balance/internal/bucket"
+	"github.com/DullJZ/s3-balance/internal/config"
+	"github.com/gorilla/mux"
+)
+
+// AdminHandler 管理API处理器
+type AdminHandler struct {
+	bucketManager *bucket.Manager
+	balancer      *balancer.Balancer
+	config        *config.Config
+}
+
+// NewAdminHandler 创建新的管理API处理器
+func NewAdminHandler(
+	bucketManager *bucket.Manager,
+	balancer *balancer.Balancer,
+	cfg *config.Config,
+) *AdminHandler {
+	return &AdminHandler{
+		bucketManager: bucketManager,
+		balancer:      balancer,
+		config:        cfg,
+	}
+}
+
+// BucketResponse 存储桶响应结构
+type BucketResponse struct {
+	Name            string    `json:"name"`
+	Endpoint        string    `json:"endpoint"`
+	Region          string    `json:"region"`
+	MaxSize         string    `json:"max_size"`
+	MaxSizeBytes    int64     `json:"max_size_bytes"`
+	UsedSize        int64     `json:"used_size"`
+	AvailableSize   int64     `json:"available_size"`
+	UsagePercent    float64   `json:"usage_percent"`
+	Weight          int       `json:"weight"`
+	Enabled         bool      `json:"enabled"`
+	Available       bool      `json:"available"`
+	Virtual         bool      `json:"virtual"`
+	LastChecked     time.Time `json:"last_checked"`
+	OperationCountA int64     `json:"operation_count_a"`
+	OperationCountB int64     `json:"operation_count_b"`
+	OperationLimits struct {
+		TypeA int `json:"type_a"`
+		TypeB int `json:"type_b"`
+	} `json:"operation_limits"`
+}
+
+// BucketsListResponse 存储桶列表响应结构
+type BucketsListResponse struct {
+	Total   int                `json:"total"`
+	Buckets []BucketResponse   `json:"buckets"`
+}
+
+// HealthResponse 健康状态响应结构
+type HealthResponse struct {
+	Status          string    `json:"status"`
+	Timestamp       time.Time `json:"timestamp"`
+	LoadBalancer    string    `json:"load_balancer_strategy"`
+	TotalBuckets    int       `json:"total_buckets"`
+	AvailableBuckets int      `json:"available_buckets"`
+	Database        string    `json:"database_type"`
+}
+
+// RegisterRoutes 注册管理API路由
+func (h *AdminHandler) RegisterRoutes(router *mux.Router) {
+	router.HandleFunc("/api/buckets", h.ListBuckets).Methods(http.MethodGet)
+	router.HandleFunc("/api/buckets/{name}", h.GetBucketDetail).Methods(http.MethodGet)
+	router.HandleFunc("/api/health", h.GetHealth).Methods(http.MethodGet)
+}
+
+// ListBuckets 获取存储桶列表
+func (h *AdminHandler) ListBuckets(w http.ResponseWriter, r *http.Request) {
+	buckets := h.bucketManager.GetAllBuckets()
+
+	response := BucketsListResponse{
+		Total:   len(buckets),
+		Buckets: make([]BucketResponse, 0, len(buckets)),
+	}
+
+	for _, b := range buckets {
+		bucketResp := h.convertBucketInfo(b)
+		response.Buckets = append(response.Buckets, bucketResp)
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(response)
+}
+
+// GetBucketDetail 获取存储桶详情
+func (h *AdminHandler) GetBucketDetail(w http.ResponseWriter, r *http.Request) {
+	vars := mux.Vars(r)
+	name := vars["name"]
+
+	bucketInfo, exists := h.bucketManager.GetBucket(name)
+	if !exists {
+		http.Error(w, `{"error": "bucket not found"}`, http.StatusNotFound)
+		return
+	}
+
+	response := h.convertBucketInfo(bucketInfo)
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(response)
+}
+
+// GetHealth 获取系统健康状态
+func (h *AdminHandler) GetHealth(w http.ResponseWriter, r *http.Request) {
+	buckets := h.bucketManager.GetAllBuckets()
+	availableBuckets := h.bucketManager.GetAvailableBuckets()
+
+	status := "healthy"
+	if len(availableBuckets) == 0 {
+		status = "unhealthy"
+	} else if len(availableBuckets) < len(buckets)/2 {
+		status = "degraded"
+	}
+
+	response := HealthResponse{
+		Status:           status,
+		Timestamp:        time.Now(),
+		LoadBalancer:     h.config.Balancer.Strategy,
+		TotalBuckets:     len(buckets),
+		AvailableBuckets: len(availableBuckets),
+		Database:         h.config.Database.Type,
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(response)
+}
+
+// convertBucketInfo 转换BucketInfo为BucketResponse
+func (h *AdminHandler) convertBucketInfo(b *bucket.BucketInfo) BucketResponse {
+	resp := BucketResponse{
+		Name:            b.Config.Name,
+		Endpoint:        b.Config.Endpoint,
+		Region:          b.Config.Region,
+		MaxSize:         b.Config.MaxSize,
+		MaxSizeBytes:    b.Config.MaxSizeBytes,
+		UsedSize:        b.UsedSize,
+		Weight:          b.Config.Weight,
+		Enabled:         b.Config.Enabled,
+		Available:       b.Available,
+		Virtual:         b.Config.Virtual,
+		LastChecked:     b.LastChecked,
+		OperationCountA: b.GetOperationCount(bucket.OperationTypeA),
+		OperationCountB: b.GetOperationCount(bucket.OperationTypeB),
+	}
+
+	resp.OperationLimits.TypeA = b.Config.OperationLimits.TypeA
+	resp.OperationLimits.TypeB = b.Config.OperationLimits.TypeB
+
+	// 计算可用空间
+	if b.Config.MaxSizeBytes > 0 {
+		resp.AvailableSize = b.Config.MaxSizeBytes - b.UsedSize
+		if resp.AvailableSize < 0 {
+			resp.AvailableSize = 0
+		}
+		// 计算使用百分比
+		resp.UsagePercent = float64(b.UsedSize) / float64(b.Config.MaxSizeBytes) * 100
+	} else {
+		resp.AvailableSize = -1 // -1 表示无限制
+		resp.UsagePercent = 0
+	}
+
+	return resp
+}