5906f9ab98
## 数据完整性修复 (fix-critical-data-integrity) - 添加 error_translate.go 统一错误转换层 - 修复 nil 输入和 NotFound 错误处理 - 增强仓储层错误一致性 ## 仓储一致性修复 (fix-high-repository-consistency) - Group schema 添加 default_validity_days 字段 - Account schema 添加 proxy edge 关联 - 新增 UsageLog ent schema 定义 - 修复 UpdateBalance/UpdateConcurrency 受影响行数校验 ## 数据卫生修复 (fix-medium-data-hygiene) - UserSubscription 添加软删除支持 (SoftDeleteMixin) - RedeemCode/Setting 添加硬删除策略文档 - account_groups/user_allowed_groups 的 created_at 声明 timestamptz - 停止写入 legacy users.allowed_groups 列 - 新增迁移: 011-014 (索引优化、软删除、孤立数据审计、列清理) ## 测试补充 - 添加 UserSubscription 软删除测试 - 添加迁移回归测试 - 添加 NotFound 错误测试 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
98 lines
3.3 KiB
Go
98 lines
3.3 KiB
Go
package repository
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"errors"
|
||
"strings"
|
||
|
||
dbent "github.com/Wei-Shaw/sub2api/ent"
|
||
infraerrors "github.com/Wei-Shaw/sub2api/internal/infrastructure/errors"
|
||
"github.com/lib/pq"
|
||
)
|
||
|
||
// clientFromContext 从 context 中获取事务 client，如果不存在则返回默认 client。
|
||
//
|
||
// 这个辅助函数支持 repository 方法在事务上下文中工作：
|
||
// - 如果 context 中存在事务（通过 ent.NewTxContext 设置），返回事务的 client
|
||
// - 否则返回传入的默认 client
|
||
//
|
||
// 使用示例：
|
||
//
|
||
// func (r *someRepo) SomeMethod(ctx context.Context) error {
|
||
// client := clientFromContext(ctx, r.client)
|
||
// return client.SomeEntity.Create().Save(ctx)
|
||
// }
|
||
func clientFromContext(ctx context.Context, defaultClient *dbent.Client) *dbent.Client {
|
||
if tx := dbent.TxFromContext(ctx); tx != nil {
|
||
return tx.Client()
|
||
}
|
||
return defaultClient
|
||
}
|
||
|
||
// translatePersistenceError 将数据库层错误翻译为业务层错误。
|
||
//
|
||
// 这是 Repository 层的核心错误处理函数，确保数据库细节不会泄露到业务层。
|
||
// 通过统一的错误翻译，业务层可以使用语义明确的错误类型（如 ErrUserNotFound）
|
||
// 而不是依赖于特定数据库的错误（如 sql.ErrNoRows）。
|
||
//
|
||
// 参数：
|
||
// - err: 原始数据库错误
|
||
// - notFound: 当记录不存在时返回的业务错误（可为 nil 表示不处理）
|
||
// - conflict: 当违反唯一约束时返回的业务错误（可为 nil 表示不处理）
|
||
//
|
||
// 返回：
|
||
// - 翻译后的业务错误，或原始错误（如果不匹配任何规则）
|
||
//
|
||
// 示例：
|
||
//
|
||
// err := translatePersistenceError(dbErr, service.ErrUserNotFound, service.ErrEmailExists)
|
||
func translatePersistenceError(err error, notFound, conflict *infraerrors.ApplicationError) error {
|
||
if err == nil {
|
||
return nil
|
||
}
|
||
|
||
// 兼容 Ent ORM 和标准 database/sql 的 NotFound 行为。
|
||
// Ent 使用自定义的 NotFoundError，而标准库使用 sql.ErrNoRows。
|
||
// 这里同时处理两种情况，保持业务错误映射一致。
|
||
if notFound != nil && (errors.Is(err, sql.ErrNoRows) || dbent.IsNotFound(err)) {
|
||
return notFound.WithCause(err)
|
||
}
|
||
|
||
// 处理唯一约束冲突（如邮箱已存在、名称重复等）
|
||
if conflict != nil && isUniqueConstraintViolation(err) {
|
||
return conflict.WithCause(err)
|
||
}
|
||
|
||
// 未匹配任何规则，返回原始错误
|
||
return err
|
||
}
|
||
|
||
// isUniqueConstraintViolation 判断错误是否为唯一约束冲突。
|
||
//
|
||
// 支持多种检测方式：
|
||
// 1. PostgreSQL 特定错误码 23505（唯一约束冲突）
|
||
// 2. 错误消息中包含的通用关键词
|
||
//
|
||
// 这种多层次的检测确保了对不同数据库驱动和 ORM 的兼容性。
|
||
func isUniqueConstraintViolation(err error) bool {
|
||
if err == nil {
|
||
return false
|
||
}
|
||
|
||
// 优先检测 PostgreSQL 特定错误码（最精确）。
|
||
// 错误码 23505 对应 unique_violation。
|
||
// 参考：https://www.postgresql.org/docs/current/errcodes-appendix.html
|
||
var pgErr *pq.Error
|
||
if errors.As(err, &pgErr) {
|
||
return pgErr.Code == "23505"
|
||
}
|
||
|
||
// 回退到错误消息检测（兼容其他场景）。
|
||
// 这些关键词覆盖了 PostgreSQL、MySQL 等主流数据库的错误消息。
|
||
msg := strings.ToLower(err.Error())
|
||
return strings.Contains(msg, "duplicate key") ||
|
||
strings.Contains(msg, "unique constraint") ||
|
||
strings.Contains(msg, "duplicate entry")
|
||
}
|