From 2f640b1038fa331954f78ed1f4317212cf5bb34d Mon Sep 17 00:00:00 2001 From: chenlu <1320612696@qq.com> Date: 星期五, 07 六月 2024 16:56:06 +0800 Subject: [PATCH] Merge branch 'master' of http://10.153.19.25:10101/r/ERP_override --- north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserRoleService.java | 41 +++++++++++++++++++++++++++++++++++++++++ 1 files changed, 41 insertions(+), 0 deletions(-) diff --git a/north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserRoleService.java b/north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserRoleService.java new file mode 100644 index 0000000..f968e5b --- /dev/null +++ b/north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserRoleService.java @@ -0,0 +1,41 @@ +package com.example.erp.service.userInfo; + +import com.baomidou.dynamic.datasource.annotation.DS; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.example.erp.entity.userInfo.UserRole; +import com.example.erp.mapper.userInfo.UserRoleMapper; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; + +import java.util.ArrayList; +import java.util.List; +import java.util.Map; + +@Service +@DS("user_info") +public class UserRoleService { + private final UserRoleMapper userRoleMapper; + + public UserRoleService(UserRoleMapper userRoleMapper) { + this.userRoleMapper = userRoleMapper; + } + + public Boolean updateUserRole(Map<String, Object> params) { + Integer userId = (Integer) params.get("userId"); + List<Integer> list = (List<Integer>) params.get("roles"); + List<UserRole> userRoleList = new ArrayList<>(); + if(!list.isEmpty()) { + for (Integer roleId : list) { + UserRole userRole = new UserRole(); + userRole.setUserId(userId); + userRole.setRoleId(roleId); + userRoleList.add(userRole); + } + } + userRoleMapper.delete(new QueryWrapper<UserRole>().eq("user_id", userId)); + userRoleList.forEach(userRole -> { + userRoleMapper.insert(userRole); + }); + return true; + } +} -- Gitblit v1.8.0