| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333 |
- // 活动赠送金额池仓储(t_courses_activity_user_discount / _user_discount_log / _used_log)——AC-06。
- //
- // 依据:unified/03-migration/academic-domain-design.md §3.2 #14~#19(赠送金额池 4 SP)/ §1.5(金额)
- // / §1.7(scoping)。
- //
- // 覆盖 SP 映射:
- //
- // user_discount_usable → SumUsable(dump 3 参汇总口径:schoolid + campusid + userid)
- // user_discount_use → ListUsable(ASC) + IncUsed/IncUsedBatch + InsertDiscountLog(负值)
- // user_discount_refund → ListUsable(DESC) + IncUsed(负 delta) + InsertDiscountLog(正值)
- // user_discount_refund_user_activity → ListUsable(DESC)(同活动池优先退回)
- // use_activity_recommend → IncUsed + InsertUsedLog(老带新结算写负值)
- //
- // 金额口径(红线):库里 discount_amount / discount_amount_used 是 decimal(12,2) **字符串列**,
- // Go 侧一律转 money.Amount(int64 分)再运算;SQL 侧只用字符串字面量做加减(decimal 精确),
- // **绝不**把金额转 float64(sqlite 的 SUM() 返回 REAL,会让分位失真——见 SumUsable 实现说明)。
- package repository
- import (
- "context"
- "strings"
- "gorm.io/gorm"
- "yuansiwei/unified-backend/internal/academic/model"
- "yuansiwei/unified-backend/internal/academic/money"
- infraerrors "yuansiwei/unified-backend/internal/infra/errors"
- appscope "yuansiwei/unified-backend/internal/infra/scope"
- )
- // UsedDelta 批量累加「已使用赠送金额」的单元素(id + 增量)。
- //
- // 字段:
- // - DiscountID int64:优惠账户 ID(t_courses_activity_user_discount.id;<=0 的条目被跳过)。
- // - Delta money.Amount:discount_amount_used 的增量(正=消费/占用,负=退回冲正;单位:分)。
- type UsedDelta struct {
- // DiscountID 优惠账户 ID(DB 自增主键)。
- DiscountID int64
- // Delta 已使用金额增量(正=消费,负=退回;单位:分)。
- Delta money.Amount
- }
- // ActivityDiscountRepo 活动赠送金额池仓储。
- //
- // 表:
- // - t_courses_activity_user_discount(池:可用 = discount_amount − discount_amount_used);
- // - t_courses_activity_user_discount_log(使用/退回流水,只增,负=消费/正=退回);
- // - t_courses_activity_used_log(老带新获得/使用记录,结算写负值)。
- //
- // 作用域:三表均带 institution_id + campus_id(campus_id=0 = 机构所有自营校区)→ scopedCampusRows。
- type ActivityDiscountRepo struct {
- *BaseRepo
- }
- // NewActivityDiscountRepo 构造赠送金额池仓储。
- //
- // 入参:db *gorm.DB / sc *appscope.Scope(语义同 NewActivityRepo)。
- // 出参:*ActivityDiscountRepo(CampusScoped=false,校区维度走 IN (campus,0) 超集语义)。
- func NewActivityDiscountRepo(db *gorm.DB, sc *appscope.Scope) *ActivityDiscountRepo {
- r := &ActivityDiscountRepo{BaseRepo: NewBaseRepo(db, sc)}
- r.CampusScoped = false
- return r
- }
- // db 取查询句柄(ctx 内含 tx 时复用同一事务)。
- func (r *ActivityDiscountRepo) db(ctx context.Context) (*gorm.DB, error) {
- if r == nil || r.DB == nil {
- return nil, infraerrors.WithCause(infraerrors.ErrInternal, errString("repository: ActivityDiscountRepo.DB 为 nil"))
- }
- if tx := TxFromContext(ctx); tx != nil {
- return tx, nil
- }
- if ctx == nil {
- return r.DB, nil
- }
- return r.DB.WithContext(ctx), nil
- }
- // ListUsable 列出用户可用赠送金额池(按游标方向取用/退回顺序)。
- //
- // 入参:
- // - ctx 请求上下文;
- // - userID 学生(canonical_user_id ULID;空 → ErrBadParam);
- // - order 游标方向(**顺序不可改**,源自 dump):
- // OrderAsc = user_discount_use 使用顺序(先入先出);
- // OrderDesc = user_discount_refund 退回顺序(后入先退)。
- //
- // 出参:
- // - []model.CoursesActivityUserDiscount:可用行(`discount_amount > discount_amount_used`),
- // 金额为字符串原文,调用方用 money.Parse 转 Amount;
- // - ErrBadParam / ErrDBError。
- //
- // 可用口径(dump user_discount_usable 一致):仅累加 discount_amount > discount_amount_used 的行
- // (负赠送/已用尽的行不参与),故本方法只返回正可用行。
- func (r *ActivityDiscountRepo) ListUsable(ctx context.Context, userID string, order string) ([]model.CoursesActivityUserDiscount, error) {
- if strings.TrimSpace(userID) == "" {
- return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ListUsable 需要 user_id"))
- }
- dir, err := normalizeOrder(order)
- if err != nil {
- return nil, err
- }
- db, err := r.db(ctx)
- if err != nil {
- return nil, err
- }
- q := scopedCampusRows(db.Model(&model.CoursesActivityUserDiscount{}), r.Sc, "")
- q = q.Where("user_id = ?", userID).
- Where("discount_amount > discount_amount_used").
- Order("id " + dir)
- var rows []model.CoursesActivityUserDiscount
- if err := q.Find(&rows).Error; err != nil {
- return nil, infraerrors.WithCause(infraerrors.ErrDBError, err)
- }
- return rows, nil
- }
- // SumUsable 汇总用户可用赠送金额(**对齐 dump `usable` 的 3 参口径**:机构 + 校区 + 用户)。
- //
- // 入参:
- // - ctx 请求上下文;
- // - userID 学生(ULID;空 → ErrBadParam);
- // - institutionID 机构 ID(原 schoolid;<=0 → ErrBadParam);
- // - campusID 校区 ID(原 campusid;0 = 机构级口径,即不限定校区)。
- //
- // 出参:
- // - money.Amount:可用额度合计(单位:分;无可用行为 0);
- // - ErrBadParam / ErrUnscoped(1004,"报表参数越权")/ ErrDBError。
- //
- // 作用域(两条例外之外的**显式参数口径**,安全红线):
- //
- // dump 的 usable(schoolid, campusid, userid) 由调用方传参(报表/退费需要"指定机构校区"视角),
- // 与请求作用域可能不同;此处**不叠加**请求作用域过滤,但做一次防越权断言:
- // 非超管且 institutionID != Sc.InstitutionID → ErrUnscoped(禁止借参数跨机构读他人赠送金额)。
- //
- // 校区过滤:campusID > 0 → `campus_id IN (campusID, 0)`(0=机构所有自营校区,与 dump 一致);
- // campusID = 0 → 只按机构汇总(机构级报表视角)。
- //
- // 实现说明(金额精度):**不用 SQL SUM()**。sqlite 把 decimal 存 NUMERIC 时 SUM 返回 REAL,
- // 会把分位带进浮点;MySQL 返回 DECIMAL(字符串) 虽精确,但两库行为不一致。
- // 因此这里只让 DB 做过滤,**逐行在 Go 侧用 money.Amount(int64 分)累加**(用户池行数极小,
- // 性能可忽略),换取"零浮点"的确定性。
- func (r *ActivityDiscountRepo) SumUsable(ctx context.Context, userID string, institutionID, campusID int64) (money.Amount, error) {
- if strings.TrimSpace(userID) == "" || institutionID <= 0 {
- return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: SumUsable 需要 user_id/institution_id"))
- }
- if r != nil && r.Sc != nil && !r.Sc.SuperAdmin && r.Sc.InstitutionID != institutionID {
- return 0, infraerrors.WithCause(infraerrors.ErrUnscoped,
- errString("repository: SumUsable 机构参数越权(作用域机构与入参不一致)"))
- }
- db, err := r.db(ctx)
- if err != nil {
- return 0, err
- }
- q := db.Model(&model.CoursesActivityUserDiscount{}).
- Where("user_id = ?", userID).
- Where("institution_id = ?", institutionID).
- Where("discount_amount > discount_amount_used")
- if campusID > 0 {
- // campus_id=0 = 机构所有自营校区:与指定校区同属可见范围(dump IN (x,0))
- q = q.Where("campus_id IN ?", []int64{campusID, 0})
- }
- var rows []model.CoursesActivityUserDiscount
- if err := q.Find(&rows).Error; err != nil {
- return 0, infraerrors.WithCause(infraerrors.ErrDBError, err)
- }
- total := money.Amount(0)
- for i := range rows {
- totalAmount, perr := money.Parse(rows[i].DiscountAmount)
- if perr != nil {
- return 0, infraerrors.WithCause(infraerrors.ErrDBError, perr)
- }
- usedAmount, perr := money.Parse(rows[i].DiscountAmountUsed)
- if perr != nil {
- return 0, infraerrors.WithCause(infraerrors.ErrDBError, perr)
- }
- total = total.Add(totalAmount.Sub(usedAmount))
- }
- return total, nil
- }
- // IncUsed 累加优惠账户的「已使用赠送金额」(单账户原子增减)。
- //
- // 入参:
- // - ctx 请求上下文;
- // - tx 事务句柄(**必传**);
- // - id 优惠账户 ID(<=0 → ErrBadParam);
- // - delta 增量(正=消费,负=退回冲正;单位:分;0 → 空操作)。
- //
- // 出参:
- // - int64:受影响行数(0 = 账户不存在/不在作用域内);
- // - ErrBadParam / ErrDBError。
- //
- // 红线:
- // - `discount_amount_used <= discount_amount` 的业务约束由 **Service** 在扣减前用可用额校验
- // (不足 → 3109),本方法**不做截断**——静默截断会破坏"池余额 = 流水累计"的对账恒等式;
- // - 与流水写入(InsertDiscountLog)必须同事务。
- func (r *ActivityDiscountRepo) IncUsed(ctx context.Context, tx *gorm.DB, id int64, delta money.Amount) (int64, error) {
- if err := requireTx(tx, "IncUsed"); err != nil {
- return 0, err
- }
- if id <= 0 {
- return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: IncUsed 需要 discount id"))
- }
- if delta.IsZero() {
- return 0, nil
- }
- if err := delta.Validate(); err != nil {
- return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
- }
- q := scopedCampusRows(tx.Model(&model.CoursesActivityUserDiscount{}), r.Sc, "")
- res := q.Where("id = ?", id).
- UpdateColumn("discount_amount_used", gorm.Expr("discount_amount_used + ?", delta.String()))
- if res.Error != nil {
- return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
- }
- return res.RowsAffected, nil
- }
- // IncUsedBatch 批量累加多个优惠账户的「已使用赠送金额」(**单语句 CASE WHEN**,禁止逐行 UPDATE)。
- //
- // 入参:
- // - ctx 请求上下文;
- // - tx 事务句柄(**必传**);
- // - items 增量集合(同一 id 出现多次时**后者覆盖前者**;<=0 的 id 与零增量条目被忽略)。
- //
- // 出参:
- // - int64:受影响行数;
- // - ErrBadParam / ErrDBError(空集合 → 0, nil,不发 SQL)。
- //
- // 为什么必须单语句(性能/正确性双约束):
- // - 一次跨多个池扣减若逐行 UPDATE,N 个账户 = N 次往返 + N 个独立执行计划;
- // - 单语句在事务内原子生效,减少锁持有时间(并发退费场景)。
- //
- // SQL:
- //
- // UPDATE t_courses_activity_user_discount
- // SET discount_amount_used = discount_amount_used + CASE id WHEN ? THEN ? ... ELSE 0 END
- // WHERE id IN (...) [AND institution_id = ? AND campus_id IN (?,0)]
- //
- // 红线:金额以**字符串**字面量入场(decimal 精确),禁止 float。
- func (r *ActivityDiscountRepo) IncUsedBatch(ctx context.Context, tx *gorm.DB, items []UsedDelta) (int64, error) {
- if err := requireTx(tx, "IncUsedBatch"); err != nil {
- return 0, err
- }
- // 归一:去重(后者覆盖)+ 过滤非法 id/零增量,保持入参顺序(CASE 分支顺序可预期,便于断言)
- order := make([]int64, 0, len(items))
- deltaByID := make(map[int64]money.Amount, len(items))
- for _, it := range items {
- if it.DiscountID <= 0 || it.Delta.IsZero() {
- continue
- }
- if err := it.Delta.Validate(); err != nil {
- return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
- }
- if _, ok := deltaByID[it.DiscountID]; !ok {
- order = append(order, it.DiscountID)
- }
- deltaByID[it.DiscountID] = it.Delta
- }
- if len(order) == 0 {
- return 0, nil
- }
- var sqlText strings.Builder
- sqlText.WriteString("discount_amount_used + CASE id")
- caseArgs := make([]any, 0, len(order)*2)
- for _, id := range order {
- sqlText.WriteString(" WHEN ? THEN ?")
- caseArgs = append(caseArgs, id, deltaByID[id].String())
- }
- sqlText.WriteString(" ELSE 0 END")
- q := scopedCampusRows(tx.Model(&model.CoursesActivityUserDiscount{}), r.Sc, "")
- res := q.Where("id IN ?", order).UpdateColumn("discount_amount_used", gorm.Expr(sqlText.String(), caseArgs...))
- if res.Error != nil {
- return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
- }
- return res.RowsAffected, nil
- }
- // InsertDiscountLog 写入赠送金额使用/退回流水(**只增**)。
- //
- // 入参:
- // - ctx 请求上下文;
- // - tx 事务句柄(**必传**);
- // - row 流水行(ActivityUserDiscountID>0;DiscountAmountUsed 负=消费/正=退回)。
- //
- // 出参:nil 成功;ErrBadParam / ErrDBError。
- //
- // 红线:流水与池余额变动(IncUsed)必须同事务;流水禁止 UPDATE/DELETE(审计底线)。
- func (r *ActivityDiscountRepo) InsertDiscountLog(ctx context.Context, tx *gorm.DB, row *model.CoursesActivityUserDiscountLog) error {
- if err := requireTx(tx, "InsertDiscountLog"); err != nil {
- return err
- }
- if row == nil {
- return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: discount log 为 nil"))
- }
- if row.ActivityUserDiscountID <= 0 {
- return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: discount log 缺少 activity_user_discount_id"))
- }
- if err := tx.WithContext(ctxOrBackground(ctx)).Create(row).Error; err != nil {
- return infraerrors.WithCause(infraerrors.ErrDBError, err)
- }
- return nil
- }
- // InsertUsedLog 写入老带新活动金额获得/使用记录(**只增**)。
- //
- // 入参:
- // - ctx 请求上下文;
- // - tx 事务句柄(**必传**);
- // - row 记录行(ActivityID>0、UserID 非空;老带新结算写**负值** activity_money_used)。
- //
- // 出参:nil 成功;ErrBadParam / ErrDBError。
- //
- // 红线:符号方向固定(结算负值);撤销报名前置校验按 (activity_id, user_id) 判存在(3108),
- // 故本表写入必须与 signed 状态推进同事务。
- func (r *ActivityDiscountRepo) InsertUsedLog(ctx context.Context, tx *gorm.DB, row *model.CoursesActivityUsedLog) error {
- if err := requireTx(tx, "InsertUsedLog"); err != nil {
- return err
- }
- if row == nil {
- return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: used log 为 nil"))
- }
- if row.ActivityID <= 0 || strings.TrimSpace(row.UserID) == "" {
- return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: used log 缺少 activity_id/user_id"))
- }
- if err := tx.WithContext(ctxOrBackground(ctx)).Create(row).Error; err != nil {
- return infraerrors.WithCause(infraerrors.ErrDBError, err)
- }
- return nil
- }
|