team/hm-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
liangzai
2025-10-24 15:39:20 +08:00
parent c72d0a7aa6
commit a258c360c2
28 changed files with 1335 additions and 351 deletions
Download Patch File Download Diff File Expand all files Collapse all files

270
app/main/api/internal/service/agentService.go
View File

@@ -12,6 +12,7 @@ import (
type AgentService struct {
config config.Config
OrderModel model.OrderModel
AgentModel model.AgentModel
AgentAuditModel model.AgentAuditModel
AgentClosureModel model.AgentClosureModel
@@ -30,7 +31,7 @@ type AgentService struct {
AgentWithdrawalModel model.AgentWithdrawalModel
}
func NewAgentService(c config.Config, agentModel model.AgentModel, agentAuditModel model.AgentAuditModel,
func NewAgentService(c config.Config, orderModel model.OrderModel, agentModel model.AgentModel, agentAuditModel model.AgentAuditModel,
agentClosureModel model.AgentClosureModel, agentCommissionModel model.AgentCommissionModel,
agentCommissionDeductionModel model.AgentCommissionDeductionModel, agentWalletModel model.AgentWalletModel, agentLinkModel model.AgentLinkModel, agentOrderModel model.AgentOrderModel, agentRewardsModel model.AgentRewardsModel,
agentMembershipConfigModel model.AgentMembershipConfigModel,
@@ -41,6 +42,7 @@ func NewAgentService(c config.Config, agentModel model.AgentModel, agentAuditMod
return &AgentService{
config: c,
OrderModel: orderModel,
AgentModel: agentModel,
AgentAuditModel: agentAuditModel,
AgentClosureModel: agentClosureModel,
@@ -99,10 +101,10 @@ func (l *AgentService) AgentProcess(ctx context.Context, order *model.Order) err
if AgentClosureModel != nil {
AncestorId := AgentClosureModel.AncestorId
AncestorModel, findAgentModelErr := l.AgentModel.FindOne(transCtx, AncestorId)
if findAgentModelErr != nil != errors.Is(findAgentModelErr, model.ErrNotFound) {
if findAgentModelErr != nil && !errors.Is(findAgentModelErr, model.ErrNotFound) {
return findAgentModelErr
}
if AgentClosureModel != nil {
if AncestorModel != nil {
if AncestorModel.LevelName == "" {
AncestorModel.LevelName = model.AgentLeveNameNormal
}
@@ -271,6 +273,10 @@ func (l *AgentService) CommissionCost(ctx context.Context, descendantId int64, A
// 拥有则查看该上级设定的成本
agentMembershipUserConfigModel, findAgentMembershipUserConfigModelErr := l.AgentMembershipUserConfigModel.FindOneByAgentIdProductId(ctx, AncestorId, productID)
if findAgentMembershipUserConfigModelErr != nil {
// 如果上级没有配置该产品的定价规则，则跳过成本计算
if errors.Is(findAgentMembershipUserConfigModelErr, model.ErrNotFound) {
return 0, nil
}
return 0, findAgentMembershipUserConfigModelErr
}
@@ -301,6 +307,10 @@ func (l *AgentService) CommissionPricing(ctx context.Context, descendantId int64
// 拥有则查看该上级设定的成本
agentMembershipUserConfigModel, findAgentMembershipUserConfigModelErr := l.AgentMembershipUserConfigModel.FindOneByAgentIdProductId(ctx, AncestorId, productID)
if findAgentMembershipUserConfigModelErr != nil {
// 如果上级没有配置该产品的定价规则，则跳过成本计算
if errors.Is(findAgentMembershipUserConfigModelErr, model.ErrNotFound) {
return 0, nil
}
return 0, findAgentMembershipUserConfigModelErr
}
@@ -332,13 +342,247 @@ func (l *AgentService) CommissionPricing(ctx context.Context, descendantId int64
return 0, nil
}
//func (l *AgentService) UpgradeVip(ctx context.Context, agentID int64, leve string, session sqlx.Session) error {
// agentModel, err := l.AgentModel.FindOne(ctx, agentID)
// if err != nil {
// return err
// }
// if agentModel.LevelName != model.AgentLeveNameNormal {
// return fmt.Errorf("已经是会员")
// }
// return nil
//}
// GiveUpgradeReward 给上级代理发放下级升级奖励
func (l *AgentService) GiveUpgradeReward(ctx context.Context, agentID int64, oldLevel, newLevel string, session sqlx.Session) error {
// 查找上级代理
agentClosureModel, err := l.AgentClosureModel.FindOneByDescendantIdDepth(ctx, agentID, 1)
if err != nil {
if errors.Is(err, model.ErrNotFound) {
// 没有上级代理，直接返回
return nil
}
return err
}
ancestorID := agentClosureModel.AncestorId
ancestorModel, err := l.AgentModel.FindOne(ctx, ancestorID)
if err != nil {
if errors.Is(err, model.ErrNotFound) {
// 上级代理不存在，直接返回
return nil
}
return err
}
if ancestorModel == nil {
// 上级代理不存在，直接返回
return nil
}
// 获取上级代理的等级配置
if ancestorModel.LevelName == "" {
ancestorModel.LevelName = model.AgentLeveNameNormal
}
agentMembershipConfigModel, err := l.AgentMembershipConfigModel.FindOneByLevelName(ctx, ancestorModel.LevelName)
if err != nil {
return err
}
// 根据升级路径计算奖励金额差额
var rewardAmount float64
var rewardType string
// 获取各等级的奖励金额
var vipRewardAmount float64
var svipRewardAmount float64
if agentMembershipConfigModel.LowerConvertVipReward.Valid {
vipRewardAmount = agentMembershipConfigModel.LowerConvertVipReward.Float64
}
if agentMembershipConfigModel.LowerConvertSvipReward.Valid {
svipRewardAmount = agentMembershipConfigModel.LowerConvertSvipReward.Float64
}
// 根据升级路径计算实际奖励金额
switch {
case oldLevel == "" || oldLevel == model.AgentLeveNameNormal:
// 普通代理升级
switch newLevel {
case model.AgentLeveNameVIP:
rewardAmount = vipRewardAmount
rewardType = model.AgentRewardsTypeDescendantUpgradeVip
case model.AgentLeveNameSVIP:
rewardAmount = svipRewardAmount
rewardType = model.AgentRewardsTypeDescendantUpgradeSvip
default:
// 无效的升级路径，直接返回
return nil
}
case oldLevel == model.AgentLeveNameVIP && newLevel == model.AgentLeveNameSVIP:
// VIP升级到SVIP，发放差额奖励
rewardAmount = svipRewardAmount - vipRewardAmount
rewardType = model.AgentRewardsTypeDescendantUpgradeSvip
// 如果差额为负数或零，不发放奖励
if rewardAmount <= 0 {
return nil
}
default:
// 其他无效的升级路径（如SVIP降级等），直接返回
return nil
}
// 如果有奖励金额，则发放奖励
if rewardAmount > 0 {
// 创建奖励记录
agentRewards := model.AgentRewards{
AgentId: ancestorID,
Amount: rewardAmount,
RelationAgentId: lzUtils.Int64ToNullInt64(agentID),
Type: rewardType,
}
_, err = l.AgentRewardsModel.Insert(ctx, session, &agentRewards)
if err != nil {
return err
}
// 更新上级代理钱包
ancestorWallet, err := l.AgentWalletModel.FindOneByAgentId(ctx, ancestorID)
if err != nil {
return err
}
ancestorWallet.Balance += rewardAmount
ancestorWallet.TotalEarnings += rewardAmount
err = l.AgentWalletModel.UpdateWithVersion(ctx, session, ancestorWallet)
if err != nil {
return err
}
}
return nil
}
// GiveWithdrawReward 给上级代理发放下级提现奖励
func (l *AgentService) GiveWithdrawReward(ctx context.Context, agentID int64, withdrawAmount float64, session sqlx.Session) error {
// 验证提现金额
if withdrawAmount <= 0 {
return nil
}
// 查找上级代理
agentClosureModel, err := l.AgentClosureModel.FindOneByDescendantIdDepth(ctx, agentID, 1)
if err != nil {
if errors.Is(err, model.ErrNotFound) {
// 没有上级代理，直接返回
return nil
}
return err
}
ancestorID := agentClosureModel.AncestorId
ancestorModel, err := l.AgentModel.FindOne(ctx, ancestorID)
if err != nil {
if errors.Is(err, model.ErrNotFound) {
// 上级代理不存在，直接返回
return nil
}
return err
}
if ancestorModel == nil {
// 上级代理不存在，直接返回
return nil
}
// 获取上级代理的等级配置
if ancestorModel.LevelName == "" {
ancestorModel.LevelName = model.AgentLeveNameNormal
}
agentMembershipConfigModel, err := l.AgentMembershipConfigModel.FindOneByLevelName(ctx, ancestorModel.LevelName)
if err != nil {
return err
}
// 计算提现奖励金额
if agentMembershipConfigModel.LowerWithdrawRewardRatio.Valid {
rewardRatio := agentMembershipConfigModel.LowerWithdrawRewardRatio.Float64
// 验证奖励比例的有效性（0-1之间）
if rewardRatio < 0 || rewardRatio > 1 {
// 无效的奖励比例，直接返回
return nil
}
rewardAmount := withdrawAmount * rewardRatio
if rewardAmount > 0 {
// 创建奖励记录
agentRewards := model.AgentRewards{
AgentId: ancestorID,
Amount: rewardAmount,
RelationAgentId: lzUtils.Int64ToNullInt64(agentID),
Type: model.AgentRewardsTypeDescendantWithdraw,
}
_, err = l.AgentRewardsModel.Insert(ctx, session, &agentRewards)
if err != nil {
return err
}
// 更新上级代理钱包
ancestorWallet, err := l.AgentWalletModel.FindOneByAgentId(ctx, ancestorID)
if err != nil {
return err
}
ancestorWallet.Balance += rewardAmount
ancestorWallet.TotalEarnings += rewardAmount
err = l.AgentWalletModel.UpdateWithVersion(ctx, session, ancestorWallet)
if err != nil {
return err
}
}
}
return nil
}
// CheckAgentProcessStatus 检查代理处理事务是否已成功
func (l *AgentService) CheckAgentProcessStatus(ctx context.Context, orderID int64) (bool, error) {
// 检查是否存在代理订单记录
_, err := l.AgentOrderModel.FindOneByOrderId(ctx, orderID)
if err != nil {
if errors.Is(err, model.ErrNotFound) {
// 没有代理订单记录，说明不是代理推广订单
return true, nil
}
return false, err
}
// 检查是否存在代理佣金记录
// 使用SelectBuilder查询该订单的佣金记录
selectBuilder := l.AgentCommissionModel.SelectBuilder()
selectBuilder = selectBuilder.Where("order_id = ?", orderID)
selectBuilder = selectBuilder.Where("del_state = ?", 0) // 未删除
commissions, err := l.AgentCommissionModel.FindAll(ctx, selectBuilder, "")
if err != nil {
return false, err
}
// 如果存在佣金记录，说明代理处理已成功
return len(commissions) > 0, nil
}
// RetryAgentProcess 重新执行代理处理事务
func (l *AgentService) RetryAgentProcess(ctx context.Context, orderID int64) error {
// 首先检查订单是否存在
order, err := l.OrderModel.FindOne(ctx, orderID)
if err != nil {
return err
}
// 检查订单状态是否为已支付
if order.Status != "paid" {
return errors.New("订单状态不是已支付，无法执行代理处理")
}
// 检查代理处理是否已经成功
alreadyProcessed, err := l.CheckAgentProcessStatus(ctx, orderID)
if err != nil {
return err
}
if alreadyProcessed {
return errors.New("代理处理已经成功，无需重新执行")
}
// 执行代理处理
return l.AgentProcess(ctx, order)
}