com.didiglobal.logi.security.service.impl.UserRoleServiceImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of logi-security-spring-boot-starter Show documentation
logi-common/logi-security
The newest version!
package com.didiglobal.logi.security.service.impl;

import com.didiglobal.logi.security.common.entity.UserRole;
import com.didiglobal.logi.security.dao.UserRoleDao;
import com.didiglobal.logi.security.service.UserRoleService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.List;

/**
 * @author cjm
 */
@Service("logiSecurityUserRoleServiceImpl")
public class UserRoleServiceImpl implements UserRoleService {

    @Autowired
    private UserRoleDao userRoleDao;

    @Override
    public List getUserIdListByRoleId(Integer roleId) {
        if(roleId == null) {
            return new ArrayList<>();
        }
        return userRoleDao.selectUserIdListByRoleId(roleId);
    }

    @Override
    public List getRoleIdListByUserId(Integer userId) {
        if(userId == null) {
            return new ArrayList<>();
        }
        return userRoleDao.selectRoleIdListByUserId(userId);
    }

    @Override
    public void updateUserRoleByUserId(Integer userId, List roleIdList) {
        if(userId == null) {
            return;
        }

        // 删除old的全部角色用户关联信息
        userRoleDao.deleteByUserIdOrRoleId(userId, null);

        if(CollectionUtils.isEmpty(roleIdList)) {
            return;
        }

        // 插入new的角色与用户关联关系
        userRoleDao.insertBatch(getUserRoleList(true, userId, roleIdList));
    }

    @Override
    public void updateUserRoleByRoleId(Integer roleId, List userIdList) {
        if(roleId == null) {
            return;
        }

        // 删除old的全部角色用户关联信息
        userRoleDao.deleteByUserIdOrRoleId(null, roleId);

        if(CollectionUtils.isEmpty(userIdList)) {
            return;
        }

        // 插入new的角色与用户关联关系
        userRoleDao.insertBatch(getUserRoleList(false, roleId, userIdList));
    }

    @Override
    public int getUserRoleCountByRoleId(Integer roleId) {
        if(roleId == 0) {
            return 0;
        }
        return userRoleDao.selectCountByRoleId(roleId);
    }

    private List getUserRoleList(boolean isUserId, Integer id, List idList) {
        List result = new ArrayList<>();
        for(Integer id2 : idList) {
            result.add(isUserId ? new UserRole(id, id2) : new UserRole(id2, id));
        }
        return result;
    }
}