app-cloudep-wallet-service/pkg/usecase/wallet.go

package usecase

import (
	"code.30cm.net/digimon/app-cloudep-wallet-service/pkg/domain/entity"
	"code.30cm.net/digimon/app-cloudep-wallet-service/pkg/domain/repository"
	"code.30cm.net/digimon/app-cloudep-wallet-service/pkg/domain/usecase"
	"code.30cm.net/digimon/app-cloudep-wallet-service/pkg/domain/wallet"
	"code.30cm.net/digimon/library-go/errs"
	"context"
	"errors"
	"github.com/go-sql-driver/mysql"
	"github.com/sirupsen/logrus"
	"sync"
)

type WalletUseCaseParam struct {
	WalletRepo            repository.WalletRepository
	TransactionRepo       repository.TransactionRepository
	WalletTransactionRepo repository.WalletTransactionRepo
}

type WalletUseCase struct {
	WalletUseCaseParam

	// 內存讀寫所記錄有哪些玩家已經確認有存在錢包過了，減少確認錢包是否存在頻率
	sync.RWMutex
	existUIDAsset map[string]struct{}
}

func (use *WalletUseCase) Process(ctx context.Context, req usecase.WalletTransferRequest) error {
	//TODO implement me
	panic("implement me")
}

// Withdraw 提幣
// 1. 新增一筆提幣交易
// 2. 錢包減少可用餘額
// 3. 錢包變化新增一筆減少可用餘額資料
func (use *WalletUseCase) Withdraw(ctx context.Context, tx usecase.WalletTransferRequest) error {
	if !tx.Amount.IsPositive() {
		return errs.InvalidRange("failed to get correct amount")
	}
	tx.TxType = wallet.Withdraw

	return use.ProcessTransaction(ctx, tx, userWalletFlow{
		UID:     tx.FromUID,
		Asset:   tx.Asset,
		Actions: []walletActionOption{use.withLockAvailable(), use.withSubAvailable()},
	})

}

// Deposit 充值
// 1. 新增一筆充值交易
// 2. 錢包增加可用餘額
// 3. 錢包變化新增一筆增加可用餘額資料
func (use *WalletUseCase) Deposit(ctx context.Context, tx usecase.WalletTransferRequest) error {
	// 確認錢包新增或減少的餘額是否正確
	if !tx.Amount.IsPositive() {
		return errs.InvalidRange("failed to get correct amount")
	}

	tx.TxType = wallet.Deposit

	uidAsset := uidAssetKey{
		uid:   tx.FromUID,
		asset: tx.Asset,
	}

	exists := use.checkWalletExistence(uidAsset)

	withLockAvailable := func(ctx context.Context, tx *usecase.WalletTransferRequest, w repository.UserWalletService) error {
		if !exists {
			checkWallet, err := w.HasAvailableBalance(ctx)
			if err != nil {
				return err
			}

			// 錢包不存在要做新增
			if !checkWallet {
				if _, err := w.InitializeWallets(ctx, tx.Brand); err != nil {
					var mysqlErr *mysql.MySQLError
					// 解析是否被其他 transaction insert 了，是的話嘗試取得 insert 後的鎖，不是的話需要直接回傳錯誤
					if errors.As(err, &mysqlErr) && mysqlErr.Number == 1062 {
						logrus.WithFields(logrus.Fields{
							"err": err,
							"uid": tx.FromUID,
						}).Warn("Deposit.Create.Wallet")
					} else {
						return err
					}
				} else {
					// 因為是透過 transaction 新增，所以不用上鎖
					return nil
				}
			} else {
				exists = true
				use.markWalletAsExisting(uidAsset)
			}
		}

		// 確認有 wallet 再 lock for update，避免 deadlock
		_, err := w.GetBalancesForUpdate(ctx, []wallet.Types{wallet.TypeAvailable})
		if err != nil {
			return err
		}

		return nil
	}

	if err := use.ProcessTransaction(
		ctx, tx, userWalletFlow{
			UID:     tx.FromUID,
			Asset:   tx.Asset,
			Actions: []walletActionOption{withLockAvailable, use.withAddAvailable()},
		}); err != nil {
		return err
	}

	return nil
}

