代码初始化

dao/wallet/wallet_dao.go

@@ -0,0 +1,134 @@
+package dao
+
+import (
+	"context"
+	"errors"
+
+	"gitea.com/red-future/common/db/gfdb"
+	"github.com/gogf/gf/v2/database/gdb"
+	"github.com/gogf/gf/v2/util/gconv"
+
+	"shop-user-trade/consts/public"
+	walletConsts "shop-user-trade/consts/wallet"
+	walletDto "shop-user-trade/model/dto/wallet"
+	walletEntity "shop-user-trade/model/entity/wallet"
+)
+
+var Wallet = new(walletDao)
+
+type walletDao struct{}
+
+// GetByUserID 根据用户ID获取钱包
+func (d *walletDao) GetByUserID(ctx context.Context, userID int64) (res *walletEntity.Wallet, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameWallet).
+		Where(walletEntity.WalletCol.UserID, userID).
+		WhereNot(walletEntity.WalletCol.Status, int(walletConsts.WalletStatusFrozen)).
+		One()
+	if err != nil {
+		return
+	}
+	if r.IsEmpty() {
+		return nil, nil
+	}
+	err = r.Struct(&res)
+	return
+}
+
+// GetByID 根据ID获取钱包
+func (d *walletDao) GetByID(ctx context.Context, walletID int64) (res *walletEntity.Wallet, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameWallet).
+		Where(walletEntity.WalletCol.Id, walletID).
+		One()
+	if err != nil {
+		return
+	}
+	if r.IsEmpty() {
+		return nil, nil
+	}
+	err = r.Struct(&res)
+	return
+}
+
+// Create 创建钱包
+func (d *walletDao) Create(ctx context.Context, req *walletDto.CreateWalletReq) (id int64, err error) {
+	entity := &walletEntity.Wallet{
+		UserID:   req.UserId,
+		Balance:  0,
+		Currency: req.Currency,
+		Status:   walletConsts.WalletStatusEnabled,
+		Version:  1,
+	}
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameWallet).Data(entity).Insert()
+	if err != nil {
+		return
+	}
+	return r.LastInsertId()
+}
+
+// UpdateBalance 更新余额（乐观锁）
+func (d *walletDao) UpdateBalance(ctx context.Context, walletID int64, amount int64, opType string, version int64) (bool, error) {
+	var updateData gdb.Map
+	switch opType {
+	case "income":
+		updateData = gdb.Map{
+			"balance": gdb.Raw("balance + " + gconv.String(amount)),
+			"version": gdb.Raw("version + 1"),
+		}
+	case "expense":
+		updateData = gdb.Map{
+			"balance": gdb.Raw("balance - " + gconv.String(amount)),
+			"version": gdb.Raw("version + 1"),
+		}
+	case "freeze":
+		updateData = gdb.Map{
+			"status": int(walletConsts.WalletStatusFrozen),
+		}
+	case "unfreeze":
+		updateData = gdb.Map{
+			"status": int(walletConsts.WalletStatusEnabled),
+		}
+	default:
+		return false, errors.New("unsupported operation type")
+	}
+
+	result, err := gfdb.DB(ctx).Model(ctx, public.TableNameWallet).
+		Data(updateData).
+		Where(walletEntity.WalletCol.Id, walletID).
+		Where(walletEntity.WalletCol.Version, version).
+		Update()
+	if err != nil {
+		return false, err
+	}
+	rows, err := result.RowsAffected()
+	if err != nil {
+		return false, err
+	}
+	return rows > 0, nil
+}
+
+// CreateLog 创建钱包日志
+func (d *walletDao) CreateLog(ctx context.Context, req *walletDto.CreateWalletLogReq) (id int64, err error) {
+	var entity *walletEntity.WalletLog
+	if err = gconv.Struct(req, &entity); err != nil {
+		return
+	}
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameWalletLog).Data(entity).Insert()
+	if err != nil {
+		return
+	}
+	return r.LastInsertId()
+}
+
+// ListLogs 获取钱包日志列表
+func (d *walletDao) ListLogs(ctx context.Context, userID int64, page, pageSize int) (res []walletEntity.WalletLog, total int, err error) {
+	r, total, err := gfdb.DB(ctx).Model(ctx, public.TableNameWalletLog).
+		Where(walletEntity.WalletLogCol.UserID, userID).
+		OrderDesc(walletEntity.WalletLogCol.CreatedAt).
+		Page(page, pageSize).
+		AllAndCount(false)
+	if err != nil {
+		return
+	}
+	err = r.Structs(&res)
+	return
+}