117 lines
6.5 KiB
Go
117 lines
6.5 KiB
Go
// Copyright © 2023 OpenIM open source community. All rights reserved.
|
||
//
|
||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
// you may not use this file except in compliance with the License.
|
||
// You may obtain a copy of the License at
|
||
//
|
||
// http://www.apache.org/licenses/LICENSE-2.0
|
||
//
|
||
// Unless required by applicable law or agreed to in writing, software
|
||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
// See the License for the specific language governing permissions and
|
||
// limitations under the License.
|
||
|
||
package chat
|
||
|
||
import (
|
||
"context"
|
||
"time"
|
||
|
||
"github.com/openimsdk/tools/db/pagination"
|
||
)
|
||
|
||
// BalanceRecordType 余额变动类型
|
||
const (
|
||
BalanceRecordTypeRecharge = 1 // 充值
|
||
BalanceRecordTypeWithdraw = 2 // 提现/提款
|
||
BalanceRecordTypeConsume = 3 // 消费
|
||
BalanceRecordTypeRefund = 4 // 退款
|
||
BalanceRecordTypeReward = 5 // 奖励
|
||
BalanceRecordTypeAdminRecharge = 6 // 后台充值
|
||
BalanceRecordTypeSendRedPacket = 7 // 发红包（减少余额）
|
||
BalanceRecordTypeGrabRedPacket = 8 // 抢红包（增加余额）
|
||
BalanceRecordTypeOther = 99 // 其他
|
||
)
|
||
|
||
// WithdrawAccountType 提现账号类型
|
||
const (
|
||
WithdrawAccountTypeAlipay = 1 // 支付宝
|
||
WithdrawAccountTypeWeChat = 2 // 微信
|
||
WithdrawAccountTypeBankCard = 3 // 银行卡
|
||
)
|
||
|
||
// RealNameAuth 实名认证信息
|
||
type RealNameAuth struct {
|
||
IDCard string `bson:"id_card"` // 身份证号
|
||
IDCardPhotoFront string `bson:"id_card_photo_front"` // 身份证正面照片URL
|
||
IDCardPhotoBack string `bson:"id_card_photo_back"` // 身份证反面照片URL
|
||
Name string `bson:"name"` // 真实姓名
|
||
AuditStatus int32 `bson:"audit_status"` // 审核状态：0-未审核，1-审核通过，2-审核拒绝
|
||
}
|
||
|
||
// Wallet 钱包模型
|
||
type Wallet struct {
|
||
UserID string `bson:"user_id"` // 用户ID
|
||
Balance int64 `bson:"balance"` // 用户余额（单位：分）
|
||
PaymentPassword string `bson:"payment_password"` // 支付密码（加密存储）
|
||
WithdrawAccount string `bson:"withdraw_account"` // 提现账号
|
||
WithdrawAccountType int32 `bson:"withdraw_account_type"` // 提现账号类型：1-支付宝，2-微信，3-银行卡
|
||
RealNameAuth RealNameAuth `bson:"real_name_auth"` // 实名认证信息
|
||
WithdrawReceiveAccount string `bson:"withdraw_receive_account"` // 提现收款账号
|
||
CreateTime time.Time `bson:"create_time"` // 创建时间
|
||
UpdateTime time.Time `bson:"update_time"` // 更新时间
|
||
}
|
||
|
||
func (Wallet) TableName() string {
|
||
return "wallets"
|
||
}
|
||
|
||
type WalletInterface interface {
|
||
Create(ctx context.Context, wallets ...*Wallet) error
|
||
Take(ctx context.Context, userID string) (*Wallet, error)
|
||
Find(ctx context.Context, userIDs []string) ([]*Wallet, error)
|
||
Update(ctx context.Context, userID string, data map[string]any) error
|
||
UpdateBalance(ctx context.Context, userID string, balance int64) error
|
||
IncrementBalance(ctx context.Context, userID string, amount int64) (beforeBalance int64, afterBalance int64, err error) // 原子更新余额，返回更新前后的余额
|
||
UpdatePaymentPassword(ctx context.Context, userID string, paymentPassword string) error
|
||
UpdateWithdrawAccount(ctx context.Context, userID string, withdrawAccount string) error // 更新提款账号（兼容旧接口）
|
||
UpdateWithdrawAccountWithType(ctx context.Context, userID string, withdrawAccount string, accountType int32) error // 更新提款账号（带类型）
|
||
UpdateRealNameAuth(ctx context.Context, userID string, realNameAuth RealNameAuth) error // 更新实名认证信息
|
||
Delete(ctx context.Context, userIDs []string) error
|
||
Page(ctx context.Context, pagination pagination.Pagination) (int64, []*Wallet, error)
|
||
PageByRealNameAuthAuditStatus(ctx context.Context, auditStatus int32, userID string, pagination pagination.Pagination) (int64, []*Wallet, error) // 按实名认证审核状态分页查询（支持用户ID搜索）
|
||
SearchByRealNameAuth(ctx context.Context, realNameKeyword string, idCardKeyword string) ([]string, error) // 按实名认证信息搜索钱包（返回userIDs）
|
||
}
|
||
|
||
// WalletBalanceRecord 钱包余额记录
|
||
type WalletBalanceRecord struct {
|
||
ID string `bson:"_id"` // 记录ID
|
||
UserID string `bson:"user_id"` // 用户ID
|
||
Amount int64 `bson:"amount"` // 变动金额（单位：分，正数表示增加，负数表示减少）
|
||
Type int32 `bson:"type"` // 变动类型：1-充值，2-提现/提款，3-消费，4-退款，5-奖励，6-后台充值，7-发红包，8-抢红包，99-其他
|
||
BeforeBalance int64 `bson:"before_balance"` // 变动前余额（单位：分）
|
||
AfterBalance int64 `bson:"after_balance"` // 变动后余额（单位：分）
|
||
OrderID string `bson:"order_id"` // 关联订单ID（可选）
|
||
TransactionID string `bson:"transaction_id"` // 交易ID（可选）
|
||
RedPacketID string `bson:"red_packet_id"` // 红包ID（用于发红包和抢红包记录关联，可选）
|
||
Remark string `bson:"remark"` // 备注
|
||
CreateTime time.Time `bson:"create_time"` // 创建时间
|
||
}
|
||
|
||
func (WalletBalanceRecord) TableName() string {
|
||
return "wallet_balance_records"
|
||
}
|
||
|
||
type WalletBalanceRecordInterface interface {
|
||
Create(ctx context.Context, records ...*WalletBalanceRecord) error
|
||
Take(ctx context.Context, recordID string) (*WalletBalanceRecord, error)
|
||
FindByUserID(ctx context.Context, userID string, pagination pagination.Pagination) (int64, []*WalletBalanceRecord, error)
|
||
FindByUserIDAndType(ctx context.Context, userID string, recordType int32, pagination pagination.Pagination) (int64, []*WalletBalanceRecord, error)
|
||
FindByOrderID(ctx context.Context, orderID string) (*WalletBalanceRecord, error)
|
||
FindByTransactionID(ctx context.Context, transactionID string) (*WalletBalanceRecord, error)
|
||
FindByRedPacketID(ctx context.Context, redPacketID string) ([]*WalletBalanceRecord, error)
|
||
GetUserBalanceHistory(ctx context.Context, userID string, startTime, endTime *time.Time, pagination pagination.Pagination) (int64, []*WalletBalanceRecord, error)
|
||
CountByUserID(ctx context.Context, userID string) (int64, error)
|
||
}
|