526 lines
16 KiB
Go
526 lines
16 KiB
Go
|
package handler
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"math"
|
|||
|
|
"net/http"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"soul-api/internal/database"
|
|||
|
|
"soul-api/internal/model"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const defaultBindingDays = 30
|
|||
|
|
|
|||
|
|
// ReferralBind POST /api/referral/bind 推荐码绑定（新绑定/续期/切换）
|
|||
|
|
func ReferralBind(c *gin.Context) {
|
|||
|
|
var req struct {
|
|||
|
|
UserID string `json:"userId"`
|
|||
|
|
ReferralCode string `json:"referralCode" binding:"required"`
|
|||
|
|
OpenID string `json:"openId"`
|
|||
|
|
Source string `json:"source"`
|
|||
|
|
}
|
|||
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|||
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "用户ID和推荐码不能为空"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
effectiveUserID := req.UserID
|
|||
|
|
if effectiveUserID == "" && req.OpenID != "" {
|
|||
|
|
effectiveUserID = "user_" + req.OpenID[len(req.OpenID)-8:]
|
|||
|
|
}
|
|||
|
|
if effectiveUserID == "" || req.ReferralCode == "" {
|
|||
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "用户ID和推荐码不能为空"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
db := database.DB()
|
|||
|
|
bindingDays := defaultBindingDays
|
|||
|
|
var cfg model.SystemConfig
|
|||
|
|
if err := db.Where("config_key = ?", "referral_config").First(&cfg).Error; err == nil {
|
|||
|
|
var config map[string]interface{}
|
|||
|
|
if _ = json.Unmarshal(cfg.ConfigValue, &config); config["bindingDays"] != nil {
|
|||
|
|
if v, ok := config["bindingDays"].(float64); ok {
|
|||
|
|
bindingDays = int(v)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var referrer model.User
|
|||
|
|
if err := db.Where("referral_code = ?", req.ReferralCode).First(&referrer).Error; err != nil {
|
|||
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "推荐码无效"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if referrer.ID == effectiveUserID {
|
|||
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "不能使用自己的推荐码"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var user model.User
|
|||
|
|
if err := db.Where("id = ?", effectiveUserID).First(&user).Error; err != nil {
|
|||
|
|
if req.OpenID != "" {
|
|||
|
|
if err := db.Where("open_id = ?", req.OpenID).First(&user).Error; err != nil {
|
|||
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "用户不存在"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "用户不存在"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
expiryDate := time.Now().AddDate(0, 0, bindingDays)
|
|||
|
|
var existing model.ReferralBinding
|
|||
|
|
err := db.Where("referee_id = ? AND status = ?", user.ID, "active").Order("binding_date DESC").First(&existing).Error
|
|||
|
|
action := "new"
|
|||
|
|
var oldReferrerID interface{}
|
|||
|
|
|
|||
|
|
if err == nil {
|
|||
|
|
if existing.ReferrerID == referrer.ID {
|
|||
|
|
action = "renew"
|
|||
|
|
db.Model(&existing).Updates(map[string]interface{}{
|
|||
|
|
"expiry_date": expiryDate,
|
|||
|
|
"binding_date": time.Now(),
|
|||
|
|
})
|
|||
|
|
} else {
|
|||
|
|
action = "switch"
|
|||
|
|
oldReferrerID = existing.ReferrerID
|
|||
|
|
db.Model(&existing).Update("status", "cancelled")
|
|||
|
|
bindID := fmt.Sprintf("bind_%d_%s", time.Now().UnixNano(), randomStr(6))
|
|||
|
|
db.Create(&model.ReferralBinding{
|
|||
|
|
ID: bindID,
|
|||
|
|
ReferrerID: referrer.ID,
|
|||
|
|
RefereeID: user.ID,
|
|||
|
|
ReferralCode: req.ReferralCode,
|
|||
|
|
Status: refString("active"),
|
|||
|
|
ExpiryDate: expiryDate,
|
|||
|
|
BindingDate: time.Now(),
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
bindID := fmt.Sprintf("bind_%d_%s", time.Now().UnixNano(), randomStr(6))
|
|||
|
|
db.Create(&model.ReferralBinding{
|
|||
|
|
ID: bindID,
|
|||
|
|
ReferrerID: referrer.ID,
|
|||
|
|
RefereeID: user.ID,
|
|||
|
|
ReferralCode: req.ReferralCode,
|
|||
|
|
Status: refString("active"),
|
|||
|
|
ExpiryDate: expiryDate,
|
|||
|
|
BindingDate: time.Now(),
|
|||
|
|
})
|
|||
|
|
db.Model(&model.User{}).Where("id = ?", referrer.ID).UpdateColumn("referral_count", gorm.Expr("COALESCE(referral_count, 0) + 1"))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
msg := "绑定成功"
|
|||
|
|
if action == "renew" {
|
|||
|
|
msg = "绑定已续期"
|
|||
|
|
} else if action == "switch" {
|
|||
|
|
msg = "推荐人已切换"
|
|||
|
|
}
|
|||
|
|
c.JSON(http.StatusOK, gin.H{
|
|||
|
|
"success": true,
|
|||
|
|
"message": msg,
|
|||
|
|
"data": gin.H{
|
|||
|
|
"action": action,
|
|||
|
|
"referrer": gin.H{"id": referrer.ID, "nickname": getStringValue(referrer.Nickname)},
|
|||
|
|
"expiryDate": expiryDate,
|
|||
|
|
"bindingDays": bindingDays,
|
|||
|
|
"oldReferrerId": oldReferrerID,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func refString(s string) *string { return &s }
|
|||
|
|
func randomStr(n int) string {
|
|||
|
|
const letters = "abcdefghijklmnopqrstuvwxyz0123456789"
|
|||
|
|
b := make([]byte, n)
|
|||
|
|
for i := range b {
|
|||
|
|
b[i] = letters[time.Now().UnixNano()%int64(len(letters))]
|
|||
|
|
}
|
|||
|
|
return string(b)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ReferralData GET /api/referral/data 获取分销数据统计
|
|||
|
|
func ReferralData(c *gin.Context) {
|
|||
|
|
userId := c.Query("userId")
|
|||
|
|
if userId == "" {
|
|||
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "用户ID不能为空"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
db := database.DB()
|
|||
|
|
|
|||
|
|
// 获取分销配置（与 soul-admin 推广设置一致）
|
|||
|
|
distributorShare := 0.9
|
|||
|
|
minWithdrawAmount := 10.0
|
|||
|
|
bindingDays := defaultBindingDays
|
|||
|
|
userDiscount := 5
|
|||
|
|
withdrawFee := 5.0
|
|||
|
|
|
|||
|
|
var cfg model.SystemConfig
|
|||
|
|
if err := db.Where("config_key = ?", "referral_config").First(&cfg).Error; err == nil {
|
|||
|
|
var config map[string]interface{}
|
|||
|
|
if err := json.Unmarshal(cfg.ConfigValue, &config); err == nil {
|
|||
|
|
if share, ok := config["distributorShare"].(float64); ok {
|
|||
|
|
distributorShare = share / 100
|
|||
|
|
}
|
|||
|
|
if minAmount, ok := config["minWithdrawAmount"].(float64); ok {
|
|||
|
|
minWithdrawAmount = minAmount
|
|||
|
|
}
|
|||
|
|
if days, ok := config["bindingDays"].(float64); ok && days > 0 {
|
|||
|
|
bindingDays = int(days)
|
|||
|
|
}
|
|||
|
|
if discount, ok := config["userDiscount"].(float64); ok {
|
|||
|
|
userDiscount = int(discount)
|
|||
|
|
}
|
|||
|
|
if fee, ok := config["withdrawFee"].(float64); ok {
|
|||
|
|
withdrawFee = fee
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 1. 查询用户基本信息
|
|||
|
|
var user model.User
|
|||
|
|
if err := db.Where("id = ?", userId).First(&user).Error; err != nil {
|
|||
|
|
c.JSON(http.StatusNotFound, gin.H{"success": false, "error": "用户不存在"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 绑定统计
|
|||
|
|
var totalBindings int64
|
|||
|
|
db.Model(&model.ReferralBinding{}).Where("referrer_id = ?", userId).Count(&totalBindings)
|
|||
|
|
|
|||
|
|
var activeBindings int64
|
|||
|
|
db.Model(&model.ReferralBinding{}).Where(
|
|||
|
|
"referrer_id = ? AND status = 'active' AND expiry_date > ?",
|
|||
|
|
userId, time.Now(),
|
|||
|
|
).Count(&activeBindings)
|
|||
|
|
|
|||
|
|
var convertedBindings int64
|
|||
|
|
db.Model(&model.ReferralBinding{}).Where(
|
|||
|
|
"referrer_id = ? AND status = 'active' AND purchase_count > 0",
|
|||
|
|
userId,
|
|||
|
|
).Count(&convertedBindings)
|
|||
|
|
|
|||
|
|
var expiredBindings int64
|
|||
|
|
db.Model(&model.ReferralBinding{}).Where(
|
|||
|
|
"referrer_id = ? AND (status IN ('expired', 'cancelled') OR (status = 'active' AND expiry_date <= ?))",
|
|||
|
|
userId, time.Now(),
|
|||
|
|
).Count(&expiredBindings)
|
|||
|
|
|
|||
|
|
// 3. 付款统计
|
|||
|
|
var paidOrders []model.Order
|
|||
|
|
db.Where("referrer_id = ? AND status = ?", userId, "paid").Find(&paidOrders)
|
|||
|
|
|
|||
|
|
totalAmount := 0.0
|
|||
|
|
totalCommission := 0.0
|
|||
|
|
uniqueUsers := make(map[string]bool)
|
|||
|
|
for i := range paidOrders {
|
|||
|
|
totalAmount += paidOrders[i].Amount
|
|||
|
|
totalCommission += computeOrderCommission(db, &paidOrders[i], nil)
|
|||
|
|
uniqueUsers[paidOrders[i].UserID] = true
|
|||
|
|
}
|
|||
|
|
uniquePaidCount := len(uniqueUsers)
|
|||
|
|
|
|||
|
|
// 4. 访问统计
|
|||
|
|
totalVisits := int(totalBindings)
|
|||
|
|
var visitCount int64
|
|||
|
|
if err := db.Model(&model.ReferralVisit{}).
|
|||
|
|
Select("COUNT(DISTINCT visitor_id) as count").
|
|||
|
|
Where("referrer_id = ?", userId).
|
|||
|
|
Count(&visitCount).Error; err == nil {
|
|||
|
|
totalVisits = int(visitCount)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 5. 提现统计（与小程序可提现逻辑一致：可提现 = 累计佣金 - 已提现 - 待审核）
|
|||
|
|
// 待审核 = pending + processing + pending_confirm，与 /api/withdraw/pending-confirm 口径一致
|
|||
|
|
var pendingWithdraw struct{ Total float64 }
|
|||
|
|
db.Model(&model.Withdrawal{}).
|
|||
|
|
Select("COALESCE(SUM(amount), 0) as total").
|
|||
|
|
Where("user_id = ? AND status IN ?", userId, []string{"pending", "processing", "pending_confirm"}).
|
|||
|
|
Scan(&pendingWithdraw)
|
|||
|
|
|
|||
|
|
var successWithdraw struct{ Total float64 }
|
|||
|
|
db.Model(&model.Withdrawal{}).
|
|||
|
|
Select("COALESCE(SUM(amount), 0) as total").
|
|||
|
|
Where("user_id = ? AND status = ?", userId, "success").
|
|||
|
|
Scan(&successWithdraw)
|
|||
|
|
|
|||
|
|
pendingWithdrawAmount := pendingWithdraw.Total
|
|||
|
|
withdrawnFromTable := successWithdraw.Total
|
|||
|
|
|
|||
|
|
// 6. 获取活跃绑定用户列表
|
|||
|
|
var activeBindingsList []model.ReferralBinding
|
|||
|
|
db.Where("referrer_id = ? AND status = 'active' AND expiry_date > ?", userId, time.Now()).
|
|||
|
|
Order("binding_date DESC").
|
|||
|
|
Limit(20).
|
|||
|
|
Find(&activeBindingsList)
|
|||
|
|
|
|||
|
|
activeUsers := []gin.H{}
|
|||
|
|
for _, b := range activeBindingsList {
|
|||
|
|
var referee model.User
|
|||
|
|
db.Where("id = ?", b.RefereeID).First(&referee)
|
|||
|
|
|
|||
|
|
daysRemaining := int(time.Until(b.ExpiryDate).Hours() / 24)
|
|||
|
|
if daysRemaining < 0 {
|
|||
|
|
daysRemaining = 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
activeUsers = append(activeUsers, gin.H{
|
|||
|
|
"id": b.RefereeID,
|
|||
|
|
"nickname": getStringValue(referee.Nickname),
|
|||
|
|
"avatar": getStringValue(referee.Avatar),
|
|||
|
|
"daysRemaining": daysRemaining,
|
|||
|
|
"hasFullBook": getBoolValue(referee.HasFullBook),
|
|||
|
|
"bindingDate": b.BindingDate,
|
|||
|
|
"status": "active",
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 7. 获取已转化用户列表
|
|||
|
|
var convertedBindingsList []model.ReferralBinding
|
|||
|
|
db.Where("referrer_id = ? AND status = 'active' AND purchase_count > 0", userId).
|
|||
|
|
Order("last_purchase_date DESC").
|
|||
|
|
Limit(20).
|
|||
|
|
Find(&convertedBindingsList)
|
|||
|
|
|
|||
|
|
convertedUsers := []gin.H{}
|
|||
|
|
for _, b := range convertedBindingsList {
|
|||
|
|
var referee model.User
|
|||
|
|
db.Where("id = ?", b.RefereeID).First(&referee)
|
|||
|
|
|
|||
|
|
commission := 0.0
|
|||
|
|
if b.TotalCommission != nil {
|
|||
|
|
commission = *b.TotalCommission
|
|||
|
|
}
|
|||
|
|
orderAmount := commission / distributorShare
|
|||
|
|
|
|||
|
|
convertedUsers = append(convertedUsers, gin.H{
|
|||
|
|
"id": b.RefereeID,
|
|||
|
|
"nickname": getStringValue(referee.Nickname),
|
|||
|
|
"avatar": getStringValue(referee.Avatar),
|
|||
|
|
"commission": commission,
|
|||
|
|
"orderAmount": orderAmount,
|
|||
|
|
"purchaseCount": getIntValue(b.PurchaseCount),
|
|||
|
|
"conversionDate": b.LastPurchaseDate,
|
|||
|
|
"status": "converted",
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 8. 获取已过期用户列表
|
|||
|
|
var expiredBindingsList []model.ReferralBinding
|
|||
|
|
db.Where(
|
|||
|
|
"referrer_id = ? AND (status = 'expired' OR (status = 'active' AND expiry_date <= ?))",
|
|||
|
|
userId, time.Now(),
|
|||
|
|
).Order("expiry_date DESC").Limit(20).Find(&expiredBindingsList)
|
|||
|
|
|
|||
|
|
expiredUsers := []gin.H{}
|
|||
|
|
for _, b := range expiredBindingsList {
|
|||
|
|
var referee model.User
|
|||
|
|
db.Where("id = ?", b.RefereeID).First(&referee)
|
|||
|
|
|
|||
|
|
expiredUsers = append(expiredUsers, gin.H{
|
|||
|
|
"id": b.RefereeID,
|
|||
|
|
"nickname": getStringValue(referee.Nickname),
|
|||
|
|
"avatar": getStringValue(referee.Avatar),
|
|||
|
|
"bindingDate": b.BindingDate,
|
|||
|
|
"expiryDate": b.ExpiryDate,
|
|||
|
|
"status": "expired",
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 9. 获取收益明细
|
|||
|
|
var earningsDetailsList []model.Order
|
|||
|
|
db.Where("referrer_id = ? AND status = 'paid'", userId).
|
|||
|
|
Order("pay_time DESC").
|
|||
|
|
Limit(20).
|
|||
|
|
Find(&earningsDetailsList)
|
|||
|
|
|
|||
|
|
earningsDetails := []gin.H{}
|
|||
|
|
for i := range earningsDetailsList {
|
|||
|
|
e := &earningsDetailsList[i]
|
|||
|
|
var buyer model.User
|
|||
|
|
db.Where("id = ?", e.UserID).First(&buyer)
|
|||
|
|
|
|||
|
|
commission := computeOrderCommission(db, e, nil)
|
|||
|
|
earningsDetails = append(earningsDetails, gin.H{
|
|||
|
|
"id": e.ID,
|
|||
|
|
"orderSn": e.OrderSN,
|
|||
|
|
"amount": e.Amount,
|
|||
|
|
"commission": commission,
|
|||
|
|
"productType": e.ProductType,
|
|||
|
|
"productId": getStringValue(e.ProductID),
|
|||
|
|
"description": getStringValue(e.Description),
|
|||
|
|
"buyerNickname": getStringValue(buyer.Nickname),
|
|||
|
|
"buyerAvatar": getStringValue(buyer.Avatar),
|
|||
|
|
"payTime": e.PayTime,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 计算收益（totalCommission 已按订单逐条计算）
|
|||
|
|
estimatedEarnings := totalCommission
|
|||
|
|
availableEarnings := totalCommission - withdrawnFromTable - pendingWithdrawAmount
|
|||
|
|
if availableEarnings < 0 {
|
|||
|
|
availableEarnings = 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 计算即将过期用户数（7天内）
|
|||
|
|
sevenDaysLater := time.Now().Add(7 * 24 * time.Hour)
|
|||
|
|
expiringCount := 0
|
|||
|
|
for _, b := range activeBindingsList {
|
|||
|
|
if b.ExpiryDate.After(time.Now()) && b.ExpiryDate.Before(sevenDaysLater) {
|
|||
|
|
expiringCount++
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
c.JSON(http.StatusOK, gin.H{
|
|||
|
|
"success": true,
|
|||
|
|
"data": gin.H{
|
|||
|
|
// 核心可见数据
|
|||
|
|
"bindingCount": activeBindings,
|
|||
|
|
"visitCount": totalVisits,
|
|||
|
|
"paidCount": uniquePaidCount,
|
|||
|
|
"expiredCount": expiredBindings,
|
|||
|
|
|
|||
|
|
// 收益数据
|
|||
|
|
"totalCommission": round(totalCommission, 2),
|
|||
|
|
"availableEarnings": round(availableEarnings, 2),
|
|||
|
|
"pendingWithdrawAmount": round(pendingWithdrawAmount, 2),
|
|||
|
|
"withdrawnEarnings": withdrawnFromTable,
|
|||
|
|
"earnings": getFloatValue(user.Earnings),
|
|||
|
|
"pendingEarnings": getFloatValue(user.PendingEarnings),
|
|||
|
|
"estimatedEarnings": round(estimatedEarnings, 2),
|
|||
|
|
"shareRate": int(distributorShare * 100),
|
|||
|
|
"minWithdrawAmount": minWithdrawAmount,
|
|||
|
|
"bindingDays": bindingDays,
|
|||
|
|
"userDiscount": userDiscount,
|
|||
|
|
"withdrawFee": withdrawFee,
|
|||
|
|
|
|||
|
|
// 推荐码
|
|||
|
|
"referralCode": getStringValue(user.ReferralCode),
|
|||
|
|
"referralCount": getIntValue(user.ReferralCount),
|
|||
|
|
|
|||
|
|
// 详细统计
|
|||
|
|
"stats": gin.H{
|
|||
|
|
"totalBindings": totalBindings,
|
|||
|
|
"activeBindings": activeBindings,
|
|||
|
|
"convertedBindings": convertedBindings,
|
|||
|
|
"expiredBindings": expiredBindings,
|
|||
|
|
"expiringCount": expiringCount,
|
|||
|
|
"totalPaymentAmount": totalAmount,
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
// 用户列表
|
|||
|
|
"activeUsers": activeUsers,
|
|||
|
|
"convertedUsers": convertedUsers,
|
|||
|
|
"expiredUsers": expiredUsers,
|
|||
|
|
|
|||
|
|
// 收益明细
|
|||
|
|
"earningsDetails": earningsDetails,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// round 四舍五入保留小数
|
|||
|
|
func round(val float64, precision int) float64 {
|
|||
|
|
ratio := math.Pow(10, float64(precision))
|
|||
|
|
return math.Round(val*ratio) / ratio
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MyEarnings GET /api/miniprogram/earnings 仅返回「我的收益」卡片所需数据（累计、可提现、推荐人数），用于我的页展示与刷新
|
|||
|
|
func MyEarnings(c *gin.Context) {
|
|||
|
|
userId := c.Query("userId")
|
|||
|
|
if userId == "" {
|
|||
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "用户ID不能为空"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
db := database.DB()
|
|||
|
|
var user model.User
|
|||
|
|
if err := db.Where("id = ?", userId).First(&user).Error; err != nil {
|
|||
|
|
c.JSON(http.StatusNotFound, gin.H{"success": false, "error": "用户不存在"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
var paidOrders []model.Order
|
|||
|
|
db.Where("referrer_id = ? AND status = ?", userId, "paid").Find(&paidOrders)
|
|||
|
|
totalCommission := 0.0
|
|||
|
|
for i := range paidOrders {
|
|||
|
|
totalCommission += computeOrderCommission(db, &paidOrders[i], nil)
|
|||
|
|
}
|
|||
|
|
var pendingWithdraw struct{ Total float64 }
|
|||
|
|
db.Model(&model.Withdrawal{}).
|
|||
|
|
Select("COALESCE(SUM(amount), 0) as total").
|
|||
|
|
Where("user_id = ? AND status IN ?", userId, []string{"pending", "processing", "pending_confirm"}).
|
|||
|
|
Scan(&pendingWithdraw)
|
|||
|
|
var successWithdraw struct{ Total float64 }
|
|||
|
|
db.Model(&model.Withdrawal{}).
|
|||
|
|
Select("COALESCE(SUM(amount), 0) as total").
|
|||
|
|
Where("user_id = ? AND status = ?", userId, "success").
|
|||
|
|
Scan(&successWithdraw)
|
|||
|
|
pendingWithdrawAmount := pendingWithdraw.Total
|
|||
|
|
withdrawnFromTable := successWithdraw.Total
|
|||
|
|
availableEarnings := totalCommission - withdrawnFromTable - pendingWithdrawAmount
|
|||
|
|
if availableEarnings < 0 {
|
|||
|
|
availableEarnings = 0
|
|||
|
|
}
|
|||
|
|
c.JSON(http.StatusOK, gin.H{
|
|||
|
|
"success": true,
|
|||
|
|
"data": gin.H{
|
|||
|
|
"totalCommission": round(totalCommission, 2),
|
|||
|
|
"availableEarnings": round(availableEarnings, 2),
|
|||
|
|
"referralCount": getIntValue(user.ReferralCount),
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ReferralVisit POST /api/referral/visit 记录推荐访问（不需登录）
|
|||
|
|
func ReferralVisit(c *gin.Context) {
|
|||
|
|
var req struct {
|
|||
|
|
ReferralCode string `json:"referralCode" binding:"required"`
|
|||
|
|
VisitorOpenID string `json:"visitorOpenId"`
|
|||
|
|
VisitorID string `json:"visitorId"`
|
|||
|
|
Source string `json:"source"`
|
|||
|
|
Page string `json:"page"`
|
|||
|
|
}
|
|||
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|||
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "推荐码不能为空"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
db := database.DB()
|
|||
|
|
var referrer model.User
|
|||
|
|
if err := db.Where("referral_code = ?", req.ReferralCode).First(&referrer).Error; err != nil {
|
|||
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "推荐码无效"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
source := req.Source
|
|||
|
|
if source == "" {
|
|||
|
|
source = "miniprogram"
|
|||
|
|
}
|
|||
|
|
visitorID := req.VisitorID
|
|||
|
|
if visitorID == "" {
|
|||
|
|
visitorID = ""
|
|||
|
|
}
|
|||
|
|
vOpenID := req.VisitorOpenID
|
|||
|
|
vPage := req.Page
|
|||
|
|
err := db.Create(&model.ReferralVisit{
|
|||
|
|
ReferrerID: referrer.ID,
|
|||
|
|
VisitorID: strPtrOrNil(visitorID),
|
|||
|
|
VisitorOpenID: strPtrOrNil(vOpenID),
|
|||
|
|
Source: strPtrOrNil(source),
|
|||
|
|
Page: strPtrOrNil(vPage),
|
|||
|
|
}).Error
|
|||
|
|
if err != nil {
|
|||
|
|
c.JSON(http.StatusOK, gin.H{"success": true, "message": "已处理"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
c.JSON(http.StatusOK, gin.H{"success": true, "message": "访问已记录"})
|
|||
|
|
}
|
|||
|
|
func strPtrOrNil(s string) *string {
|
|||
|
|
if s == "" {
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
return &s
|
|||
|
|
}
|