Function FetchUpstreamRatios

controller/ratio_sync.go:34–269

(c *gin.Context)

}

func FetchUpstreamRatios(c *gin.Context) {
    var req dto.UpstreamRequest
    if err := c.ShouldBindJSON(&req); err != nil {
        c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": err.Error()})
        return
    }

    if req.Timeout <= 0 {
        req.Timeout = defaultTimeoutSeconds
    }

    var upstreams []dto.UpstreamDTO

    if len(req.Upstreams) > 0 {
        for _, u := range req.Upstreams {
            if strings.HasPrefix(u.BaseURL, "http") {
                if u.Endpoint == "" {
                    u.Endpoint = defaultEndpoint
                }
                u.BaseURL = strings.TrimRight(u.BaseURL, "/")
                upstreams = append(upstreams, u)
            }
        }
    } else if len(req.ChannelIDs) > 0 {
        intIds := make([]int, 0, len(req.ChannelIDs))
        for _, id64 := range req.ChannelIDs {
            intIds = append(intIds, int(id64))
        }
        dbChannels, err := model.GetChannelsByIds(intIds)
        if err != nil {
            common.LogError(c.Request.Context(), "failed to query channels: "+err.Error())
            c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": "查询渠道失败"})
            return
        }
        for _, ch := range dbChannels {
            if base := ch.GetBaseURL(); strings.HasPrefix(base, "http") {
                upstreams = append(upstreams, dto.UpstreamDTO{
                    ID:       ch.Id,
                    Name:     ch.Name,
                    BaseURL:  strings.TrimRight(base, "/"),
                    Endpoint: "",
                })
            }
        }
    }

    if len(upstreams) == 0 {
        c.JSON(http.StatusOK, gin.H{"success": false, "message": "无有效上游渠道"})
        return
    }

    var wg sync.WaitGroup
    ch := make(chan upstreamResult, len(upstreams))

    sem := make(chan struct{}, maxConcurrentFetches)

    client := &http.Client{Transport: &http.Transport{MaxIdleConns: 100, IdleConnTimeout: 90 * time.Second, TLSHandshakeTimeout: 10 * time.Second, ExpectContinueTimeout: 1 * time.Second}}