feat: 品牌重命名 Sub2API -> TianShuAPI
- 前端: 所有界面显示、i18n 文本、组件中的品牌名称 - 后端: 服务层、设置默认值、邮件模板、安装向导 - 数据库: 迁移脚本注释 - 保持功能完全一致，仅更改品牌名称 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
huangzhenpc
@@ -1,97 +1,97 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"errors"
|
||||
"strings"
|
||||
|
||||
dbent "github.com/Wei-Shaw/sub2api/ent"
|
||||
infraerrors "github.com/Wei-Shaw/sub2api/internal/pkg/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")
|
||||
}
|
||||
package repository
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"errors"
|
||||
"strings"
|
||||
|
||||
dbent "github.com/Wei-Shaw/sub2api/ent"
|
||||
infraerrors "github.com/Wei-Shaw/sub2api/internal/pkg/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")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user