Compare commits

..

4 Commits

Author SHA1 Message Date
daniel.w e9c5818157 update const 2024-10-31 12:05:22 +08:00
daniel.w 3c1c611f96 add wallet repo 2024-10-30 19:13:56 +08:00
daniel.w fc748fb66f add wallet repo 2024-10-29 22:49:47 +08:00
daniel.w 8687e467c9 add wallet database 2024-10-29 16:17:31 +08:00
45 changed files with 1528 additions and 122 deletions

View File

@ -62,3 +62,14 @@ mock-gen: # 建立 mock 資料
.PHONY: migrate-database
migrate-database:
migrate -source file://generate/database/mongodb -database 'mongodb://127.0.0.1:27017/digimon_order' up
.PHONY: gen-my-sql-model
gen-my-sql-model: # 建立 rpc 資料庫
goctl model mysql ddl -s ./generate/database/mysql/20230626063528_create_wallet_table.up.sql --style $(GO_ZERO_STYLE) -d ./internal/model -i ''
goctl model mysql ddl -s ./generate/database/mysql/20230626065705_create_transaction_table.up.sql --style $(GO_ZERO_STYLE) -d ./internal/model -i ''
goctl model mysql ddl -s ./generate/database/mysql/20230913072305_create_wallet_transaction.up.sql --style $(GO_ZERO_STYLE) -d ./internal/model -i ''
@echo "Generate mysql model files successfully"
.PHONY: migrate-database
migrate-database:
migrate -source file://generate/database/mysql -database 'mysql://root:yytt@tcp(127.0.0.1:3306)/digimon_wallet' up

View File

@ -27,3 +27,15 @@ Mongo:
MaxPoolSize: 30
MinPoolSize: 10
MaxConnIdleTime: 30m
DB:
Host: 127.0.0.1
Port: 3306
User: root
Password: yytt
name: permission
MaxIdleConns: 10
MaxOpenConns: 200
ConnMaxLifetime: 10s
InterpolateParams: false
Database: digimon_wallet

View File

@ -0,0 +1 @@
DROP TABLE IF EXISTS `wallet`;

View File

