com.didiglobal.logi.security.dao.impl.UserResourceDaoImpl 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.dao.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.didiglobal.logi.security.common.dto.resource.ControlLevelQueryDTO;
import com.didiglobal.logi.security.common.dto.resource.UserResourceQueryDTO;
import com.didiglobal.logi.security.common.entity.UserResource;
import com.didiglobal.logi.security.common.enums.resource.ControlLevelCode;
import com.didiglobal.logi.security.common.po.UserResourcePO;
import com.didiglobal.logi.security.dao.UserResourceDao;
import com.didiglobal.logi.security.dao.mapper.UserResourceMapper;
import com.didiglobal.logi.security.util.CopyBeanUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import java.util.List;
import java.util.stream.Collectors;

/**
 * @author cjm
 */
@Component
public class UserResourceDaoImpl extends BaseDaoImpl implements UserResourceDao {

    @Autowired
    private UserResourceMapper userResourceMapper;

    /**
     * 封装UserResource的查询条件
     * @param userId 用户id
     * @param queryDTO 查询条件
     * @return QueryWrapper
     */
    private QueryWrapper wrapQueryCriteria(Integer userId, UserResourceQueryDTO queryDTO) {
        QueryWrapper queryWrapper = getQueryWrapper();
        Integer resourceTypeId = queryDTO.getResourceTypeId();
        queryWrapper
                .eq(FieldConstant.CONTROL_LEVEL, queryDTO.getControlLevel())
                .eq(userId != null, FieldConstant.USER_ID, userId)
                .eq(queryDTO.getProjectId() != null, FieldConstant.PROJECT_ID, queryDTO.getProjectId())
                .eq(resourceTypeId != null, FieldConstant.RESOURCE_TYPE_ID, resourceTypeId)
                .eq(queryDTO.getResourceId() != null, FieldConstant.RESOURCE_ID, queryDTO.getResourceId());
        return queryWrapper;
    }

    @Override
    public int selectCountByUserId(Integer userId, UserResourceQueryDTO queryDTO) {
        return userResourceMapper.selectCount(wrapQueryCriteria(userId, queryDTO));
    }

    @Override
    public void deleteByUserId(Integer userId, UserResourceQueryDTO queryDTO) {
        userResourceMapper.delete(wrapQueryCriteria(userId, queryDTO));
    }

    @Override
    public void deleteByControlLevel(ControlLevelCode controlLevel) {
        QueryWrapper queryWrapper = getQueryWrapper();
        queryWrapper.eq(FieldConstant.CONTROL_LEVEL, controlLevel.getType());
        userResourceMapper.delete(queryWrapper);
    }

    @Override
    public void insert(UserResource userResource) {
        userResourceMapper.insert(CopyBeanUtil.copy(userResource, UserResourcePO.class));
    }

    @Override
    public void insertBatch(List userResourceList) {
        if(CollectionUtils.isEmpty(userResourceList)) {
            return;
        }
        List userResourcePOList = CopyBeanUtil.copyList(userResourceList, UserResourcePO.class);
        for(UserResourcePO userResourcePO : userResourcePOList) {
            userResourceMapper.insert(userResourcePO);
        }

    }

    @Override
    public void deleteByUserIdList(List userIdList, UserResourceQueryDTO queryDTO) {
        QueryWrapper queryWrapper = wrapQueryCriteria(null, queryDTO);
        queryWrapper.in(FieldConstant.USER_ID, userIdList);
        userResourceMapper.delete(queryWrapper);
    }

    @Override
    public void deleteByProjectIdList(List projectIdList, UserResourceQueryDTO queryDTO) {
        QueryWrapper queryWrapper = wrapQueryCriteria(null, queryDTO);
        queryWrapper.in(FieldConstant.PROJECT_ID, projectIdList);
        userResourceMapper.delete(queryWrapper);
    }

    @Override
    public void deleteByResourceTypeIdList(List resourceTypeIdList, UserResourceQueryDTO queryDTO) {
        QueryWrapper queryWrapper = wrapQueryCriteria(null, queryDTO);
        queryWrapper.in(FieldConstant.RESOURCE_TYPE_ID, resourceTypeIdList);
        userResourceMapper.delete(queryWrapper);
    }

    @Override
    public void deleteByResourceIdList(List resourceIdList, UserResourceQueryDTO queryDTO) {
        QueryWrapper queryWrapper = wrapQueryCriteria(null, queryDTO);
        queryWrapper.in(FieldConstant.RESOURCE_ID, resourceIdList);
        userResourceMapper.delete(queryWrapper);
    }

    @Override
    public int selectCountByUserIdAndControlLevel(Integer userId, ControlLevelCode controlLevel) {
        QueryWrapper queryWrapper = getQueryWrapper();
        queryWrapper
                .eq(userId != null, FieldConstant.USER_ID, userId)
                .eq(FieldConstant.CONTROL_LEVEL, controlLevel.getType());
        return userResourceMapper.selectCount(queryWrapper);
    }

    @Override
    public int selectCount(UserResourceQueryDTO queryDTO) {
        return userResourceMapper.selectCount(wrapQueryCriteria(null, queryDTO));
    }

    @Override
    public List selectResourceIdListByUserId(Integer userId, UserResourceQueryDTO queryDTO) {
        QueryWrapper queryWrapper = wrapQueryCriteria(userId, queryDTO);
        queryWrapper.select(FieldConstant.RESOURCE_ID);
        List