divide_service.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569
  1. // 分账服务(Nested Set 树维护 + 上行分成 + 撤销 + 额度级联)——AC-09。
  2. //
  3. // 依据:unified/03-migration/academic-domain-design.md §3.3 #22~#25(分账 4 SP)
  4. // / §1.7(全局无作用域表)/ §10.3(事务边界只在 Service)/ §11.1 拍板 11-12。
  5. //
  6. // 四个 SP 的 Go 化落点:
  7. //
  8. // divide_into_add → AddSubordinate(**服务端算左右值**,dump 2 参语义:只传上下级)
  9. // divide_into_log_add → AddLog(上行链入账;**递归改迭代 BFS**,深度上限 100)
  10. // divide_into_log_delete → DeleteLog(两分支:改绑 / 回滚删除,dump 2 参语义:studentid + orderid)
  11. // divide_into_update_quota → UpdateQuota(改额度 + 子树级联下降,子不得超父 → 3111)
  12. //
  13. // 事务纪律:所有写路径走 `repository.TxRunner.WithinTx`;事务内下游调用一律传
  14. // `repository.WithTxContext(ctx, tx)` 注入的 ctx(读方法自动复用同一 tx,见仓储 db(ctx))。
  15. // 禁止在本服务内直接调用 `db.Transaction`(嵌套只允许经 TxRunner 透传)。
  16. //
  17. // 金额纪律:库里 quota / total_income / money 均为 decimal(12,2) 字符串列,
  18. // 一律 `money.Parse` 转 money.Amount(int64 分)后运算,**零 float64**。
  19. package service
  20. import (
  21. "context"
  22. "errors"
  23. "strconv"
  24. "strings"
  25. "gorm.io/gorm"
  26. "yuansiwei/unified-backend/internal/academic/apperr"
  27. "yuansiwei/unified-backend/internal/academic/model"
  28. "yuansiwei/unified-backend/internal/academic/money"
  29. "yuansiwei/unified-backend/internal/academic/repository"
  30. infraerrors "yuansiwei/unified-backend/internal/infra/errors"
  31. )
  32. // MaxDivideDepth 分账树深度上限(根 layer=1;超限 → apperr.ErrDivideQuotaExceeded 3111)。
  33. //
  34. // 依据:模型注释 layer「深度上限 100(超限 → 3111,防递归栈溢出)」+ dump 的层数保护。
  35. // BFS 迭代在本上限内必然终止(即使数据成环也不会死循环)。
  36. const MaxDivideDepth int32 = 100
  37. // remarkRebind 撤销改绑分支写入的备注文案(dump 原样:"推广分成")。
  38. const remarkRebind = "推广分成"
  39. // FirstChargedRecordReader 首次扣费上课记录只读端口(跨聚合依赖,本服务**只定义契约**)。
  40. //
  41. // 为什么用端口而不是直接 JOIN:`log_delete` 的改绑分支需要判断"学生是否已有首次扣费上课记录",
  42. // 该判定属于课次/打卡聚合(AC-05/AC-10 范围)。AC-09 不实现它,只声明只读契约 + 允许注入;
  43. // 未注入时**所有撤销走回滚删除分支**(保守:不让分成悬空在多订单上)。
  44. //
  45. // 契约(**取数口径已写死,实现方不得自选**,dump `yuansiwei_finance.sql:763811`):
  46. // - **4 个谓词缺一不可**:`student_id`(本人)、`cost > 0`(真扣费,排除免费/试听)、
  47. // `end_time IS NOT NULL`(已结课,未结课不算"首次扣费")、`ORDER BY start_time ASC LIMIT 1`(最早);
  48. // - **返回的是 `recordid` 列,不是 `record_student.id`**(dump 原文取 recordid;
  49. // 取错列会让撤销时的 order_no 与明细侧对不上 → 改绑失效、分成悬空);
  50. // - 只读,禁止写任何表;
  51. // - 返回 (recordID, found, err):found=false 表示学生尚无扣费记录 → 走回滚删除分支;
  52. // - err 仅 DB 异常(infraerrors.ErrDBError),不得返回业务错误。
  53. //
  54. // 参考 SQL(口径以此为准):
  55. //
  56. // SELECT recordid FROM t_courses_record_student
  57. // WHERE student_id = ? AND cost > 0 AND end_time IS NOT NULL
  58. // ORDER BY start_time ASC LIMIT 1;
  59. type FirstChargedRecordReader interface {
  60. // FirstChargedRecordID 取该学生"首次扣费上课记录"的 ID。
  61. //
  62. // 入参:ctx 请求上下文;studentID 报课学生(canonical_user_id ULID)。
  63. // 出参:recordID int64(上课记录 ID);found bool;error。
  64. FirstChargedRecordID(ctx context.Context, studentID string) (recordID int64, found bool, err error)
  65. }
  66. // DivideService 分账服务(全局推广体系;越权防线在 Handler 角色白名单,本服务不做租户裁剪)。
  67. type DivideService struct {
  68. // repo 分账树仓储(全局数据,不继承 BaseRepo)。
  69. repo *repository.DivideRepo
  70. // runner 事务运行器(顶层开事务,事务内透传)。
  71. runner *repository.TxRunner
  72. // firstCharged 首次扣费记录端口(可 nil:未注入时撤销一律走回滚删除)。
  73. firstCharged FirstChargedRecordReader
  74. }
  75. // NewDivideService 构造分账服务(repo 与 runner 绑定同一 DB 句柄,避免跨连接池事务失效)。
  76. //
  77. // 入参:
  78. // - db *gorm.DB:数据库句柄(必填)。
  79. // - firstCharged FirstChargedRecordReader:首次扣费记录端口(可 nil)。
  80. //
  81. // 出参:*DivideService。
  82. func NewDivideService(db *gorm.DB, firstCharged FirstChargedRecordReader) *DivideService {
  83. return &DivideService{
  84. repo: repository.NewDivideRepo(db),
  85. runner: repository.NewTxRunner(db),
  86. firstCharged: firstCharged,
  87. }
  88. }
  89. // AddSubordinate 把 subUserID 挂为 parentUserID 的下级(dump `divide_into_add` 2 参语义)。
  90. //
  91. // 入参:
  92. // - ctx 请求上下文;
  93. // - parentUserID 上级用户(ULID;必须已是树节点,否则 3001);
  94. // - subUserID 下级用户(ULID;== parentUserID → 3140,禁止自环)。
  95. //
  96. // 出参:
  97. // - nil:成功(新建节点,或"已存在节点"分支改挂成功);
  98. // - ErrNotFound(3001):上级不存在或不在分账体系;
  99. // - ErrDivideQuotaExceeded(3111):层级超上限(父 layer+1 > 100);
  100. // - ErrBusinessParam(3140):入参非法。
  101. //
  102. // 左右值算法(Nested Set 插入最右子节点,**服务端计算**,调用方不传 left/right):
  103. //
  104. // ① 父节点右值 R = parent.right_val(插入点)
  105. // ② ShiftRight(R, +2):right_val >= R 全体 +2 → 父节点及其**全部祖先**区间右移
  106. // ③ ShiftLeft (R, +2):left_val >= R 全体 +2 → 位于插入点右侧的兄弟及其后代右移
  107. // ④ 新节点 (left, right) = (R, R+1),layer = parent.layer+1,root = 父的树根
  108. //
  109. // 不变式:①②③④ 必须在同一事务内完成(否则中间态树不可用);完成后恒有 left < right。
  110. //
  111. // 已存在节点分支(audit D4,dump `yuansiwei_finance.sql:763695`):
  112. //
  113. // dump 的判断是 `if exists(select 1 from t_user_divide_into where userid = subid) then ... end if;`
  114. // 分支体内**没有**任何改挂父链的动作 —— 即"已存在 = 完全 no-op(幂等成功,返回 nil)"。
  115. // 本实现按 dump 对齐:不再改 parent/root/layer(改挂会凭空移动整棵子树且不重排左右值,
  116. // 直接破坏嵌套集不变式)。
  117. func (s *DivideService) AddSubordinate(ctx context.Context, parentUserID, subUserID string) error {
  118. parentUserID = strings.TrimSpace(parentUserID)
  119. subUserID = strings.TrimSpace(subUserID)
  120. if parentUserID == "" || subUserID == "" {
  121. return infraerrors.WithCause(apperr.ErrBusinessParam, errString("service: AddSubordinate 需要 parent/sub user id"))
  122. }
  123. if parentUserID == subUserID {
  124. return infraerrors.WithCause(apperr.ErrBusinessParam, errString("service: 下级不能是自身(禁止成环)"))
  125. }
  126. if err := s.assertReady(); err != nil {
  127. return err
  128. }
  129. return s.runner.WithinTx(ctx, func(tx *gorm.DB) error {
  130. txCtx := repository.WithTxContext(ctx, tx)
  131. parent, err := s.repo.GetByUserID(txCtx, parentUserID)
  132. if err != nil {
  133. return err // ErrNotFound(3001) 原样上抛:上级未加入分账体系
  134. }
  135. nextLayer := parent.Layer + 1
  136. if nextLayer > MaxDivideDepth {
  137. return infraerrors.WithCause(apperr.ErrDivideQuotaExceeded,
  138. errString("service: 分账层级超上限 "+strconv.Itoa(int(MaxDivideDepth))))
  139. }
  140. rootID := resolveRoot(parent)
  141. _, err = s.repo.GetByUserID(txCtx, subUserID)
  142. switch {
  143. case err == nil:
  144. // ── 已存在节点:完全 no-op(dump 763695 的分支体没有任何写动作)
  145. return nil
  146. case !errors.Is(err, infraerrors.ErrNotFound):
  147. return err
  148. }
  149. // ── 新建节点分支:嵌套集"插入最右子节点"
  150. insertAt := parent.RightVal
  151. // root 限定(audit V-02):位移只作用于本树,否则会串改同库其它树的左右值区间
  152. if _, serr := s.repo.ShiftRight(txCtx, tx, rootID, insertAt, 2); serr != nil {
  153. return serr
  154. }
  155. if _, serr := s.repo.ShiftLeft(txCtx, tx, rootID, insertAt, 2); serr != nil {
  156. return serr
  157. }
  158. parentID := parentUserID
  159. node := &model.UserDivideInto{
  160. UserID: subUserID,
  161. // audit D6:新节点固定「一次付清(1)」,**不继承父**(dump 763714 的 INSERT 不写 type,
  162. // 落 DDL 默认值 1)。继承父会把"按课次"格式传染整条下级链 → 幂等维度(1 vs 2)跟着错。
  163. Type: model.DivideTypeOneOff,
  164. Quota: "0.00", // 新节点额度由后台 UpdateQuota 配置(默认 0,且不得超父)
  165. TotalIncome: "0.00",
  166. TotalIncomePayed: "0.00",
  167. RootUserID: &rootID,
  168. ParentUserID: &parentID,
  169. LeftVal: insertAt,
  170. RightVal: insertAt + 1,
  171. Layer: nextLayer,
  172. }
  173. if ierr := s.repo.InsertNode(txCtx, tx, node); ierr != nil {
  174. return ierr
  175. }
  176. return nil
  177. })
  178. }
  179. // AddLog 上行分成入账(dump `divide_into_log_add`;**递归改迭代 BFS**)。
  180. //
  181. // 入参:
  182. // - ctx 请求上下文;
  183. // - findUserID 起始用户(ULID;通常是直接推荐人);
  184. // - stuID 报课学生(ULID;幂等键组成列);
  185. // - recordID 上课记录 ID(int64;写入 order_no 作为 type=2 的幂等键,dump 传 orderid 语义);
  186. // - remark 备注(可空;空则落 NULL)。
  187. //
  188. // 出参:
  189. // - nil:成功(含"起始用户不是树节点 → 不入账"的空操作、以及幂等命中);
  190. // - ErrDivideQuotaExceeded(3111):上行层数超 100(数据成环/异常链);
  191. // - ErrBusinessParam / ErrDBError;
  192. // - 若节点 quota 金额被写坏(非十进制)→ ErrDBError(数据异常,不静默归零)。
  193. //
  194. // 上行链语义(**逐字对齐 dump `yuansiwei_finance.sql:763744`**):
  195. //
  196. // DECLARE cur CURSOR FOR SELECT userid,type,parent_userid,quota FROM t_user_divide_into
  197. // WHERE userid = (SELECT parent_userid FROM t_user_divide_into WHERE userid = findUserid);
  198. //
  199. // 起点 = **findUserid 的上级**(本人不参与分成,本人只是"被推荐人"),随后逐级上溯 parent_user_id,
  200. // 每级生成一条明细(money = 该级节点的 quota 快照)并累加该节点 total_income,到根(parent 为 NULL)终止。
  201. // findUserid 是根(其 parent_user_id 为 NULL)或本身不是树节点 → 子查询为 NULL → **无明细且不报错**
  202. // (静默不入账,与 dump 的子查询语义一致)。
  203. //
  204. // 幂等(两级,设计 §3.3 #22,**同事务内逐级判定**):
  205. // - 节点 type=1(一次付清):同 (user, student) 只生成一条 → 后续不同 recordID 也不重复入账;
  206. // - 节点 type=2(按课次) :同 (user, student, order_no) 只生成一条。
  207. //
  208. // 为什么用迭代:链深可达 100,递归在极端脏数据下会栈溢出;迭代 + 显式层数计数可兜底成环。
  209. func (s *DivideService) AddLog(ctx context.Context, findUserID, stuID string, recordID int64, remark string) error {
  210. findUserID = strings.TrimSpace(findUserID)
  211. stuID = strings.TrimSpace(stuID)
  212. if findUserID == "" || stuID == "" {
  213. return infraerrors.WithCause(apperr.ErrBusinessParam, errString("service: AddLog 需要 findUserID/stuID"))
  214. }
  215. if recordID <= 0 {
  216. return infraerrors.WithCause(apperr.ErrBusinessParam, errString("service: AddLog 需要 recordID > 0"))
  217. }
  218. if err := s.assertReady(); err != nil {
  219. return err
  220. }
  221. orderNo := strconv.FormatInt(recordID, 10)
  222. return s.runner.WithinTx(ctx, func(tx *gorm.DB) error {
  223. txCtx := repository.WithTxContext(ctx, tx)
  224. // ① 定位 findUserid 的**上级**(dump 子查询口径):本人不参与分成
  225. self, err := s.repo.GetByUserID(txCtx, findUserID)
  226. if errors.Is(err, infraerrors.ErrNotFound) {
  227. // 红线(模型注释):必须是分账树节点才入账;非节点 → 静默不入账(不是错误)
  228. return nil
  229. }
  230. if err != nil {
  231. return err
  232. }
  233. if self.ParentUserID == nil || strings.TrimSpace(*self.ParentUserID) == "" {
  234. // findUserid 是根:dump 子查询返回 NULL → 无明细(幂等空操作)
  235. return nil
  236. }
  237. startID := strings.TrimSpace(*self.ParentUserID)
  238. if startID == self.UserID {
  239. return infraerrors.WithCause(apperr.ErrDivideQuotaExceeded,
  240. errString("service: 分账树自环(parent 指向自身)"))
  241. }
  242. cur, err := s.repo.GetByUserID(txCtx, startID)
  243. if errors.Is(err, infraerrors.ErrNotFound) {
  244. return nil // 断链(上级已不在树中):无明细(与 dump 子查询 NULL 同效)
  245. }
  246. if err != nil {
  247. return err
  248. }
  249. for depth := int32(0); cur != nil; depth++ {
  250. if depth >= MaxDivideDepth {
  251. return infraerrors.WithCause(apperr.ErrDivideQuotaExceeded,
  252. errString("service: 上行分成层级超上限(疑似成环) "+strconv.Itoa(int(MaxDivideDepth))))
  253. }
  254. if err := s.appendShare(txCtx, tx, cur, stuID, orderNo, remark); err != nil {
  255. return err
  256. }
  257. if cur.ParentUserID == nil || strings.TrimSpace(*cur.ParentUserID) == "" {
  258. break // 到根
  259. }
  260. parentID := strings.TrimSpace(*cur.ParentUserID)
  261. if parentID == cur.UserID {
  262. return infraerrors.WithCause(apperr.ErrDivideQuotaExceeded,
  263. errString("service: 分账树自环(parent 指向自身)"))
  264. }
  265. next, perr := s.repo.GetByUserID(txCtx, parentID)
  266. if errors.Is(perr, infraerrors.ErrNotFound) {
  267. break // 断链(上级已不在树中):终止上行,已入账部分保留
  268. }
  269. if perr != nil {
  270. return perr
  271. }
  272. cur = next
  273. }
  274. return nil
  275. })
  276. }
  277. // appendShare 为单个节点生成分成明细并累加 total_income(AddLog 的迭代体)。
  278. //
  279. // 入参:txCtx(含 tx 的 ctx);tx;node 当前节点;stuID 学生;orderNo 订单号;remark 备注。
  280. // 出参:nil 成功/幂等命中;ErrDBError。
  281. //
  282. // 幂等判定经 repository.FindLog(两级键口径唯一入口),命中即跳过(**不返回错误**,
  283. // 对外语义是"重复请求幂等成功",与设计 §10.1 的 3114 仅内部日志口径一致)。
  284. func (s *DivideService) appendShare(txCtx context.Context, tx *gorm.DB, node *model.UserDivideInto, stuID, orderNo, remark string) error {
  285. if node == nil {
  286. return nil
  287. }
  288. _, err := s.repo.FindLog(txCtx, node.UserID, stuID, node.Type, orderNo)
  289. switch {
  290. case err == nil:
  291. return nil // 幂等命中:该 (user, student[, order]) 已有分成
  292. case !errors.Is(err, infraerrors.ErrNotFound):
  293. return err
  294. }
  295. amount, perr := parseAmount(node.Quota)
  296. if perr != nil {
  297. return perr
  298. }
  299. // audit D2(dump 763760):额度 <= 0 的节点**不生成明细**(也不累加 total_income)。
  300. // 否则会给 0 元分成写一条"流水",把"明细条数 = 实际发钱次数"这个对账恒等式打破。
  301. if !amount.GreaterThan(0) {
  302. return nil
  303. }
  304. row := &model.UserDivideIntoLog{
  305. Type: node.Type,
  306. UserID: node.UserID,
  307. StudentID: stuID,
  308. OrderNo: stringPtr(orderNo),
  309. Money: amount.String(),
  310. Remark: optionalString(remark),
  311. }
  312. if ierr := s.repo.InsertLog(txCtx, tx, row); ierr != nil {
  313. return ierr
  314. }
  315. if _, ierr := s.repo.IncTotalIncome(txCtx, tx, node.UserID, amount); ierr != nil {
  316. return ierr
  317. }
  318. return nil
  319. }
  320. // DeleteLog 撤销某「学生 + 订单」产生的全部分成(dump `divide_into_log_delete` 2 参语义)。
  321. //
  322. // 入参:
  323. // - ctx 请求上下文;
  324. // - stuID 报课学生(ULID);
  325. // - orderID 订单/上课记录 ID(int64;dump 传 orderid)。
  326. //
  327. // 出参:
  328. // - nil:成功(含"无对应明细"的幂等空操作);
  329. // - ErrBusinessParam / ErrDBError(含端口返回的 DB 异常)。
  330. //
  331. // 两分支(audit D11 按 dump 定稿,**顺序不可颠倒**):
  332. //
  333. // ① 改绑(保留分成):**只要**学生存在首次扣费上课记录 first
  334. // → 该组明细的 order_no 一律改为 first、create_time 重置为 now()、备注写"推广分成",
  335. // total_income **不回滚**;
  336. // ② 回滚删除:**仅**在无 first 记录时 → total_income -= money(冲正)并物理删除明细行。
  337. //
  338. // 关键改动(D11):去掉旧实现里的 `first != orderID` 条件 —— dump 762547 的改绑分支不做这个比较,
  339. // 撤的正是"当前这单",而它自己就是首扣记录;带条件会把它判成"无可改绑"→ 误走回滚删除,
  340. // 把真实产生的分成整组冲掉(少发钱)。
  341. //
  342. // 说明:同一订单的分成会沿上行链生成多行(本人 + 各级上线),故按 (student, order) 取全量
  343. // 逐行处理,而不是只处理一行(否则上级分成会悬空)。
  344. func (s *DivideService) DeleteLog(ctx context.Context, stuID string, orderID int64) error {
  345. stuID = strings.TrimSpace(stuID)
  346. if stuID == "" || orderID <= 0 {
  347. return infraerrors.WithCause(apperr.ErrBusinessParam, errString("service: DeleteLog 需要 stuID/orderID"))
  348. }
  349. if err := s.assertReady(); err != nil {
  350. return err
  351. }
  352. orderNo := strconv.FormatInt(orderID, 10)
  353. return s.runner.WithinTx(ctx, func(tx *gorm.DB) error {
  354. txCtx := repository.WithTxContext(ctx, tx)
  355. rows, err := s.repo.ListLogsByStudentOrder(txCtx, stuID, orderNo)
  356. if err != nil {
  357. return err
  358. }
  359. if len(rows) == 0 {
  360. return nil // 已撤销/从未生成:幂等空操作
  361. }
  362. // 首次扣费记录(端口未注入 → 一律走回滚删除分支)
  363. var firstID int64
  364. var hasFirst bool
  365. if s.firstCharged != nil {
  366. firstID, hasFirst, err = s.firstCharged.FirstChargedRecordID(txCtx, stuID)
  367. if err != nil {
  368. return err
  369. }
  370. }
  371. rebindTo := ""
  372. if hasFirst && firstID > 0 {
  373. // D11:有首扣记录就改绑,**不与当前 orderID 比较**(dump 762547)
  374. rebindTo = strconv.FormatInt(firstID, 10)
  375. }
  376. for i := range rows {
  377. row := &rows[i]
  378. if rebindTo != "" {
  379. // ① 改绑分支:保留分成,仅改幂等键指向首次扣费订单
  380. if _, uerr := s.repo.UpdateLogOrderNo(txCtx, tx, row.ID, rebindTo, remarkRebind); uerr != nil {
  381. return uerr
  382. }
  383. continue
  384. }
  385. // ② 回滚删除分支:冲正 total_income 后物理删除
  386. amount, perr := parseAmount(row.Money)
  387. if perr != nil {
  388. return perr
  389. }
  390. if _, ierr := s.repo.IncTotalIncome(txCtx, tx, row.UserID, amount.Neg()); ierr != nil {
  391. return ierr
  392. }
  393. if derr := s.repo.DeleteLog(txCtx, tx, row.ID); derr != nil {
  394. return derr
  395. }
  396. }
  397. return nil
  398. })
  399. }
  400. // UpdateQuota 修改节点分成额度并级联下降子树(dump `divide_into_update_quota`)。
  401. //
  402. // 入参:
  403. // - ctx 请求上下文;
  404. // - userID 节点用户(ULID;必须是树节点,否则 3001);
  405. // - quota 新额度(单位:分;负值/越界 → 3140)。
  406. //
  407. // 出参:
  408. // - nil:成功(含"子树内无超标节点"的空更新);
  409. // - ErrNotFound(3001):节点不存在;
  410. // - ErrDivideQuotaExceeded(3111):新额度**超过上级**额度(子不得超父);
  411. // - ErrBusinessParam / ErrDBError。
  412. //
  413. // 级联规则(audit A6,dump `yuansiwei_finance.sql:763870-763874` **两个前置条件**):
  414. //
  415. // UPDATE ... SET quota=updateQuota WHERE userid=findUserid; -- 本节点先落新值
  416. // IF (uQuota > 0) THEN -- ① 节点**原** quota>0 才级联
  417. // UPDATE ... WHERE left_val>uLeft AND right_val<uRight
  418. // AND quota>updateQuota AND type=uType; -- ② 只压**同分成类型**的子节点
  419. // END IF;
  420. //
  421. // 即:① 原 quota=0 的节点不级联(尚未配置额度的节点不该"顺手"压低子树);
  422. //
  423. // ② 级联只在同一分成类型内传播(一次付清/按课次两条线的额度互不影响)。
  424. //
  425. // 事务:本节点更新与子树压制**同事务**(否则出现"父已降、子仍高于父"的越权窗口)。
  426. func (s *DivideService) UpdateQuota(ctx context.Context, userID string, quota money.Amount) error {
  427. userID = strings.TrimSpace(userID)
  428. if userID == "" {
  429. return infraerrors.WithCause(apperr.ErrBusinessParam, errString("service: UpdateQuota 需要 userID"))
  430. }
  431. if quota.IsNegative() {
  432. return infraerrors.WithCause(apperr.ErrBusinessParam, errString("service: 分成额度不能为负"))
  433. }
  434. if verr := quota.Validate(); verr != nil {
  435. return infraerrors.WithCause(apperr.ErrBusinessParam, verr)
  436. }
  437. if err := s.assertReady(); err != nil {
  438. return err
  439. }
  440. return s.runner.WithinTx(ctx, func(tx *gorm.DB) error {
  441. txCtx := repository.WithTxContext(ctx, tx)
  442. node, err := s.repo.GetByUserID(txCtx, userID)
  443. if err != nil {
  444. return err // ErrNotFound(3001) 原样上抛
  445. }
  446. // 上限校验:子不得超父(父额度是硬上限,超限 → 3111)
  447. if node.ParentUserID != nil && strings.TrimSpace(*node.ParentUserID) != "" {
  448. parent, perr := s.repo.GetByUserID(txCtx, strings.TrimSpace(*node.ParentUserID))
  449. if perr == nil {
  450. parentQuota, aerr := parseAmount(parent.Quota)
  451. if aerr != nil {
  452. return aerr
  453. }
  454. if quota.GreaterThan(parentQuota) {
  455. return infraerrors.WithCause(apperr.ErrDivideQuotaExceeded,
  456. errString("service: 分成额度不得超过上级("+quota.String()+" > "+parentQuota.String()+")"))
  457. }
  458. } else if !errors.Is(perr, infraerrors.ErrNotFound) {
  459. return perr
  460. }
  461. // 上级不在树中(断链)→ 不做上限校验(不阻断后台修复数据)
  462. }
  463. // A6 前置①:取**原** quota(必须在更新本节点之前读,否则拿到的已是新值)
  464. origQuota, aerr := parseAmount(node.Quota)
  465. if aerr != nil {
  466. return aerr
  467. }
  468. if _, uerr := s.repo.UpdateNodeQuota(txCtx, tx, userID, quota); uerr != nil {
  469. return uerr
  470. }
  471. if !origQuota.GreaterThan(0) {
  472. // 原额度为 0 → dump 不级联(只落本节点新值)
  473. return nil
  474. }
  475. // A6 前置②:只压同一分成类型的子节点
  476. if _, cerr := s.repo.ClampSubtreeQuota(txCtx, tx, node.LeftVal, node.RightVal, node.Type, quota); cerr != nil {
  477. return cerr
  478. }
  479. return nil
  480. })
  481. }
  482. // assertReady 装配期自检(repo/runner 未构造 → ErrInternal,避免 nil 解引用)。
  483. func (s *DivideService) assertReady() error {
  484. if s == nil || s.repo == nil || s.runner == nil {
  485. return infraerrors.WithCause(infraerrors.ErrInternal, errString("service: DivideService 未装配 repo/runner"))
  486. }
  487. return nil
  488. }
  489. // resolveRoot 解析节点的所属树根(自身为根时返回自身 user_id)。
  490. //
  491. // 入参:node 分账节点(可 nil)。出参:根 user_id 字符串。
  492. //
  493. // 修正点(模型注释 ★):旧 SP 在"subid 已存在"分支可能留下 root_user_id = NULL,
  494. // 因此这里必须容忍 nil/空串并回退到节点自身,再回退到 parent(由调用方兜底)。
  495. func resolveRoot(node *model.UserDivideInto) string {
  496. if node == nil {
  497. return ""
  498. }
  499. if node.RootUserID != nil && strings.TrimSpace(*node.RootUserID) != "" {
  500. return strings.TrimSpace(*node.RootUserID)
  501. }
  502. return node.UserID
  503. }
  504. // parseAmount 把库里的 decimal(12,2) 字符串转 money.Amount(空串视为 0.00,容错迁移期脏值)。
  505. //
  506. // 入参:s 十进制"元"字符串。出参:money.Amount;解析失败 → ErrDBError(数据异常不静默归零)。
  507. func parseAmount(s string) (money.Amount, error) {
  508. trimmed := strings.TrimSpace(s)
  509. if trimmed == "" {
  510. return 0, nil
  511. }
  512. amount, err := money.Parse(trimmed)
  513. if err != nil {
  514. return 0, infraerrors.WithCause(infraerrors.ErrDBError, err)
  515. }
  516. return amount, nil
  517. }
  518. // stringPtr 返回字符串指针(空串返回 nil,便于落 NULL)。
  519. func stringPtr(s string) *string {
  520. trimmed := strings.TrimSpace(s)
  521. if trimmed == "" {
  522. return nil
  523. }
  524. return &trimmed
  525. }
  526. // optionalString 空串返回 nil 的指针构造(备注列语义:空 → NULL)。
  527. func optionalString(s string) *string { return stringPtr(s) }
  528. // errString 实现 error 的最小字符串类型(本包不引入额外依赖,语义同 repository.errString)。
  529. type errString string
  530. // Error 实现 error 接口。
  531. func (e errString) Error() string { return string(e) }