@ -0,0 +1,13 @@
CREATE TABLE `wallet` (
`id` BIGINT(20) AUTO_INCREMENT NOT NULL COMMENT '錢包流水號',
`uid` VARCHAR(50) NOT NULL COMMENT '用戶ID',
`brand` VARCHAR(50) DEFAULT '' NOT NULL COMMENT '品牌名稱',
`currency` VARCHAR(20) NOT NULL COMMENT '幣別(或平台點數)',
`balance` DECIMAL(30, 18) UNSIGNED DEFAULT 0 NOT NULL COMMENT '錢包餘額',
`wallet_type` TINYINT NOT NULL COMMENT '錢包種類: 1=可用, 2=凍結, 3=限制(僅出金)',
`created_at` BIGINT NOT NULL COMMENT '創建時間',
`updated_at` BIGINT NOT NULL COMMENT '更新時間',
PRIMARY KEY (`id`),
UNIQUE INDEX `uq_user_currency_type` (`uid`, `currency`, `wallet_type`),
INDEX `idx_user_currency` (`uid`, `currency`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='用戶錢包';

View File

@ -0,0 +1 @@
DROP TABLE IF EXISTS `transaction`;

View File

@ -0,0 +1,20 @@
CREATE TABLE `transaction` (
`id` BIGINT(20) AUTO_INCREMENT NOT NULL COMMENT '交易流水號',
`tx_id` VARCHAR(200) NOT NULL COMMENT '交易 ID',
`order_id` VARCHAR(200) NOT NULL COMMENT '訂單 ID',
`brand` VARCHAR(50) NOT NULL COMMENT '品牌名稱',
`uid` VARCHAR(50) NOT NULL COMMENT '發起 UID',
`to_uid` VARCHAR(50) NOT NULL DEFAULT '' COMMENT '接收方UID',
`type` TINYINT NOT NULL COMMENT '交易類型: 1=提現, 2=充值',
`business_type` TINYINT NOT NULL COMMENT '業務類型',
`currency` VARCHAR(20) NOT NULL COMMENT '幣種',
`amount` DECIMAL(30, 18) UNSIGNED DEFAULT 0 NOT NULL COMMENT '交易金額',
`balance` DECIMAL(30, 18) UNSIGNED DEFAULT 0 NOT NULL COMMENT '交易後餘額',
`before_balance` DECIMAL(30, 18) UNSIGNED DEFAULT 0 NOT NULL COMMENT '交易前餘額',
`status` TINYINT NOT NULL DEFAULT 0 COMMENT '結算狀態',
`create_time` BIGINT NOT NULL COMMENT '創建時間Unix 時間戳,毫秒)',
PRIMARY KEY (`id`),
UNIQUE INDEX `uk_order_id_type` (`order_id`, `type`),
INDEX `idx_uid` (`uid`),
INDEX `idx_order_id` (`order_id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='用戶錢包交異動紀錄';

View File

@ -0,0 +1 @@
DROP TABLE IF EXISTS `wallet_transaction`;

View File

@ -0,0 +1,19 @@
CREATE TABLE `wallet_journal` (
`id` BIGINT(20) AUTO_INCREMENT NOT NULL COMMENT '錢包賬本流水號',
`transaction_id` BIGINT(20) NOT NULL COMMENT '交易 ID對應 transaction 表',
`order_id` VARCHAR(200) NOT NULL COMMENT '訂單 ID對應 order 表',
`brand` VARCHAR(50) NOT NULL COMMENT '品牌名稱',
`uid` VARCHAR(50) NOT NULL COMMENT '用戶 ID',
`wallet_type` TINYINT NOT NULL COMMENT '餘額種類: 1=可用, 2=凍結, 3=限制(僅出金)',
`currency` VARCHAR(20) NOT NULL COMMENT '幣種或平台點數',
`transaction_amount` DECIMAL(30, 18) UNSIGNED DEFAULT 0 NOT NULL COMMENT '交易金額',
`post_transaction_balance` DECIMAL(30, 18) UNSIGNED DEFAULT 0 NOT NULL COMMENT '交易後餘額',
`business_type` TINYINT NOT NULL COMMENT '業務類型',
`status` TINYINT NOT NULL DEFAULT 0 COMMENT '狀態',
`due_time` BIGINT NOT NULL DEFAULT 0 COMMENT 'T+N 執行時間',
`created_at` BIGINT NOT NULL COMMENT '創建時間Unix 時間戳,毫秒)',
PRIMARY KEY (`id`),
KEY `idx_user_brand` (`uid`, `brand`),
KEY `idx_order_id` (`order_id`),
KEY `idx_status` (`status`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='用戶錢包變動';

View File

@ -0,0 +1 @@
DROP DATABASE IF EXISTS `digimon_wallet`;

View File

@ -0,0 +1 @@
CREATE DATABASE IF NOT EXISTS `digimon_wallet`;

7
go.mod
View File

@ -13,9 +13,12 @@ require (
go.uber.org/mock v0.5.0
google.golang.org/grpc v1.67.1
google.golang.org/protobuf v1.35.1
gorm.io/driver/sqlite v1.5.6
gorm.io/gorm v1.25.12
)
require (
filippo.io/edwards25519 v1.1.0 // indirect
github.com/beorn7/perks v1.0.1 // indirect
github.com/cenkalti/backoff/v4 v4.3.0 // indirect
github.com/cespare/xxhash/v2 v2.3.0 // indirect
@ -33,6 +36,7 @@ require (
github.com/go-openapi/swag v0.22.4 // indirect
github.com/go-playground/locales v0.14.1 // indirect
github.com/go-playground/universal-translator v0.18.1 // indirect
github.com/go-sql-driver/mysql v1.8.1 // indirect
github.com/gogo/protobuf v1.3.2 // indirect
github.com/golang/mock v1.6.0 // indirect
github.com/golang/protobuf v1.5.4 // indirect
@ -42,6 +46,8 @@ require (
github.com/google/gofuzz v1.2.0 // indirect
github.com/google/uuid v1.6.0 // indirect
github.com/grpc-ecosystem/grpc-gateway/v2 v2.20.0 // indirect
github.com/jinzhu/inflection v1.0.0 // indirect
github.com/jinzhu/now v1.1.5 // indirect
github.com/josharian/intern v1.0.0 // indirect
github.com/json-iterator/go v1.1.12 // indirect
github.com/klauspost/compress v1.17.9 // indirect
@ -49,6 +55,7 @@ require (
github.com/mailru/easyjson v0.7.7 // indirect
github.com/mattn/go-colorable v0.1.13 // indirect
github.com/mattn/go-isatty v0.0.20 // indirect
github.com/mattn/go-sqlite3 v1.14.22 // indirect
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
github.com/modern-go/reflect2 v1.0.2 // indirect
github.com/montanaflynn/stats v0.7.1 // indirect

View File

@ -24,4 +24,16 @@ type Config struct {
// Compressors []string
// EnableStandardReadWriteSplitMode bool
}
DB struct {
User string
Password string
Host string
Port int64
Database string
MaxIdleConns int
MaxOpenConns int
ConnMaxLifetime time.Duration
InterpolateParams bool
}
}

View File

@ -23,6 +23,14 @@ const (
TimeoutOrderErrorCode
)
// Error Code 統一這邊改
const (
_ ErrorCode = 20 + iota
CreateWalletErrorCode
WalletTxErrorCode
WalletBalanceNotFound
)
const (
_ ErrorCode = 10 + iota
DataNotFoundErrorCode

View File

@ -1,32 +0,0 @@
package domain
type OrderStatus int64
func (o *OrderStatus) ToInt64() int64 {
return int64(*o)
}
const (
OrderStatusCreated OrderStatus = 0 // 建立訂單
OrderStatusFailed OrderStatus = 1 // 建單失敗
OrderStatusReviewing OrderStatus = 2 // 審核中
OrderStatusPaying OrderStatus = 3 // 付款中
OrderStatusPaid OrderStatus = 4 // 已付款
OrderStatusPendingTransfer OrderStatus = 5 // 已付款待轉帳
OrderStatusDisputing OrderStatus = 6 // 申訴中
OrderStatusCompleted OrderStatus = 7 // 交易完成
OrderStatusFailedTrade OrderStatus = 8 // 交易失敗
OrderStatusCancelled OrderStatus = 9 // 交易取消
OrderStatusAbnormal OrderStatus = 10 // 交易異常
OrderStatusTimeout OrderStatus = 11 // 交易超時
)
type OrderType int64
const (
OrderTypeTest OrderType = 0 // 測試訂單
)
func (o *OrderType) ToInt() int {
return int(*o)
}

View File

@ -0,0 +1,25 @@
package order
// Status 表示訂單狀態
type Status int64
// ToInt64 將訂單狀態轉為 int64
func (s Status) ToInt64() int64 {
return int64(s)
}
// 訂單狀態常量
const (
Created Status = iota // 訂單已建立
Failed // 建單失敗
UnderReview // 審核中
Processing // 付款中
Paid // 已付款
AwaitingTransfer // 待轉帳
InDispute // 申訴中
Completed // 交易完成
FailedTrade // 交易失敗
Cancelled // 交易取消
Abnormal // 交易異常
TimedOut // 交易超時
)

View File

@ -0,0 +1,15 @@
package order
// Type 表示訂單類型
type Type int64
// ToInt 將訂單類型轉為 int
func (t Type) ToInt() int {
return int(t)
}
// 訂單類型常量
const (
TestType Type = iota // 測試訂單
)

View File

@ -0,0 +1,21 @@
package repository
import (
"app-cloudep-trade-service/internal/domain/wallet"
"app-cloudep-trade-service/internal/model"
"context"
"github.com/shopspring/decimal"
)
// 如果有需要Select for update 的話,請在 Option 當中加上鎖並且傳入 sqlx.conn)
// UserWalletOperator 針對使用者的錢包基本操作接口
type UserWalletOperator interface {
// Balances 取得多種類別餘額
Balances(ctx context.Context, balanceType []wallet.BalanceType, opts ...WalletOperatorOption) ([]model.Wallet, error)
// LocalBalance 取得本地錢包的數額
LocalBalance(balanceType wallet.BalanceType) decimal.Decimal
// GetBalancesByID 取得錢包的數額 ByID
GetBalancesByID(ctx context.Context, ids []int64, opts ...WalletOperatorOption) ([]model.Wallet, error)
}

View File

@ -0,0 +1,30 @@
package repository
import (
"app-cloudep-trade-service/internal/domain/wallet"
"app-cloudep-trade-service/internal/model"
"context"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
// WalletRepository 錢包基礎操作(可能有平台,使用者等多元的錢包)
type WalletRepository interface {
// Create 建立錢包(某 id 的可用,凍結,限制三種)
Create(ctx context.Context, uid, currency, brand string) ([]*model.Wallet, error)
// Balances 取得某個錢包的餘額
Balances(ctx context.Context, req BalanceReq) ([]model.Wallet, error)
// GetUserWalletOperator 取得使用者錢包操作看使否需要使用 transaction
GetUserWalletOperator(uid, currency string, opts ...Option) UserWalletOperator
// Transaction 把 tx 暴露出來
Transaction(ctx context.Context, fn func(tx sqlx.Session) error) error
}
// BalanceReq 取得全部的,因為一個人錢包種類的不會太多,故全撈
type BalanceReq struct {
UID []string
Currency []string
BalanceType []wallet.BalanceType
}
type Option func() sqlx.SqlConn

View File

@ -0,0 +1,67 @@
package repository
import (
"app-cloudep-trade-service/internal/domain/wallet"
"github.com/zeromicro/go-zero/core/stores/sqlx"
"github.com/shopspring/decimal"
)
// WalletOperatorOption 選項模式
type WalletOperatorOption func(*WalletOptions)
type WalletOptions struct {
WithLock bool
OrderID string
Amount decimal.Decimal
BalanceType wallet.BalanceType
Business wallet.BusinessLogic
Tx *sqlx.SqlConn
}
// ApplyOptions 將多個 WalletOperatorOption 應用到一個 walletOptions 中
func ApplyOptions(opts ...WalletOperatorOption) WalletOptions {
options := WalletOptions{}
for _, opt := range opts {
opt(&options)
}
return options
}
func WithLock() WalletOperatorOption {
return func(opts *WalletOptions) {
opts.WithLock = true
}
}
func WithOrderID(orderID string) WalletOperatorOption {
return func(opts *WalletOptions) {
opts.OrderID = orderID
}
}
func WithAmount(amount decimal.Decimal) WalletOperatorOption {
return func(opts *WalletOptions) {
opts.Amount = amount
}
}
func WithKind(balanceType wallet.BalanceType) WalletOperatorOption {
return func(opts *WalletOptions) {
opts.BalanceType = balanceType
}
}
func WithBusiness(business wallet.BusinessLogic) WalletOperatorOption {
return func(opts *WalletOptions) {
opts.Business = business
}
}
func WithSession(session *sqlx.SqlConn) WalletOperatorOption {
return func(opts *WalletOptions) {
opts.Tx = session
}
}

View File

@ -1,7 +1,7 @@
package usecase
import (
"app-cloudep-trade-service/internal/domain"
"app-cloudep-trade-service/internal/domain/order"
"context"
"github.com/shopspring/decimal"
@ -38,7 +38,7 @@ type ModifyOrderQuery struct {
// CancelOrderQuery 1.建單失敗 9.交易取消 10.交易異常
type CancelOrderQuery struct {
BusinessID string
Status domain.OrderStatus
Status order.Status
}
// DeleteOrderQuery 刪除訂單(軟刪除)
@ -52,42 +52,42 @@ type GetOrderQuery struct {
}
type GetOrderResp struct {
BusinessID string // 訂單業務流水號
OrderType domain.OrderType `json:"order_type"` // 訂單類型
OrderStatus domain.OrderStatus `json:"order_status"` // 訂單狀態
Brand string `json:"brand"` // 下單平台
OrderUID string `json:"order_uid"` // 下單用戶 UID
ReferenceID string `json:"reference_id"` // 訂單來源
Count string `json:"count"` // 訂單數量 (decimal to string)
OrderFee string `json:"order_fee"` // 訂單手續費 (decimal to string)
Amount string `json:"amount"` // 單價 (decimal to string)
ReferenceBrand *string `json:"reference_brand,omitempty"` // 訂單來源平台
ReferenceUID *string `json:"reference_uid,omitempty"` // 訂單來源用戶 UID
WalletStatus *int64 `json:"wallet_status,omitempty"` // 交易金額狀態
ThreePartyStatus *int64 `json:"three_party_status,omitempty"` // 三方請求狀態
DirectionType *int64 `json:"direction_type,omitempty"` // 交易方向
CryptoType *string `json:"crypto_type,omitempty"` // 交易幣種
ThirdPartyFee *string `json:"third_party_fee,omitempty"` // 第三方手續費 (decimal to string)
CryptoToUsdtRate *string `json:"crypto_to_usdt_rate,omitempty"` // 交易幣種對 USDT 匯率 (decimal to string)
FiatToUsdRate *string `json:"fiat_to_usd_rate,omitempty"` // 法幣對 USD 匯率 (decimal to string)
FeeCryptoToUsdtRate *string `json:"fee_crypto_to_usdt_rate,omitempty"` // 手續費幣種對 USDT 匯率 (decimal to string)
UsdtToCryptoTypeRate *string `json:"usdt_to_crypto_type_rate,omitempty"` // USDT 對交易幣種匯率 (decimal to string)
PaymentFiat *string `json:"payment_fiat,omitempty"` // 支付法幣
PaymentUnitPrice *string `json:"payment_unit_price,omitempty"` // crypto 單價 (decimal to string)
PaymentTemplateID *string `json:"payment_template_id,omitempty"` // 支付方式配置 ID
OrderArrivalTime *int64 `json:"order_arrival_time,omitempty"` // 訂單到帳時間
OrderPaymentTime *int64 `json:"order_payment_time,omitempty"` // 訂單付款時間
UnpaidTimeoutSecond *int64 `json:"unpaid_timeout_second,omitempty"` // 支付期限秒數
ChainType *string `json:"chain_type,omitempty"` // 主網類型
TxHash *string `json:"tx_hash,omitempty,omitempty"` // 交易哈希
FromAddress *string `json:"from_address,omitempty,omitempty"` // 來源地址
ToAddress *string `json:"to_address,omitempty,omitempty"` // 目標地址
ChainFee *string `json:"chain_fee,omitempty"` // 鏈上交易手續費 (decimal to string)
ChainFeeCrypto *string `json:"chain_fee_crypto,omitempty"` // 鏈上手續費使用幣別
Memo *string `json:"memo,omitempty"` // 鏈上備註
OrderNote *string `json:"order_note,omitempty"` // 訂單交易備註
CreateTime int64 `json:"create_time,omitempty"` // 建立時間
UpdateTime int64 `json:"update_time,omitempty"` // 更新時間
BusinessID string // 訂單業務流水號
OrderType order.Type `json:"order_type"` // 訂單類型
OrderStatus order.Status `json:"order_status"` // 訂單狀態
Brand string `json:"brand"` // 下單平台
OrderUID string `json:"order_uid"` // 下單用戶 UID
ReferenceID string `json:"reference_id"` // 訂單來源
Count string `json:"count"` // 訂單數量 (decimal to string)
OrderFee string `json:"order_fee"` // 訂單手續費 (decimal to string)
Amount string `json:"amount"` // 單價 (decimal to string)
ReferenceBrand *string `json:"reference_brand,omitempty"` // 訂單來源平台
ReferenceUID *string `json:"reference_uid,omitempty"` // 訂單來源用戶 UID
WalletStatus *int64 `json:"wallet_status,omitempty"` // 交易金額狀態
ThreePartyStatus *int64 `json:"three_party_status,omitempty"` // 三方請求狀態
DirectionType *int64 `json:"direction_type,omitempty"` // 交易方向
CryptoType *string `json:"crypto_type,omitempty"` // 交易幣種
ThirdPartyFee *string `json:"third_party_fee,omitempty"` // 第三方手續費 (decimal to string)
CryptoToUsdtRate *string `json:"crypto_to_usdt_rate,omitempty"` // 交易幣種對 USDT 匯率 (decimal to string)
FiatToUsdRate *string `json:"fiat_to_usd_rate,omitempty"` // 法幣對 USD 匯率 (decimal to string)
FeeCryptoToUsdtRate *string `json:"fee_crypto_to_usdt_rate,omitempty"` // 手續費幣種對 USDT 匯率 (decimal to string)
UsdtToCryptoTypeRate *string `json:"usdt_to_crypto_type_rate,omitempty"` // USDT 對交易幣種匯率 (decimal to string)
PaymentFiat *string `json:"payment_fiat,omitempty"` // 支付法幣
PaymentUnitPrice *string `json:"payment_unit_price,omitempty"` // crypto 單價 (decimal to string)
PaymentTemplateID *string `json:"payment_template_id,omitempty"` // 支付方式配置 ID
OrderArrivalTime *int64 `json:"order_arrival_time,omitempty"` // 訂單到帳時間
OrderPaymentTime *int64 `json:"order_payment_time,omitempty"` // 訂單付款時間
UnpaidTimeoutSecond *int64 `json:"unpaid_timeout_second,omitempty"` // 支付期限秒數
ChainType *string `json:"chain_type,omitempty"` // 主網類型
TxHash *string `json:"tx_hash,omitempty,omitempty"` // 交易哈希
FromAddress *string `json:"from_address,omitempty,omitempty"` // 來源地址
ToAddress *string `json:"to_address,omitempty,omitempty"` // 目標地址
ChainFee *string `json:"chain_fee,omitempty"` // 鏈上交易手續費 (decimal to string)
ChainFeeCrypto *string `json:"chain_fee_crypto,omitempty"` // 鏈上手續費使用幣別
Memo *string `json:"memo,omitempty"` // 鏈上備註
OrderNote *string `json:"order_note,omitempty"` // 訂單交易備註
CreateTime int64 `json:"create_time,omitempty"` // 建立時間
UpdateTime int64 `json:"update_time,omitempty"` // 更新時間
}
type GetOrderListReq struct {
@ -98,7 +98,7 @@ type GetOrderListReq struct {
ReferenceUID string
BusinessID string
UID string
OrderType domain.OrderType
OrderType order.Type
DirectionType []int64
OrderStatus []int64
@ -128,16 +128,16 @@ type ListOrderResp struct {
type CreateOrderReq struct {
BusinessID string
OrderType domain.OrderType // 訂單類型
OrderStatus domain.OrderStatus // 訂單狀態
Brand string // 下單平台
OrderUID string // 下單用戶 UID
ReferenceID string // 訂單來源
Count decimal.Decimal // 訂單數量
OrderFee decimal.Decimal // 訂單手續費
Amount decimal.Decimal // 單價
WalletStatus int64 // 交易金額狀態
DirectionType int64 // 交易方向
OrderType order.Type // 訂單類型
OrderStatus order.Status // 訂單狀態
Brand string // 下單平台
OrderUID string // 下單用戶 UID
ReferenceID string // 訂單來源
Count decimal.Decimal // 訂單數量
OrderFee decimal.Decimal // 訂單手續費
Amount decimal.Decimal // 單價
WalletStatus int64 // 交易金額狀態
DirectionType int64 // 交易方向
// 以上為必要欄位,下面是區塊鏈時才需要
ReferenceBrand *string // 訂單來源平台
ReferenceUID *string // 訂單來源用戶 UID

View File

@ -0,0 +1,81 @@
package usecase
import (
"app-cloudep-trade-service/internal/domain/wallet"
"context"
"github.com/shopspring/decimal"
)
// WalletQueryUseCase 定義所有查詢行為(餘額查詢、檢查、歷史記錄)的行為
type WalletQueryUseCase interface {
// Balance 用戶餘額
Balance(ctx context.Context, req BalanceReq) ([]Balance, error)
// CheckBalance 根據tx檢查用戶餘額是否足夠
CheckBalance(ctx context.Context, tx Transaction) error
// HistoryBalance 歷史餘額變化
HistoryBalance(ctx context.Context, req BalanceReq) ([]Balance, error)
}
// WalletOperationUseCase 定義所有錢包操作(提款、充值、凍結等)的行為
type WalletOperationUseCase interface {
// Withdraw 提款
Withdraw(ctx context.Context, tx Transaction) error
// Deposit 充值
Deposit(ctx context.Context, tx Transaction) error
// DepositUnconfirmed 增加限制餘額
DepositUnconfirmed(ctx context.Context, tx Transaction) error
// Freeze 凍結
Freeze(ctx context.Context, tx Transaction) error
// AppendFreeze 追加凍結金額
AppendFreeze(ctx context.Context, tx Transaction) error
// UnFreeze 解凍
UnFreeze(ctx context.Context, tx Transaction) error
// RollbackFreeze Rollback 凍結,不可指定金額(剩餘order凍結金額)
RollbackFreeze(ctx context.Context, tx Transaction) error
// RollbackFreezeAddAvailable Rollback剩餘凍結可指定金額(剩餘order凍結金額)
RollbackFreezeAddAvailable(ctx context.Context, tx Transaction) error
// CancelFreeze 取消凍結,可指定金額
CancelFreeze(ctx context.Context, tx Transaction) error
// Unconfirmed 增加限制
Unconfirmed(ctx context.Context, tx Transaction) error
}
// WalletUseCase 基礎操作類別
type WalletUseCase interface {
WalletOperationUseCase
WalletQueryUseCase
}
// Transaction 交易
type Transaction struct {
OrderID string // 交易訂單
UID string // 交易發起人
ToUID string // 交易接收人
Currency string // 幣別
Amount decimal.Decimal // 交易金額
BeforeBalance decimal.Decimal // 交易前餘額
Type wallet.TransactionType // 交易種類
BusinessType wallet.BusinessLogic // 商業種類
Brand string // 轉帳平台
From wallet.BalanceType // 從哪種錢包類型
To wallet.BalanceType // 到哪種錢包類型
}
type BalanceReq struct {
UID string // 用戶 UID
Currency string // 幣值
BeforeHour int // 在某個時段之前
}
type Balance struct {
Currency string `json:"currency"`
Available decimal.Decimal `json:"available"`
Unavailable UnavailableBalance `json:"unavailable"`
UpdateTime int64 `json:"update_time"`
}
type UnavailableBalance struct {
Freeze decimal.Decimal `json:"freeze"`
Unconfirmed decimal.Decimal `json:"unconfirmed"`
}

View File

@ -1,8 +1,8 @@
package domain
package wallet
type WalletStatus int64
type Status int64
func (o *WalletStatus) ToInt() int {
func (o *Status) ToInt() int {
return int(*o)
}

View File

@ -0,0 +1,52 @@
package wallet
// ===================商業邏輯類型===================
type BusinessLogic string
// 定義商業邏輯名稱
const (
// NoLogic 無商業邏輯
NoLogic BusinessLogic = ""
// OrderLogic 訂單相關業務邏輯
OrderLogic BusinessLogic = "order"
// SystemTransferLogic 系統劃轉邏輯
SystemTransferLogic BusinessLogic = "system_transfer"
)
// 定義商業邏輯類型
const (
// NoLogicType 無商業邏輯類型
NoLogicType int8 = iota
// OrderLogicType 訂單業務邏輯類型
OrderLogicType
// SystemTransferLogicType 系統劃轉業務邏輯類型
SystemTransferLogicType
)
// 定義名稱和類型之間的映射
var logicNameToType = map[BusinessLogic]int8{
OrderLogic: OrderLogicType,
SystemTransferLogic: SystemTransferLogicType,
}
var logicTypeToName = map[int8]BusinessLogic{
OrderLogicType: OrderLogic,
SystemTransferLogicType: SystemTransferLogic,
}
// ToInt8 將 BusinessLogic 轉換為對應的 int8 類型
func (b BusinessLogic) ToInt8() int8 {
if val, ok := logicNameToType[b]; ok {
return val
}
return NoLogicType
}
// LogicTypeToName 將 int8 類型轉換為對應的 BusinessLogic 名稱
func LogicTypeToName(b int8) BusinessLogic {
if val, ok := logicTypeToName[b]; ok {
return val
}
return NoLogic
}

View File

@ -0,0 +1,49 @@
package wallet
// ===================交易類型===================
type TransactionType int64
// 定義交易類型
const (
// Deposit 增加可用餘額的充值交易
Deposit TransactionType = iota + 1
// Withdraw 減少可用餘額的提現交易
Withdraw
// Freeze 將可用餘額轉入凍結餘額的凍結交易
Freeze
// Unfreeze 減少凍結餘額的解凍交易
Unfreeze
// RollbackFreeze 回滾凍結:減少凍結餘額並恢復至可用餘額,不指定金額
RollbackFreeze
// Unconfirmed 限制交易:減少凍結餘額並增加他人限制餘額
Unconfirmed
// CancelFreeze 取消凍結:減少凍結餘額並恢復至可用餘額,允許指定金額
CancelFreeze
// DepositToUnconfirmed 增加限制餘額的充值交易
DepositToUnconfirmed
// AppendFreeze 追加凍結:減少可用餘額並增加凍結餘額
AppendFreeze
// RollbackFreezeToAvailable 回滾凍結:指定金額回滾凍結餘額並增加至可用餘額
RollbackFreezeToAvailable
// PlatformDistribution 平台分發交易
PlatformDistribution
// SystemTransfer 系統劃轉交易
SystemTransfer
)
// ToInt 將交易類型轉換為 int64
func (t TransactionType) ToInt() int64 {
return int64(t)
}

View File

@ -0,0 +1,58 @@
package wallet
// ===================錢包餘額類型===================
// BalanceType 表示錢包餘額的類型
type BalanceType int64
// 錢包餘額類型
const (
// AvailableBalanceType 表示可動用的餘額
AvailableBalanceType BalanceType = iota + 1
// FrozenBalanceType 表示在交易過程中凍結的餘額
FrozenBalanceType
// PendingBalanceType 表示已提交但尚未確認完成的交易餘額
PendingBalanceType
// ContractAvailableBalanceType 表示合約或交易可用的餘額
ContractAvailableBalanceType
// ContractFrozenBalanceType 表示合約或交易凍結的餘額
ContractFrozenBalanceType
)
// balanceTypeToName 將餘額類型映射到其名稱
var balanceTypeToName = map[BalanceType]string{
AvailableBalanceType: "available",
FrozenBalanceType: "frozen",
PendingBalanceType: "pending",
ContractAvailableBalanceType: "contract_available",
ContractFrozenBalanceType: "contract_frozen",
}
// nameToBalanceType 將名稱映射到餘額類型
var nameToBalanceType = map[string]BalanceType{
"available": AvailableBalanceType,
"frozen": FrozenBalanceType,
"pending": PendingBalanceType,
"contract_available": ContractAvailableBalanceType,
"contract_frozen": ContractFrozenBalanceType,
}
// Name 返回餘額類型的名稱
func (b BalanceType) Name() string {
return balanceTypeToName[b]
}
// GetBalanceTypeByName 根據名稱返回對應的餘額類型
func GetBalanceTypeByName(name string) BalanceType {
return nameToBalanceType[name]
}
// AllBalanceTypes 包含目前所有的錢包餘額類型
var AllBalanceTypes = []BalanceType{
AvailableBalanceType, FrozenBalanceType, PendingBalanceType,
ContractAvailableBalanceType, ContractFrozenBalanceType,
}

View File

@ -1,7 +1,7 @@
package orderservicelogic
import (
"app-cloudep-trade-service/internal/domain"
"app-cloudep-trade-service/internal/domain/order"
"app-cloudep-trade-service/internal/domain/usecase"
"context"
@ -46,7 +46,7 @@ func (l *CancelOrderLogic) CancelOrder(in *trade.CancelOrderReq) (*trade.OKResp,
err := l.svcCtx.OrderUseCase.CancelOrder(l.ctx, usecase.CancelOrderQuery{
BusinessID: in.GetBusinessId(),
Status: domain.OrderStatus(in.GetStatus()),
Status: order.Status(in.GetStatus()),
})
if err != nil {
return nil, err

View File

@ -2,7 +2,7 @@ package orderservicelogic
import (
"app-cloudep-trade-service/gen_result/pb/trade"
"app-cloudep-trade-service/internal/domain"
"app-cloudep-trade-service/internal/domain/order"
"app-cloudep-trade-service/internal/domain/usecase"
"app-cloudep-trade-service/internal/svc"
"context"
@ -169,8 +169,8 @@ func buildCreateOrderReq(in *trade.CreateOrderReq) *createOrderReq {
func toCreateOrderUseCase(req *createOrderReq) usecase.CreateOrderReq {
return usecase.CreateOrderReq{
BusinessID: req.BusinessID,
OrderType: domain.OrderType(req.OrderType),
OrderStatus: domain.OrderStatus(req.OrderStatus),
OrderType: order.Type(req.OrderType),
OrderStatus: order.Status(req.OrderStatus),
Brand: req.Brand,
OrderUID: req.OrderUID,
ReferenceID: req.ReferenceID,

View File

@ -1,7 +1,7 @@
package orderservicelogic
import (
"app-cloudep-trade-service/internal/domain"
"app-cloudep-trade-service/internal/domain/order"
"app-cloudep-trade-service/internal/domain/usecase"
"context"
@ -61,7 +61,7 @@ func toGetOrderListReq(req *trade.ListOrderReq) usecase.GetOrderListReq {
ReferenceUID: req.ReferenceUid,
BusinessID: req.BusinessId,
UID: req.Uid,
OrderType: domain.OrderType(req.OrderType),
OrderType: order.Type(req.OrderType),
DirectionType: i32To64(req.DirectionType),
OrderStatus: i32To64(req.OrderStatus),
StartCreateTime: req.StartCreateTime,

View File

@ -2,23 +2,8 @@ package orderservicelogic
import (
"github.com/shopspring/decimal"
"github.com/zeromicro/go-zero/core/logx"
)
func decimalPtrFromString(val string) *decimal.Decimal {
if val == "" {
return nil
}
dec, err := decimal.NewFromString(val)
if err != nil {
logx.Errorf("Failed to convert string to decimal: %v", err)
return nil
}
return &dec
}
// getInt64Value 將 *int64 的值返回,如果為 nil 則返回 0
func getInt64Value(val *int64) int64 {
if val == nil {

View File

@ -1,7 +1,7 @@
package model
import (
"app-cloudep-trade-service/internal/domain"
"app-cloudep-trade-service/internal/domain/order"
"context"
"errors"
"time"
@ -99,7 +99,7 @@ func (m *customOrderModel) UpdateTimeoutOrder(ctx context.Context, req UpdateTim
// 構建過濾條件,選擇創建時間在指定時間之前且狀態為 0 (創建) 的項目
filter := bson.M{
"create_time": bson.M{"$lt": req.CreateTimeBefore},
"status": domain.OrderStatusCreated,
"status": order.Cancelled,
"$or": []bson.M{
{"delete_time": bson.M{"$exists": false}},
{"delete_time": 0},
@ -109,7 +109,7 @@ func (m *customOrderModel) UpdateTimeoutOrder(ctx context.Context, req UpdateTim
// 更新內容,將狀態設置為 11並更新 update_time
updates := bson.M{
"$set": bson.M{
"status": domain.OrderStatusTimeout,
"status": order.TimedOut,
"update_time": time.Now().UTC().UnixNano(),
},
}

View File

@ -1,7 +1,7 @@
package model
import (
"app-cloudep-trade-service/internal/domain"
"app-cloudep-trade-service/internal/domain/order"
"github.com/shopspring/decimal"
"go.mongodb.org/mongo-driver/bson/primitive"
@ -12,8 +12,8 @@ type Order struct {
UpdateTime int64 `bson:"update_time"`
CreateTime int64 `bson:"create_time"`
BusinessID string `bson:"business_id"` // 訂單業務流水號
OrderType domain.OrderType `bson:"order_type"` // 訂單類型
OrderStatus domain.OrderStatus `bson:"order_status"` // 訂單狀態
OrderType order.Type `bson:"order_type"` // 訂單類型
OrderStatus order.Status `bson:"order_status"` // 訂單狀態
Brand string `bson:"brand"` // 下單平台
OrderUID string `bson:"order_uid"` // 下單用戶 UID
ReferenceID string `bson:"reference_id"` // 訂單來源

View File

@ -0,0 +1,37 @@
package model
import (
"fmt"
"strings"
)
// queryBindINBuilder 用來動態生成 WHERE 條件和參數
func queryBindINBuilder(baseQuery string, conditions map[string][]any) (string, []any) {
args := make([]any, 0)
whereClause := make([]string, 0)
for column, values := range conditions {
if len(values) > 0 {
placeholders := strings.Repeat("?,", len(values))
placeholders = placeholders[:len(placeholders)-1]
whereClause = append(whereClause, fmt.Sprintf("%s IN (%s)", column, placeholders))
args = append(args, values...)
}
}
// 將 WHERE 條件添加到查詢中
if len(whereClause) > 0 {
baseQuery += " WHERE " + strings.Join(whereClause, " AND ")
}
return baseQuery, args
}
func convertSliceToInterface[T any](slice []T) []any {
result := make([]any, 0, len(slice))
for _, v := range slice {
result = append(result, v)
}
return result
}

View File

@ -0,0 +1,24 @@
package model
import "github.com/zeromicro/go-zero/core/stores/sqlx"
var _ TransactionModel = (*customTransactionModel)(nil)
type (
// TransactionModel is an interface to be customized, add more methods here,
// and implement the added methods in customTransactionModel.
TransactionModel interface {
transactionModel
}
customTransactionModel struct {
*defaultTransactionModel
}
)
// NewTransactionModel returns a model for the database table.
func NewTransactionModel(conn sqlx.SqlConn) TransactionModel {
return &customTransactionModel{
defaultTransactionModel: newTransactionModel(conn),
}
}

View File

@ -0,0 +1,118 @@
// Code generated by goctl. DO NOT EDIT.
package model
import (
"context"
"database/sql"
"fmt"
"strings"
"github.com/zeromicro/go-zero/core/stores/builder"
"github.com/zeromicro/go-zero/core/stores/sqlc"
"github.com/zeromicro/go-zero/core/stores/sqlx"
"github.com/zeromicro/go-zero/core/stringx"
)
var (
transactionFieldNames = builder.RawFieldNames(&Transaction{})
transactionRows = strings.Join(transactionFieldNames, ",")
transactionRowsExpectAutoSet = strings.Join(stringx.Remove(transactionFieldNames, "`id`"), ",")
transactionRowsWithPlaceHolder = strings.Join(stringx.Remove(transactionFieldNames, "`id`"), "=?,") + "=?"
)
type (
transactionModel interface {
Insert(ctx context.Context, data *Transaction) (sql.Result, error)
FindOne(ctx context.Context, id int64) (*Transaction, error)
FindOneByOrderIdType(ctx context.Context, orderId string, tp int64) (*Transaction, error)
Update(ctx context.Context, data *Transaction) error
Delete(ctx context.Context, id int64) error
}
defaultTransactionModel struct {
conn sqlx.SqlConn
table string
}
Transaction struct {
Id int64 `db:"id"` // 交易流水號
TxId string `db:"tx_id"` // 交易 ID
OrderId string `db:"order_id"` // 訂單 ID
Brand string `db:"brand"` // 品牌名稱
Uid string `db:"uid"` // 發起 UID
ToUid string `db:"to_uid"` // 接收方UID
Type int64 `db:"type"` // 交易類型: 1=提現, 2=充值
BusinessType int64 `db:"business_type"` // 業務類型
Currency string `db:"currency"` // 幣種
Amount float64 `db:"amount"` // 交易金額
Balance float64 `db:"balance"` // 交易後餘額
BeforeBalance float64 `db:"before_balance"` // 交易前餘額
Status int64 `db:"status"` // 結算狀態
CreateTime int64 `db:"create_time"` // 創建時間Unix 時間戳,毫秒)
}
)
func newTransactionModel(conn sqlx.SqlConn) *defaultTransactionModel {
return &defaultTransactionModel{
conn: conn,
table: "`transaction`",
}
}
func (m *defaultTransactionModel) withSession(session sqlx.Session) *defaultTransactionModel {
return &defaultTransactionModel{
conn: sqlx.NewSqlConnFromSession(session),
table: "`transaction`",
}
}
func (m *defaultTransactionModel) Delete(ctx context.Context, id int64) error {
query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
_, err := m.conn.ExecCtx(ctx, query, id)
return err
}
func (m *defaultTransactionModel) FindOne(ctx context.Context, id int64) (*Transaction, error) {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", transactionRows, m.table)
var resp Transaction
err := m.conn.QueryRowCtx(ctx, &resp, query, id)
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultTransactionModel) FindOneByOrderIdType(ctx context.Context, orderId string, tp int64) (*Transaction, error) {
var resp Transaction
query := fmt.Sprintf("select %s from %s where `order_id` = ? and `type` = ? limit 1", transactionRows, m.table)
err := m.conn.QueryRowCtx(ctx, &resp, query, orderId, tp)
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultTransactionModel) Insert(ctx context.Context, data *Transaction) (sql.Result, error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, transactionRowsExpectAutoSet)
ret, err := m.conn.ExecCtx(ctx, query, data.TxId, data.OrderId, data.Brand, data.Uid, data.ToUid, data.Type, data.BusinessType, data.Currency, data.Amount, data.Balance, data.BeforeBalance, data.Status, data.CreateTime)
return ret, err
}
func (m *defaultTransactionModel) Update(ctx context.Context, newData *Transaction) error {
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, transactionRowsWithPlaceHolder)
_, err := m.conn.ExecCtx(ctx, query, newData.TxId, newData.OrderId, newData.Brand, newData.Uid, newData.ToUid, newData.Type, newData.BusinessType, newData.Currency, newData.Amount, newData.Balance, newData.BeforeBalance, newData.Status, newData.CreateTime, newData.Id)
return err
}
func (m *defaultTransactionModel) tableName() string {
return m.table
}

5
internal/model/vars.go Normal file
View File

@ -0,0 +1,5 @@
package model
import "github.com/zeromicro/go-zero/core/stores/sqlx"
var ErrNotFound = sqlx.ErrNotFound

View File

@ -0,0 +1,37 @@
package model
import (
"context"
"database/sql"
"fmt"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
var _ WalletJournalModel = (*customWalletJournalModel)(nil)
type (
// WalletJournalModel is an interface to be customized, add more methods here,
// and implement the added methods in customWalletJournalModel.
WalletJournalModel interface {
walletJournalModel
InsertWithSession(ctx context.Context, tx sqlx.Session, data *WalletJournal) (sql.Result, error)
}
customWalletJournalModel struct {
*defaultWalletJournalModel
}
)
// NewWalletJournalModel returns a model for the database table.
func NewWalletJournalModel(conn sqlx.SqlConn) WalletJournalModel {
return &customWalletJournalModel{
defaultWalletJournalModel: newWalletJournalModel(conn),
}
}
func (m *customWalletJournalModel) InsertWithSession(ctx context.Context, tx sqlx.Session, data *WalletJournal) (sql.Result, error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, walletJournalRowsExpectAutoSet)
ret, err := tx.ExecCtx(ctx, query, data.TransactionId, data.OrderId, data.Brand, data.Uid, data.WalletType, data.Currency, data.TransactionAmount, data.PostTransactionBalance, data.BusinessType, data.Status, data.DueTime, data.CreatedAt)
return ret, err
}

View File

@ -0,0 +1,102 @@
// Code generated by goctl. DO NOT EDIT.
package model
import (
"context"
"database/sql"
"fmt"
"strings"
"github.com/zeromicro/go-zero/core/stores/builder"
"github.com/zeromicro/go-zero/core/stores/sqlc"
"github.com/zeromicro/go-zero/core/stores/sqlx"
"github.com/zeromicro/go-zero/core/stringx"
)
var (
walletJournalFieldNames = builder.RawFieldNames(&WalletJournal{})
walletJournalRows = strings.Join(walletJournalFieldNames, ",")
walletJournalRowsExpectAutoSet = strings.Join(stringx.Remove(walletJournalFieldNames, "`id`"), ",")
walletJournalRowsWithPlaceHolder = strings.Join(stringx.Remove(walletJournalFieldNames, "`id`"), "=?,") + "=?"
)
type (
walletJournalModel interface {
Insert(ctx context.Context, data *WalletJournal) (sql.Result, error)
FindOne(ctx context.Context, id int64) (*WalletJournal, error)
Update(ctx context.Context, data *WalletJournal) error
Delete(ctx context.Context, id int64) error
}
defaultWalletJournalModel struct {
conn sqlx.SqlConn
table string
}
WalletJournal struct {
Id int64 `db:"id"` // 錢包賬本流水號
TransactionId int64 `db:"transaction_id"` // 交易 ID對應 transaction 表
OrderId string `db:"order_id"` // 訂單 ID對應 order 表
Brand string `db:"brand"` // 品牌名稱
Uid string `db:"uid"` // 用戶 ID
WalletType int64 `db:"wallet_type"` // 餘額種類: 1=可用, 2=凍結, 3=限制(僅出金)
Currency string `db:"currency"` // 幣種或平台點數
TransactionAmount float64 `db:"transaction_amount"` // 交易金額
PostTransactionBalance float64 `db:"post_transaction_balance"` // 交易後餘額
BusinessType int64 `db:"business_type"` // 業務類型
Status int64 `db:"status"` // 狀態
DueTime int64 `db:"due_time"` // T+N 執行時間
CreatedAt int64 `db:"created_at"` // 創建時間Unix 時間戳,毫秒)
}
)
func newWalletJournalModel(conn sqlx.SqlConn) *defaultWalletJournalModel {
return &defaultWalletJournalModel{
conn: conn,
table: "`wallet_journal`",
}
}
func (m *defaultWalletJournalModel) withSession(session sqlx.Session) *defaultWalletJournalModel {
return &defaultWalletJournalModel{
conn: sqlx.NewSqlConnFromSession(session),
table: "`wallet_journal`",
}
}
func (m *defaultWalletJournalModel) Delete(ctx context.Context, id int64) error {
query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
_, err := m.conn.ExecCtx(ctx, query, id)
return err
}
func (m *defaultWalletJournalModel) FindOne(ctx context.Context, id int64) (*WalletJournal, error) {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", walletJournalRows, m.table)
var resp WalletJournal
err := m.conn.QueryRowCtx(ctx, &resp, query, id)
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultWalletJournalModel) Insert(ctx context.Context, data *WalletJournal) (sql.Result, error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, walletJournalRowsExpectAutoSet)
ret, err := m.conn.ExecCtx(ctx, query, data.TransactionId, data.OrderId, data.Brand, data.Uid, data.WalletType, data.Currency, data.TransactionAmount, data.PostTransactionBalance, data.BusinessType, data.Status, data.DueTime, data.CreatedAt)
return ret, err
}
func (m *defaultWalletJournalModel) Update(ctx context.Context, data *WalletJournal) error {
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, walletJournalRowsWithPlaceHolder)
_, err := m.conn.ExecCtx(ctx, query, data.TransactionId, data.OrderId, data.Brand, data.Uid, data.WalletType, data.Currency, data.TransactionAmount, data.PostTransactionBalance, data.BusinessType, data.Status, data.DueTime, data.CreatedAt, data.Id)
return err
}
func (m *defaultWalletJournalModel) tableName() string {
return m.table
}

125
internal/model/wallet_model.go Executable file
View File

@ -0,0 +1,125 @@
package model
import (
"app-cloudep-trade-service/internal/domain/wallet"
"context"
"database/sql"
"errors"
"fmt"
"strings"
"github.com/zeromicro/go-zero/core/stores/sqlc"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
var _ WalletModel = (*customWalletModel)(nil)
type (
// WalletModel is an interface to be customized, add more methods here,
// and implement the added methods in customWalletModel.
WalletModel interface {
walletModel
InsertMany(ctx context.Context, wallets []*Wallet) (sql.Result, error)
// Balances 給他 select for update 的鎖先加上去
Balances(ctx context.Context, req BalanceReq, withLock bool, tx sqlx.Session) ([]Wallet, error)
// BalancesByIDs 給他 select for update 的鎖先加上去
BalancesByIDs(ctx context.Context, ids []int64, withLock bool, tx sqlx.Session) ([]Wallet, error)
}
customWalletModel struct {
*defaultWalletModel
}
BalanceReq struct {
UID []string
Currency []string
Kind []wallet.BalanceType
}
)
// NewWalletModel returns a model for the database table.
func NewWalletModel(conn sqlx.SqlConn) WalletModel {
return &customWalletModel{
defaultWalletModel: newWalletModel(conn),
}
}
func (m *customWalletModel) InsertMany(ctx context.Context, wallets []*Wallet) (sql.Result, error) {
if len(wallets) == 0 {
return nil, fmt.Errorf("no data to insert")
}
// 構建多條記錄的佔位符,例如: (?, ?, ?, ?, ?, ?, ?), (?, ?, ?, ?, ?, ?, ?), ...
valueStrings := make([]string, 0, len(wallets))
valueArgs := make([]interface{}, 0, len(wallets)*7) // 每條記錄有7個值
for _, wallet := range wallets {
valueStrings = append(valueStrings, "(?, ?, ?, ?, ?, ?, ?)")
valueArgs = append(valueArgs, wallet.Uid, wallet.Brand, wallet.Currency, wallet.Balance, wallet.WalletType, wallet.CreatedAt, wallet.UpdatedAt)
}
// 構建批量插入的 SQL 語句
query := fmt.Sprintf("insert into %s (%s) values %s", m.table, walletRowsExpectAutoSet, strings.Join(valueStrings, ","))
// 使用單一連線執行批量插入
return m.conn.ExecCtx(ctx, query, valueArgs...)
}
func (m *customWalletModel) Balances(ctx context.Context, req BalanceReq, isLock bool, tx sqlx.Session) ([]Wallet, error) {
baseQuery := fmt.Sprintf("SELECT `id`, `currency`, `balance`, `wallet_type` FROM %s", m.table)
// 構建條件字典
conditions := map[string][]any{
"`uid`": convertSliceToInterface(req.UID),
"`currency`": convertSliceToInterface(req.Currency),
"`wallet_type`": convertSliceToInterface(req.Kind),
}
// 使用 queryBuilder 構建完整查詢
query, args := queryBindINBuilder(baseQuery, conditions)
if isLock {
// 加上排他鎖
query += " FOR UPDATE"
}
// 執行查詢
var wallets []Wallet
err := tx.QueryRowsPartialCtx(ctx, &wallets, query, args...)
switch {
case err == nil:
return wallets, nil
case errors.Is(err, sqlc.ErrNotFound):
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *customWalletModel) BalancesByIDs(ctx context.Context, ids []int64, withLock bool, tx sqlx.Session) ([]Wallet, error) {
baseQuery := fmt.Sprintf("SELECT `id`, `currency`, `balance`, `wallet_type` FROM %s", m.table)
// 構建條件字典
conditions := map[string][]any{
"`id`": convertSliceToInterface(ids),
}
// 使用 queryBuilder 構建完整查詢
query, args := queryBindINBuilder(baseQuery, conditions)
if withLock {
// 加上排他鎖
query += " FOR UPDATE"
}
// 執行查詢
var wallets []Wallet
err := tx.QueryRowsPartialCtx(ctx, &wallets, query, args...)
switch {
case err == nil:
return wallets, nil
case errors.Is(err, sqlc.ErrNotFound):
return nil, ErrNotFound
default:
return nil, err
}
}

View File

@ -0,0 +1,115 @@
// Code generated by goctl. DO NOT EDIT.
package model
import (
"app-cloudep-trade-service/internal/domain/wallet"
"context"
"database/sql"
"fmt"
"strings"
"github.com/shopspring/decimal"
"github.com/zeromicro/go-zero/core/stores/builder"
"github.com/zeromicro/go-zero/core/stores/sqlc"
"github.com/zeromicro/go-zero/core/stores/sqlx"
"github.com/zeromicro/go-zero/core/stringx"
)
var (
walletFieldNames = builder.RawFieldNames(&Wallet{})
walletRows = strings.Join(walletFieldNames, ",")
walletRowsExpectAutoSet = strings.Join(stringx.Remove(walletFieldNames, "`id`"), ",")
walletRowsWithPlaceHolder = strings.Join(stringx.Remove(walletFieldNames, "`id`"), "=?,") + "=?"
)
type (
walletModel interface {
Insert(ctx context.Context, data *Wallet) (sql.Result, error)
FindOne(ctx context.Context, id int64) (*Wallet, error)
FindOneByUidCurrencyWalletType(ctx context.Context, uid string, currency string, walletType int64) (*Wallet, error)
Update(ctx context.Context, data *Wallet) error
Delete(ctx context.Context, id int64) error
}
defaultWalletModel struct {
conn sqlx.SqlConn
table string
}
Wallet struct {
Id int64 `db:"id"` // 錢包流水號
Uid string `db:"uid"` // 用戶ID
Brand string `db:"brand"` // 品牌名稱
Currency string `db:"currency"` // 幣別(或平台點數)
Balance decimal.Decimal `db:"balance"` // 錢包餘額
WalletType wallet.BalanceType `db:"wallet_type"` // 錢包種類: 1=可用, 2=凍結, 3=限制(僅出金)
CreatedAt int64 `db:"created_at"` // 創建時間
UpdatedAt int64 `db:"updated_at"` // 更新時間
}
)
func newWalletModel(conn sqlx.SqlConn) *defaultWalletModel {
return &defaultWalletModel{
conn: conn,
table: "`wallet`",
}
}
func (m *defaultWalletModel) withSession(session sqlx.Session) *defaultWalletModel {
return &defaultWalletModel{
conn: sqlx.NewSqlConnFromSession(session),
table: "`wallet`",
}
}
func (m *defaultWalletModel) Delete(ctx context.Context, id int64) error {
query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
_, err := m.conn.ExecCtx(ctx, query, id)
return err
}
func (m *defaultWalletModel) FindOne(ctx context.Context, id int64) (*Wallet, error) {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", walletRows, m.table)
var resp Wallet
err := m.conn.QueryRowCtx(ctx, &resp, query, id)
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultWalletModel) FindOneByUidCurrencyWalletType(ctx context.Context, uid string, currency string, walletType int64) (*Wallet, error) {
var resp Wallet
query := fmt.Sprintf("select %s from %s where `uid` = ? and `currency` = ? and `wallet_type` = ? limit 1", walletRows, m.table)
err := m.conn.QueryRowCtx(ctx, &resp, query, uid, currency, walletType)
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultWalletModel) Insert(ctx context.Context, data *Wallet) (sql.Result, error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?)", m.table, walletRowsExpectAutoSet)
ret, err := m.conn.ExecCtx(ctx, query, data.Uid, data.Brand, data.Currency, data.Balance, data.WalletType, data.CreatedAt, data.UpdatedAt)
return ret, err
}
func (m *defaultWalletModel) Update(ctx context.Context, newData *Wallet) error {
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, walletRowsWithPlaceHolder)
_, err := m.conn.ExecCtx(ctx, query, newData.Uid, newData.Brand, newData.Currency, newData.Balance, newData.WalletType, newData.CreatedAt, newData.UpdatedAt, newData.Id)
return err
}
func (m *defaultWalletModel) tableName() string {
return m.table
}

View File

@ -0,0 +1,95 @@
package repository
import (
"app-cloudep-trade-service/internal/domain/repository"
"app-cloudep-trade-service/internal/domain/wallet"
"app-cloudep-trade-service/internal/model"
"context"
"github.com/shopspring/decimal"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
// 用戶某個幣種餘額
type userLocalWallet struct {
wm model.WalletModel
txConn sqlx.SqlConn
uid string
currency string
// local wallet 相關計算的餘額存在這裡
walletBalance map[wallet.BalanceType]model.Wallet
// local order wallet 相關計算的餘額存在這裡
localOrderBalance map[int64]decimal.Decimal
// local wallet 內所有餘額變化紀錄
transactions []model.WalletJournal
}
// GetBalancesByID 使用 Wallet ID 取得餘額,如果需要上鎖,可以在 option 內加上 tx 以及 lock可以參考 TestBalanceUseCase
func (use *userLocalWallet) GetBalancesByID(ctx context.Context, ids []int64, opts ...repository.WalletOperatorOption) ([]model.Wallet, error) {
o := repository.ApplyOptions(opts...)
tx := use.txConn
if o.Tx != nil {
tx = *o.Tx
}
wallets, err := use.wm.BalancesByIDs(ctx, ids, o.WithLock, tx)
if err != nil {
return nil, err
}
for _, w := range wallets {
use.walletBalance[w.WalletType] = w
}
return wallets, nil
}
// LocalBalance 內存餘額
func (use *userLocalWallet) LocalBalance(BalanceType wallet.BalanceType) decimal.Decimal {
w, ok := use.walletBalance[BalanceType]
if !ok {
return decimal.Zero
}
return w.Balance
}
// Balances 取得餘額如果需要上鎖可以在option 內加上 tx 以及 lock可以參考 TestBalanceUseCase
func (use *userLocalWallet) Balances(ctx context.Context, balanceType []wallet.BalanceType, opts ...repository.WalletOperatorOption) ([]model.Wallet, error) {
o := repository.ApplyOptions(opts...)
tx := use.txConn
if o.Tx != nil {
tx = *o.Tx
}
wallets, err := use.wm.Balances(ctx, model.BalanceReq{
UID: []string{use.uid},
Currency: []string{use.currency},
Kind: balanceType,
}, o.WithLock, tx)
if err != nil {
return nil, err
}
for _, wallet := range wallets {
use.walletBalance[wallet.WalletType] = wallet
}
return wallets, nil
}
func NewUserWalletOperator(uid, currency string, wm model.WalletModel, txConn sqlx.SqlConn) repository.UserWalletOperator {
return &userLocalWallet{
wm: wm,
txConn: txConn,
uid: uid,
currency: currency,
walletBalance: make(map[wallet.BalanceType]model.Wallet, len(wallet.AllBalanceTypes)),
localOrderBalance: make(map[int64]decimal.Decimal, len(wallet.AllBalanceTypes)),
}
}

View File

@ -0,0 +1,161 @@
package repository
import (
"app-cloudep-trade-service/internal/domain"
"app-cloudep-trade-service/internal/domain/repository"
"app-cloudep-trade-service/internal/domain/wallet"
"app-cloudep-trade-service/internal/model"
"context"
"database/sql"
"errors"
"fmt"
"time"
"github.com/zeromicro/go-zero/core/stores/sqlc"
"github.com/shopspring/decimal"
"github.com/zeromicro/go-zero/core/logx"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
type WalletRepositoryParam struct {
WalletModel model.WalletModel
TxConn sqlx.SqlConn
}
type WalletRepository struct {
WalletRepositoryParam
}
func (repo *WalletRepository) Transaction(ctx context.Context, fn func(tx sqlx.Session) error) error {
// 取得原生的 *sql.DB
db, err := repo.TxConn.RawDB()
if err != nil {
return err
}
// 定義事務選項
txOptions := &sql.TxOptions{
Isolation: sql.LevelReadCommitted,
ReadOnly: false,
}
// 開啟事務
tx, err := db.BeginTx(ctx, txOptions)
if err != nil {
return err
}
// 使用 sqlx.Session 包裹事務
session := sqlx.NewSessionFromTx(tx)
// 執行傳入的操作
if err := fn(session); err != nil {
// 若有錯誤則回滾
err := tx.Rollback()
if err != nil {
// 錯誤代碼 06-021-22
e := domain.CommentErrorL(
domain.WalletTxErrorCode,
logx.WithContext(ctx),
[]logx.LogField{
{Key: "func", Value: "WalletModel.Transaction.Rollback"},
{Key: "err", Value: err},
},
"failed to find balance into mongo:").Wrap(err)
return e
}
// 裡面那一層會紀錄
return err
}
// 提交事務
return tx.Commit()
}
// GetUserWalletOperator 取得本地操作使用者錢包的操作運算元
func (repo *WalletRepository) GetUserWalletOperator(uid, currency string, opts ...repository.Option) repository.UserWalletOperator {
db := repo.TxConn
// 看是否有最新的DB 連線要傳入,做 tx
for _, fn := range opts {
db = fn()
}
return NewUserWalletOperator(uid, currency, repo.WalletModel, db)
}
// Create 創建錢包,如果有相同的就跳過不建立
func (repo *WalletRepository) Create(ctx context.Context, uid, currency, brand string) ([]*model.Wallet, error) {
wallets := make([]*model.Wallet, 0, len(wallet.AllBalanceTypes))
// 建立個人所有種類的錢包
now := time.Now().UTC().UnixNano()
for _, item := range wallet.AllBalanceTypes {
balance := decimal.Zero
wallets = append(wallets, &model.Wallet{
Brand: brand,
Currency: currency,
Uid: uid,
Balance: balance,
WalletType: item,
CreatedAt: now,
UpdatedAt: now,
})
}
_, err := repo.WalletModel.InsertMany(ctx, wallets)
if err != nil {
// 錯誤代碼 06-021-20
e := domain.CommentErrorL(
domain.CreateWalletErrorCode,
logx.WithContext(ctx),
[]logx.LogField{
{Key: "param", Value: fmt.Sprintf("uid: %s, currency:%s, brand:%s", uid, currency, brand)},
{Key: "func", Value: "WalletModel.InsertMany"},
{Key: "err", Value: err},
},
"failed to insert wallet into mysql:").Wrap(err)
return nil, e
}
return wallets, nil
}
func (repo *WalletRepository) Balances(ctx context.Context, req repository.BalanceReq) ([]model.Wallet, error) {
data, err := repo.WalletModel.Balances(ctx, model.BalanceReq{
UID: req.UID,
Currency: req.Currency,
Kind: req.BalanceType,
}, false, repo.TxConn)
if err != nil {
if errors.Is(sqlc.ErrNotFound, err) {
// 錯誤代碼 06-031-23
return nil, domain.NotFoundError(domain.WalletBalanceNotFound, "balance not found")
}
// 錯誤代碼 06-021-20
e := domain.CommentErrorL(
domain.CreateWalletErrorCode,
logx.WithContext(ctx),
[]logx.LogField{
{Key: "param", Value: req},
{Key: "func", Value: "WalletModel.Balances"},
{Key: "err", Value: err},
},
"failed to get balance")
return nil, e
}
return data, nil
}
func NewWalletRepository(param WalletRepositoryParam) repository.WalletRepository {
return &WalletRepository{
WalletRepositoryParam: param,
}
}

View File

@ -0,0 +1,81 @@
package repository
import (
"app-cloudep-trade-service/internal/domain/wallet"
"app-cloudep-trade-service/internal/model"
"context"
"fmt"
"testing"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
func TestByMe(t *testing.T) {
sqlConn := sqlx.NewSqlConn("mysql",
"root:yytt@tcp(127.0.0.1:3306)/digimon_wallet?parseTime=true&interpolateParams=false")
WalletRepo := NewWalletRepository(WalletRepositoryParam{
model.NewWalletModel(sqlConn),
sqlConn,
})
ctx := context.Background()
wo := WalletRepo.GetUserWalletOperator("OOOOOOOK", "USD")
balances, err := wo.Balances(ctx, wallet.AllBalanceTypes)
if err != nil {
return
}
fmt.Println(balances)
// create, err := WalletRepo.Create(ctx, "OOOOOOOK", "USD", "Digimon")
// if err != nil {
// return
// }
// fmt.Println(create)
// balances, err := WalletRepo.Balances(ctx, repository.BalanceReq{
// UID: []string{"OOOOOOOK"},
// Currency: []string{"USD"},
// // Kind: make([]domain.WalletType, 0),
// })
// if err != nil {
// fmt.Println(err)
// return
// }
// err := WalletRepo.Transaction(ctx, func(tx sqlx.Session) error {
// wm := model.NewWalletJournalModel(sqlConn)
// _, err := wm.InsertWithSession(ctx, tx, &model.WalletJournal{
// Id: 1,
// TransactionId: 10001,
// OrderId: "ORD123456789",
// Brand: "BrandX",
// Uid: "user123",
// WalletType: 1, // 1=可用
// Currency: "USD",
// TransactionAmount: 500.00,
// PostTransactionBalance: 1500.00,
// BusinessType: 2, // 假設 2 表示特定業務類型
// Status: 1, // 假設 1 表示成功狀態
// DueTime: 1698289200000, // T+N 執行時間
// CreatedAt: 1698192800000, // 創建時間 (Unix 時間戳,毫秒)
// })
//
// if err != nil {
// return err
// }
//
// return nil
// })
//
// if err != nil {
// return
// }
// fmt.Println(balances)
}
func TestBalanceUseCase(t *testing.T) {}

View File

@ -0,0 +1,38 @@
package svc
import (
"app-cloudep-trade-service/internal/config"
"database/sql"
"fmt"
"github.com/zeromicro/go-zero/core/logx"
)
func mustDSN(c config.Config) string {
// 構建 DSN 字符串
return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?parseTime=true&interpolateParams=%t",
c.DB.User,
c.DB.Password,
c.DB.Host,
c.DB.Port,
c.DB.Database,
c.DB.InterpolateParams,
)
}
func newDatabase(c config.Config) (*sql.DB, error) {
// 創建基礎的 *sql.DB 連接
db, err := sql.Open("mysql", mustDSN(c))
if err != nil {
logx.Error(fmt.Errorf("failed to connect to database: %w", err))
return nil, err
}
// 設置數據庫連接池參數
db.SetMaxIdleConns(c.DB.MaxIdleConns)
db.SetMaxOpenConns(c.DB.MaxOpenConns)
db.SetConnMaxLifetime(c.DB.ConnMaxLifetime)
return db, nil
}

View File

@ -5,6 +5,8 @@ import (
duc "app-cloudep-trade-service/internal/domain/usecase"
"app-cloudep-trade-service/internal/usecase"
"github.com/zeromicro/go-zero/core/stores/sqlx"
ers "code.30cm.net/digimon/library-go/errs"
"code.30cm.net/digimon/library-go/errs/code"
vi "code.30cm.net/digimon/library-go/validator"
@ -15,6 +17,7 @@ type ServiceContext struct {
Validate vi.Validate
OrderUseCase duc.OrderUseCase
SQLConn sqlx.SqlConn
}
func NewServiceContext(c config.Config) *ServiceContext {
@ -26,13 +29,20 @@ func NewServiceContext(c config.Config) *ServiceContext {
OrderModel: om,
})
mysql, err := newDatabase(c)
if err != nil {
panic("failed to connect to wallet")
}
// 創建 SQL 連線並返回
sqlConn := sqlx.NewSqlConnFromDB(mysql)
return &ServiceContext{
Config: c,
Validate: vi.MustValidator(
WithDecimalGt(),
WithDecimalGte(),
),
SQLConn: sqlConn,
OrderUseCase: orderUseCase,
}
}

View File

@ -1,7 +1,7 @@
package usecase
import (
"app-cloudep-trade-service/internal/domain"
"app-cloudep-trade-service/internal/domain/order"
"app-cloudep-trade-service/internal/domain/usecase"
mockmodel "app-cloudep-trade-service/internal/mock/model"
model "app-cloudep-trade-service/internal/model/mongo"
@ -30,7 +30,7 @@ func TestOrderUseCase_CancelOrder(t *testing.T) {
cancelOrderQuery := usecase.CancelOrderQuery{
BusinessID: "business_123",
Status: domain.OrderStatusCancelled, // 使用模擬的狀態
Status: order.Cancelled, // 使用模擬的狀態
}
tests := []struct {
@ -89,8 +89,8 @@ func TestOrderUseCase_CreateOrder(t *testing.T) {
// 構建測試參數
createOrderReq := usecase.CreateOrderReq{
BusinessID: "business_123",
OrderType: domain.OrderTypeTest,
OrderStatus: domain.OrderStatusCreated,
OrderType: order.TestType,
OrderStatus: order.Created,
Brand: "test_brand",
OrderUID: "user_123",
ReferenceID: "reference_123",
@ -222,8 +222,8 @@ func TestOrderUseCase_GetOrder(t *testing.T) {
UpdateTime: 123456789,
CreateTime: 123456789,
BusinessID: "business_123",
OrderType: domain.OrderTypeTest,
OrderStatus: domain.OrderStatusCreated,
OrderType: order.TestType,
OrderStatus: order.Created,
Brand: "test_brand",
OrderUID: "user_123",
ReferenceID: "reference_123",
@ -343,14 +343,14 @@ func TestOrderUseCase_ListOrder(t *testing.T) {
PageIndex: 1,
PageSize: 10,
BusinessID: "business_123",
OrderStatus: []int64{int64(domain.OrderStatusCreated)},
OrderStatus: []int64{int64(order.Created)},
}
mockOrders := []model.Order{
{
BusinessID: "business_123",
OrderUID: "user_123",
OrderStatus: domain.OrderStatusCreated,
OrderStatus: order.Created,
// 其他欄位根據需要填充
},
}
@ -414,7 +414,7 @@ func TestOrderUseCase_ModifyOrderStatus(t *testing.T) {
modifyOrderQuery := &usecase.ModifyOrderQuery{
BusinessID: "business_123",
Status: int64(domain.OrderStatusCancelled),
Status: int64(order.Cancelled),
}
tests := []struct {