divide_service.go 25 KB

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