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