主人 2026-10-07 裁定:按最优解加全局限流。 - 新增 t1RateGate(滑动窗口、可注入时钟,~70 行);默认 120 次 / 60 分钟(现网 700-1000 条新记忆/天 ≈ 25-42/小时, 取 120 留约 3-5 倍余量) - 过闸摆在「t1Enabled + 有灰区候选」两个守卫之后、发请求之前 ⇒ 不调模型就不占额度;放行才记账 - 超限语义与既有失败路径同源:跳过 T1、决定仍是 create、零改写, 只多一个 reason=t1_throttled 进留痕 + ReviewStats.T1Throttled 计数 - 配置键 t1PerWindow / t1WindowSec 放 $DATA_DIR/config/memory-tidy.json (零新 env 键,启动期读一次);坏配置回落默认 + WARN,不启动失败 - 每窗口只记一条 WARN(超限是预期行为,一天几百条会淹掉真故障) - IngestRuleVersion 仍为 d3.2:判定规则集与解析契约没变,变的只是 额度状态,reason 列已可分辨(升版会有 §17.6 记过的幂等键重复行副作用) 自研而非复用 session.NewRateLimiter 的理由(design/12 §N C.5 自己登记过 这条取舍):既有件 Allow() 内部写死 time.Now(),没有时钟注入缝, 而本切片必须能注入可控时钟;且复用会把 channel 层概念拖进 memory 包。
214 lines
8.0 KiB
Go
214 lines
8.0 KiB
Go
package runtimeconfig
|
||
|
||
import (
|
||
"errors"
|
||
"os"
|
||
"path/filepath"
|
||
"testing"
|
||
)
|
||
|
||
// 本文件锁定 `$DATA_DIR/config/memory-tidy.json`(memorytidy.go)的读取口径:
|
||
//
|
||
// ① 文件缺失 / 空白 ⇒ 零值 + nil(「还没配过」不是错误);
|
||
// ② `autoApply` **没写** ⇒ 不覆盖(AutoApplyOff=false)—— 这是指针存在的理由:
|
||
// 「没配」与「显式配成 false」必须可区分,否则只写上限的文件会静默关掉自动档;
|
||
// ③ `autoApply: false` ⇒ AutoApplyOff=true(一键回到只产建议);
|
||
// ④ 文件存在但解不开 ⇒ 分类错误(调用方据此记 WARN 并回落代码默认值);
|
||
// ⑤ 落点常量与 backthink 那一族同一目录、不同文件名(不共用常量,避免静默搬家)。
|
||
|
||
func TestLoadMemoryTidyTuning_MissingIsNotAnError(t *testing.T) {
|
||
t.Parallel()
|
||
path := filepath.Join(t.TempDir(), "config", "memory-tidy.json")
|
||
got, err := LoadMemoryTidyTuning(path)
|
||
if err != nil {
|
||
t.Fatalf("缺失文件的 err = %v, want nil", err)
|
||
}
|
||
if got.AutoApplyOff() || got.IngestGateOff() || got.AutoApplyPerRound != 0 {
|
||
t.Fatalf("缺失文件应当给零值:%+v", got)
|
||
}
|
||
}
|
||
|
||
func TestLoadMemoryTidyTuning_BlankIsNotAnError(t *testing.T) {
|
||
t.Parallel()
|
||
path := writeMemoryTidyFile(t, " \n\t")
|
||
got, err := LoadMemoryTidyTuning(path)
|
||
if err != nil {
|
||
t.Fatalf("空白文件的 err = %v, want nil", err)
|
||
}
|
||
if got.AutoApplyOff() {
|
||
t.Fatal("空白文件不该被读成「关掉自动档」")
|
||
}
|
||
}
|
||
|
||
func TestLoadMemoryTidyTuning_AbsentAutoApplyKeepsDefault(t *testing.T) {
|
||
t.Parallel()
|
||
path := writeMemoryTidyFile(t, `{"autoApplyPerRound": 7}`)
|
||
got, err := LoadMemoryTidyTuning(path)
|
||
if err != nil {
|
||
t.Fatalf("LoadMemoryTidyTuning: %v", err)
|
||
}
|
||
if got.AutoApply != nil {
|
||
t.Fatalf("没写 autoApply ⇒ 指针应为 nil,实际 %v", *got.AutoApply)
|
||
}
|
||
if got.AutoApplyOff() {
|
||
t.Fatal("只写了上限的文件不得把自动档关掉(代码默认是开)")
|
||
}
|
||
if got.AutoApplyPerRound != 7 {
|
||
t.Fatalf("autoApplyPerRound = %d, want 7", got.AutoApplyPerRound)
|
||
}
|
||
}
|
||
|
||
func TestLoadMemoryTidyTuning_ExplicitFalseTurnsAutoApplyOff(t *testing.T) {
|
||
t.Parallel()
|
||
path := writeMemoryTidyFile(t, `{"autoApply": false}`)
|
||
got, err := LoadMemoryTidyTuning(path)
|
||
if err != nil {
|
||
t.Fatalf("LoadMemoryTidyTuning: %v", err)
|
||
}
|
||
if !got.AutoApplyOff() {
|
||
t.Fatal("显式 false 必须被读成「关掉自动档」")
|
||
}
|
||
|
||
on := writeMemoryTidyFile(t, `{"autoApply": true, "autoApplyPerRound": 3}`)
|
||
got, err = LoadMemoryTidyTuning(on)
|
||
if err != nil {
|
||
t.Fatalf("LoadMemoryTidyTuning(on): %v", err)
|
||
}
|
||
if got.AutoApplyOff() || got.AutoApplyPerRound != 3 {
|
||
t.Fatalf("显式 true 的回读 = %+v", got)
|
||
}
|
||
}
|
||
|
||
func TestLoadMemoryTidyTuning_BrokenIsClassifiedError(t *testing.T) {
|
||
t.Parallel()
|
||
path := writeMemoryTidyFile(t, `{"autoApply": "yes"}`)
|
||
if _, err := LoadMemoryTidyTuning(path); !errors.Is(err, ErrInvalidMemoryTidy) {
|
||
t.Fatalf("坏配置的 err = %v, want ErrInvalidMemoryTidy", err)
|
||
}
|
||
}
|
||
|
||
// TestLoadMemoryTidyTuning_IngestGate 钉住**同步入库闸**的开关口径
|
||
// (2026-10-07 主人裁定):它与自动档同一个文件、同一个「指针 = 没配」语义,
|
||
// 且**零新 env 键**(`.env` 已冻结,design/13 §2.1)。
|
||
func TestLoadMemoryTidyTuning_IngestGate(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
// ① 没写 ingestGate ⇒ 不覆盖(代码默认是「开」)。
|
||
absent := writeMemoryTidyFile(t, `{"autoApplyPerRound": 3}`)
|
||
got, err := LoadMemoryTidyTuning(absent)
|
||
if err != nil {
|
||
t.Fatalf("LoadMemoryTidyTuning: %v", err)
|
||
}
|
||
if got.IngestGate != nil || got.IngestGateOff() {
|
||
t.Fatalf("没写 ingestGate 不得把闸关掉:%+v", got)
|
||
}
|
||
|
||
// ② 显式 false ⇒ 一键回到「只插不审」。
|
||
off := writeMemoryTidyFile(t, `{"ingestGate": false}`)
|
||
got, err = LoadMemoryTidyTuning(off)
|
||
if err != nil {
|
||
t.Fatalf("LoadMemoryTidyTuning(off): %v", err)
|
||
}
|
||
if !got.IngestGateOff() {
|
||
t.Fatal("显式 false 必须被读成「关掉入库闸」")
|
||
}
|
||
// 闸的开关与自动档的开关互不影响:一个键不能顺手改另一件事。
|
||
if got.AutoApplyOff() {
|
||
t.Fatal("写 ingestGate 不得顺手把自动档也关掉")
|
||
}
|
||
|
||
// ③ 显式 true ⇒ 与默认同义(写它只是为了自解释)。
|
||
on := writeMemoryTidyFile(t, `{"ingestGate": true, "autoApply": false}`)
|
||
got, err = LoadMemoryTidyTuning(on)
|
||
if err != nil {
|
||
t.Fatalf("LoadMemoryTidyTuning(on): %v", err)
|
||
}
|
||
if got.IngestGateOff() || !got.AutoApplyOff() {
|
||
t.Fatalf("两个开关必须各自独立:%+v", got)
|
||
}
|
||
|
||
// ④ 类型错 ⇒ 分类错误(调用方据此记 WARN 并回落代码默认值,绝不静默)。
|
||
broken := writeMemoryTidyFile(t, `{"ingestGate": "nope"}`)
|
||
if _, err := LoadMemoryTidyTuning(broken); !errors.Is(err, ErrInvalidMemoryTidy) {
|
||
t.Fatalf("坏 ingestGate 的 err = %v, want ErrInvalidMemoryTidy", err)
|
||
}
|
||
}
|
||
|
||
// TestLoadMemoryTidyTuning_T1RateLimit 钉住 **T1 全局限速**的两个数值键
|
||
// (design/23 §15.4,主人 2026-10-07 裁定):它们与 `t1Judge` 同一个文件、
|
||
// **零新 env 键**,且**零值只能是「没配」**——绝不能是「无上限」。
|
||
func TestLoadMemoryTidyTuning_T1RateLimit(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
// ① 没写 ⇒ 保持零值(调用方据此回落代码默认额,不是「不限流」)。
|
||
absent := writeMemoryTidyFile(t, `{"t1Judge": true}`)
|
||
got, err := LoadMemoryTidyTuning(absent)
|
||
if err != nil {
|
||
t.Fatalf("LoadMemoryTidyTuning: %v", err)
|
||
}
|
||
if got.T1PerWindow != 0 || got.T1WindowSec != 0 {
|
||
t.Fatalf("没写限速键应当给零值:%+v", got)
|
||
}
|
||
|
||
// ② 写了 ⇒ 如实回读,且不连带改动别的键(一个键不能顺手关掉另一件事)。
|
||
set := writeMemoryTidyFile(t, `{"t1PerWindow": 240, "t1WindowSec": 1800}`)
|
||
got, err = LoadMemoryTidyTuning(set)
|
||
if err != nil {
|
||
t.Fatalf("LoadMemoryTidyTuning(set): %v", err)
|
||
}
|
||
if got.T1PerWindow != 240 || got.T1WindowSec != 1800 {
|
||
t.Fatalf("限速键回读 = %d/%d, want 240/1800", got.T1PerWindow, got.T1WindowSec)
|
||
}
|
||
if got.T1JudgeOff() || got.IngestGateOff() || got.AutoApplyOff() {
|
||
t.Fatalf("限速键顺手改了别的开关:%+v", got)
|
||
}
|
||
|
||
// ③ 显式 0 ⇒ 仍是零值(「没法用 0 关掉限流」是刻意的:那会是一处静默的失效面)。
|
||
zero := writeMemoryTidyFile(t, `{"t1PerWindow": 0, "t1WindowSec": 0}`)
|
||
got, err = LoadMemoryTidyTuning(zero)
|
||
if err != nil {
|
||
t.Fatalf("LoadMemoryTidyTuning(zero): %v", err)
|
||
}
|
||
if got.T1PerWindow != 0 || got.T1WindowSec != 0 {
|
||
t.Fatalf("显式 0 的回读 = %+v", got)
|
||
}
|
||
|
||
// ④ 类型错 ⇒ 分类错误(调用方据此记 WARN 并回落代码默认额,绝不静默不限流)。
|
||
for _, body := range []string{`{"t1PerWindow": "many"}`, `{"t1WindowSec": "1h"}`} {
|
||
broken := writeMemoryTidyFile(t, body)
|
||
if _, err := LoadMemoryTidyTuning(broken); !errors.Is(err, ErrInvalidMemoryTidy) {
|
||
t.Fatalf("坏配置 %s 的 err = %v, want ErrInvalidMemoryTidy", body, err)
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestDefaultMemoryTidyPathIsDataDirConfig(t *testing.T) {
|
||
t.Parallel()
|
||
got := DefaultMemoryTidyPath(filepath.Join("root", "data"))
|
||
want := filepath.Join("root", "data", "config", "memory-tidy.json")
|
||
if got != want {
|
||
t.Fatalf("落点 = %q, want %q", got, want)
|
||
}
|
||
if MemoryTidyConfigRelPath != "config/memory-tidy.json" {
|
||
t.Fatalf("相对路径 = %q, want config/memory-tidy.json", MemoryTidyConfigRelPath)
|
||
}
|
||
// 与 backthink 那一族同目录、不同文件:共用常量会在某一族搬家时静默带走另一族。
|
||
if filepath.Dir(MemoryTidyConfigRelPath) != filepath.Dir(BackthinkConfigRelPath) {
|
||
t.Fatalf("两族应当同目录:%q vs %q", MemoryTidyConfigRelPath, BackthinkConfigRelPath)
|
||
}
|
||
}
|
||
|
||
// writeMemoryTidyFile 在临时目录里写一份 memory-tidy.json 并返回路径。
|
||
func writeMemoryTidyFile(t *testing.T, body string) string {
|
||
t.Helper()
|
||
dir := filepath.Join(t.TempDir(), "config")
|
||
if err := os.MkdirAll(dir, 0o700); err != nil {
|
||
t.Fatalf("mkdir: %v", err)
|
||
}
|
||
path := filepath.Join(dir, memoryTidyFileName)
|
||
if err := os.WriteFile(path, []byte(body), 0o600); err != nil {
|
||
t.Fatalf("write: %v", err)
|
||
}
|
||
return path
|
||
}
|