重构白银节点分配方式,使用手动接口精确配置节点

This commit is contained in:
2025-12-08 14:22:30 +08:00
parent 9e237be21e
commit 983dbb4564
25 changed files with 651 additions and 630 deletions

View File

@@ -5,18 +5,11 @@ import (
"encoding/json"
"fmt"
"log/slog"
"platform/pkg/env"
"platform/pkg/u"
"platform/web/events"
g "platform/web/globals"
m "platform/web/models"
q "platform/web/queries"
s "platform/web/services"
"strings"
"time"
"github.com/hibiken/asynq"
"gorm.io/datatypes"
)
func HandleCompleteTrade(_ context.Context, task *asynq.Task) (err error) {
@@ -51,99 +44,3 @@ func HandleRemoveChannel(_ context.Context, task *asynq.Task) (err error) {
}
return nil
}
func HandleFlushGateway(_ context.Context, task *asynq.Task) error {
start := time.Now()
defer func() {
duration := time.Since(start)
if duration > time.Second {
slog.Warn("更新代理后备配置耗时过长", "time", duration.String())
}
}()
// 获取所有网关:配置组
proxies, err := s.Proxy.AllProxies(m.ProxyTypeBaiYin, true)
if err != nil {
return fmt.Errorf("获取网关失败: %w", err)
}
for _, proxy := range proxies {
// 获取当前后备配置
locals := map[string]int{}
for _, channel := range proxy.Channels {
isp := channel.FilterISP.String()
prov := u.Z(channel.FilterProv)
city := u.Z(channel.FilterCity)
locals[fmt.Sprintf("%s:%s:%s", isp, prov, city)]++
}
// 获取之前的后备配置
remotes := map[string]int{}
if proxy.Meta != nil {
meta, ok := proxy.Meta.Data().([]any)
if !ok {
return fmt.Errorf("解析网关数据失败: %T", proxy.Meta.Data())
}
for _, rawM := range meta {
m, ok := rawM.(map[string]any)
if !ok {
return fmt.Errorf("解析网关数据失败: %T", rawM)
}
remotes[fmt.Sprintf("%s:%s:%s", m["isp"], m["province"], m["city"])] = int(m["count"].(float64))
}
}
// 检查是否需要更新
pass := true
for k, local := range locals {
remote, ok := remotes[k]
if !ok {
pass = false
} else {
local, remote := float64(local), float64(remote)
if remote < local*1.5 || remote > local*3 {
pass = false
}
}
}
if pass {
continue
}
// 更新后备配置
configs := make([]g.AutoConfig, 0)
for k, local := range locals {
arr := strings.Split(k, ":")
isp, prov, city := arr[0], arr[1], arr[2]
configs = append(configs, g.AutoConfig{
Isp: isp,
Province: prov,
City: city,
Count: local * 2,
})
}
if env.DebugExternalChange {
err := g.Cloud.CloudConnect(g.CloudConnectReq{
Uuid: proxy.Mac,
AutoConfig: configs,
})
if err != nil {
slog.Error("提交代理后备配置失败", "error", err)
}
} else {
bytes, _ := json.Marshal(configs)
slog.Debug("更新代理后备配置", "proxy", proxy.IP.String(), "config", string(bytes))
}
_, err := q.Proxy.
Where(q.Proxy.ID.Eq(proxy.ID)).
UpdateSimple(q.Proxy.Meta.Value(datatypes.NewJSONType(configs)))
if err != nil {
slog.Error("更新代理后备配置失败", "error", err)
}
}
return nil
}