// DepositUnconfirmed 增加限制餘額
// 1. 新增一筆充值限制交易
// 2. 錢包新增限制餘額
// 3. 錢包變化新增一筆增加限制餘額資料
func (use *WalletUseCase) DepositUnconfirmed(ctx context.Context, tx usecase.WalletTransferRequest) error {
	// 確認錢包新增或減少的餘額是否正確
	if !tx.Amount.IsPositive() {
		return errs.InvalidRange("failed to get correct amount")
	}

	uidAsset := uidAssetKey{
		uid:   tx.FromUID,
		asset: tx.Asset,
	}

	exists := use.checkWalletExistence(uidAsset)
	tx.TxType = wallet.DepositUnconfirmed

	withLockUnconfirmed := func(ctx context.Context, tx *usecase.WalletTransferRequest, w repository.UserWalletService) error {
		if !exists {
			checkWallet, err := w.HasAvailableBalance(ctx)
			if err != nil {
				return err
			}

			// 錢包不存在要做新增
			if !checkWallet {
				if _, err := w.InitializeWallets(ctx, tx.Brand); err != nil {
					var mysqlErr *mysql.MySQLError
					// 解析是否被其他 transaction insert 了，是的話嘗試取得 insert 後的鎖，不是的話需要直接回傳錯誤
					if errors.As(err, &mysqlErr) && mysqlErr.Number == 1062 {
						logrus.WithFields(logrus.Fields{
							"err": err,
							"uid": tx.FromUID,
						}).Warn("Deposit.Create.Wallet")
					} else {
						return err
					}
				} else {
					// 因為是透過 transaction 新增，所以不用上鎖
					return nil
				}
			} else {
				exists = true
				use.markWalletAsExisting(uidAsset)
			}
		}

		// 確認有 wallet 再 lock for update，避免 deadlock
		_, err := w.GetBalancesForUpdate(ctx, []wallet.Types{wallet.TypeUnconfirmed})
		if err != nil {
			return err
		}

		return nil
	}

	if err := use.ProcessTransaction(
		ctx, tx, userWalletFlow{
			UID:     tx.FromUID,
			Asset:   tx.Asset,
			Actions: []walletActionOption{withLockUnconfirmed, use.withAddUnconfirmed()},
		}); err != nil {
		return err
	}

	return nil
}

// Freeze 凍結
// 1. 新增一筆凍結交易
// 2. 錢包減少可用餘額
// 3. 錢包增加凍結餘額
// 4. 錢包變化新增一筆減少可用餘額資料
// 5. 錢包變化新增一筆增加凍結餘額資料
// 6. 訂單錢包新增一筆資料，餘額是凍結金額
func (use *WalletUseCase) Freeze(ctx context.Context, tx usecase.WalletTransferRequest) error {
	// 確認錢包新增或減少的餘額是否正確
	if !tx.Amount.IsPositive() {
		return errs.InvalidRange("failed to get correct amount")
	}

	tx.TxType = wallet.DepositUnconfirmed

	return use.ProcessTransaction(ctx, tx, userWalletFlow{
		UID:     tx.FromUID,
		Asset:   tx.Asset,
		Actions: []walletActionOption{use.withLockAvailableAndFreeze(), use.withSubAvailable(), use.withAddFreeze()},
	})
}

// AppendFreeze 追加凍結金額
// 1. 新增一筆凍結交易
// 2. 錢包減少可用餘額
// 3. 錢包增加凍結餘額
// 4. 錢包變化新增一筆減少可用餘額資料
// 5. 錢包變化新增一筆增加凍結餘額資料
// 6. 原凍結金額上追加凍結金額
func (use *WalletUseCase) AppendFreeze(ctx context.Context, tx usecase.WalletTransferRequest) error {
	// 確認錢包新增或減少的餘額是否正確
	if !tx.Amount.IsPositive() {
		return errs.InvalidRange("failed to get correct amount")
	}

	tx.TxType = wallet.DepositUnconfirmed

	return use.ProcessTransaction(ctx, tx, userWalletFlow{
		UID:     tx.FromUID,
		Asset:   tx.Asset,
		Actions: []walletActionOption{use.withLockAvailableAndFreeze(), use.withSubAvailable(), use.withAddFreeze()},
	})
}

