divide_repo.go 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640
  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. "time"
  25. "gorm.io/gorm"
  26. "yuansiwei/unified-backend/internal/academic/model"
  27. "yuansiwei/unified-backend/internal/academic/money"
  28. infraerrors "yuansiwei/unified-backend/internal/infra/errors"
  29. )
  30. // DivideRepo 分账树仓储(全局数据,**不继承 BaseRepo**,见文件头)。
  31. type DivideRepo struct {
  32. // DB 数据库句柄(连接池由 infra/db 统一建立;nil 时任何查询返回 ErrInternal)。
  33. DB *gorm.DB
  34. }
  35. // NewDivideRepo 构造分账树仓储。
  36. //
  37. // 入参:db *gorm.DB——连接池句柄(必填)。
  38. // 出参:*DivideRepo。
  39. func NewDivideRepo(db *gorm.DB) *DivideRepo { return &DivideRepo{DB: db} }
  40. // db 取查询句柄(ctx 内含 tx 时复用同一事务,保证事务内读一致)。
  41. //
  42. // 入参:ctx context.Context(可 nil)。出参:(*gorm.DB, error);未装配 DB → ErrInternal。
  43. func (r *DivideRepo) db(ctx context.Context) (*gorm.DB, error) {
  44. if r == nil || r.DB == nil {
  45. return nil, infraerrors.WithCause(infraerrors.ErrInternal, errString("repository: DivideRepo.DB 为 nil"))
  46. }
  47. if tx := TxFromContext(ctx); tx != nil {
  48. return tx, nil
  49. }
  50. if ctx == nil {
  51. return r.DB, nil
  52. }
  53. return r.DB.WithContext(ctx), nil
  54. }
  55. // GetByUserID 取分账树节点(主键 user_id)。
  56. //
  57. // 入参:ctx 请求上下文;userID 节点用户(ULID;空 → ErrBadParam)。
  58. // 出参:
  59. // - *model.UserDivideInto:节点(left_val/right_val/layer/quota 原样返回,金额为字符串);
  60. // - ErrNotFound(3001):该用户尚未进入分账体系(**不是**错误路径,调用方按业务分支处理);
  61. // - ErrBadParam / ErrDBError。
  62. func (r *DivideRepo) GetByUserID(ctx context.Context, userID string) (*model.UserDivideInto, error) {
  63. if strings.TrimSpace(userID) == "" {
  64. return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: GetByUserID 需要 user_id"))
  65. }
  66. db, err := r.db(ctx)
  67. if err != nil {
  68. return nil, err
  69. }
  70. var out model.UserDivideInto
  71. if err := db.Where("user_id = ?", userID).First(&out).Error; err != nil {
  72. return nil, wrapQueryErr(err, "GetByUserID(t_user_divide_into)")
  73. }
  74. return &out, nil
  75. }
  76. // GetSubtree 取节点子树(严格位于 (leftVal, rightVal) 区间内的后代)。
  77. //
  78. // 入参:
  79. // - ctx 请求上下文;
  80. // - leftVal/rightVal int32:父节点的左右值(Nested Set 区间;rightVal <= leftVal → ErrBadParam);
  81. // - divideType int8:分成类型过滤(model.DivideTypeOneOff / DivideTypePerLesson;0 = 不筛类型)。
  82. //
  83. // 出参:
  84. // - []model.UserDivideInto:后代节点,按 left_val 升序(= 树的深度优先前序,便于逐层遍历);
  85. // - ErrBadParam / ErrDBError。
  86. //
  87. // 用途:额度级联下降(update_quota)/ 子树统计。**不含**父节点自身。
  88. func (r *DivideRepo) GetSubtree(ctx context.Context, leftVal, rightVal int32, divideType int8) ([]model.UserDivideInto, error) {
  89. if leftVal < 0 || rightVal <= leftVal {
  90. return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: GetSubtree 左右值区间非法"))
  91. }
  92. db, err := r.db(ctx)
  93. if err != nil {
  94. return nil, err
  95. }
  96. q := db.Model(&model.UserDivideInto{}).
  97. Where("left_val > ? AND right_val < ?", leftVal, rightVal)
  98. if divideType > 0 {
  99. q = q.Where("type = ?", divideType)
  100. }
  101. var rows []model.UserDivideInto
  102. if err := q.Order("left_val asc").Find(&rows).Error; err != nil {
  103. return nil, infraerrors.WithCause(infraerrors.ErrDBError, err)
  104. }
  105. return rows, nil
  106. }
  107. // ShiftRight 右值扩张:`right_val = right_val + delta WHERE root_user_id = ? AND right_val >= threshold`。
  108. //
  109. // 入参:
  110. // - ctx 请求上下文;
  111. // - tx 事务句柄(**必传**);
  112. // - rootUserID **所属树根**(ULID;**必填**,见下方红线);
  113. // - threshold int32:扩张阈值(新节点插入点 = 父节点右值;父与全部祖先的 right 均 >= 它,一次覆盖);
  114. // - delta int32:增量(插入子节点固定 +2;撤销固定 -2)。
  115. //
  116. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  117. //
  118. // 红线(audit V-02,dump `yuansiwei_finance.sql:763707-763708`):
  119. //
  120. // dump 原文的位移语句带 `WHERE root_userid = rootUserid AND left_val >= parentRight`。
  121. // **漏掉 root 限定的后果**:多棵树共用一张表(各自 root 的左右值区间可能重叠/相邻),
  122. // 同输入会产出不同的 left/right → 双跑快照 diff 必然不一致,且会污染别的树的区间。
  123. // (同一 dump 的另一处 left 位移同理,行号 763708。)
  124. //
  125. // 与 ShiftLeft **必须同一事务、同一 rootUserID 与 threshold**(两步各自只动一列,
  126. // 合成标准嵌套集插入算法;拆事务会让树在中间态不可用)。
  127. func (r *DivideRepo) ShiftRight(ctx context.Context, tx *gorm.DB, rootUserID string, threshold int32, delta int32) (int64, error) {
  128. return r.shift(ctx, tx, "right_val", rootUserID, threshold, delta)
  129. }
  130. // ShiftLeft 左值扩张:`left_val = left_val + delta WHERE root_user_id = ? AND left_val >= threshold`。
  131. //
  132. // 入参:语义同 ShiftRight(只作用于 left_val 列——即"位于插入点右侧的兄弟及其后代")。
  133. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  134. func (r *DivideRepo) ShiftLeft(ctx context.Context, tx *gorm.DB, rootUserID string, threshold int32, delta int32) (int64, error) {
  135. return r.shift(ctx, tx, "left_val", rootUserID, threshold, delta)
  136. }
  137. // shift 左右值扩张的公共实现(列名由内部常量传入,**不接受外部字符串**,防注入)。
  138. //
  139. // 入参:col 列名("left_val"/"right_val");rootUserID 树根(必填);
  140. // threshold/delta 语义见 ShiftRight。
  141. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  142. func (r *DivideRepo) shift(ctx context.Context, tx *gorm.DB, col, rootUserID string, threshold int32, delta int32) (int64, error) {
  143. if err := requireTx(tx, "shift("+col+")"); err != nil {
  144. return 0, err
  145. }
  146. if col != "left_val" && col != "right_val" {
  147. return 0, infraerrors.WithCause(infraerrors.ErrInternal, errString("repository: shift 列名非法"))
  148. }
  149. if strings.TrimSpace(rootUserID) == "" {
  150. // root 限定不可省:漏了会串改其它树的区间(audit V-02)
  151. return 0, infraerrors.WithCause(infraerrors.ErrBadParam,
  152. errString("repository: shift 需要 root_user_id(多树隔离,dump 763707)"))
  153. }
  154. if threshold < 0 {
  155. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: shift 阈值不能为负"))
  156. }
  157. if delta == 0 {
  158. return 0, nil
  159. }
  160. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  161. Where("root_user_id = ?", strings.TrimSpace(rootUserID)).
  162. Where(col+" >= ?", threshold).
  163. UpdateColumn(col, gorm.Expr(col+" + ?", delta))
  164. if res.Error != nil {
  165. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  166. }
  167. return res.RowsAffected, nil
  168. }
  169. // InsertNode 插入分账树节点(新节点左右值由 Service 用嵌套集算法算出)。
  170. //
  171. // 入参:
  172. // - ctx 请求上下文;
  173. // - tx 事务句柄(**必传**);
  174. // - node 节点(UserID 非空;必须满足 LeftVal < RightVal,否则 ErrBadParam)。
  175. //
  176. // 出参:nil 成功;ErrBadParam / ErrDBError。
  177. //
  178. // 红线:与 ShiftRight/ShiftLeft **必须同事务**;已存在同 user_id 节点 → 唯一键冲突(ErrDBError,
  179. // 调用方按"已存在分支"重入,不得重试插入)。
  180. func (r *DivideRepo) InsertNode(ctx context.Context, tx *gorm.DB, node *model.UserDivideInto) error {
  181. if err := requireTx(tx, "InsertNode"); err != nil {
  182. return err
  183. }
  184. if node == nil || strings.TrimSpace(node.UserID) == "" {
  185. return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: InsertNode 需要 user_id"))
  186. }
  187. if node.LeftVal >= node.RightVal {
  188. return infraerrors.WithCause(infraerrors.ErrBadParam,
  189. errString("repository: InsertNode 违反 Nested Set 不变式 left_val < right_val"))
  190. }
  191. if err := tx.WithContext(ctxOrBackground(ctx)).Create(node).Error; err != nil {
  192. return infraerrors.WithCause(infraerrors.ErrDBError, err)
  193. }
  194. return nil
  195. }
  196. // InsertLog 写入分成明细(**只增**)。
  197. //
  198. // 入参:
  199. // - ctx 请求上下文;
  200. // - tx 事务句柄(**必传**);
  201. // - log 明细(Type∈{1,2}、UserID/StudentID 非空、Money 为元字符串)。
  202. //
  203. // 出参:nil 成功;ErrBadParam / ErrDBError。
  204. //
  205. // 红线:与节点 total_income 累加(IncTotalIncome)必须同事务(否则"明细有、收入没加")。
  206. func (r *DivideRepo) InsertLog(ctx context.Context, tx *gorm.DB, log *model.UserDivideIntoLog) error {
  207. if err := requireTx(tx, "InsertLog"); err != nil {
  208. return err
  209. }
  210. if log == nil || strings.TrimSpace(log.UserID) == "" || strings.TrimSpace(log.StudentID) == "" {
  211. return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: InsertLog 需要 user_id/student_id"))
  212. }
  213. if log.Type != model.DivideTypeOneOff && log.Type != model.DivideTypePerLesson {
  214. return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: InsertLog 分成类型非法"))
  215. }
  216. if err := tx.WithContext(ctxOrBackground(ctx)).Create(log).Error; err != nil {
  217. return infraerrors.WithCause(infraerrors.ErrDBError, err)
  218. }
  219. return nil
  220. }
  221. // DeleteLog 物理删除分成明细(撤销分支二:回滚 total_income 后删行)。
  222. //
  223. // 入参:
  224. // - ctx 请求上下文;
  225. // - tx 事务句柄(**必传**);
  226. // - id 明细 ID(<=0 → ErrBadParam)。
  227. //
  228. // 出参:nil 成功/行不存在(幂等);ErrBadParam / ErrDBError。
  229. //
  230. // 红线(设计 §3.3 #23):旧语义允许物理删除本表行(与"流水只增"纪律的唯一例外),
  231. // 但必须与 total_income 回滚同事务;删除前须判定"是否已有首次扣费上课记录"(改绑分支优先)。
  232. func (r *DivideRepo) DeleteLog(ctx context.Context, tx *gorm.DB, id int64) error {
  233. if err := requireTx(tx, "DeleteLog"); err != nil {
  234. return err
  235. }
  236. if id <= 0 {
  237. return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: DeleteLog 需要 id"))
  238. }
  239. if err := tx.WithContext(ctxOrBackground(ctx)).Where("id = ?", id).
  240. Delete(&model.UserDivideIntoLog{}).Error; err != nil {
  241. return infraerrors.WithCause(infraerrors.ErrDBError, err)
  242. }
  243. return nil
  244. }
  245. // FindLog 按**幂等键**查分成明细(两级幂等判定唯一入口)。
  246. //
  247. // 入参:
  248. // - ctx 请求上下文;
  249. // - userID 获得分成的用户(ULID);studentID 报课学生(ULID);divideType 分成类型;
  250. // - orderNo 订单号(type=2 的上课记录 id 字符串;**type=1 时忽略**——一次付清不区分订单)。
  251. //
  252. // 出参:
  253. // - *model.UserDivideIntoLog:命中行;
  254. // - ErrNotFound:未命中(调用方据此判定"未生成",非异常);
  255. // - ErrBadParam / ErrDBError。
  256. //
  257. // 幂等键(dump divide_into_log_add 口径,**不可改**):
  258. // - type=1(一次付清):(user_id, student_id, type);
  259. // - type=2(按课次) :(user_id, student_id, type, order_no)。
  260. func (r *DivideRepo) FindLog(ctx context.Context, userID, studentID string, divideType int8, orderNo string) (*model.UserDivideIntoLog, error) {
  261. if strings.TrimSpace(userID) == "" || strings.TrimSpace(studentID) == "" {
  262. return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: FindLog 需要 user_id/student_id"))
  263. }
  264. db, err := r.db(ctx)
  265. if err != nil {
  266. return nil, err
  267. }
  268. q := db.Model(&model.UserDivideIntoLog{}).
  269. Where("user_id = ? AND student_id = ? AND type = ?", userID, studentID, divideType)
  270. if divideType == model.DivideTypePerLesson {
  271. q = q.Where("order_no = ?", strings.TrimSpace(orderNo))
  272. }
  273. var out model.UserDivideIntoLog
  274. if err := q.Order("id asc").First(&out).Error; err != nil {
  275. return nil, wrapQueryErr(err, "FindLog(t_user_divide_into_log)")
  276. }
  277. return &out, nil
  278. }
  279. // ListLogsByStudentOrder 取某「学生 + 订单」产生的**全部**分成明细(撤销的两参入口)。
  280. //
  281. // 入参:ctx 请求上下文;studentID 报课学生(ULID);orderNo 订单号(上课记录 id 字符串)。
  282. // 出参:
  283. // - []model.UserDivideIntoLog:按 id 升序(生成先后;空切片表示无记录,幂等空操作);
  284. // - ErrBadParam / ErrDBError。
  285. //
  286. // 语义(dump `divide_into_log_delete(studentid, orderid)` 两参):同一订单的分成会沿上行链
  287. // 生成多行(本人 + 各级上线),撤销时必须**整组处理**,故按 (student_id, order_no) 取全量,
  288. // 而非只取一行。
  289. func (r *DivideRepo) ListLogsByStudentOrder(ctx context.Context, studentID, orderNo string) ([]model.UserDivideIntoLog, error) {
  290. if strings.TrimSpace(studentID) == "" || strings.TrimSpace(orderNo) == "" {
  291. return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ListLogsByStudentOrder 需要 student_id/order_no"))
  292. }
  293. db, err := r.db(ctx)
  294. if err != nil {
  295. return nil, err
  296. }
  297. var rows []model.UserDivideIntoLog
  298. if err := db.Model(&model.UserDivideIntoLog{}).
  299. Where("student_id = ? AND order_no = ?", studentID, orderNo).
  300. Order("id asc").
  301. Find(&rows).Error; err != nil {
  302. return nil, infraerrors.WithCause(infraerrors.ErrDBError, err)
  303. }
  304. return rows, nil
  305. }
  306. // UpdateNodeParent 改挂节点父链(只改父/根/层级,**不重排左右值**)。
  307. //
  308. // ⚠ **当前无业务调用方**(audit D4:dump `:763695` 的 "subid 已存在" 分支是**完全 no-op**,
  309. // 分支体内没有任何写动作)。本方法因此**不属于 dump 流程**,仅保留给后台数据修复;
  310. // 若将来要用于业务链路,必须先评审——它不重排左右值,改挂自带子树的节点会产生区间交叠。
  311. //
  312. // 入参:
  313. // - ctx 请求上下文;tx 事务句柄(**必传**);
  314. // - userID 被改挂节点;parentUserID 新上级;rootUserID 所属树根;layer 新层级(根=1)。
  315. //
  316. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  317. //
  318. // 红线(修正 dump 缺陷,模型注释 §UserDivideInto.root_user_id 已标注):
  319. // 旧 SP 在该分支**漏写 root_user_id**(留 NULL,导致子树查询/级联失效);本方法强制显式写根。
  320. // 遗留:不重排左右值(与原 SP 一致)——若被改挂节点自带子树,区间可能出现交叠,
  321. // 已在交付说明中登记为遗留项(AC-09 范围内不引入区间重排,避免与迁移期数据冲突)。
  322. func (r *DivideRepo) UpdateNodeParent(ctx context.Context, tx *gorm.DB, userID, parentUserID, rootUserID string, layer int32) (int64, error) {
  323. if err := requireTx(tx, "UpdateNodeParent"); err != nil {
  324. return 0, err
  325. }
  326. if strings.TrimSpace(userID) == "" || strings.TrimSpace(parentUserID) == "" || strings.TrimSpace(rootUserID) == "" {
  327. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateNodeParent 需要 user/parent/root"))
  328. }
  329. if layer < 1 {
  330. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateNodeParent layer 必须 >= 1"))
  331. }
  332. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  333. Where("user_id = ?", userID).
  334. Updates(map[string]any{
  335. "parent_user_id": parentUserID,
  336. "root_user_id": rootUserID,
  337. "layer": layer,
  338. })
  339. if res.Error != nil {
  340. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  341. }
  342. return res.RowsAffected, nil
  343. }
  344. // UpdateNodeQuota 更新节点分成额度(不触碰左右值/收入)。
  345. //
  346. // 入参:ctx;tx(**必传**);userID 节点;quota 新额度(单位:分;负 → ErrBadParam)。
  347. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  348. //
  349. // 红线:**必须**与 ClampSubtreeQuota 同事务(否则出现"父降了、子仍高于父"的越权额度窗口)。
  350. func (r *DivideRepo) UpdateNodeQuota(ctx context.Context, tx *gorm.DB, userID string, quota money.Amount) (int64, error) {
  351. if err := requireTx(tx, "UpdateNodeQuota"); err != nil {
  352. return 0, err
  353. }
  354. if strings.TrimSpace(userID) == "" {
  355. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateNodeQuota 需要 user_id"))
  356. }
  357. if err := quota.Validate(); err != nil {
  358. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
  359. }
  360. if quota.IsNegative() {
  361. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: 分成额度不能为负"))
  362. }
  363. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  364. Where("user_id = ?", userID).
  365. UpdateColumn("quota", quota.String())
  366. if res.Error != nil {
  367. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  368. }
  369. return res.RowsAffected, nil
  370. }
  371. // ClampSubtreeQuota 额度级联下降:把子树内**同分成类型**且**超过**上限的节点统一压到上限。
  372. //
  373. // 入参:
  374. // - ctx;tx(**必传**);
  375. // - leftVal/rightVal int32 父节点区间(rightVal <= leftVal → ErrBadParam);
  376. // - divideType 分成类型过滤(model.DivideTypeOneOff/PerLesson;**0 = 不筛**,仅 admin 全量场景);
  377. // - quota 上限(单位:分;负 → ErrBadParam)。
  378. //
  379. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  380. //
  381. // SQL(audit A6 / dump `yuansiwei_finance.sql:763872-763873` 原文条件,**缺一不可**):
  382. //
  383. // UPDATE t_user_divide_into SET quota=updateQuota
  384. // WHERE left_val>uLeft AND right_val<uRight AND quota>updateQuota AND type=uType;
  385. //
  386. // 语义("子不得超父"):只降不升——低于上限的节点保持原额度(分别配置的阶梯额度不被抹平)。
  387. // 红线:`type` 过滤不可省 —— 否则一次付清(1)与按课次(2)两条线的额度会被跨类型抹平(漏实现 = 缺陷)。
  388. func (r *DivideRepo) ClampSubtreeQuota(ctx context.Context, tx *gorm.DB, leftVal, rightVal int32, divideType int8, quota money.Amount) (int64, error) {
  389. if err := requireTx(tx, "ClampSubtreeQuota"); err != nil {
  390. return 0, err
  391. }
  392. if leftVal < 0 || rightVal <= leftVal {
  393. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ClampSubtreeQuota 左右值区间非法"))
  394. }
  395. if divideType != 0 && divideType != model.DivideTypeOneOff && divideType != model.DivideTypePerLesson {
  396. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ClampSubtreeQuota 分成类型非法"))
  397. }
  398. if err := quota.Validate(); err != nil {
  399. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
  400. }
  401. if quota.IsNegative() {
  402. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: 分成额度不能为负"))
  403. }
  404. q := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  405. Where("left_val > ? AND right_val < ?", leftVal, rightVal).
  406. Where("quota > ?", quota.String())
  407. if divideType != 0 {
  408. q = q.Where("type = ?", divideType)
  409. }
  410. res := q.UpdateColumn("quota", quota.String())
  411. if res.Error != nil {
  412. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  413. }
  414. return res.RowsAffected, nil
  415. }
  416. // IncTotalIncome 累加节点分成总收入(上行分成 +,撤销回滚 −)。
  417. //
  418. // 入参:ctx;tx(**必传**);userID 节点;delta 增量(单位:分;0 → 空操作)。
  419. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  420. //
  421. // 红线:与 InsertLog/DeleteLog 同事务;金额字符串入场(decimal 精确,零 float64)。
  422. func (r *DivideRepo) IncTotalIncome(ctx context.Context, tx *gorm.DB, userID string, delta money.Amount) (int64, error) {
  423. if err := requireTx(tx, "IncTotalIncome"); err != nil {
  424. return 0, err
  425. }
  426. if strings.TrimSpace(userID) == "" {
  427. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: IncTotalIncome 需要 user_id"))
  428. }
  429. if delta.IsZero() {
  430. return 0, nil
  431. }
  432. if err := delta.Validate(); err != nil {
  433. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
  434. }
  435. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  436. Where("user_id = ?", userID).
  437. UpdateColumn("total_income", gorm.Expr("total_income + ?", delta.String()))
  438. if res.Error != nil {
  439. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  440. }
  441. return res.RowsAffected, nil
  442. }
  443. // UpdateLogOrderNo 改绑分成明细的订单号(撤销分支一:保留分成,改指首次扣费上课记录)。
  444. //
  445. // 入参:ctx;tx(**必传**);id 明细 ID;orderNo 新订单号(上课记录 id 字符串);remark 备注(可空)。
  446. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  447. //
  448. // 红线(dump 口径):改绑**不改 money**、**不回滚 total_income**(分成保留);
  449. // order_no 是 type=2 的幂等键组成列,改绑即改幂等语义 → 必须写审计备注(默认"推广分成")。
  450. func (r *DivideRepo) UpdateLogOrderNo(ctx context.Context, tx *gorm.DB, id int64, orderNo, remark string) (int64, error) {
  451. if err := requireTx(tx, "UpdateLogOrderNo"); err != nil {
  452. return 0, err
  453. }
  454. if id <= 0 || strings.TrimSpace(orderNo) == "" {
  455. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateLogOrderNo 需要 id/order_no"))
  456. }
  457. // create_time 重置为 DB 当前时间(dump 改绑分支:`create_time=now()`,让改绑后的明细排到最新)
  458. fields := map[string]any{"order_no": orderNo, "create_time": gorm.Expr("CURRENT_TIMESTAMP")}
  459. if strings.TrimSpace(remark) != "" {
  460. fields["remark"] = remark
  461. }
  462. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideIntoLog{}).
  463. Where("id = ?", id).
  464. Updates(fields)
  465. if res.Error != nil {
  466. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  467. }
  468. return res.RowsAffected, nil
  469. }
  470. // IsNotFound 判断错误是否为仓储 ErrNotFound(供 Service 分支判断,避免各处抄 errors.Is)。
  471. //
  472. // 入参:err error(可 nil)。出参:true 表示"记录不存在"。
  473. func IsNotFound(err error) bool { return err != nil && errors.Is(err, infraerrors.ErrNotFound) }
  474. // ---------------------------------------------------------------- AC-13 查询原语(端点 38 / 43)
  475. // ListAllNodes 取全部分账节点(Nested Set,按 `root_user_id, left_val` 排序)。
  476. //
  477. // 入参:ctx。
  478. // 出参:
  479. // - []model.UserDivideInto:稳定顺序(同 root 内按 left 升序 ⇒ 天然是树的先序);
  480. // - ErrDBError。
  481. //
  482. // 作用域:`t_user_divide_into` **无** institution_id / campus_id(全局推广体系,设计 §1.7 例外一),
  483. // 故不叠 Scoped;可见性由端点角色白名单(super_admin/finance/market)控制,见 handler 注释。
  484. //
  485. // 为什么不在此拼树:拼树 + 树不变式校验属业务规则(layer = parent.layer+1、
  486. // 区间嵌套或相离),放 Service 才能单测(`service.divide_tree.go`)。
  487. func (r *DivideRepo) ListAllNodes(ctx context.Context, rootUserID string) ([]model.UserDivideInto, error) {
  488. db, err := r.db(ctx)
  489. if err != nil {
  490. return nil, err
  491. }
  492. q := db.Model(&model.UserDivideInto{})
  493. if strings.TrimSpace(rootUserID) != "" {
  494. // 端点 38 的 `rootUserId?`:只取某一棵树(多树共存时前端按需拉取,避免全量森林)
  495. q = q.Where("root_user_id = ?", strings.TrimSpace(rootUserID))
  496. }
  497. rows := make([]model.UserDivideInto, 0, 64)
  498. if qErr := q.Order("root_user_id ASC").Order("left_val ASC").Find(&rows).Error; qErr != nil {
  499. return nil, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
  500. }
  501. return rows, nil
  502. }
  503. // ListAncestorIDs 取某节点的**祖先链**(端点 39 的 `{ancestorIds:[]}`,设计 §3.3 #21 末尾原语)。
  504. //
  505. // 入参:ctx;userID 新加入的下线;leftVal/rightVal/rootUserID 其 Nested Set 区间与树根。
  506. // 出参:[]string 祖先 user_id(**不含自身**;由内向外或由外向内按 `left_val ASC` = 由外向内);
  507. //
  508. // ErrDBError。
  509. //
  510. // dump 原文:`SELECT userid ... WHERE left<userLeft AND right>userRight AND root_userid=root`
  511. // ⇒ 天然含"区间包含该节点"的所有祖先;**root 限定不可省**(D5:跨树区间数值可能重叠,
  512. // 缺 root 会把别的树的节点算成本节点祖先)。
  513. func (r *DivideRepo) ListAncestorIDs(
  514. ctx context.Context, leftVal, rightVal int32, rootUserID string,
  515. ) ([]string, error) {
  516. db, err := r.db(ctx)
  517. if err != nil {
  518. return nil, err
  519. }
  520. ids := make([]string, 0, 8)
  521. q := db.Model(&model.UserDivideInto{}).
  522. Where("left_val < ? AND right_val > ?", leftVal, rightVal)
  523. if strings.TrimSpace(rootUserID) != "" {
  524. q = q.Where("root_user_id = ?", strings.TrimSpace(rootUserID))
  525. }
  526. if qErr := q.Order("left_val ASC").Pluck("user_id", &ids).Error; qErr != nil {
  527. return nil, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
  528. }
  529. return ids, nil
  530. }
  531. // ListSubtreeUserIDsWithQuotaAtLeast 取子树内 `quota >= 阈值` 的 user_id(端点 40 的 `{affectedUserIds:[]}`)。
  532. //
  533. // 入参:ctx;leftVal/rightVal 子树区间(含自身);quota 新额度(调用方已转为分)。
  534. // 出参:[]string 受影响 user_id(`left_val ASC` 由外向内);ErrDBError。
  535. //
  536. // dump 原文:`SELECT userid ... WHERE left>=uLeft AND right<=uRight AND quota>=newQuota`(含自身)。
  537. // 该查询在 `update_quota` **末尾**执行 ⇒ 本方法应在 UpdateQuota 之后调用(读已提交状态)。
  538. func (r *DivideRepo) ListSubtreeUserIDsWithQuotaAtLeast(
  539. ctx context.Context, leftVal, rightVal int32, quota money.Amount,
  540. ) ([]string, error) {
  541. db, err := r.db(ctx)
  542. if err != nil {
  543. return nil, err
  544. }
  545. ids := make([]string, 0, 8)
  546. if qErr := db.Model(&model.UserDivideInto{}).
  547. Where("left_val >= ? AND right_val <= ? AND quota >= ?", leftVal, rightVal, quota.String()).
  548. Order("left_val ASC").Pluck("user_id", &ids).Error; qErr != nil {
  549. return nil, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
  550. }
  551. return ids, nil
  552. }
  553. // DivideLogFilter 分账流水筛选(端点 43)。
  554. //
  555. // 字段语义(nil/空 = 不筛):
  556. // - FindUserID 分成获得者(**表列是 `user_id`**;契约响应字段名才叫 `findUserId`);StudentID 学生;
  557. // - From/To:`create_time >= from AND create_time < to`(**右开**,与契约 §1.3 一致)。
  558. type DivideLogFilter struct {
  559. // FindUserID 分成获得者(ULID;空 = 不筛)。
  560. FindUserID string
  561. // StudentID 学生(ULID;空 = 不筛)。
  562. StudentID string
  563. // From 起始时间(含;nil = 不筛)。
  564. From *time.Time
  565. // To 结束时间(**不含**;nil = 不筛)。
  566. To *time.Time
  567. }
  568. // ListLogsFiltered 分页查分账流水(端点 43,带用户/学生/时间窗筛选,附总条数)。
  569. //
  570. // 入参:ctx;f DivideLogFilter;page PageQuery(内部钳制 pageSize ∈ [1,200])。
  571. // 出参:
  572. // - []model.UserDivideIntoLog:`create_time DESC, id DESC`;空结果非 nil 切片;
  573. // - int64:同条件总条数(**与列表逐字同 WHERE**,同快照);
  574. // - ErrDBError。
  575. //
  576. // ⚠ 该端点**无 dump SP 对照**(dump 只有 add/delete/update_quota 三个 SP)⇒ 排序与窗口由 Go 定,
  577. // 已登记为「Go 新增查询」(契约 v1.4 端点 43 注)。
  578. func (r *DivideRepo) ListLogsFiltered(
  579. ctx context.Context, f DivideLogFilter, page PageQuery,
  580. ) ([]model.UserDivideIntoLog, int64, error) {
  581. db, err := r.db(ctx)
  582. if err != nil {
  583. return nil, 0, err
  584. }
  585. base := db.Model(&model.UserDivideIntoLog{})
  586. if f.FindUserID != "" {
  587. base = base.Where("user_id = ?", f.FindUserID)
  588. }
  589. if f.StudentID != "" {
  590. base = base.Where("student_id = ?", f.StudentID)
  591. }
  592. if f.From != nil {
  593. base = base.Where("create_time >= ?", *f.From)
  594. }
  595. if f.To != nil {
  596. base = base.Where("create_time < ?", *f.To)
  597. }
  598. var total int64
  599. if cErr := base.Count(&total).Error; cErr != nil {
  600. return nil, 0, infraerrors.WithCause(infraerrors.ErrDBError, cErr)
  601. }
  602. pageSize := page.Limit()
  603. rows := make([]model.UserDivideIntoLog, 0, pageSize)
  604. if qErr := base.Order("create_time DESC").Order("id DESC").
  605. Limit(pageSize).Offset(page.Offset()).Find(&rows).Error; qErr != nil {
  606. return nil, 0, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
  607. }
  608. if rows == nil {
  609. rows = make([]model.UserDivideIntoLog, 0)
  610. }
  611. return rows, total, nil
  612. }