浏览代码

Merge branch 'feature-V2.0' into 'master'

Feature v2.0

See merge request crowd-2019/crowd-test-service-backend!146
郭超 5 年之前
父节点
当前提交
c85d8530da
共有 1 个文件被更改,包括 18 次插入0 次删除
  1. 18 0
      site/src/main/java/com/mooctest/crowd/site/service/impl/CrowdProjectServiceImpl.java

+ 18 - 0
site/src/main/java/com/mooctest/crowd/site/service/impl/CrowdProjectServiceImpl.java

@@ -344,6 +344,24 @@ public class CrowdProjectServiceImpl implements CrowdProjectService {
             reportForProject.setName(project.getName() + "测试报告");
             reportForProject.setType(project.getType());
             reportForProject.setTestObject(project.getName());
+
+            // 更新当前用户的接包数量
+            Optional<UserTaskCountPO> userTaskCountPOOptional = userTaskCountDao.findByUserId(user.getId());
+            if (!userTaskCountPOOptional.isPresent()) {
+                if (user.getRoleList().stream().anyMatch(role -> role.getName().equals(RoleType.EVALUATION_USER.getName()))) {
+                    Long roleType = RoleType.EVALUATION_USER.getId();
+                    UserTaskCount userTaskCount = new UserTaskCount(user.getId(), 1L, roleType);
+                    userTaskCountDao.save(Converter.convert(UserTaskCountPO.class, userTaskCount));
+                } else if (user.getRoleList().stream().anyMatch(role -> role.getName().equals(RoleType.AGENCY.getName()))) {
+                    Long roleType = RoleType.AGENCY.getId();
+                    UserTaskCount userTaskCount = new UserTaskCount(user.getId(), 1L, roleType);
+                    userTaskCountDao.save(Converter.convert(UserTaskCountPO.class, userTaskCount));
+                }
+            } else {
+                UserTaskCountPO userTaskCountPO = userTaskCountPOOptional.get();
+                userTaskCountPO.setCount(userTaskCountPO.getCount() + 1);
+                userTaskCountDao.save(userTaskCountPO);
+            }
         });
         return projects;
     }