From cd8755a6ac55137176666a6f4fbfd15210389d77 Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Fri, 19 Mar 2021 17:33:26 +0800
Subject: [PATCH] 登陆验证码,组分配,菜单分配

---
 screen-api/src/main/java/com/moral/api/service/impl/UserGroupServiceImpl.java |   30 ++++++++++++++++++++++++++++++
 1 files changed, 30 insertions(+), 0 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 8ab1c0d..230e3f4 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,9 +1,18 @@
 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.UserGroupMapper;
 import com.moral.api.service.UserGroupService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.moral.constant.Constants;
+import com.moral.util.TokenUtils;
+
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 /**
@@ -17,4 +26,25 @@
 @Service
 public class UserGroupServiceImpl extends ServiceImpl<UserGroupMapper, UserGroup> implements UserGroupService {
 
+    @Autowired
+    private UserGroupMapper userGroupMapper;
+
+    @Override
+    public void allotGroups(Map<String, Object> parameters, String token) {
+        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);
+        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);
+            }
+        }
+    }
 }

--
Gitblit v1.8.0