1

src/main/java/com/fdkankan/agent/common/ResultCode.java

@@ -33,6 +33,7 @@ public enum ResultCode {
     AGENT_STOP_USE(4030, "当前账号已停用，请联系管理员"),
     AGENT_NOT_SUB(4031, "当前经销商无法新增下级"),
     AUTH_ERROR(4032, "权限不足"),
+    AUTH_CODE_EXIST(4033, "添加失败，存在重复授权码"),
 
     ;
 

src/main/java/com/fdkankan/agent/controller/AgentAuthorizeModelingController.java

@@ -7,13 +7,18 @@ import com.fdkankan.agent.common.ResultData;
 import com.fdkankan.agent.entity.AgentAuthorizeModeling;
 import com.fdkankan.agent.exception.BusinessException;
 import com.fdkankan.agent.mapper.IAgentAuthorizeModelingMapper;
+import com.fdkankan.agent.request.AuthModelingParam;
 import com.fdkankan.agent.request.AuthorizeParam;
 import com.fdkankan.agent.service.IAgentAuthorizeModelingService;
 import com.fdkankan.agent.service.IExcelService;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.poi.util.StringUtil;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
+import java.util.List;
+
 /**
  * <p>
  *  前端控制器
@@ -40,8 +45,19 @@ public class AgentAuthorizeModelingController extends BaseController {
 
     @PostMapping("/addOrUpdate")
     public ResultData addOrUpdate(@RequestBody AgentAuthorizeModeling param){
+        if(StringUtils.isBlank(param.getAuthorizeKey())){
+            throw new BusinessException(ResultCode.PARAM_MISS);
+        }
+        List<AgentAuthorizeModeling> authModelings = authorizeModelingService.getByAuthorizeKey(param.getAuthorizeKey());
         if(param.getId() == null){
             param.setCreateAgentId(getAgent().getId());
+            if(!authModelings.isEmpty()){
+                throw new BusinessException(ResultCode.AUTH_CODE_EXIST);
+            }
+        }else {
+            if(!authModelings.get(0).getId().equals(param.getId())){
+                throw new BusinessException(ResultCode.AUTH_CODE_EXIST);
+            }
         }
         return ResultData.ok( authorizeModelingService.addOrUpdate(param));
     }

src/main/java/com/fdkankan/agent/service/IAgentAuthorizeModelingService.java

@@ -24,4 +24,6 @@ public interface IAgentAuthorizeModelingService extends IService<AgentAuthorizeM
     void checkExpTime();
 
     void addByParam(List<AuthModelingParam> params);
+
+    List<AgentAuthorizeModeling> getByAuthorizeKey(String authorizeKey);
 }

src/main/java/com/fdkankan/agent/service/impl/AgentAuthorizeModelingServiceImpl.java

@@ -156,4 +156,11 @@ public class AgentAuthorizeModelingServiceImpl extends ServiceImpl<IAgentAuthori
         list.forEach( e-> map.put(e.getAuthorizeKey(),e));
         return map;
     }
+
+    @Override
+    public List<AgentAuthorizeModeling> getByAuthorizeKey(String authorizeKey) {
+        LambdaQueryWrapper<AgentAuthorizeModeling> wrapper = new LambdaQueryWrapper<>();
+        wrapper.eq(AgentAuthorizeModeling::getAuthorizeKey,authorizeKey);
+        return this.list(wrapper);
+    }
 }