divide_repo.go 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648
  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, rootUserID string, 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 strings.TrimSpace(rootUserID) == "" {
  393. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ClampSubtreeQuota 需要 root_user_id"))
  394. }
  395. if leftVal < 0 || rightVal <= leftVal {
  396. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ClampSubtreeQuota 左右值区间非法"))
  397. }
  398. if divideType != 0 && divideType != model.DivideTypeOneOff && divideType != model.DivideTypePerLesson {
  399. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ClampSubtreeQuota 分成类型非法"))
  400. }
  401. if err := quota.Validate(); err != nil {
  402. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
  403. }
  404. if quota.IsNegative() {
  405. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: 分成额度不能为负"))
  406. }
  407. q := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  408. // root_user_id 过滤**必须**在:`left_val`/`right_val` 只在**同一棵树内**唯一。
  409. // 每棵树的 left 都从 1 起 ⇒ 多个树根的区间在数值上必然互相重叠
  410. //(真实数据 26 棵树、跨树重叠 325 对 = C(26,2) 满重叠)。
  411. // 缺这一条 ⇒ 调 A 树的额会压低 B 树里区间数值重叠的节点(台账 F25,P0)。
  412. Where("root_user_id = ?", strings.TrimSpace(rootUserID)).
  413. Where("left_val > ? AND right_val < ?", leftVal, rightVal).
  414. Where("quota > ?", quota.String())
  415. if divideType != 0 {
  416. q = q.Where("type = ?", divideType)
  417. }
  418. res := q.UpdateColumn("quota", quota.String())
  419. if res.Error != nil {
  420. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  421. }
  422. return res.RowsAffected, nil
  423. }
  424. // IncTotalIncome 累加节点分成总收入(上行分成 +,撤销回滚 −)。
  425. //
  426. // 入参:ctx;tx(**必传**);userID 节点;delta 增量(单位:分;0 → 空操作)。
  427. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  428. //
  429. // 红线:与 InsertLog/DeleteLog 同事务;金额字符串入场(decimal 精确,零 float64)。
  430. func (r *DivideRepo) IncTotalIncome(ctx context.Context, tx *gorm.DB, userID string, delta money.Amount) (int64, error) {
  431. if err := requireTx(tx, "IncTotalIncome"); err != nil {
  432. return 0, err
  433. }
  434. if strings.TrimSpace(userID) == "" {
  435. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: IncTotalIncome 需要 user_id"))
  436. }
  437. if delta.IsZero() {
  438. return 0, nil
  439. }
  440. if err := delta.Validate(); err != nil {
  441. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
  442. }
  443. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
  444. Where("user_id = ?", userID).
  445. UpdateColumn("total_income", gorm.Expr("total_income + ?", delta.String()))
  446. if res.Error != nil {
  447. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  448. }
  449. return res.RowsAffected, nil
  450. }
  451. // UpdateLogOrderNo 改绑分成明细的订单号(撤销分支一:保留分成,改指首次扣费上课记录)。
  452. //
  453. // 入参:ctx;tx(**必传**);id 明细 ID;orderNo 新订单号(上课记录 id 字符串);remark 备注(可空)。
  454. // 出参:int64 受影响行数;ErrBadParam / ErrDBError。
  455. //
  456. // 红线(dump 口径):改绑**不改 money**、**不回滚 total_income**(分成保留);
  457. // order_no 是 type=2 的幂等键组成列,改绑即改幂等语义 → 必须写审计备注(默认"推广分成")。
  458. func (r *DivideRepo) UpdateLogOrderNo(ctx context.Context, tx *gorm.DB, id int64, orderNo, remark string) (int64, error) {
  459. if err := requireTx(tx, "UpdateLogOrderNo"); err != nil {
  460. return 0, err
  461. }
  462. if id <= 0 || strings.TrimSpace(orderNo) == "" {
  463. return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateLogOrderNo 需要 id/order_no"))
  464. }
  465. // create_time 重置为 DB 当前时间(dump 改绑分支:`create_time=now()`,让改绑后的明细排到最新)
  466. fields := map[string]any{"order_no": orderNo, "create_time": gorm.Expr("CURRENT_TIMESTAMP")}
  467. if strings.TrimSpace(remark) != "" {
  468. fields["remark"] = remark
  469. }
  470. res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideIntoLog{}).
  471. Where("id = ?", id).
  472. Updates(fields)
  473. if res.Error != nil {
  474. return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
  475. }
  476. return res.RowsAffected, nil
  477. }
  478. // IsNotFound 判断错误是否为仓储 ErrNotFound(供 Service 分支判断,避免各处抄 errors.Is)。
  479. //
  480. // 入参:err error(可 nil)。出参:true 表示"记录不存在"。
  481. func IsNotFound(err error) bool { return err != nil && errors.Is(err, infraerrors.ErrNotFound) }
  482. // ---------------------------------------------------------------- AC-13 查询原语(端点 38 / 43)
  483. // ListAllNodes 取全部分账节点(Nested Set,按 `root_user_id, left_val` 排序)。
  484. //
  485. // 入参:ctx。
  486. // 出参:
  487. // - []model.UserDivideInto:稳定顺序(同 root 内按 left 升序 ⇒ 天然是树的先序);
  488. // - ErrDBError。
  489. //
  490. // 作用域:`t_user_divide_into` **无** institution_id / campus_id(全局推广体系,设计 §1.7 例外一),
  491. // 故不叠 Scoped;可见性由端点角色白名单(super_admin/finance/market)控制,见 handler 注释。
  492. //
  493. // 为什么不在此拼树:拼树 + 树不变式校验属业务规则(layer = parent.layer+1、
  494. // 区间嵌套或相离),放 Service 才能单测(`service.divide_tree.go`)。
  495. func (r *DivideRepo) ListAllNodes(ctx context.Context, rootUserID string) ([]model.UserDivideInto, error) {
  496. db, err := r.db(ctx)
  497. if err != nil {
  498. return nil, err
  499. }
  500. q := db.Model(&model.UserDivideInto{})
  501. if strings.TrimSpace(rootUserID) != "" {
  502. // 端点 38 的 `rootUserId?`:只取某一棵树(多树共存时前端按需拉取,避免全量森林)
  503. q = q.Where("root_user_id = ?", strings.TrimSpace(rootUserID))
  504. }
  505. rows := make([]model.UserDivideInto, 0, 64)
  506. if qErr := q.Order("root_user_id ASC").Order("left_val ASC").Find(&rows).Error; qErr != nil {
  507. return nil, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
  508. }
  509. return rows, nil
  510. }
  511. // ListAncestorIDs 取某节点的**祖先链**(端点 39 的 `{ancestorIds:[]}`,设计 §3.3 #21 末尾原语)。
  512. //
  513. // 入参:ctx;userID 新加入的下线;leftVal/rightVal/rootUserID 其 Nested Set 区间与树根。
  514. // 出参:[]string 祖先 user_id(**不含自身**;由内向外或由外向内按 `left_val ASC` = 由外向内);
  515. //
  516. // ErrDBError。
  517. //
  518. // dump 原文:`SELECT userid ... WHERE left<userLeft AND right>userRight AND root_userid=root`
  519. // ⇒ 天然含"区间包含该节点"的所有祖先;**root 限定不可省**(D5:跨树区间数值可能重叠,
  520. // 缺 root 会把别的树的节点算成本节点祖先)。
  521. func (r *DivideRepo) ListAncestorIDs(
  522. ctx context.Context, leftVal, rightVal int32, rootUserID string,
  523. ) ([]string, error) {
  524. db, err := r.db(ctx)
  525. if err != nil {
  526. return nil, err
  527. }
  528. ids := make([]string, 0, 8)
  529. q := db.Model(&model.UserDivideInto{}).
  530. Where("left_val < ? AND right_val > ?", leftVal, rightVal)
  531. if strings.TrimSpace(rootUserID) != "" {
  532. q = q.Where("root_user_id = ?", strings.TrimSpace(rootUserID))
  533. }
  534. if qErr := q.Order("left_val ASC").Pluck("user_id", &ids).Error; qErr != nil {
  535. return nil, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
  536. }
  537. return ids, nil
  538. }
  539. // ListSubtreeUserIDsWithQuotaAtLeast 取子树内 `quota >= 阈值` 的 user_id(端点 40 的 `{affectedUserIds:[]}`)。
  540. //
  541. // 入参:ctx;leftVal/rightVal 子树区间(含自身);quota 新额度(调用方已转为分)。
  542. // 出参:[]string 受影响 user_id(`left_val ASC` 由外向内);ErrDBError。
  543. //
  544. // dump 原文:`SELECT userid ... WHERE left>=uLeft AND right<=uRight AND quota>=newQuota`(含自身)。
  545. // 该查询在 `update_quota` **末尾**执行 ⇒ 本方法应在 UpdateQuota 之后调用(读已提交状态)。
  546. func (r *DivideRepo) ListSubtreeUserIDsWithQuotaAtLeast(
  547. ctx context.Context, leftVal, rightVal int32, quota money.Amount,
  548. ) ([]string, error) {
  549. db, err := r.db(ctx)
  550. if err != nil {
  551. return nil, err
  552. }
  553. ids := make([]string, 0, 8)
  554. if qErr := db.Model(&model.UserDivideInto{}).
  555. Where("left_val >= ? AND right_val <= ? AND quota >= ?", leftVal, rightVal, quota.String()).
  556. Order("left_val ASC").Pluck("user_id", &ids).Error; qErr != nil {
  557. return nil, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
  558. }
  559. return ids, nil
  560. }
  561. // DivideLogFilter 分账流水筛选(端点 43)。
  562. //
  563. // 字段语义(nil/空 = 不筛):
  564. // - FindUserID 分成获得者(**表列是 `user_id`**;契约响应字段名才叫 `findUserId`);StudentID 学生;
  565. // - From/To:`create_time >= from AND create_time < to`(**右开**,与契约 §1.3 一致)。
  566. type DivideLogFilter struct {
  567. // FindUserID 分成获得者(ULID;空 = 不筛)。
  568. FindUserID string
  569. // StudentID 学生(ULID;空 = 不筛)。
  570. StudentID string
  571. // From 起始时间(含;nil = 不筛)。
  572. From *time.Time
  573. // To 结束时间(**不含**;nil = 不筛)。
  574. To *time.Time
  575. }
  576. // ListLogsFiltered 分页查分账流水(端点 43,带用户/学生/时间窗筛选,附总条数)。
  577. //
  578. // 入参:ctx;f DivideLogFilter;page PageQuery(内部钳制 pageSize ∈ [1,200])。
  579. // 出参:
  580. // - []model.UserDivideIntoLog:`create_time DESC, id DESC`;空结果非 nil 切片;
  581. // - int64:同条件总条数(**与列表逐字同 WHERE**,同快照);
  582. // - ErrDBError。
  583. //
  584. // ⚠ 该端点**无 dump SP 对照**(dump 只有 add/delete/update_quota 三个 SP)⇒ 排序与窗口由 Go 定,
  585. // 已登记为「Go 新增查询」(契约 v1.4 端点 43 注)。
  586. func (r *DivideRepo) ListLogsFiltered(
  587. ctx context.Context, f DivideLogFilter, page PageQuery,
  588. ) ([]model.UserDivideIntoLog, int64, error) {
  589. db, err := r.db(ctx)
  590. if err != nil {
  591. return nil, 0, err
  592. }
  593. base := db.Model(&model.UserDivideIntoLog{})
  594. if f.FindUserID != "" {
  595. base = base.Where("user_id = ?", f.FindUserID)
  596. }
  597. if f.StudentID != "" {
  598. base = base.Where("student_id = ?", f.StudentID)
  599. }
  600. if f.From != nil {
  601. base = base.Where("create_time >= ?", *f.From)
  602. }
  603. if f.To != nil {
  604. base = base.Where("create_time < ?", *f.To)
  605. }
  606. var total int64
  607. if cErr := base.Count(&total).Error; cErr != nil {
  608. return nil, 0, infraerrors.WithCause(infraerrors.ErrDBError, cErr)
  609. }
  610. pageSize := page.Limit()
  611. rows := make([]model.UserDivideIntoLog, 0, pageSize)
  612. if qErr := base.Order("create_time DESC").Order("id DESC").
  613. Limit(pageSize).Offset(page.Offset()).Find(&rows).Error; qErr != nil {
  614. return nil, 0, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
  615. }
  616. if rows == nil {
  617. rows = make([]model.UserDivideIntoLog, 0)
  618. }
  619. return rows, total, nil
  620. }