divide_service.go 31 KB

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