voucher/internal/data/repoimpl/order.go

package repoimpl

import (
	"context"
	"gorm.io/gorm"
	"time"
	"voucher/internal/biz/bo"
	"voucher/internal/biz/repo"
	"voucher/internal/biz/vo"
	"voucher/internal/data"
	"voucher/internal/data/model"
)

// OrderRepoImpl .
type OrderRepoImpl struct {
	Base[model.Order, bo.OrderBo]
	db *data.Db
}

// NewOrderRepoImpl .
func NewOrderRepoImpl(db *data.Db) repo.OrderRepo {
	return &OrderRepoImpl{db: db}
}

func (p *OrderRepoImpl) DB(ctx context.Context) *gorm.DB {
	return p.db.DB(ctx).Model(model.Order{})
}

func (p *OrderRepoImpl) Create(ctx context.Context, req *bo.OrderBo) (*bo.OrderBo, error) {
	now := time.Now()

	info := &model.Order{
		OrderNo:     req.OrderNo,
		OutBizNo:    req.OutBizNo,
		ProductNo:   req.ProductNo,
		BatchNo:     req.BatchNo,
		Account:     req.Account,
		AccountType: req.AccountType.GetValue(),
		Status:      vo.OrderStatusWait.GetValue(),
		Type:        req.Type.GetValue(),
		AppID:       req.AppID,
		MerchantNo:  req.MerchantNo,
		Channel:     req.Channel.GetValue(),
		NotifyUrl:   req.NotifyUrl,
		CreateTime:  &now,
		UpdateTime:  &now,
	}

	tx := p.db.DB(ctx).Create(info)
	if tx.Error != nil {
		return nil, tx.Error
	}

	return p.ToBo(info), nil
}

func (p *OrderRepoImpl) GetByOutBizNo(ctx context.Context, t vo.OrderType, appId, outBizNo string) (*bo.OrderBo, error) {
	info := &model.Order{}

	tx := p.DB(ctx).Where(model.Order{Type: t.GetValue(), AppID: appId, OutBizNo: outBizNo}).Find(&info)

	if tx.Error != nil {
		return nil, tx.Error
	}

	if tx.RowsAffected == 0 {
		return nil, gorm.ErrRecordNotFound
	}

	return p.ToBo(info), nil
}

func (p *OrderRepoImpl) GetByID(ctx context.Context, id uint64) (*bo.OrderBo, error) {
	info := &model.Order{}

	tx := p.DB(ctx).Where(model.Order{ID: id}).Find(&info)

	if tx.Error != nil {
		return nil, tx.Error
	}

	if tx.RowsAffected == 0 {
		return nil, gorm.ErrRecordNotFound
	}

	return p.ToBo(info), nil
}

func (p *OrderRepoImpl) GetByOrderNo(ctx context.Context, orderNo string) (*bo.OrderBo, error) {
	info := &model.Order{}

	tx := p.DB(ctx).Where(model.Order{OrderNo: orderNo}).Find(&info)

	if tx.Error != nil {
		return nil, tx.Error
	}

	if tx.RowsAffected == 0 {
		return nil, gorm.ErrRecordNotFound
	}

	return p.ToBo(info), nil
}

func (p *OrderRepoImpl) Ing(ctx context.Context, id uint64) error {
	now := time.Now()

	res := p.db.DB(ctx).
		Where(model.Order{
			ID:     id,
			Status: vo.OrderStatusWait.GetValue(),
		}).
		Updates(model.Order{
			Status:     vo.OrderStatusIng.GetValue(),
			UpdateTime: &now,
		})

	if res.Error != nil {
		return res.Error
	}

	return nil
}

func (p *OrderRepoImpl) Success(ctx context.Context, id uint64) error {
	now := time.Now()

	res := p.db.DB(ctx).
		Where(model.Order{
			ID:     id,
			Status: vo.OrderStatusIng.GetValue(),
		}).
		Updates(model.Order{
			Status:     vo.OrderStatusSuccess.GetValue(),
			UpdateTime: &now,
		})

	if res.Error != nil {
		return res.Error
	}

	return nil
}

func (p *OrderRepoImpl) Fail(ctx context.Context, id uint64) error {
	now := time.Now()

	res := p.db.DB(ctx).
		Where(model.Order{
			ID:     id,
			Status: vo.OrderStatusIng.GetValue(),
		}).
		Updates(model.Order{
			Status:     vo.OrderStatusFail.GetValue(),
			UpdateTime: &now,
		})

	if res.Error != nil {
		return res.Error
	}

	return nil
}

func (p *OrderRepoImpl) Used(ctx context.Context, id uint64) error {
	now := time.Now()

	res := p.db.DB(ctx).
		Where(model.Order{
			ID:     id,
			Status: vo.OrderStatusSuccess.GetValue(),
		}).
		Updates(model.Order{
			Status:     vo.OrderStatusUse.GetValue(),
			UpdateTime: &now,
		})

	if res.Error != nil {
		return res.Error
	}

	return nil
}

func (p *OrderRepoImpl) Expired(ctx context.Context, id uint64) error {
	now := time.Now()

	res := p.db.DB(ctx).
		Where(model.Order{
			ID:     id,
			Status: vo.OrderStatusSuccess.GetValue(),
		}).
		Updates(model.Order{
			Status:     vo.OrderStatusExpired.GetValue(),
			UpdateTime: &now,
		})

	if res.Error != nil {
		return res.Error
	}

	return nil
}