杭州明婳科技
6595e94677
经 deepseek/GLM/Claude 三轮对抗性评审 + 交叉核验 + 终审（见 last_report.md、cross_review_assessment.md），逐条回读源码核实并修复。 go build / vet / test -race ./... 全绿。 安全（P0）： - JWT alg confusion：WithValidMethods 固定 HMAC + keyfunc 方法断言 - JWT 空密钥 / 不支持算法 fail-closed（ErrEmptySecret / ErrUnsupportedAlgorithm） - 上传大小实测封顶（enforceUploadSize），不信任客户端 file.Size - HTTP headers/cookies map 竞态（写锁+快照）+ SSRF 防护（opt-in dialer.Control） 并发/资源（主线A 闭合）： - 包级可变全局一律 atomic.Pointer：DefaultRedis / DefaultStorage / DefaultManager / Validator / DefaultCache 等，消除裸指针数据竞争 - trace.Close 去 sync.Once（防 exporter 泄漏）；app OnReady 失败走 Shutdown - ws.Hub.Stop stopOnce 防 double-close；cron 接入 App 生命周期（WithCron） - ratelimit failClosed atomic.Bool；Recover Written() 守卫 - cron checkAndRun 锁内收集锁外 spawn（wg.Add 在锁内防 Stop 竞态） 终审剩余项收口（H-A~M-H + L 系列）： - 副本连接池 MaxOpenConns/2 截断修复（replicaMaxOpenConns） - JWT 黑名单 1s 超时 + 显式错误；ratelimit GetRedis 取一次复用 - GetPage 深分页上限；HashFile 流式；ReadFile 去 TOCTOU - config Clone() 深拷贝；cache 锁操作返 ErrRedisNotReady - compress 解压残留清理；正则预编译；EqualsIgnoreCase→EqualFold - 删 redisLimiters 死代码；CLI 输入校验 + 回滚 文档： - README/GUIDE/CHANGELOG 对齐当前 API（含破坏性变更迁移说明） - docs/README 索引修正；config 注释修正（RS256 已不支持） 破坏性变更详见 CHANGELOG.md [Unreleased]。项目处于研发初期、无下游用户。 Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
39 lines
1.8 KiB
Go
39 lines
1.8 KiB
Go
package database
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
)
|
|
|
|
func TestIsTransientDBError(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
err error
|
|
want bool
|
|
}{
|
|
{"nil", nil, true},
|
|
{"access denied", errors.New("Error 1045: Access denied for user 'root'@'localhost'"), false},
|
|
{"auth plugin", errors.New("authentication plugin 'caching_sha2_password' cannot be loaded"), false},
|
|
{"unknown database", errors.New("Error 1049: Unknown database 'foo'"), false},
|
|
{"invalid DSN", errors.New("invalid DSN: missing the slash separating the database name"), false},
|
|
{"unknown driver", errors.New("sql: unknown driver \"foobar\" (forgotten import?)"), false},
|
|
{"connection refused (transient)", errors.New("dial tcp 127.0.0.1:3306: connect: connection refused"), true},
|
|
{"i/o timeout (transient)", errors.New("dial tcp 10.0.0.1:3306: i/o timeout"), true},
|
|
{"empty msg", errors.New(""), true},
|
|
// P1 #12：PG 目标库不存在应视为非瞬态（不重试）。
|
|
{"pg database not exist", errors.New(`FATAL: database "app" does not exist (SQLSTATE 3D000)`), false},
|
|
{"pg password auth failed", errors.New("FATAL: password authentication failed for user \"app\""), false},
|
|
// P1 #12 回归核心：含 "database" 但是瞬态的错误必须仍被判为瞬态（可重试），
|
|
// 修复前独立 "database" 子串会把它误判为非瞬态而放弃重试。
|
|
{"pg starting up (transient)", errors.New("FATAL: the database system is starting up (SQLSTATE 57P03)"), true},
|
|
{"pg in recovery (transient)", errors.New("FATAL: the database system is in recovery mode"), true},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := isTransientDBError(tt.err); got != tt.want {
|
|
t.Errorf("isTransientDBError(%v) = %v, want %v", tt.err, got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|