PermissionsServiceImpl.java 1.52 KB
package com.zhongzhi.service.impl;

import com.zhongzhi.model.PermissionsDO;
import com.zhongzhi.dao.PermissionsDAO;
import com.zhongzhi.service.PermissionsService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zhongzhi.vo.permission.PermissionVO;
import org.springframework.stereotype.Service;

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

/**
 * <p>
 * 权限表 服务实现类
 * </p>
 *
 * @author DengMin
 * @since 2021-04-28
 */
@Service
public class PermissionsServiceImpl extends ServiceImpl<PermissionsDAO, PermissionsDO> implements PermissionsService {

    @Override
    public List<PermissionVO> getPermissionById(Long id) {
        return this.baseMapper.getPermissionById(id);
    }

    @Override
    public List<PermissionVO> allPermissions(){
        List<PermissionVO> list = this.getPermissionById(null);
        List<PermissionVO> treeList = new ArrayList();
        list.forEach(permissions -> {
            if (permissions.getParentId() == null) {
                treeList.add(permissions);
            }

            list.forEach(p -> {
                if (p.getParentId() != null) {
                    if (p.getParentId().equals(permissions.getId())) {
                        if (permissions.getChildNodeList() == null) {
                            permissions.setChildNodeList(new ArrayList<>());
                        }
                        permissions.getChildNodeList().add(p);
                    }
                }
            });
        });
        return treeList;
    }

}