gpt

internal/mysqldao/benefit_revoke.go

@@ -0,0 +1,67 @@
+package mysqldao
+
+import (
+	"context"
+
+	"git.lzcwy.cn/zhengchiliu/zt-closet-backend/global"
+	"git.lzcwy.cn/zhengchiliu/zt-closet-backend/internal/mysqlmodel"
+	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
+)
+
+type BenefitRevokeTaskDao struct{ db *gorm.DB }
+
+func NewBenefitRevokeTaskDao() *BenefitRevokeTaskDao {
+	return &BenefitRevokeTaskDao{db: global.MysqlEngine}
+}
+func (d *BenefitRevokeTaskDao) GetByOrderNo(ctx context.Context, orderNo string) (*mysqlmodel.BenefitRevokeTask, error) {
+	var task mysqlmodel.BenefitRevokeTask
+	err := d.db.WithContext(ctx).Where("order_no = ?", orderNo).First(&task).Error
+	return &task, err
+}
+func (d *BenefitRevokeTaskDao) ListPending(ctx context.Context, limit int) ([]mysqlmodel.BenefitRevokeTask, error) {
+	var list []mysqlmodel.BenefitRevokeTask
+	err := d.db.WithContext(ctx).Where("status = ?", 0).Order("id ASC").Limit(limit).Find(&list).Error
+	return list, err
+}
+func (d *BenefitRevokeTaskDao) Claim(ctx context.Context, id int64) (bool, error) {
+	r := d.db.WithContext(ctx).Model(&mysqlmodel.BenefitRevokeTask{}).Where("id = ? AND status IN ?", id, []int8{0, 2}).Update("status", 3)
+	return r.RowsAffected > 0, r.Error
+}
+func (d *BenefitRevokeTaskDao) Update(ctx context.Context, id int64, updates map[string]any) error {
+	return d.db.WithContext(ctx).Model(&mysqlmodel.BenefitRevokeTask{}).Where("id = ?", id).Updates(updates).Error
+}
+func (d *BenefitRevokeTaskDao) RecoverStale(ctx context.Context, before int64) error {
+	return d.db.WithContext(ctx).Model(&mysqlmodel.BenefitRevokeTask{}).Where("status = ? AND updated_at < ?", 3, before).Update("status", 0).Error
+}
+
+// RefundSettleWithRevokeTask 原子写入退款订单终态、取消未执行发放并创建回收任务。
+func (d *BenefitRevokeTaskDao) RefundSettleWithRevokeTask(ctx context.Context, orderNo string, refundAmount, refundTime int64) (bool, error) {
+	settled := false
+	err := d.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		var order mysqlmodel.Order
+		if err := tx.Clauses(clause.Locking{Strength: "UPDATE"}).Where("order_no = ?", orderNo).First(&order).Error; err != nil {
+			return err
+		}
+		if order.OrderStatus == 5 {
+			return nil
+		}
+		if order.OrderStatus != 2 && order.OrderStatus != 3 {
+			return nil
+		}
+		if err := tx.Model(&mysqlmodel.Order{}).Where("id = ?", order.ID).Updates(map[string]any{"order_status": 5, "refund_time": refundTime, "refund_amount": refundAmount}).Error; err != nil {
+			return err
+		}
+		if err := tx.Model(&mysqlmodel.BenefitGrantTask{}).Where("order_no = ? AND status IN ?", orderNo, []int8{0, 2}).Update("status", 4).Error; err != nil {
+			return err
+		}
+		if order.OrderStatus == 2 {
+			if err := tx.Clauses(clause.OnConflict{DoNothing: true}).Create(&mysqlmodel.BenefitRevokeTask{OrderNo: orderNo, UserID: order.UserID, MaxRetry: 10}).Error; err != nil {
+				return err
+			}
+		}
+		settled = true
+		return nil
+	})
+	return settled, err
+}