xiaolajiaoyyds
6f71df4849
多平台账号批量注册系统，支持 OpenAI / Grok / Kiro / Gemini 平台的全自动注册。 Features: - 用户名/密码注册登录（首个用户自动管理员） - JWT 认证 + bcrypt 密码加密 - 积分系统与多用户管理 - HuggingFace Space 弹性 Worker 节点管理 - Cloudflare Worker 负载均衡 - 实时日志推送（SSE/WebSocket） - 管理后台（用户管理、系统设置、代理池、卡池） Tech Stack: Go 1.25 / Gin / GORM / PostgreSQL / Vue 3 / Vite / TailwindCSS
76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
package worker
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// QueueStatus CF Worker /api/v1/status 聚合的远程节点池排队状态
|
|
type QueueStatus struct {
|
|
TotalActive int `json:"total_active"`
|
|
TotalWaiting int `json:"total_waiting"`
|
|
MaxConcurrent int `json:"max_concurrent"`
|
|
AvgSeconds int `json:"avg_seconds"`
|
|
HealthyNodes int `json:"healthy_nodes"`
|
|
TotalNodes int `json:"total_nodes"`
|
|
}
|
|
|
|
// ETA 估算排队等待时间（秒）
|
|
// 公式：平均耗时 × 排队人数 ÷ 总处理槽位
|
|
func (qs *QueueStatus) ETA() int {
|
|
if qs.TotalWaiting <= 0 {
|
|
return 0
|
|
}
|
|
throughput := qs.MaxConcurrent
|
|
if throughput < 1 {
|
|
throughput = 1
|
|
}
|
|
return qs.AvgSeconds * qs.TotalWaiting / throughput
|
|
}
|
|
|
|
// queuePollClient 排队状态轮询专用 HTTP 客户端（短超时，不影响主请求）
|
|
var queuePollClient = &http.Client{Timeout: 5 * time.Second}
|
|
|
|
// PollQueueStatus 查询 CF Worker 聚合的排队状态
|
|
// statusURL 格式: https://cf-worker.xxx/api/v1/status?t=openai
|
|
func PollQueueStatus(ctx context.Context, statusURL string) (*QueueStatus, error) {
|
|
req, err := http.NewRequestWithContext(ctx, "GET", statusURL, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
resp, err := queuePollClient.Do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
var qs QueueStatus
|
|
if err := json.NewDecoder(resp.Body).Decode(&qs); err != nil {
|
|
return nil, err
|
|
}
|
|
return &qs, nil
|
|
}
|
|
|
|
// FormatQueueLog 将排队状态格式化为用户可读的日志行
|
|
func FormatQueueLog(qs *QueueStatus, elapsed int) string {
|
|
if qs.TotalWaiting > 0 {
|
|
eta := qs.ETA()
|
|
return fmt.Sprintf("[…] 排队中 %ds | 处理中 %d · 排队 %d · 节点 %d | 预计还需 %d 秒",
|
|
elapsed, qs.TotalActive, qs.TotalWaiting, qs.HealthyNodes, eta)
|
|
}
|
|
return fmt.Sprintf("[…] 远程注册进行中 (%ds)...", elapsed)
|
|
}
|
|
|
|
// BuildQueueStatusURL 构建排队状态查询 URL
|
|
// 多节点逗号分隔时，只取第一个（CF Worker），直连节点没有 /api/v1/status 路由
|
|
func BuildQueueStatusURL(serviceURL, platform string) string {
|
|
if idx := strings.Index(serviceURL, ","); idx > 0 {
|
|
serviceURL = serviceURL[:idx]
|
|
}
|
|
return strings.TrimRight(serviceURL, "/") + "/api/v1/status?t=" + platform
|
|
}
|