chenxin
144415247e
- 神识 Go：store 双实现（PG 自动建表+60s 就绪重试 / 内存回退）承载 usage_logs·byok_keys·quota_balances；BYOK 密钥库（AES-GCM 加密、API 脱敏、撤回即失效、主密钥未配置 fail-closed）；配额三段式（预扣 max_tokens 钳位 1M→结算实际 token→全败幂等退款，HJ-2004）；计量归属（IR user_id→usage_logs，/me/usage 用户维度聚合+配额总览，/admin/usage/sample 对账抽样）；Anthropic/Gemini 出站适配器（SSE 解析、system 提升、用量映射）；stub 端点补名修复计量空 endpoint - IR 契约：user_id/byok_key_id 双端同步（Go types.go + Py types.py） - 灵台：runs.owner_id（迁移 005，server_default 兼容存量）经 kernel 全链透传为 IR user_id；shenshi_client 扩展用户面方法（X-Hunjing-User-Id 代理，Echo 桩带内存实现）；/me/byok-keys 登记·列表·撤回与 /me/usage 端点（4 用例） - 前端神识阁（/models 重建）：偏好（生成参数表单）/自带密钥（登记加密保存、脱敏列表、撤回二次确认）/用量（聚合表+配额进度条）三区；me API 库+mock+5 用例；侧栏恢复入口 - 配置与部署：.env.example 增 SHENSHI_DATABASE_URL/SHENSHI_BYOK_MASTER_KEY/SHENSHI_USER_BUDGET_TOKENS；compose 神识容器接自有库（init-shenshi.sql 预建 hunjing_shenshi，与灵台不共库共表） - 证据：Go 34 用例（BYOK 独占/配额拒绝/结算退款/归属聚合）、后端 532、前端 210、BFF 4 全绿；跨进程 E2E 四场景（灵台 HttpShenshiClient→真 Go 神识：登记脱敏/BYOK 独占不可达整体失败/用量归属聚合/撤回即失效）通过；rewrite-plan M4 标记完成、能力矩阵神识行升已验证
63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
// 神识（Shenshi）入口：魂镜模型网关（rewrite/1.0）。
|
||
// 数据面 POST /v1/chat/completions（IR 入站，SSE 出站）；健康面 /healthz /readyz；
|
||
// 计量面 /me/usage、对账抽样 /admin/usage/sample；BYOK 面 /me/byok-keys。
|
||
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
"net/http"
|
||
"os"
|
||
|
||
"hunjing/shenshi/internal/byok"
|
||
"hunjing/shenshi/internal/quota"
|
||
"hunjing/shenshi/internal/route"
|
||
"hunjing/shenshi/internal/server"
|
||
"hunjing/shenshi/internal/store"
|
||
)
|
||
|
||
func envOr(key, fallback string) string {
|
||
if v := os.Getenv(key); v != "" {
|
||
return v
|
||
}
|
||
return fallback
|
||
}
|
||
|
||
func main() {
|
||
st, err := store.Open(os.Getenv("SHENSHI_DATABASE_URL"))
|
||
if err != nil {
|
||
log.Fatalf("神识存储初始化失败: %v", err)
|
||
}
|
||
|
||
cfg := server.Config{
|
||
ServiceToken: os.Getenv("SHENSHI_SERVICE_TOKEN"),
|
||
Store: st,
|
||
}
|
||
|
||
// BYOK：主密钥未配置时整体禁用（fail-closed）
|
||
if master := os.Getenv("SHENSHI_BYOK_MASTER_KEY"); master != "" {
|
||
vault, err := byok.NewVault(master, st)
|
||
if err != nil {
|
||
log.Fatalf("BYOK 主密钥配置无效: %v", err)
|
||
}
|
||
cfg.Vault = vault
|
||
log.Printf("BYOK 密钥库就绪")
|
||
}
|
||
|
||
// 配额：用户级软配额（token 口径；0 = 不限制）
|
||
var budget int64
|
||
if raw := os.Getenv("SHENSHI_USER_BUDGET_TOKENS"); raw != "" {
|
||
if _, err := fmt.Sscanf(raw, "%d", &budget); err != nil {
|
||
log.Fatalf("SHENSHI_USER_BUDGET_TOKENS 须为整数: %v", err)
|
||
}
|
||
cfg.Quota = quota.New(st, budget)
|
||
}
|
||
|
||
srv := server.New(cfg, route.New())
|
||
addr := ":" + envOr("SHENSHI_PORT", "8080")
|
||
log.Printf("神识就绪：http://0.0.0.0%s（端点目录 SHENSHI_ENDPOINTS，未配置时回声桩兜底）", addr)
|
||
if err := http.ListenAndServe(addr, srv); err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
}
|