Sakurasan
8c12207478
按性能分析逐项落地，实测对比（16 并发、mock 上游零延迟、SQLite）： - 每请求 SQL 10 → 2：模型/渠道目录（模型行、绑定、渠道行）进程内缓存 10s， 运行期状态（冷却、健康）仍每次实时判定；管理端写操作经中间件、 健康状态变化经回调立即失效。新增 TestProxyRequestSQLBudget 钉住预算。 - 端到端 p50 1.3–1.6ms → 0.66–0.70ms，RPS 3.5–8.9k → 16–18k。 - 流式正文累积：字符串 += 改 strings.Builder（原为 O(n²)）， 逐行 JSON 提取改字节快路径，并加廉价的子串预过滤、拿到输出 usage 后停累积。 500 块：949µs/7.9MB 分配 → 83µs/136KB。 - 请求体定点改写：模型映射从整包 unmarshal/marshal 改为顶层字段替换 （100KB：585µs → 29µs），stream_options 注入同理；requestText 改为 只在「上游没报用量」时按需提取，亲和键改用原始体前缀。 - 渠道满载背压：新增 Acquire 有界排队（OT_PROXY_CHANNEL_ACQUIRE_WAIT， 默认 150ms）；排队仍满时返回 503 + Retry-After 而不是 502。 实测渠道上限 4、16 并发：无排队 81 次 503；排队 150ms 后 400/400 成功。 - 记账按批聚合：同批同用户一次扣款、流水批量插入、日表按 (user,model,日期) 聚合，替代每行 5 条语句；幂等仍靠 ref_id 唯一索引兜底。 - gorm 开启 PrepareStmt（SQL 解析此前占 CPU 15%）；tiktoken 编码器按模型复用。 - 管理端用量列表不再返回原始请求/响应体（可能几十 MB）， 改为 has_raw 标记 + GET /admin/usage/:id 按需拉取，前端弹窗打开时取。 验证：go vet、go test ./...、go test -race（proxy/channel/usage/tokenizer）、 vue-tsc、eslint、vite build 全绿；本地起 mock 上游实测列表/详情接口与 管理页弹窗按需加载。
64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
// Package tokenizer 按模型估算 token 数（tiktoken-go）。
|
|
// 用于流式中断时对已生成内容做近似计费；未知模型回退 cl100k_base。
|
|
package tokenizer
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/tiktoken-go/tokenizer"
|
|
)
|
|
|
|
// 编码器缓存：tokenizer.Get/ForModel 每次调用都新建 codec，其中包含 regexp2
|
|
// 正则编译（回溯引擎，编译昂贵）与 map 分配。实测短文本 6.9µs、18KB 文本
|
|
// 1.05ms，绝大部分是这套固定开销而非真正的分词。
|
|
//
|
|
// codec 构建后只读（vocabulary/splitRegexp 不再变），regexp2.Regexp 自身带锁，
|
|
// 因此缓存实例可并发复用。键是模型名：调用方模型名有限，缓存自然有界。
|
|
var codecs sync.Map // string → tokenizer.Codec
|
|
|
|
// Count 估算文本 token 数；空文本返回 0。
|
|
func Count(text, model string) int {
|
|
if text == "" {
|
|
return 0
|
|
}
|
|
enc := codecFor(model)
|
|
if enc == nil {
|
|
return 0
|
|
}
|
|
toks, _, err := enc.Encode(text)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
return len(toks)
|
|
}
|
|
|
|
// codecFor 取（必要时构建并缓存）该模型对应的 codec；未知模型回退 cl100k_base。
|
|
func codecFor(model string) tokenizer.Codec {
|
|
if v, ok := codecs.Load(model); ok {
|
|
return v.(tokenizer.Codec)
|
|
}
|
|
enc, err := tokenizer.ForModel(tokenizer.Model(model))
|
|
if err != nil || enc == nil {
|
|
enc = fallback()
|
|
}
|
|
if enc == nil {
|
|
return nil
|
|
}
|
|
codecs.Store(model, enc)
|
|
return enc
|
|
}
|
|
|
|
// fallback 取 cl100k_base 编码器（单独缓存，避免每次未知模型都重建）。
|
|
func fallback() tokenizer.Codec {
|
|
const key = "\x00cl100k"
|
|
if v, ok := codecs.Load(key); ok {
|
|
return v.(tokenizer.Codec)
|
|
}
|
|
enc, err := tokenizer.Get(tokenizer.Cl100kBase)
|
|
if err != nil || enc == nil {
|
|
return nil
|
|
}
|
|
codecs.Store(key, enc)
|
|
return enc
|
|
}
|