zt-closet-backend/internal/mysqldao/order.go

package mysqldao

import (
	"context"

	"git.lzcwy.cn/zhengchiliu/zt-closet-backend/global"
	"git.lzcwy.cn/zhengchiliu/zt-closet-backend/internal/mysqlmodel"
	"gorm.io/gorm"
)

// OrderDao 订单数据访问
type OrderDao struct {
	db *gorm.DB
}

// NewOrderDao 创建订单 DAO
func NewOrderDao() *OrderDao {
	return &OrderDao{db: global.MysqlEngine}
}

// OrderCreate 创建订单
func (d *OrderDao) OrderCreate(ctx context.Context, order *mysqlmodel.Order) error {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.OrderCreate")
	defer span.End()

	if err := d.db.WithContext(ctx).Create(order).Error; err != nil {
		span.RecordError(err)
		return err
	}
	return nil
}

// OrderUpdateByOrderNo 按订单号更新订单字段
func (d *OrderDao) OrderUpdateByOrderNo(ctx context.Context, orderNo string, updates map[string]any) error {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.OrderUpdateByOrderNo")
	defer span.End()

	if err := d.db.WithContext(ctx).Model(&mysqlmodel.Order{}).Where("order_no = ?", orderNo).Updates(updates).Error; err != nil {
		span.RecordError(err)
		return err
	}
	return nil
}

// OrderUpdateStatusByOrderNo 条件流转订单状态（防重复回调），返回是否更新成功
func (d *OrderDao) OrderUpdateStatusByOrderNo(ctx context.Context, orderNo string, fromStatus, toStatus int8) (bool, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.OrderUpdateStatusByOrderNo")
	defer span.End()

	res := d.db.WithContext(ctx).Model(&mysqlmodel.Order{}).
		Where("order_no = ? AND order_status = ?", orderNo, fromStatus).
		Update("order_status", toStatus)
	if res.Error != nil {
		span.RecordError(res.Error)
		return false, res.Error
	}
	return res.RowsAffected > 0, nil
}

// OrderExpireByTime 将超过支付有效期的待支付订单批量置为已过期，返回更新条数。
// 状态字面量：1=待支付 4=已过期（与 service 层 OrderStatusPending/OrderStatusExpired 一致）。
func (d *OrderDao) OrderExpireByTime(ctx context.Context, now int64) (int64, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.OrderExpireByTime")
	defer span.End()

	res := d.db.WithContext(ctx).Model(&mysqlmodel.Order{}).
		Where("order_status = ? AND expire_time < ?", 1, now).
		Update("order_status", 4)
	if res.Error != nil {
		span.RecordError(res.Error)
		return 0, res.Error
	}
	return res.RowsAffected, nil
}

// OrderGetByID 按主键 ID 查询订单
func (d *OrderDao) OrderGetByID(ctx context.Context, id int64) (*mysqlmodel.Order, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.OrderGetByID")
	defer span.End()

	var order mysqlmodel.Order
	if err := d.db.WithContext(ctx).First(&order, id).Error; err != nil {
		span.RecordError(err)
		return nil, err
	}
	return &order, nil
}

// OrderGetByOrderNo 按商户订单号查询订单
func (d *OrderDao) OrderGetByOrderNo(ctx context.Context, orderNo string) (*mysqlmodel.Order, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.OrderGetByOrderNo")
	defer span.End()

	var order mysqlmodel.Order
	if err := d.db.WithContext(ctx).Where("order_no = ?", orderNo).First(&order).Error; err != nil {
		span.RecordError(err)
		return nil, err
	}
	return &order, nil
}

// OrderListByUserID 分页查询用户订单（status 为 0 不筛选）
func (d *OrderDao) OrderListByUserID(ctx context.Context, userID int64, status int8, page, pageSize int) ([]mysqlmodel.Order, int64, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.OrderListByUserID")
	defer span.End()

	var list []mysqlmodel.Order
	var total int64
	q := d.db.WithContext(ctx).Model(&mysqlmodel.Order{}).Where("user_id = ?", userID)
	if status != 0 {
		q = q.Where("order_status = ?", status)
	}
	if err := q.Count(&total).Error; err != nil {
		span.RecordError(err)
		return nil, 0, err
	}
	if err := q.Offset((page - 1) * pageSize).Limit(pageSize).Order("id DESC").Find(&list).Error; err != nil {
		span.RecordError(err)
		return nil, 0, err
	}
	return list, total, nil
}

// RefundFlowLogDao 退款流水数据访问
type RefundFlowLogDao struct {
	db *gorm.DB
}

// NewRefundFlowLogDao 创建退款流水 DAO
func NewRefundFlowLogDao() *RefundFlowLogDao {
	return &RefundFlowLogDao{db: global.MysqlEngine}
}

// RefundFlowLogCreate 创建退款流水
func (d *RefundFlowLogDao) RefundFlowLogCreate(ctx context.Context, log *mysqlmodel.RefundFlowLog) error {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.RefundFlowLogCreate")
	defer span.End()

	if err := d.db.WithContext(ctx).Create(log).Error; err != nil {
		span.RecordError(err)
		return err
	}
	return nil
}

// RefundFlowLogUpdateStatus 按退款单号更新退款流水字段
func (d *RefundFlowLogDao) RefundFlowLogUpdateStatus(ctx context.Context, refundNo string, updates map[string]any) error {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.RefundFlowLogUpdateStatus")
	defer span.End()

	if err := d.db.WithContext(ctx).Model(&mysqlmodel.RefundFlowLog{}).Where("refund_no = ?", refundNo).Updates(updates).Error; err != nil {
		span.RecordError(err)
		return err
	}
	return nil
}

// RefundFlowLogUpdateStatusByFrom 条件流转退款流水状态（防并发/乱序回调覆盖），返回是否更新成功。
func (d *RefundFlowLogDao) RefundFlowLogUpdateStatusByFrom(ctx context.Context, refundNo string, fromStatus int8, updates map[string]any) (bool, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.RefundFlowLogUpdateStatusByFrom")
	defer span.End()

	res := d.db.WithContext(ctx).Model(&mysqlmodel.RefundFlowLog{}).
		Where("refund_no = ? AND status = ?", refundNo, fromStatus).
		Updates(updates)
	if res.Error != nil {
		span.RecordError(res.Error)
		return false, res.Error
	}
	return res.RowsAffected > 0, nil
}

// RefundFlowLogGetByRefundNo 按退款单号查询退款流水
func (d *RefundFlowLogDao) RefundFlowLogGetByRefundNo(ctx context.Context, refundNo string) (*mysqlmodel.RefundFlowLog, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.RefundFlowLogGetByRefundNo")
	defer span.End()

	var log mysqlmodel.RefundFlowLog
	if err := d.db.WithContext(ctx).Where("refund_no = ?", refundNo).First(&log).Error; err != nil {
		span.RecordError(err)
		return nil, err
	}
	return &log, nil
}

// RefundFlowLogGetByOrderNo 按商户订单号查询退款流水
func (d *RefundFlowLogDao) RefundFlowLogGetByOrderNo(ctx context.Context, orderNo string) (*mysqlmodel.RefundFlowLog, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.RefundFlowLogGetByOrderNo")
	defer span.End()

	var log mysqlmodel.RefundFlowLog
	if err := d.db.WithContext(ctx).Where("order_no = ?", orderNo).First(&log).Error; err != nil {
		span.RecordError(err)
		return nil, err
	}
	return &log, nil
}