From ba55d44ea9f61a007f5c63fb8689488cc35d48ee Mon Sep 17 00:00:00 2001
From: chen_xi <276999030@qq.com>
Date: Mon, 26 Sep 2022 15:05:56 +0800
Subject: [PATCH] 多站单参接口修改

---
 screen-api/src/main/java/com/moral/api/service/impl/UserGroupServiceImpl.java |   51 ++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 36 insertions(+), 15 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 230e3f4..a4007ee 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
@@ -1,19 +1,22 @@
 package com.moral.api.service.impl;
 
-import java.util.ArrayList;
-import java.util.List;
 import java.util.Map;
 
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.moral.api.entity.UserGroup;
+import com.moral.api.mapper.GroupMapper;
 import com.moral.api.mapper.UserGroupMapper;
+import com.moral.api.mapper.UserMapper;
 import com.moral.api.service.UserGroupService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.moral.api.utils.OperationLogUtils;
 import com.moral.constant.Constants;
 import com.moral.util.TokenUtils;
 
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.ObjectUtils;
 
 /**
  * <p>
@@ -29,22 +32,40 @@
     @Autowired
     private UserGroupMapper userGroupMapper;
 
+    @Autowired
+    private UserMapper userMapper;
+
+    @Autowired
+    private GroupMapper groupMapper;
+
+    @Autowired
+    private OperationLogUtils operationLogUtils;
+
     @Override
-    public void allotGroups(Map<String, Object> parameters, String token) {
+    @Transactional
+    public void allotGroups(Map<String, Object> parameters) {
+        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> currentUserInfo = (Map<String, Object>) TokenUtils.getUserInfo();
+        Map<String, Object> orgInfo = (Map<String, Object>) currentUserInfo.get("organization");
+
         UpdateWrapper<UserGroup> deleteWrapper = new UpdateWrapper<>();
-        deleteWrapper.set("is_delete", Constants.DELETE).eq("user_id", userId);
-        userGroupMapper.update(null, deleteWrapper);
-        if (!groupIds.isEmpty()) {
-            for (Integer groupId : groupIds) {
-                UserGroup userGroup = new UserGroup();
-                userGroup.setUserId(userId);
-                userGroup.setGroupId(groupId);
-                userGroup.setOrganizationId((Integer) currentUserInfo.get("organizationId"));
-                userGroupMapper.insert(userGroup);
-            }
+        deleteWrapper.eq("user_id", userId);
+        userGroupMapper.delete(deleteWrapper);
+
+        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(content, Constants.UPDATE_OPERATE_TYPE);
         }
     }
 }

--
Gitblit v1.8.0