Browse Source

Merge branch 'develop' of http://132.232.206.88:3000/lift-manager/lift-server into huangyuan-user

黄远 5 years ago
parent
commit
838fa79f39

+ 8 - 5
lift-business-service/src/main/java/cn/com/ty/lift/business/contract/service/ContractService.java

@@ -76,7 +76,8 @@ public class ContractService extends ServiceImpl<ContractsMapper, Contracts> {
      * @description 保存项目信息
      * @date 2020/1/14 4:30 下午
      */
-    public Project saveProject(Contracts contracts, ContractsExtend extend, Project project) {
+    public Project saveProject(Contracts contracts, ContractsExtend extend) {
+        Project project = new Project();
         project.setProjectName(extend.getProjectName());
         project.setAddress(extend.getProjectAddress());
         project.setNum(contracts.getLiftNum());
@@ -95,20 +96,22 @@ public class ContractService extends ServiceImpl<ContractsMapper, Contracts> {
     @Transactional(rollbackFor = Exception.class)
     public RestResponse add(ContractsExtend extend) {
         Contracts contracts = extend.getContracts();
-        Project project = new Project();
-        contracts.setProjectId(project.getId());
-        if (ObjectUtil.isEmpty(extend.getProjectId())) {
+        Long projectId = extend.getProjectId();
+        Project info;
+        if (ObjectUtil.isEmpty(projectId)) {
             //插入项目信息
-            Project info = saveProject(contracts, extend, project);
+            info = saveProject(contracts, extend);
             if (ObjectUtil.isEmpty(info)) {
                 return RestResponse.fail(MessageUtils.get("msg.add.fail"));
             }
+            contracts.setProjectId(info.getId());
         }
         if (ObjectUtil.isEmpty(contracts.getCode())) {
             //按当前日期时间戳自动生成合同编号
             String contractCode = DateUtils.generateCode();
             contracts.setCode("HT" + contractCode);
         }
+        contracts.setProjectId(projectId);
         //插入合同信息
         int result = contractsMapper.insert(contracts);
         if (result < 0) {

+ 1 - 0
lift-business-service/src/main/java/cn/com/ty/lift/business/project/service/ProjectUserService.java

@@ -37,6 +37,7 @@ public class ProjectUserService extends ServiceImpl<ProjectUserMapper, ProjectUs
         List<ProjectUserResponse> userList = baseMapper.findUserListById(projectId, mtCompanyId);
         return userList.stream().filter(user -> (
                 CommonEnum.DefaultRole.MTWORK.getCode().equals(user.getUserRole())
+                || CommonEnum.DefaultRole.REGION_DIRECTOR.getCode().equals(user.getUserRole())
         )).sorted(Comparator.comparing(ProjectUserResponse::getUserRole)).collect(Collectors.toList());
     }