From 373f08dd87acf015d0c70356cd6a9ad44e5cc470 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 15 Jun 2020 17:48:46 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/modules/agent/service/impl/AgentServiceImpl.java |   16 ++++++++++++----
 1 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/agent/service/impl/AgentServiceImpl.java b/src/main/java/com/xcong/excoin/modules/agent/service/impl/AgentServiceImpl.java
index bc8fccd..b07af97 100644
--- a/src/main/java/com/xcong/excoin/modules/agent/service/impl/AgentServiceImpl.java
+++ b/src/main/java/com/xcong/excoin/modules/agent/service/impl/AgentServiceImpl.java
@@ -4,6 +4,7 @@
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.xcong.excoin.common.entity.FebsConstant;
+import com.xcong.excoin.common.entity.FebsResponse;
 import com.xcong.excoin.common.entity.QueryRequest;
 import com.xcong.excoin.common.exception.FebsException;
 import com.xcong.excoin.common.utils.Md5Util;
@@ -131,6 +132,15 @@
     @Transactional(rollbackFor = Exception.class)
     @Override
     public void editAgent(AgentUser agentUser, User user) {
+        if (FebsConstant.USER_TYPE_ADMIN.equals(user.getType())) {
+            user.setInviteId(FebsConstant.DEFAULT_REFERER_ID);
+        }
+
+        AgentFriendRelationEntity agentFriendRelationEntity = agentFriendRelationMapper.selectAgentFriendRelationByUserId(agentUser.getId());
+        if (!user.getInviteId().equals(agentFriendRelationEntity.getRefererId())) {
+            throw new FebsException("不是该用户直接上级,无法修改");
+        }
+
         if (!FebsConstant.USER_TYPE_ADMIN.equals(user.getType())) {
             AgentFriendRelationEntity friendRelationEntity = agentFriendRelationMapper.selectAgentFriendRelationByUserId(user.getUserId());
             if (agentUser.getReturnRatio().compareTo(friendRelationEntity.getReturnRatio()) > 0) {
@@ -144,9 +154,7 @@
         editUser.setAgentName(agentUser.getName());
         userMapper.updateById(editUser);
 
-        AgentFriendRelationEntity relationEntity = new AgentFriendRelationEntity();
-        relationEntity.setUserId(agentUser.getId());
-        relationEntity.setReturnRatio(agentUser.getReturnRatio());
-        agentFriendRelationMapper.updateByUserId(relationEntity);
+        agentFriendRelationEntity.setReturnRatio(agentUser.getReturnRatio());
+        agentFriendRelationMapper.updateById(agentFriendRelationEntity);
     }
 }

--
Gitblit v1.9.1