blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
chenlu
2024-06-07 2f640b1038fa331954f78ed1f4317212cf5bb34d 
 north-glass-erp/src/main/java/com/example/erp/service/userInfo/UserRoleService.java


New file


@@ -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;


    }


}