// UnFreeze 解凍
// 1. 新增一筆解凍交易
// 2. 減少order餘額
// 3. 錢包減少凍結餘額
// 4. 錢包變化新增一筆減少凍結餘額資料
func (use *WalletUseCase) UnFreeze(ctx context.Context, tx usecase.WalletTransferRequest) error {
	// 確認錢包新增或減少的餘額是否正確
	if !tx.Amount.IsPositive() {
		return errs.InvalidRange("failed to get correct amount")
	}

	tx.TxType = wallet.UnFreeze

	return use.ProcessTransaction(ctx, tx, userWalletFlow{
		UID:     tx.FromUID,
		Asset:   tx.Asset,
		Actions: []walletActionOption{use.withSubOrder(nil), use.withLockFreeze(), use.withSubFreeze()},
	})
}

// RollbackFreeze Rollback 凍結，不可指定金額(剩餘order凍結金額)
// 1. 新增一筆 Rollback 凍結交易
// 2. 減少 order 餘額
// 3. 錢包減少凍結餘額
// 4. 錢包增加可用餘額
// 5. 錢包變化新增一筆減少凍結餘額資料
// 6. 錢包變化新增一筆增加可用餘額資料
func (use *WalletUseCase) RollbackFreeze(ctx context.Context, tx usecase.WalletTransferRequest) error {
	tx.TxType = wallet.RollbackFreeze

	return use.ProcessTransaction(ctx, tx, userWalletFlow{
		UID:   tx.FromUID,
		Asset: tx.Asset,
		Actions: []walletActionOption{use.withSubOrder(func(tx *usecase.Transaction, order entity.Transaction) {
			tx.Amount = order.Balance
		}), use.withLockAvailableAndFreeze(), use.withSubFreeze(), use.withAddAvailable()},
	})

}

func (use *WalletUseCase) RollbackFreezeAddAvailable(ctx context.Context, tx usecase.WalletTransferRequest) error {
	//TODO implement me
	panic("implement me")
}

func (use *WalletUseCase) CancelFreeze(ctx context.Context, tx usecase.WalletTransferRequest) error {
	//TODO implement me
	panic("implement me")
}

func (use *WalletUseCase) Unconfirmed(ctx context.Context, tx usecase.WalletTransferRequest) error {
	//TODO implement me
	panic("implement me")
}

func (use *WalletUseCase) Balance(ctx context.Context, req usecase.BalanceReq) ([]usecase.Balance, error) {
	//TODO implement me
	panic("implement me")
}

func (use *WalletUseCase) HistoryBalance(ctx context.Context, req usecase.BalanceReq) ([]usecase.Balance, error) {
	//TODO implement me
	panic("implement me")
}

func (use *WalletUseCase) BalanceByAssets(ctx context.Context, uid, cryptoCode string, walletTypes []wallet.Types) (usecase.BalanceAssetsResp, error) {
	//TODO implement me
	panic("implement me")
}

func (use *WalletUseCase) CheckBalance(ctx context.Context, tx usecase.WalletTransferRequest) error {
	//TODO implement me
	panic("implement me")
}

func (use *WalletUseCase) GetTodayWithdraw(ctx context.Context, uid, toCrypto string) (usecase.TodayWithdrawResp, error) {
	//TODO implement me
	panic("implement me")
}

func MustWalletUseCase(param WalletUseCaseParam) usecase.WalletTransferUseCase {
	return &WalletUseCase{
		WalletUseCaseParam: param,
		existUIDAsset:      make(map[string]struct{}),
	}
}