Update Context for store

internal/domain/proxy/handler.go

@@ -2,6 +2,7 @@
 package proxy
 
 import (
+	"context"
 	"encoding/json"
 	"gemini-balancer/internal/errors"
 	"gemini-balancer/internal/models"
@@ -49,7 +50,6 @@ func (h *handler) registerRoutes(rg *gin.RouterGroup) {
 	}
 }
 
-// --- 请求 DTO ---
 type CreateProxyConfigRequest struct {
 	Address     string `json:"address" binding:"required"`
 	Protocol    string `json:"protocol" binding:"required,oneof=http https socks5"`
@@ -64,12 +64,10 @@ type UpdateProxyConfigRequest struct {
 	Description *string `json:"description"`
 }
 
-// 单个检测的请求体 (与前端JS对齐)
 type CheckSingleProxyRequest struct {
 	Proxy string `json:"proxy" binding:"required"`
 }
 
-// 批量检测的请求体
 type CheckAllProxiesRequest struct {
 	Proxies []string `json:"proxies" binding:"required"`
 }
@@ -84,7 +82,7 @@ func (h *handler) CreateProxyConfig(c *gin.Context) {
 	}
 
 	if req.Status == "" {
-		req.Status = "active" // 默认状态
+		req.Status = "active"
 	}
 
 	proxyConfig := models.ProxyConfig{
@@ -98,7 +96,6 @@ func (h *handler) CreateProxyConfig(c *gin.Context) {
 		response.Error(c, errors.ParseDBError(err))
 		return
 	}
-	// 写操作后，发布事件并使缓存失效
 	h.publishAndInvalidate(proxyConfig.ID, "created")
 	response.Created(c, proxyConfig)
 }
@@ -199,17 +196,16 @@ func (h *handler) DeleteProxyConfig(c *gin.Context) {
 	response.NoContent(c)
 }
 
-// publishAndInvalidate 统一事件发布和缓存失效逻辑
 func (h *handler) publishAndInvalidate(proxyID uint, action string) {
 	go h.manager.invalidate()
 	go func() {
+		ctx := context.Background()
 		event := models.ProxyStatusChangedEvent{ProxyID: proxyID, Action: action}
 		eventData, _ := json.Marshal(event)
-		_ = h.store.Publish(models.TopicProxyStatusChanged, eventData)
+		_ = h.store.Publish(ctx, models.TopicProxyStatusChanged, eventData)
 	}()
 }
 
-// 新的 Handler 方法和 DTO
 type SyncProxiesRequest struct {
 	Proxies []string `json:"proxies"`
 }
@@ -220,14 +216,12 @@ func (h *handler) SyncProxies(c *gin.Context) {
 		response.Error(c, errors.NewAPIError(errors.ErrInvalidJSON, err.Error()))
 		return
 	}
-	taskStatus, err := h.manager.SyncProxiesInBackground(req.Proxies)
+
+	taskStatus, err := h.manager.SyncProxiesInBackground(c.Request.Context(), req.Proxies)
 	if err != nil {
-
 		if errors.Is(err, ErrTaskConflict) {
-
 			response.Error(c, errors.NewAPIError(errors.ErrTaskInProgress, err.Error()))
 		} else {
-
 			response.Error(c, errors.NewAPIError(errors.ErrInternalServer, err.Error()))
 		}
 		return
@@ -262,7 +256,7 @@ func (h *handler) CheckAllProxies(c *gin.Context) {
 
 	concurrency := cfg.ProxyCheckConcurrency
 	if concurrency <= 0 {
-		concurrency = 5 // 如果配置不合法，提供一个安全的默认值
+		concurrency = 5
 	}
 	results := h.manager.CheckMultipleProxies(req.Proxies, timeout, concurrency)
 	response.Success(c, results)