yml/hfb_sys
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化

Browse Source
This commit is contained in:
yml
2026-06-10 02:12:02 +08:00
parent 992d4ed234
commit c517349db6
6 changed files with 388 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/internal/modules/wallet/repository.go
+38 -38
View File
@@ -1,6 +1,7 @@
package wallet
import (
"context"
"crypto/rand"
"fmt"
"time"
@@ -32,9 +33,9 @@ func NewRepository(db *gorm.DB) *Repository {
return &Repository{db: db}
}
func (r *Repository) Account(userID uint64) (*AccountDTO, error) {
func (r *Repository) Account(ctx context.Context, userID uint64) (*AccountDTO, error) {
var account model.WalletAccount
err := r.db.Transaction(func(tx *gorm.DB) error {
err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
if err := ensureAccount(tx, userID); err != nil {
return err
}
@@ -46,14 +47,14 @@ func (r *Repository) Account(userID uint64) (*AccountDTO, error) {
return toAccountDTO(account), nil
}
func (r *Repository) Ledger(userID uint64, page, pageSize int) (*PaginatedResult, error) {
func (r *Repository) Ledger(ctx context.Context, userID uint64, page, pageSize int) (*PaginatedResult, error) {
var total int64
if err := r.db.Model(&model.WalletLedger{}).Where("user_id = ?", userID).Count(&total).Error; err != nil {
if err := r.db.WithContext(ctx).Model(&model.WalletLedger{}).Where("user_id = ?", userID).Count(&total).Error; err != nil {
return nil, err
}
offset := (page - 1) * pageSize
var rows []model.WalletLedger
if err := r.db.Where("user_id = ?", userID).Order("id DESC").Offset(offset).Limit(pageSize).Find(&rows).Error; err != nil {
if err := r.db.WithContext(ctx).Where("user_id = ?", userID).Order("id DESC").Offset(offset).Limit(pageSize).Find(&rows).Error; err != nil {
return nil, err
}
items := make([]LedgerDTO, 0, len(rows))
@@ -63,8 +64,8 @@ func (r *Repository) Ledger(userID uint64, page, pageSize int) (*PaginatedResult
return &PaginatedResult{Items: items, Total: total, Page: page, PageSize: pageSize}, nil
}
func (r *Repository) Recharge(userID uint64, amountCent int64) (*AccountDTO, error) {
err := r.db.Transaction(func(tx *gorm.DB) error {
func (r *Repository) Recharge(ctx context.Context, userID uint64, amountCent int64) (*AccountDTO, error) {
err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
return AppendEntries(tx, Entry{
UserID: userID,
Direction: "in",
@@ -78,14 +79,14 @@ func (r *Repository) Recharge(userID uint64, amountCent int64) (*AccountDTO, err
if err != nil {
return nil, err
}
return r.Account(userID)
return r.Account(ctx, userID)
}
func (r *Repository) ConfirmRechargeFromChannel(userID uint64, bizNo string, amountCent int64) error {
func (r *Repository) ConfirmRechargeFromChannel(ctx context.Context, userID uint64, bizNo string, amountCent int64) error {
if userID == 0 || amountCent <= 0 || bizNo == "" {
return ErrInvalidAmount
}
return r.db.Transaction(func(tx *gorm.DB) error {
return r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
if err := ensureAccount(tx, userID); err != nil {
return err
}
@@ -117,11 +118,11 @@ func (r *Repository) ConfirmRechargeFromChannel(userID uint64, bizNo string, amo
}
// Withdraw 保留仓库能力；当前公开提现入口在 service 层标记为待开发，不会调用到这里。
func (r *Repository) Withdraw(userID uint64, amountCent int64) (*AccountDTO, error) {
func (r *Repository) Withdraw(ctx context.Context, userID uint64, amountCent int64) (*AccountDTO, error) {
if userID == 0 || amountCent <= 0 {
return nil, ErrInvalidAmount
}
err := r.db.Transaction(func(tx *gorm.DB) error {
err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
return AppendEntries(tx, Entry{
UserID: userID,
Direction: "out",
@@ -135,11 +136,11 @@ func (r *Repository) Withdraw(userID uint64, amountCent int64) (*AccountDTO, err
if err != nil {
return nil, err
}
return r.Account(userID)
return r.Account(ctx, userID)
}
func (r *Repository) AdminLedger(query AdminLedgerQuery) (*PaginatedResult, error) {
db := r.db.Table("wallet_ledger AS wl").
func (r *Repository) AdminLedger(ctx context.Context, query AdminLedgerQuery) (*PaginatedResult, error) {
db := r.db.WithContext(ctx).Table("wallet_ledger AS wl").
Select(`wl.id, wl.ledger_no, wl.user_id, COALESCE(u.phone, '') AS user_phone,
COALESCE(u.nickname, '') AS user_nickname, wl.order_id, COALESCE(ro.order_no, '') AS order_no,
wl.direction, wl.amount_cent, wl.balance_after_cent, wl.balance_type, wl.biz_type, wl.biz_no,
@@ -147,7 +148,7 @@ func (r *Repository) AdminLedger(query AdminLedgerQuery) (*PaginatedResult, erro
Joins("LEFT JOIN users AS u ON u.id = wl.user_id").
Joins("LEFT JOIN rental_orders AS ro ON ro.id = wl.order_id")
countDB := r.db.Model(&model.WalletLedger{})
countDB := r.db.WithContext(ctx).Model(&model.WalletLedger{})
if query.UserID > 0 {
db = db.Where("wl.user_id = ?", query.UserID)
countDB = countDB.Where("user_id = ?", query.UserID)
@@ -249,12 +250,12 @@ func applyEntry(account *model.WalletAccount, entry Entry) (int64, error) {
}
return account.FrozenBalanceCent, nil
default:
return 0, fmt.Errorf("unsupported balance type: %s", entry.BalanceType)
return 0, fmt.Errorf("unknown balance type: %s", entry.BalanceType)
}
}
func roundWalletMoney(value float64) float64 {
return money.Round(value)
func roundWalletMoney(yuan float64) float64 {
return money.Round(yuan)
}
func toAccountDTO(account model.WalletAccount) *AccountDTO {
@@ -266,31 +267,30 @@ func toAccountDTO(account model.WalletAccount) *AccountDTO {
}
}
func toLedgerDTO(row model.WalletLedger) LedgerDTO {
func toLedgerDTO(ledger model.WalletLedger) LedgerDTO {
return LedgerDTO{
ID: row.ID,
LedgerNo: row.LedgerNo,
UserID: row.UserID,
OrderID: row.OrderID,
Direction: row.Direction,
AmountCent: row.AmountCent,
BalanceAfterCent: row.BalanceAfterCent,
BalanceType: row.BalanceType,
BizType: row.BizType,
BizNo: row.BizNo,
Remark: row.Remark,
CreatedAt: row.CreatedAt,
ID: ledger.ID,
LedgerNo: ledger.LedgerNo,
UserID: ledger.UserID,
OrderID: ledger.OrderID,
Direction: ledger.Direction,
AmountCent: ledger.AmountCent,
BalanceAfterCent: ledger.BalanceAfterCent,
BalanceType: ledger.BalanceType,
BizType: ledger.BizType,
BizNo: ledger.BizNo,
Remark: ledger.Remark,
CreatedAt: ledger.CreatedAt,
}
}
func newLedgerNo() (string, error) {
// 生成格式：WL + YYYYMMDDHHMMSS + 毫秒 + 4位随机数
// 例如：WL202606051234561230456，便于用户和客服识别。
now := timeutil.ShanghaiNow()
buf := make([]byte, 2)
if _, err := rand.Read(buf); err != nil {
prefix := "WL" + now.Format("20060102150405")
randomBytes := make([]byte, 4)
if _, err := rand.Read(randomBytes); err != nil {
return "", err
}
randomNum := (int(buf[0])<<8 | int(buf[1])) % 10000
return fmt.Sprintf("WL%s%03d%04d", now.Format("20060102150405"), now.Nanosecond()/1_000_000, randomNum), nil
suffix := fmt.Sprintf("%08d", uint32(randomBytes[0])<<24|uint32(randomBytes[1])<<16|uint32(randomBytes[2])<<8|uint32(randomBytes[3]))
return prefix + suffix[:7], nil
}