101 lines
3.8 KiB
Go
101 lines
3.8 KiB
Go
package dispatch
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"regexp"
|
|
)
|
|
|
|
// SessionSource 会话键来源标记，用于观测各来源占比（dispatch_session_source_total）。
|
|
type SessionSource string
|
|
|
|
const (
|
|
SourceHeader SessionSource = "hdr" // 显式 X-Session-Id 头
|
|
SourceAnthropic SessionSource = "anthropic" // Anthropic metadata.user_id 中的 session 段
|
|
SourceOpenAI SessionSource = "openai" // OpenAI Responses previous_response_id / conversation_id
|
|
SourceIdentity SessionSource = "ident" // 身份四元组回退
|
|
)
|
|
|
|
// SessionSignals 从请求中提取的会话信号原始值（提取由 handler 完成，本包只做解析决策）。
|
|
type SessionSignals struct {
|
|
HeaderSessionID string // X-Session-Id 头
|
|
AnthropicUserID string // Anthropic 格式的 metadata.user_id
|
|
PreviousResponseID string // OpenAI Responses previous_response_id
|
|
ConversationID string // OpenAI conversation_id / thread
|
|
}
|
|
|
|
// SessionKey 解析后的会话键。Key 为哈希后的稳定标识，可直接作为 Redis 绑定 key 的一部分。
|
|
type SessionKey struct {
|
|
Source SessionSource
|
|
Key string // 形如 sk:<source>:<sha256 hex>
|
|
}
|
|
|
|
const (
|
|
maxHeaderTokenLen = 256 // 显式头最大长度，超长视为无效
|
|
maxProtocolTokenLen = 512 // 协议内信号最大长度
|
|
)
|
|
|
|
// anthropicSessionRe 提取 Claude Code metadata.user_id 中的 session UUID 段。
|
|
// 格式形如 user_<hash>_account_<uuid>_session_<uuid>，格式稳定性需抓包验证。
|
|
var anthropicSessionRe = regexp.MustCompile(`session_([0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12})`)
|
|
|
|
// ResolveSessionKey 会话键解析链：
|
|
// 显式头 → Anthropic metadata → OpenAI Responses 信号 → 身份四元组回退。
|
|
// 纯函数：相同输入永远得到相同输出。
|
|
func ResolveSessionKey(p RequestProfile, pol SessionPolicy) SessionKey {
|
|
// 1. 显式头
|
|
if validToken(p.Signals.HeaderSessionID, maxHeaderTokenLen) {
|
|
return makeSessionKey(SourceHeader, p, p.Signals.HeaderSessionID)
|
|
}
|
|
|
|
// 2a. Anthropic metadata.user_id（可配置关闭）
|
|
if pol.ParseAnthropicMetadata && validToken(p.Signals.AnthropicUserID, maxProtocolTokenLen) {
|
|
raw := p.Signals.AnthropicUserID
|
|
if m := anthropicSessionRe.FindStringSubmatch(raw); len(m) == 2 {
|
|
raw = m[1] // 提取到 session UUID 段
|
|
}
|
|
// 提取失败则整个 user_id 作为会话键（仍优于身份级）
|
|
return makeSessionKey(SourceAnthropic, p, raw)
|
|
}
|
|
|
|
// 2b. OpenAI Responses（可配置关闭）
|
|
if pol.ParseOpenAIResponses {
|
|
if validToken(p.Signals.PreviousResponseID, maxProtocolTokenLen) {
|
|
return makeSessionKey(SourceOpenAI, p, p.Signals.PreviousResponseID)
|
|
}
|
|
if validToken(p.Signals.ConversationID, maxProtocolTokenLen) {
|
|
return makeSessionKey(SourceOpenAI, p, p.Signals.ConversationID)
|
|
}
|
|
}
|
|
|
|
// 3. 身份级回退：tenant:user:apiKey:model 四元组
|
|
ident := fmt.Sprintf("%d:%d:%d:%s", p.TenantID, p.UserID, p.APIKeyID, p.Model)
|
|
return makeSessionKey(SourceIdentity, p, ident)
|
|
}
|
|
|
|
// makeSessionKey 生成最终会话键。哈希材料带上租户与模型做命名空间隔离：
|
|
// 不同租户携带相同 X-Session-Id 不会互相碰撞；同一会话在不同模型上的绑定互相独立。
|
|
func makeSessionKey(src SessionSource, p RequestProfile, raw string) SessionKey {
|
|
sum := sha256.Sum256(fmt.Appendf(nil, "%d|%s|%s|%s", p.TenantID, p.Model, src, raw))
|
|
return SessionKey{
|
|
Source: src,
|
|
Key: "sk:" + string(src) + ":" + hex.EncodeToString(sum[:]),
|
|
}
|
|
}
|
|
|
|
// validToken 校验会话信号：非空、长度受限、不含控制字符。
|
|
// 非法值直接跳过（落入下一级信号），不记录原值。
|
|
func validToken(s string, maxLen int) bool {
|
|
if s == "" || len(s) > maxLen {
|
|
return false
|
|
}
|
|
for i := 0; i < len(s); i++ {
|
|
c := s[i]
|
|
if c < 0x20 || c == 0x7f {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|