wangbo
7786692d32
将 Worker 发现、路由画像、容量与执行传输抽象为平台无关接口，新增 Kubernetes 和静态容量适配器，并以 shadow 模式接入生产配置。 实现网络与容量评分、路由防抖、池队列、同步 Worker 租约、一次性执行令牌，以及提交状态不明时禁止重复分配的安全语义。 新增 0105 兼容迁移、管理接口、指标、OpenAPI 和回归测试。已执行全量 Go 测试、go vet、OpenAPI、迁移安全、Compose 与 Kustomize 验证。
41 lines
1.3 KiB
Go
41 lines
1.3 KiB
Go
package capacitycontroller
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/easyai/easyai-ai-gateway/apps/api/internal/config"
|
|
)
|
|
|
|
func NewConfiguredAdapter(cfg config.Config) (OrchestratorAdapter, error) {
|
|
switch strings.ToLower(strings.TrimSpace(cfg.CapacityOrchestratorAdapter)) {
|
|
case "static":
|
|
pools, err := cfg.CapacityPools()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
initial := make(map[string]int, len(pools))
|
|
for _, pool := range pools {
|
|
initial[pool.ID] = pool.BootstrapReplicas
|
|
}
|
|
return NewStaticAdapter(initial), nil
|
|
case "", "kubernetes":
|
|
return NewKubernetesClient(KubernetesConfig{
|
|
Namespace: adapterEnv("AI_GATEWAY_CAPACITY_CONTROLLER_NAMESPACE", adapterEnv("POD_NAMESPACE", "easyai")),
|
|
APIServer: adapterEnv("AI_GATEWAY_CAPACITY_CONTROLLER_API_SERVER", "https://kubernetes.default.svc"),
|
|
TokenFile: adapterEnv("AI_GATEWAY_CAPACITY_CONTROLLER_TOKEN_FILE", "/var/run/secrets/kubernetes.io/serviceaccount/token"),
|
|
CAFile: adapterEnv("AI_GATEWAY_CAPACITY_CONTROLLER_CA_FILE", "/var/run/secrets/kubernetes.io/serviceaccount/ca.crt"),
|
|
})
|
|
default:
|
|
return nil, errors.New("unsupported capacity orchestrator adapter")
|
|
}
|
|
}
|
|
|
|
func adapterEnv(name, fallback string) string {
|
|
if value := strings.TrimSpace(os.Getenv(name)); value != "" {
|
|
return value
|
|
}
|
|
return fallback
|
|
}
|