From d6c509ab43bd472e56e3d9a9ddde0b335b7fd67e Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Mon, 09 Aug 2021 17:22:30 +0800
Subject: [PATCH] 5分钟数据统计任务修改

---
 screen-api/src/main/java/com/moral/api/service/impl/UserGroupServiceImpl.java |   45 +++++++++++++++++----------------------------
 1 files changed, 17 insertions(+), 28 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/service/impl/UserGroupServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/UserGroupServiceImpl.java
index 4515798..e6039cc 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/UserGroupServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/UserGroupServiceImpl.java
@@ -5,7 +5,6 @@
 import java.util.Map;
 import java.util.Objects;
 
-import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.moral.api.entity.UserGroup;
 import com.moral.api.mapper.GroupMapper;
@@ -20,6 +19,7 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.ObjectUtils;
 import org.springframework.web.context.request.RequestContextHolder;
 import org.springframework.web.context.request.ServletRequestAttributes;
 
@@ -53,39 +53,28 @@
     public void allotGroups(Map<String, Object> parameters) {
         HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
         String token = request.getHeader("token");
-
+        Object o = parameters.get("groupId");
         Integer userId = Integer.parseInt(parameters.get("userId").toString());
-        List<Integer> groupIds = (ArrayList) parameters.get("groupIds");
         Map<String, Object> currentUserInfo = (Map<String, Object>) TokenUtils.getUserInfoByToken(token);
         Map<String, Object> orgInfo = (Map<String, Object>) currentUserInfo.get("organization");
+
         UpdateWrapper<UserGroup> deleteWrapper = new UpdateWrapper<>();
         deleteWrapper.eq("user_id", userId);
         userGroupMapper.delete(deleteWrapper);
-        List<String> groups = new ArrayList<>();
-        if (groupIds != null && !groupIds.isEmpty()) {
-            groupIds.forEach(groupId -> {
-                groups.add(groupMapper.selectById(groupId).getGroupName());
-                UserGroup userGroup = new UserGroup();
-                userGroup.setUserId(userId);
-                userGroup.setGroupId(groupId);
-                userGroup.setOrganizationId((Integer) orgInfo.get("id"));
-                userGroupMapper.insert(userGroup);
-            });
+
+        if (!ObjectUtils.isEmpty(o)) {
+            int groupId = Integer.parseInt(o.toString());
+            String groupName = groupMapper.selectById(groupId).getGroupName();
+            UserGroup userGroup = new UserGroup();
+            userGroup.setUserId(userId);
+            userGroup.setGroupId(groupId);
+            userGroup.setOrganizationId((Integer) orgInfo.get("id"));
+            //user_group���insert
+            userGroupMapper.insert(userGroup);
+            //������
+            String account = userMapper.selectById((Integer) parameters.get("userId")).getAccount();
+            String content = "������������" + account + "���������������" + groupName;
+            operationLogUtils.insertLog(request, content, Constants.UPDATE_OPERATE_TYPE);
         }
-
-        //������
-        String account = userMapper.selectById((Integer) parameters.get("userId")).getAccount();
-        String content = "������������" + account + "���������������" + groups.toString();
-        operationLogUtils.insertLog(request, content, Constants.UPDATE_OPERATE_TYPE);
-    }
-
-    @Override
-    public List<Integer> getGroupIds(Integer userId) {
-        QueryWrapper<UserGroup> queryWrapper = new QueryWrapper<>();
-        queryWrapper.eq("user_id", userId).eq("is_delete", Constants.NOT_DELETE);
-        List<UserGroup> userGroups = userGroupMapper.selectList(queryWrapper);
-        List<Integer> groupIds = new ArrayList<>();
-        userGroups.forEach(userGroup -> groupIds.add(userGroup.getGroupId()));
-        return groupIds;
     }
 }

--
Gitblit v1.8.0