activity_repo.go 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452
  1. // 活动仓储(t_courses_activity / t_courses_activity_signed)——AC-06 活动报名主链路。
  2. //
  3. // 依据:unified/03-migration/academic-domain-design.md §3.2(活动 9 SP)/ §1.5(幂等语义演进)
  4. // / §1.7(scoping 与两条例外)/ §10.3(事务边界只在 Service)。
  5. //
  6. // 覆盖 SP 映射:
  7. //
  8. // signed_add → UpsertSigned(DELETE+INSERT 幂等,对齐 dump)
  9. // signed_cancel → UpdateSignedStatus + IncreaseNumberSign(服务层组合成状态机)
  10. // sign 限一次(3106) → CountSignedByUser(type4Only=true)
  11. // 退费回滚赠送金额 → ListSignedByStudentCourse(按报课记录定位报名行)
  12. //
  13. // 作用域(关键设计,与 BaseRepo.Scoped 的差异必须读):
  14. //
  15. // t_courses_activity.campus_id = 0 的语义是「本机构所有**自营**校区」(DDL §2「7.」+ 设计 §11-2),
  16. // 因此校园维度过滤**不能**用 `campus_id = ?`,否则全机构活动会被裁掉;
  17. // 本仓储统一用 scopedCampusRows:`institution_id = ? AND campus_id IN (?, 0)`(超集语义)。
  18. // → BaseRepo.CampusScoped 置 false(避免严格等值过滤),机构过滤仍由 scopedCampusRows 强制,
  19. // 安全红线(跨机构越权)不因这条例外而放松。
  20. //
  21. // 事务纪律:写方法一律接收 `tx *gorm.DB`(**仓储不开事务**);读方法经 db(ctx) 取句柄,
  22. // ctx 内已注入事务(WithTxContext)时自动复用同一 tx,保证事务内读一致。
  23. package repository
  24. import (
  25. "context"
  26. "errors"
  27. "fmt"
  28. "strings"
  29. "gorm.io/gorm"
  30. "yuansiwei/unified-backend/internal/academic/model"
  31. infraerrors "yuansiwei/unified-backend/internal/infra/errors"
  32. appscope "yuansiwei/unified-backend/internal/infra/scope"
  33. )
  34. // 游标方向常量(源自 dump 的固定排序,调用方只能二选一,禁止拼接其它 SQL 片段)。
  35. //
  36. // 语义(dump 口径,**顺序不可改**):
  37. // - OrderAsc :按 id 升序——"消费/使用"路径(先入先出,旧 SP 游标);
  38. // - OrderDesc:按 id 降序——"退回/回滚"路径(后入先退,旧 SP 游标)。
  39. //
  40. // 安全红线:方向值来自本白名单,绝不接受调用方透传的原始字符串(防 ORDER BY 注入)。
  41. const (
  42. // OrderAsc 升序(使用路径)。
  43. OrderAsc = "asc"
  44. // OrderDesc 降序(退回路径)。
  45. OrderDesc = "desc"
  46. )
  47. // normalizeOrder 校验并归一排序方向(白名单;空串按 OrderAsc 处理)。
  48. //
  49. // 入参:order string——调用方传入的方向(大小写/空白不敏感)。
  50. // 出参:归一后的 "asc"/"desc";非法值 → infraerrors.ErrBadParam(2002)。
  51. func normalizeOrder(order string) (string, error) {
  52. switch strings.ToLower(strings.TrimSpace(order)) {
  53. case "", OrderAsc:
  54. return OrderAsc, nil
  55. case OrderDesc:
  56. return OrderDesc, nil
  57. default:
  58. return "", infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: 非法排序方向 "+order))
  59. }
  60. }
  61. // scopedCampusRows 追加作用域过滤(机构强制 + 校区"含全机构池"超集)。
  62. //
  63. // 入参:
  64. // - db *gorm.DB:基础查询(nil → 返回 nil)。
  65. // - sc *appscope.Scope:请求作用域(nil → panic fail-fast,与 BaseRepo.Scoped 同口径)。
  66. // - alias string:表别名前缀(如 "signed.");JOIN 场景必须限定,否则 ambiguous column。
  67. //
  68. // 出参:*gorm.DB——追加过滤后的链式查询。
  69. //
  70. // 过滤规则:
  71. // 1. sc.SuperAdmin=true → 不加任何过滤(平台超管跨机构视角);
  72. // 2. 否则强制 `alias.institution_id = sc.InstitutionID`(安全红线,不可绕过);
  73. // 3. 仅当 sc.CampusID > 0 时追加 `alias.campus_id IN (sc.CampusID, 0)`——
  74. // 0=机构所有自营校区,属该校区可见范围(设计 §11-2)。
  75. func scopedCampusRows(db *gorm.DB, sc *appscope.Scope, alias string) *gorm.DB {
  76. if db == nil {
  77. return nil
  78. }
  79. if sc == nil {
  80. panic(errNoScope.Error())
  81. }
  82. if sc.SuperAdmin {
  83. return db
  84. }
  85. col := func(name string) string {
  86. if alias == "" {
  87. return name
  88. }
  89. return alias + "." + name
  90. }
  91. db = db.Where(col("institution_id")+" = ?", sc.InstitutionID)
  92. if sc.CampusID > 0 {
  93. // campus_id=0 = 机构所有自营校区(超集语义,不能用等值)
  94. db = db.Where(col("campus_id")+" IN ?", []int64{sc.CampusID, 0})
  95. }
  96. return db
  97. }
  98. // requireTx 校验写方法必须携带事务句柄(仓储不开事务,设计 §10.3)。
  99. //
  100. // 入参:tx *gorm.DB(可 nil)。出参:nil 合法;ErrInternal 表示调用方漏传事务。
  101. func requireTx(tx *gorm.DB, op string) error {
  102. if tx == nil {
  103. return infraerrors.WithCause(infraerrors.ErrInternal, errString("repository: "+op+" 必须传入 tx(禁止脱离事务写资金/状态)"))
  104. }
  105. return nil
  106. }
  107. // ActivityRepo 活动方案 + 报名仓储。
  108. //
  109. // 表:t_courses_activity(活动方案)/ t_courses_activity_signed(报名·UK(user_id, activity_id))。
  110. // 作用域:两表均带 institution_id + campus_id(冗余作用域列)→ 统一走 scopedCampusRows。
  111. type ActivityRepo struct {
  112. *BaseRepo
  113. }
  114. // NewActivityRepo 构造活动仓储。
  115. //
  116. // 入参:db *gorm.DB(连接池句柄)/ sc *appscope.Scope(请求作用域;系统任务可 nil,一查询即 panic)。
  117. // 出参:*ActivityRepo。
  118. //
  119. // 说明:CampusScoped 置 false —— 校区维度由 scopedCampusRows 以 `IN (campus,0)` 超集语义处理
  120. // (见文件头「作用域」段),机构过滤不受影响。
  121. func NewActivityRepo(db *gorm.DB, sc *appscope.Scope) *ActivityRepo {
  122. r := &ActivityRepo{BaseRepo: NewBaseRepo(db, sc)}
  123. r.CampusScoped = false
  124. return r
  125. }
  126. // db 取查询句柄(ctx 内含 tx 时复用同一事务,保证事务内读一致)。
  127. //
  128. // 入参:ctx context.Context(可 nil)。出参:(*gorm.DB, error);未装配 DB → ErrInternal。
  129. func (r *ActivityRepo) db(ctx context.Context) (*gorm.DB, error) {
  130. if r == nil || r.DB == nil {
  131. return nil, infraerrors.WithCause(infraerrors.ErrInternal, errString("repository: ActivityRepo.DB 为 nil"))
  132. }
  133. if tx := TxFromContext(ctx); tx != nil {
  134. return tx, nil
  135. }
  136. if ctx == nil {
  137. return r.DB, nil
  138. }
  139. return r.DB.WithContext(ctx), nil
  140. }
  141. // GetByID 按主键取单个活动(**作用域强制**:跨机构活动一律 ErrNotFound,不泄露存在性)。
  142. //
  143. // 入参:
  144. // - ctx 请求上下文(取消/超时传播);
  145. // - id 活动 ID(<=0 → ErrBadParam,避免全表扫描)。
  146. //
  147. // 出参:
  148. // - *model.CoursesActivity:命中行(含 money 字符串原文,金额换算由 Service 用 money.Amount);
  149. // - ErrNotFound(3001):不存在或不在作用域内;
  150. // - ErrDBError(5002):SQL/驱动异常(不返回 gorm 原错)。
  151. func (r *ActivityRepo) GetByID(ctx context.Context, id int64) (*model.CoursesActivity, error) {
  152. if id <= 0 {
  153. return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: activity id 必须 > 0"))
  154. }
  155. db, err := r.db(ctx)
  156. if err != nil {
  157. return nil, err
  158. }
  159. var out model.CoursesActivity
  160. q := scopedCampusRows(db.Model(&model.CoursesActivity{}), r.Sc, "")
  161. if err := q.Where("id = ?", id).First(&out).Error; err != nil {
  162. return nil, wrapQueryErr(err, "GetByID(t_courses_activity)")
  163. }
  164. return &out, nil
  165. }
  166. // GetByIDs 批量取活动(**防 N+1**:一次 IN 查询,返回 map 供调用方 O(1) 索引)。
  167. //
  168. // 入参:
  169. // - ctx 请求上下文;
  170. // - ids 活动 ID 集合(空/全非法 → 返回空 map 且**不发 SQL**)。
  171. //
  172. // 出参:
  173. // - map[int64]*model.CoursesActivity:命中行索引(**缺失的 id 不在 map 中**,调用方按需判缺失,
  174. // 语义上等价"不在作用域内或不存在",禁止据此区分存在性);
  175. // - ErrDBError(5002)。
  176. //
  177. // 实现说明:入参先去重 + 剔除 <=0(防无意义扫描);结果不保证顺序(map 语义)。
  178. func (r *ActivityRepo) GetByIDs(ctx context.Context, ids []int64) (map[int64]*model.CoursesActivity, error) {
  179. out := make(map[int64]*model.CoursesActivity, len(ids))
  180. clean := make([]int64, 0, len(ids))
  181. seen := make(map[int64]struct{}, len(ids))
  182. for _, id := range ids {
  183. if id <= 0 {
  184. continue
  185. }
  186. if _, ok := seen[id]; ok {
  187. continue
  188. }
  189. seen[id] = struct{}{}
  190. clean = append(clean, id)
  191. }
  192. if len(clean) == 0 {
  193. return out, nil
  194. }
  195. db, err := r.db(ctx)
  196. if err != nil {
  197. return nil, err
  198. }
  199. var rows []model.CoursesActivity
  200. q := scopedCampusRows(db.Model(&model.CoursesActivity{}), r.Sc, "")
  201. if err := q.Where("id IN ?", clean).Find(&rows).Error; err != nil {
  202. return nil, infraerrors.WithCause(infraerrors.ErrDBError, err)
  203. }
  204. for i := range rows {
  205. out[rows[i].ID] = &rows[i]
  206. }
  207. return out, nil
  208. }
  209. // UpsertSigned 报名落库(**DELETE+INSERT 幂等**,对齐 dump `signed_add`)。
  210. //
  211. // 依据:dump 的 signed_add 先 `DELETE FROM t_courses_activity_signed WHERE userid=? AND activityid=?`
  212. // 再 INSERT——同一 (user, activity) 只留一条,连历史 status=0 的旧行一并清掉(设计 §1.5)。
  213. // 目标态 UK(user_id, activity_id) 兜底;此处保留 DELETE+INSERT 语义(而非 OnConflict 覆盖),
  214. // 因为状态机要求"重新报名 = 全字段重置"(activity_money/recharge_amount/审核字段清零)。
  215. //
  216. // 入参:
  217. // - ctx 请求上下文;
  218. // - tx 事务句柄(**必传**,nil → ErrInternal);
  219. // - signed 报名行(ActivityID>0、UserID 非空;institution_id 为 0 时用请求作用域补齐)。
  220. //
  221. // 出参:nil 成功;ErrBadParam(参数非法)/ ErrDBError(SQL 异常,事务由调用方回滚)。
  222. //
  223. // 事务/一致性(红线):
  224. // - DELETE **不叠作用域**:UK(user_id, activity_id) 全局唯一,若历史迁移行的作用域列脏,
  225. // 叠作用域会漏删 → INSERT 撞 UK 直接失败;activity_id 的归属已由调用方 GetByID 作用域校验;
  226. // - campus_id **不由本方法补齐**:它必须是活动上的 campus_id 快照(0 有"全自营校区"语义,
  227. // 用请求作用域值覆盖会篡改语义)→ 由 Service 从活动行复制后传入;
  228. // - 报名人数增减(IncreaseNumberSign)由 Service 在同一事务内配套调用。
  229. func (r *ActivityRepo) UpsertSigned(ctx context.Context, tx *gorm.DB, signed *model.CoursesActivitySigned) error {
  230. if err := requireTx(tx, "UpsertSigned"); err != nil {
  231. return err
  232. }
  233. if signed == nil {
  234. return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: signed 为 nil"))
  235. }
  236. if signed.ActivityID <= 0 || strings.TrimSpace(signed.UserID) == "" {
  237. return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: signed 缺少 activity_id/user_id"))
  238. }
  239. if signed.InstitutionID == 0 {
  240. inst, _ := r.GetScopedID()
  241. signed.InstitutionID = inst
  242. }
  243. // ① 幂等清旧行(dump:DELETE WHERE userid AND activityid)
  244. if err := tx.Where("user_id = ? AND activity_id = ?", signed.UserID, signed.ActivityID).
  245. Delete(&model.CoursesActivitySigned{}).Error; err != nil {
  246. return infraerrors.WithCause(infraerrors.ErrDBError, err)
  247. }
  248. // ② 插入新行(状态机全字段重置)
  249. if err := tx.WithContext(ctxOrBackground(ctx)).Create(signed).Error; err != nil {
  250. return infraerrors.WithCause(infraerrors.ErrDBError, err)
  251. }
  252. return nil
  253. }
  254. // CountSignedByUser 统计用户报名数(**type=4 老带新限一次判定** / 取消报名的守卫条件)。
  255. //
  256. // 入参:
  257. // - ctx 请求上下文;
  258. // - userID 报名用户(canonical_user_id ULID;空 → ErrBadParam);
  259. // - type4Only 判定口径(**两分支语义不同,禁止合并**):
  260. // true = 老带新限一次(3106):只统计老带新活动(activity.type=4)**且活动未停用**(activity.status<>0),
  261. // **不过滤报名行状态**——取消过的报名也算"参加过";
  262. // false = 取消报名前置守卫(signed_cancel):统计该用户**当前有效**报名(signed.status<>0),
  263. // 不限活动类型、不看活动状态。
  264. //
  265. // 出参:
  266. // - int64:命中条数(>0 即命中对应业务规则);
  267. // - ErrBadParam / ErrDBError。
  268. //
  269. // SQL 口径(**逐字对齐 dump**):
  270. //
  271. // type4Only=true(dump `yuansiwei_finance.sql:762947`,老带新限一次):
  272. // SELECT COUNT(1) FROM t_courses_activity_signed t1
  273. // INNER JOIN t_courses_activity t2 ON t1.activityid=t2.id AND t2.type=4
  274. // WHERE t1.userid=? AND t2.status != 0
  275. // → **判活动 t2.status**,且对报名 t1.status **不做过滤**(取消过也算参加过,防"取消后重报再领奖励");
  276. //
  277. // type4Only=false(signed_cancel 第一道守卫):
  278. // SELECT count(*) FROM t_courses_activity_signed WHERE user_id=? AND status <> 0
  279. // → 只看报名行是否仍然有效。
  280. //
  281. // 去重语义:跨活动去重由 UK(user_id, activity_id) 保证(一活动一行)。
  282. func (r *ActivityRepo) CountSignedByUser(ctx context.Context, userID string, type4Only bool) (int64, error) {
  283. if strings.TrimSpace(userID) == "" {
  284. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: CountSignedByUser 需要 user_id"))
  285. }
  286. db, err := r.db(ctx)
  287. if err != nil {
  288. return 0, err
  289. }
  290. q := db.Table("t_courses_activity_signed AS signed").
  291. Where("signed.user_id = ?", userID)
  292. if type4Only {
  293. // dump 口径:判**活动** status(停用=0 的活动不计),报名行状态不参与过滤
  294. q = q.Joins("JOIN t_courses_activity AS activity ON activity.id = signed.activity_id").
  295. Where("activity.type = ?", model.ActivityTypeRecommend).
  296. Where("activity.status <> ?", model.ActivityStatusDisabled)
  297. } else {
  298. // signed_cancel 守卫:只看报名行是否有效
  299. q = q.Where("signed.status <> ?", model.ActivitySignedCancelled)
  300. }
  301. q = scopedCampusRows(q, r.Sc, "signed")
  302. var count int64
  303. if err := q.Count(&count).Error; err != nil {
  304. return 0, infraerrors.WithCause(infraerrors.ErrDBError, err)
  305. }
  306. return count, nil
  307. }
  308. // UpdateSignedStatus 更新报名状态(状态机推进 / 取消报名)。
  309. //
  310. // 入参:
  311. // - ctx 请求上下文;
  312. // - tx 事务句柄(**必传**);
  313. // - userID 报名用户(ULID);activityID 活动 ID;status 目标状态(model.ActivitySigned*)。
  314. //
  315. // 出参:
  316. // - int64:受影响行数(0 = 该 (user, activity) 不存在或不在作用域内,Service 据此返回 3001);
  317. // - ErrBadParam / ErrDBError。
  318. //
  319. // 说明:状态机的**合法性判定**在 Service(如 2→1 仅取消路径、已审核不可取消 3107),
  320. // 仓储只做「条件更新」,不解释状态流转规则(避免业务规则沉淀到数据层)。
  321. func (r *ActivityRepo) UpdateSignedStatus(ctx context.Context, tx *gorm.DB, userID string, activityID int64, status int32) (int64, error) {
  322. if err := requireTx(tx, "UpdateSignedStatus"); err != nil {
  323. return 0, err
  324. }
  325. if strings.TrimSpace(userID) == "" || activityID <= 0 {
  326. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateSignedStatus 需要 user_id/activity_id"))
  327. }
  328. q := scopedCampusRows(tx.Model(&model.CoursesActivitySigned{}), r.Sc, "")
  329. res := q.Where("user_id = ? AND activity_id = ?", userID, activityID).
  330. UpdateColumn("status", status)
  331. if res.Error != nil {
  332. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  333. }
  334. return res.RowsAffected, nil
  335. }
  336. // IncreaseNumberSign 报名人数原子增减(**下限 0**,防取消竞态把人数减成负数)。
  337. //
  338. // 入参:
  339. // - ctx 请求上下文;
  340. // - tx 事务句柄(**必传**);
  341. // - activityID 活动 ID(<=0 → ErrBadParam);delta 增量(报名 +1 / 取消 -1;0 → 空操作)。
  342. //
  343. // 出参:
  344. // - int64:受影响行数(0 = 活动不存在/不在作用域内);
  345. // - ErrBadParam / ErrDBError。
  346. //
  347. // SQL(跨库可移植的下限保护,不用 GREATEST/MAX 以免方言差异):
  348. //
  349. // UPDATE t_courses_activity
  350. // SET number_sign = CASE WHEN number_sign + <delta> < 0 THEN 0 ELSE number_sign + <delta> END
  351. // WHERE id = ? [AND institution_id = ? AND campus_id IN (?,0)]
  352. //
  353. // 红线:与 UpsertSigned/UpdateSignedStatus **必须同事务**(否则出现"报名行在、人数没加"的脏读)。
  354. func (r *ActivityRepo) IncreaseNumberSign(ctx context.Context, tx *gorm.DB, activityID int64, delta int32) (int64, error) {
  355. if err := requireTx(tx, "IncreaseNumberSign"); err != nil {
  356. return 0, err
  357. }
  358. if activityID <= 0 {
  359. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: IncreaseNumberSign 需要 activity_id"))
  360. }
  361. if delta == 0 {
  362. return 0, nil
  363. }
  364. expr := gorm.Expr("CASE WHEN number_sign + ? < 0 THEN 0 ELSE number_sign + ? END", delta, delta)
  365. q := scopedCampusRows(tx.Model(&model.CoursesActivity{}), r.Sc, "")
  366. res := q.Where("id = ?", activityID).UpdateColumn("number_sign", expr)
  367. if res.Error != nil {
  368. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  369. }
  370. return res.RowsAffected, nil
  371. }
  372. // CancelSigned 取消报名并**归零赠送/活动金额**(dump `yuansiwei_finance.sql:763028`)。
  373. //
  374. // 入参:
  375. // - ctx 请求上下文;
  376. // - tx 事务句柄(**必传**:与报名人数增减、赠送池回滚同事务);
  377. // - userID 报名用户(ULID;空 → ErrBadParam);activityID 活动 ID(<=0 → ErrBadParam)。
  378. //
  379. // 出参:
  380. // - int64:受影响行数(0 = 该 (user, activity) 不存在或不在作用域内 → Service 返回 3001);
  381. // - ErrBadParam / ErrDBError。
  382. //
  383. // SQL(dump 原文:一条 UPDATE 同置三列,**不可拆成多次更新**):
  384. //
  385. // UPDATE t_courses_activity_signed
  386. // SET status = 0, activity_money = 0, activity_money_used = 0
  387. // WHERE user_id = ? AND activity_id = ? [AND institution_id = ? AND campus_id IN (?,0)]
  388. //
  389. // 红线:
  390. // - 三列**必须同一条语句归零**——分步更新会在中间态出现"已取消但金额仍在",被对账/双跑读到;
  391. // - 审核推进(status 单向推进、不动金额)仍走 `UpdateSignedStatus`,两者用途不可混;
  392. // - 归零是"取消"语义的一部分(dump 明确 SET 0),金额不会回补到别处,故调用方必须已做业务校验
  393. // (如老带新已发奖励 → 3108 不可撤销,判定在 Service)。
  394. func (r *ActivityRepo) CancelSigned(ctx context.Context, tx *gorm.DB, userID string, activityID int64) (int64, error) {
  395. if err := requireTx(tx, "CancelSigned"); err != nil {
  396. return 0, err
  397. }
  398. if strings.TrimSpace(userID) == "" || activityID <= 0 {
  399. return 0, infraerrors.WithCause(infraerrors.ErrBadParam,
  400. errString("repository: CancelSigned 需要 user_id/activity_id"))
  401. }
  402. q := scopedCampusRows(tx.WithContext(ctxOrBackground(ctx)).Model(&model.CoursesActivitySigned{}), r.Sc, "")
  403. res := q.Where("user_id = ? AND activity_id = ?", userID, activityID).
  404. UpdateColumns(map[string]any{
  405. "status": model.ActivitySignedCancelled,
  406. "activity_money": "0.00",
  407. "activity_money_used": "0.00",
  408. })
  409. if res.Error != nil {
  410. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  411. }
  412. return res.RowsAffected, nil
  413. }
  414. // wrapQueryErr 把查询错误归一为 academic 错误契约(不存在 → ErrNotFound,其余 → ErrDBError)。
  415. //
  416. // 入参:err 原生错误;op 操作名(仅用于日志关联,不进入响应体)。
  417. // 出参:*infraerrors.AppError(ErrNotFound / ErrDBError)。
  418. func wrapQueryErr(err error, op string) error {
  419. if err == nil {
  420. return nil
  421. }
  422. if errors.Is(err, gorm.ErrRecordNotFound) {
  423. return infraerrors.WithCause(infraerrors.ErrNotFound, err)
  424. }
  425. return infraerrors.WithCause(infraerrors.ErrDBError, fmt.Errorf("%s: %w", op, err))
  426. }
  427. // ctxOrBackground 保证传给 gorm 的 ctx 非 nil(gorm.WithContext(nil) 会 panic)。
  428. func ctxOrBackground(ctx context.Context) context.Context {
  429. if ctx != nil {
  430. return ctx
  431. }
  432. return context.Background()
  433. }