divide_repo.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494
  1. // 分账树仓储(t_user_divide_into / t_user_divide_into_log)——AC-09。
  2. //
  3. // 依据:unified/03-migration/academic-domain-design.md §3.3 #22~#25(分账 4 SP)
  4. // / §1.7「全局无作用域表·例外二」/ §11.1 拍板 11-12。
  5. //
  6. // **本仓储不继承 BaseRepo**(设计 §11.1 拍板:全局推广体系无 institution_id/campus_id):
  7. // - 两表无租户列,`Scoped` 无从下手(叠 institution_id 会 Unknown column);
  8. // - 越权防线改为 **Handler 角色白名单(super_admin/academic)+ Service 显式数据范围参数**,
  9. // 本文件的数据访问方法因此**不接受**作用域参数,评审时必须显式标注"平台级/全局数据";
  10. // - 禁止把本仓储用于任何机构/校区维度的报表裁剪(需要裁剪时按 user_id 精确取数)。
  11. //
  12. // 表语义(Nested Set):
  13. //
  14. // t_user_divide_into 一人一节点(PK user_id);root_user_id 标记所属树;
  15. // left_val/right_val 为嵌套集区间(恒有 left < right);layer 根=1。
  16. // t_user_divide_into_log 分成明细(只增;type=1 同 user+student / type=2 同 user+student+order 幂等)。
  17. //
  18. // 事务纪律:写方法一律接收 `tx *gorm.DB`(仓储不开事务,事务边界在 DivideService)。
  19. package repository
  20. import (
  21. "context"
  22. "errors"
  23. "strings"
  24. "gorm.io/gorm"
  25. "yuansiwei/unified-backend/internal/academic/model"
  26. "yuansiwei/unified-backend/internal/academic/money"
  27. infraerrors "yuansiwei/unified-backend/internal/infra/errors"
  28. )
  29. // DivideRepo 分账树仓储(全局数据,**不继承 BaseRepo**,见文件头)。
  30. type DivideRepo struct {
  31. // DB 数据库句柄(连接池由 infra/db 统一建立;nil 时任何查询返回 ErrInternal)。
  32. DB *gorm.DB
  33. }
  34. // NewDivideRepo 构造分账树仓储。
  35. //
  36. // 入参:db *gorm.DB——连接池句柄(必填)。
  37. // 出参:*DivideRepo。
  38. func NewDivideRepo(db *gorm.DB) *DivideRepo { return &DivideRepo{DB: db} }
  39. // db 取查询句柄(ctx 内含 tx 时复用同一事务,保证事务内读一致)。
  40. //
  41. // 入参:ctx context.Context(可 nil)。出参:(*gorm.DB, error);未装配 DB → ErrInternal。
  42. func (r *DivideRepo) db(ctx context.Context) (*gorm.DB, error) {
  43. if r == nil || r.DB == nil {
  44. return nil, infraerrors.WithCause(infraerrors.ErrInternal, errString("repository: DivideRepo.DB 为 nil"))
  45. }
  46. if tx := TxFromContext(ctx); tx != nil {
  47. return tx, nil
  48. }
  49. if ctx == nil {
  50. return r.DB, nil
  51. }
  52. return r.DB.WithContext(ctx), nil
  53. }
  54. // GetByUserID 取分账树节点(主键 user_id)。
  55. //
  56. // 入参:ctx 请求上下文;userID 节点用户(ULID;空 → ErrBadParam)。
  57. // 出参:
  58. // - *model.UserDivideInto:节点(left_val/right_val/layer/quota 原样返回,金额为字符串);
  59. // - ErrNotFound(3001):该用户尚未进入分账体系(**不是**错误路径,调用方按业务分支处理);
  60. // - ErrBadParam / ErrDBError。
  61. func (r *DivideRepo) GetByUserID(ctx context.Context, userID string) (*model.UserDivideInto, error) {
  62. if strings.TrimSpace(userID) == "" {
  63. return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: GetByUserID 需要 user_id"))
  64. }
  65. db, err := r.db(ctx)
  66. if err != nil {
  67. return nil, err
  68. }
  69. var out model.UserDivideInto
  70. if err := db.Where("user_id = ?", userID).First(&out).Error; err != nil {
  71. return nil, wrapQueryErr(err, "GetByUserID(t_user_divide_into)")
  72. }
  73. return &out, nil
  74. }
  75. // GetSubtree 取节点子树(严格位于 (leftVal, rightVal) 区间内的后代)。
  76. //
  77. // 入参:
  78. // - ctx 请求上下文;
  79. // - leftVal/rightVal int32:父节点的左右值(Nested Set 区间;rightVal <= leftVal → ErrBadParam);
  80. // - divideType int8:分成类型过滤(model.DivideTypeOneOff / DivideTypePerLesson;0 = 不筛类型)。
  81. //
  82. // 出参:
  83. // - []model.UserDivideInto:后代节点,按 left_val 升序(= 树的深度优先前序,便于逐层遍历);
  84. // - ErrBadParam / ErrDBError。
  85. //
  86. // 用途:额度级联下降(update_quota)/ 子树统计。**不含**父节点自身。
  87. func (r *DivideRepo) GetSubtree(ctx context.Context, leftVal, rightVal int32, divideType int8) ([]model.UserDivideInto, error) {
  88. if leftVal < 0 || rightVal <= leftVal {
  89. return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: GetSubtree 左右值区间非法"))
  90. }
  91. db, err := r.db(ctx)
  92. if err != nil {
  93. return nil, err
  94. }
  95. q := db.Model(&model.UserDivideInto{}).
  96. Where("left_val > ? AND right_val < ?", leftVal, rightVal)
  97. if divideType > 0 {
  98. q = q.Where("type = ?", divideType)
  99. }
  100. var rows []model.UserDivideInto
  101. if err := q.Order("left_val asc").Find(&rows).Error; err != nil {
  102. return nil, infraerrors.WithCause(infraerrors.ErrDBError, err)
  103. }
  104. return rows, nil
  105. }
  106. // ShiftRight 右值扩张:`right_val = right_val + delta WHERE root_user_id = ? AND right_val >= threshold`。
  107. //
  108. // 入参:
  109. // - ctx 请求上下文;
  110. // - tx 事务句柄(**必传**);
  111. // - rootUserID **所属树根**(ULID;**必填**,见下方红线);
  112. // - threshold int32:扩张阈值(新节点插入点 = 父节点右值;父与全部祖先的 right 均 >= 它,一次覆盖);
  113. // - delta int32:增量(插入子节点固定 +2;撤销固定 -2)。
  114. //
  115. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  116. //
  117. // 红线(audit V-02,dump `yuansiwei_finance.sql:763707-763708`):
  118. //
  119. // dump 原文的位移语句带 `WHERE root_userid = rootUserid AND left_val >= parentRight`。
  120. // **漏掉 root 限定的后果**:多棵树共用一张表(各自 root 的左右值区间可能重叠/相邻),
  121. // 同输入会产出不同的 left/right → 双跑快照 diff 必然不一致,且会污染别的树的区间。
  122. // (同一 dump 的另一处 left 位移同理,行号 763708。)
  123. //
  124. // 与 ShiftLeft **必须同一事务、同一 rootUserID 与 threshold**(两步各自只动一列,
  125. // 合成标准嵌套集插入算法;拆事务会让树在中间态不可用)。
  126. func (r *DivideRepo) ShiftRight(ctx context.Context, tx *gorm.DB, rootUserID string, threshold int32, delta int32) (int64, error) {
  127. return r.shift(ctx, tx, "right_val", rootUserID, threshold, delta)
  128. }
  129. // ShiftLeft 左值扩张:`left_val = left_val + delta WHERE root_user_id = ? AND left_val >= threshold`。
  130. //
  131. // 入参:语义同 ShiftRight(只作用于 left_val 列——即"位于插入点右侧的兄弟及其后代")。
  132. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  133. func (r *DivideRepo) ShiftLeft(ctx context.Context, tx *gorm.DB, rootUserID string, threshold int32, delta int32) (int64, error) {
  134. return r.shift(ctx, tx, "left_val", rootUserID, threshold, delta)
  135. }
  136. // shift 左右值扩张的公共实现(列名由内部常量传入,**不接受外部字符串**,防注入)。
  137. //
  138. // 入参:col 列名("left_val"/"right_val");rootUserID 树根(必填);
  139. // threshold/delta 语义见 ShiftRight。
  140. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  141. func (r *DivideRepo) shift(ctx context.Context, tx *gorm.DB, col, rootUserID string, threshold int32, delta int32) (int64, error) {
  142. if err := requireTx(tx, "shift("+col+")"); err != nil {
  143. return 0, err
  144. }
  145. if col != "left_val" && col != "right_val" {
  146. return 0, infraerrors.WithCause(infraerrors.ErrInternal, errString("repository: shift 列名非法"))
  147. }
  148. if strings.TrimSpace(rootUserID) == "" {
  149. // root 限定不可省:漏了会串改其它树的区间(audit V-02)
  150. return 0, infraerrors.WithCause(infraerrors.ErrBadParam,
  151. errString("repository: shift 需要 root_user_id(多树隔离,dump 763707)"))
  152. }
  153. if threshold < 0 {
  154. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: shift 阈值不能为负"))
  155. }
  156. if delta == 0 {
  157. return 0, nil
  158. }
  159. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  160. Where("root_user_id = ?", strings.TrimSpace(rootUserID)).
  161. Where(col+" >= ?", threshold).
  162. UpdateColumn(col, gorm.Expr(col+" + ?", delta))
  163. if res.Error != nil {
  164. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  165. }
  166. return res.RowsAffected, nil
  167. }
  168. // InsertNode 插入分账树节点(新节点左右值由 Service 用嵌套集算法算出)。
  169. //
  170. // 入参:
  171. // - ctx 请求上下文;
  172. // - tx 事务句柄(**必传**);
  173. // - node 节点(UserID 非空;必须满足 LeftVal < RightVal,否则 ErrBadParam)。
  174. //
  175. // 出参:nil 成功;ErrBadParam / ErrDBError。
  176. //
  177. // 红线:与 ShiftRight/ShiftLeft **必须同事务**;已存在同 user_id 节点 → 唯一键冲突(ErrDBError,
  178. // 调用方按"已存在分支"重入,不得重试插入)。
  179. func (r *DivideRepo) InsertNode(ctx context.Context, tx *gorm.DB, node *model.UserDivideInto) error {
  180. if err := requireTx(tx, "InsertNode"); err != nil {
  181. return err
  182. }
  183. if node == nil || strings.TrimSpace(node.UserID) == "" {
  184. return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: InsertNode 需要 user_id"))
  185. }
  186. if node.LeftVal >= node.RightVal {
  187. return infraerrors.WithCause(infraerrors.ErrBadParam,
  188. errString("repository: InsertNode 违反 Nested Set 不变式 left_val < right_val"))
  189. }
  190. if err := tx.WithContext(ctxOrBackground(ctx)).Create(node).Error; err != nil {
  191. return infraerrors.WithCause(infraerrors.ErrDBError, err)
  192. }
  193. return nil
  194. }
  195. // InsertLog 写入分成明细(**只增**)。
  196. //
  197. // 入参:
  198. // - ctx 请求上下文;
  199. // - tx 事务句柄(**必传**);
  200. // - log 明细(Type∈{1,2}、UserID/StudentID 非空、Money 为元字符串)。
  201. //
  202. // 出参:nil 成功;ErrBadParam / ErrDBError。
  203. //
  204. // 红线:与节点 total_income 累加(IncTotalIncome)必须同事务(否则"明细有、收入没加")。
  205. func (r *DivideRepo) InsertLog(ctx context.Context, tx *gorm.DB, log *model.UserDivideIntoLog) error {
  206. if err := requireTx(tx, "InsertLog"); err != nil {
  207. return err
  208. }
  209. if log == nil || strings.TrimSpace(log.UserID) == "" || strings.TrimSpace(log.StudentID) == "" {
  210. return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: InsertLog 需要 user_id/student_id"))
  211. }
  212. if log.Type != model.DivideTypeOneOff && log.Type != model.DivideTypePerLesson {
  213. return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: InsertLog 分成类型非法"))
  214. }
  215. if err := tx.WithContext(ctxOrBackground(ctx)).Create(log).Error; err != nil {
  216. return infraerrors.WithCause(infraerrors.ErrDBError, err)
  217. }
  218. return nil
  219. }
  220. // DeleteLog 物理删除分成明细(撤销分支二:回滚 total_income 后删行)。
  221. //
  222. // 入参:
  223. // - ctx 请求上下文;
  224. // - tx 事务句柄(**必传**);
  225. // - id 明细 ID(<=0 → ErrBadParam)。
  226. //
  227. // 出参:nil 成功/行不存在(幂等);ErrBadParam / ErrDBError。
  228. //
  229. // 红线(设计 §3.3 #23):旧语义允许物理删除本表行(与"流水只增"纪律的唯一例外),
  230. // 但必须与 total_income 回滚同事务;删除前须判定"是否已有首次扣费上课记录"(改绑分支优先)。
  231. func (r *DivideRepo) DeleteLog(ctx context.Context, tx *gorm.DB, id int64) error {
  232. if err := requireTx(tx, "DeleteLog"); err != nil {
  233. return err
  234. }
  235. if id <= 0 {
  236. return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: DeleteLog 需要 id"))
  237. }
  238. if err := tx.WithContext(ctxOrBackground(ctx)).Where("id = ?", id).
  239. Delete(&model.UserDivideIntoLog{}).Error; err != nil {
  240. return infraerrors.WithCause(infraerrors.ErrDBError, err)
  241. }
  242. return nil
  243. }
  244. // FindLog 按**幂等键**查分成明细(两级幂等判定唯一入口)。
  245. //
  246. // 入参:
  247. // - ctx 请求上下文;
  248. // - userID 获得分成的用户(ULID);studentID 报课学生(ULID);divideType 分成类型;
  249. // - orderNo 订单号(type=2 的上课记录 id 字符串;**type=1 时忽略**——一次付清不区分订单)。
  250. //
  251. // 出参:
  252. // - *model.UserDivideIntoLog:命中行;
  253. // - ErrNotFound:未命中(调用方据此判定"未生成",非异常);
  254. // - ErrBadParam / ErrDBError。
  255. //
  256. // 幂等键(dump divide_into_log_add 口径,**不可改**):
  257. // - type=1(一次付清):(user_id, student_id, type);
  258. // - type=2(按课次) :(user_id, student_id, type, order_no)。
  259. func (r *DivideRepo) FindLog(ctx context.Context, userID, studentID string, divideType int8, orderNo string) (*model.UserDivideIntoLog, error) {
  260. if strings.TrimSpace(userID) == "" || strings.TrimSpace(studentID) == "" {
  261. return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: FindLog 需要 user_id/student_id"))
  262. }
  263. db, err := r.db(ctx)
  264. if err != nil {
  265. return nil, err
  266. }
  267. q := db.Model(&model.UserDivideIntoLog{}).
  268. Where("user_id = ? AND student_id = ? AND type = ?", userID, studentID, divideType)
  269. if divideType == model.DivideTypePerLesson {
  270. q = q.Where("order_no = ?", strings.TrimSpace(orderNo))
  271. }
  272. var out model.UserDivideIntoLog
  273. if err := q.Order("id asc").First(&out).Error; err != nil {
  274. return nil, wrapQueryErr(err, "FindLog(t_user_divide_into_log)")
  275. }
  276. return &out, nil
  277. }
  278. // ListLogsByStudentOrder 取某「学生 + 订单」产生的**全部**分成明细(撤销的两参入口)。
  279. //
  280. // 入参:ctx 请求上下文;studentID 报课学生(ULID);orderNo 订单号(上课记录 id 字符串)。
  281. // 出参:
  282. // - []model.UserDivideIntoLog:按 id 升序(生成先后;空切片表示无记录,幂等空操作);
  283. // - ErrBadParam / ErrDBError。
  284. //
  285. // 语义(dump `divide_into_log_delete(studentid, orderid)` 两参):同一订单的分成会沿上行链
  286. // 生成多行(本人 + 各级上线),撤销时必须**整组处理**,故按 (student_id, order_no) 取全量,
  287. // 而非只取一行。
  288. func (r *DivideRepo) ListLogsByStudentOrder(ctx context.Context, studentID, orderNo string) ([]model.UserDivideIntoLog, error) {
  289. if strings.TrimSpace(studentID) == "" || strings.TrimSpace(orderNo) == "" {
  290. return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ListLogsByStudentOrder 需要 student_id/order_no"))
  291. }
  292. db, err := r.db(ctx)
  293. if err != nil {
  294. return nil, err
  295. }
  296. var rows []model.UserDivideIntoLog
  297. if err := db.Model(&model.UserDivideIntoLog{}).
  298. Where("student_id = ? AND order_no = ?", studentID, orderNo).
  299. Order("id asc").
  300. Find(&rows).Error; err != nil {
  301. return nil, infraerrors.WithCause(infraerrors.ErrDBError, err)
  302. }
  303. return rows, nil
  304. }
  305. // UpdateNodeParent 改挂节点父链(只改父/根/层级,**不重排左右值**)。
  306. //
  307. // ⚠ **当前无业务调用方**(audit D4:dump `:763695` 的 "subid 已存在" 分支是**完全 no-op**,
  308. // 分支体内没有任何写动作)。本方法因此**不属于 dump 流程**,仅保留给后台数据修复;
  309. // 若将来要用于业务链路,必须先评审——它不重排左右值,改挂自带子树的节点会产生区间交叠。
  310. //
  311. // 入参:
  312. // - ctx 请求上下文;tx 事务句柄(**必传**);
  313. // - userID 被改挂节点;parentUserID 新上级;rootUserID 所属树根;layer 新层级(根=1)。
  314. //
  315. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  316. //
  317. // 红线(修正 dump 缺陷,模型注释 §UserDivideInto.root_user_id 已标注):
  318. // 旧 SP 在该分支**漏写 root_user_id**(留 NULL,导致子树查询/级联失效);本方法强制显式写根。
  319. // 遗留:不重排左右值(与原 SP 一致)——若被改挂节点自带子树,区间可能出现交叠,
  320. // 已在交付说明中登记为遗留项(AC-09 范围内不引入区间重排,避免与迁移期数据冲突)。
  321. func (r *DivideRepo) UpdateNodeParent(ctx context.Context, tx *gorm.DB, userID, parentUserID, rootUserID string, layer int32) (int64, error) {
  322. if err := requireTx(tx, "UpdateNodeParent"); err != nil {
  323. return 0, err
  324. }
  325. if strings.TrimSpace(userID) == "" || strings.TrimSpace(parentUserID) == "" || strings.TrimSpace(rootUserID) == "" {
  326. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateNodeParent 需要 user/parent/root"))
  327. }
  328. if layer < 1 {
  329. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateNodeParent layer 必须 >= 1"))
  330. }
  331. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  332. Where("user_id = ?", userID).
  333. Updates(map[string]any{
  334. "parent_user_id": parentUserID,
  335. "root_user_id": rootUserID,
  336. "layer": layer,
  337. })
  338. if res.Error != nil {
  339. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  340. }
  341. return res.RowsAffected, nil
  342. }
  343. // UpdateNodeQuota 更新节点分成额度(不触碰左右值/收入)。
  344. //
  345. // 入参:ctx;tx(**必传**);userID 节点;quota 新额度(单位:分;负 → ErrBadParam)。
  346. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  347. //
  348. // 红线:**必须**与 ClampSubtreeQuota 同事务(否则出现"父降了、子仍高于父"的越权额度窗口)。
  349. func (r *DivideRepo) UpdateNodeQuota(ctx context.Context, tx *gorm.DB, userID string, quota money.Amount) (int64, error) {
  350. if err := requireTx(tx, "UpdateNodeQuota"); err != nil {
  351. return 0, err
  352. }
  353. if strings.TrimSpace(userID) == "" {
  354. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateNodeQuota 需要 user_id"))
  355. }
  356. if err := quota.Validate(); err != nil {
  357. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
  358. }
  359. if quota.IsNegative() {
  360. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: 分成额度不能为负"))
  361. }
  362. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  363. Where("user_id = ?", userID).
  364. UpdateColumn("quota", quota.String())
  365. if res.Error != nil {
  366. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  367. }
  368. return res.RowsAffected, nil
  369. }
  370. // ClampSubtreeQuota 额度级联下降:把子树内**同分成类型**且**超过**上限的节点统一压到上限。
  371. //
  372. // 入参:
  373. // - ctx;tx(**必传**);
  374. // - leftVal/rightVal int32 父节点区间(rightVal <= leftVal → ErrBadParam);
  375. // - divideType 分成类型过滤(model.DivideTypeOneOff/PerLesson;**0 = 不筛**,仅 admin 全量场景);
  376. // - quota 上限(单位:分;负 → ErrBadParam)。
  377. //
  378. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  379. //
  380. // SQL(audit A6 / dump `yuansiwei_finance.sql:763872-763873` 原文条件,**缺一不可**):
  381. //
  382. // UPDATE t_user_divide_into SET quota=updateQuota
  383. // WHERE left_val>uLeft AND right_val<uRight AND quota>updateQuota AND type=uType;
  384. //
  385. // 语义("子不得超父"):只降不升——低于上限的节点保持原额度(分别配置的阶梯额度不被抹平)。
  386. // 红线:`type` 过滤不可省 —— 否则一次付清(1)与按课次(2)两条线的额度会被跨类型抹平(漏实现 = 缺陷)。
  387. func (r *DivideRepo) ClampSubtreeQuota(ctx context.Context, tx *gorm.DB, leftVal, rightVal int32, divideType int8, quota money.Amount) (int64, error) {
  388. if err := requireTx(tx, "ClampSubtreeQuota"); err != nil {
  389. return 0, err
  390. }
  391. if leftVal < 0 || rightVal <= leftVal {
  392. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ClampSubtreeQuota 左右值区间非法"))
  393. }
  394. if divideType != 0 && divideType != model.DivideTypeOneOff && divideType != model.DivideTypePerLesson {
  395. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ClampSubtreeQuota 分成类型非法"))
  396. }
  397. if err := quota.Validate(); err != nil {
  398. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
  399. }
  400. if quota.IsNegative() {
  401. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: 分成额度不能为负"))
  402. }
  403. q := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  404. Where("left_val > ? AND right_val < ?", leftVal, rightVal).
  405. Where("quota > ?", quota.String())
  406. if divideType != 0 {
  407. q = q.Where("type = ?", divideType)
  408. }
  409. res := q.UpdateColumn("quota", quota.String())
  410. if res.Error != nil {
  411. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  412. }
  413. return res.RowsAffected, nil
  414. }
  415. // IncTotalIncome 累加节点分成总收入(上行分成 +,撤销回滚 −)。
  416. //
  417. // 入参:ctx;tx(**必传**);userID 节点;delta 增量(单位:分;0 → 空操作)。
  418. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  419. //
  420. // 红线:与 InsertLog/DeleteLog 同事务;金额字符串入场(decimal 精确,零 float64)。
  421. func (r *DivideRepo) IncTotalIncome(ctx context.Context, tx *gorm.DB, userID string, delta money.Amount) (int64, error) {
  422. if err := requireTx(tx, "IncTotalIncome"); err != nil {
  423. return 0, err
  424. }
  425. if strings.TrimSpace(userID) == "" {
  426. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: IncTotalIncome 需要 user_id"))
  427. }
  428. if delta.IsZero() {
  429. return 0, nil
  430. }
  431. if err := delta.Validate(); err != nil {
  432. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
  433. }
  434. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  435. Where("user_id = ?", userID).
  436. UpdateColumn("total_income", gorm.Expr("total_income + ?", delta.String()))
  437. if res.Error != nil {
  438. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  439. }
  440. return res.RowsAffected, nil
  441. }
  442. // UpdateLogOrderNo 改绑分成明细的订单号(撤销分支一:保留分成,改指首次扣费上课记录)。
  443. //
  444. // 入参:ctx;tx(**必传**);id 明细 ID;orderNo 新订单号(上课记录 id 字符串);remark 备注(可空)。
  445. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  446. //
  447. // 红线(dump 口径):改绑**不改 money**、**不回滚 total_income**(分成保留);
  448. // order_no 是 type=2 的幂等键组成列,改绑即改幂等语义 → 必须写审计备注(默认"推广分成")。
  449. func (r *DivideRepo) UpdateLogOrderNo(ctx context.Context, tx *gorm.DB, id int64, orderNo, remark string) (int64, error) {
  450. if err := requireTx(tx, "UpdateLogOrderNo"); err != nil {
  451. return 0, err
  452. }
  453. if id <= 0 || strings.TrimSpace(orderNo) == "" {
  454. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateLogOrderNo 需要 id/order_no"))
  455. }
  456. // create_time 重置为 DB 当前时间(dump 改绑分支:`create_time=now()`,让改绑后的明细排到最新)
  457. fields := map[string]any{"order_no": orderNo, "create_time": gorm.Expr("CURRENT_TIMESTAMP")}
  458. if strings.TrimSpace(remark) != "" {
  459. fields["remark"] = remark
  460. }
  461. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideIntoLog{}).
  462. Where("id = ?", id).
  463. Updates(fields)
  464. if res.Error != nil {
  465. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  466. }
  467. return res.RowsAffected, nil
  468. }
  469. // IsNotFound 判断错误是否为仓储 ErrNotFound(供 Service 分支判断,避免各处抄 errors.Is)。
  470. //
  471. // 入参:err error(可 nil)。出参:true 表示"记录不存在"。
  472. func IsNotFound(err error) bool { return err != nil && errors.Is(err, infraerrors.ErrNotFound) }