From 51c36e6f6b17862714fbd5c42da43386dc05d28a Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Fri, 09 Apr 2021 14:32:29 +0800
Subject: [PATCH] Merge branch 'dev' of http://blit.7drlb.com:8888/r/moral into dev
---
screen-api/src/main/java/com/moral/api/service/impl/UserGroupServiceImpl.java | 43 +++++++++++++++++++++++++++++++++++++++++++
1 files changed, 43 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..1bf3af4 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,19 @@
package com.moral.api.service.impl;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+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.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 +27,37 @@
@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);
+ }
+ }
+ }
+
+ @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<>();
+ for (UserGroup userGroup : userGroups) {
+ groupIds.add(userGroup.getGroupId());
+ }
+ return groupIds;
+ }
}
--
Gitblit v1.8.0