changes

liuhaonan

2022-11-04 2611d27616bdbea6331f75a0147bd67c95f2eb0e

 ximon-admin/src/main/java/com/sandu/ximon/admin/service/PlatformFileService.java

@@ -1,5 +1,6 @@
package com.sandu.ximon.admin.service;

import cn.hutool.core.lang.Snowflake;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.pagehelper.PageHelper;
import com.sandu.common.execption.BusinessException;
@@ -7,6 +8,7 @@
import com.sandu.common.file.impl.AliOssFileServiceImpl;
import com.sandu.common.object.BaseConditionVO;
import com.sandu.common.service.impl.BaseServiceImpl;
import com.sandu.common.util.SpringContextHolder;
import com.sandu.ximon.admin.param.AllUserInfoParam;
import com.sandu.ximon.admin.param.EditFileParam;
import com.sandu.ximon.admin.param.PlatformFilePermissionParam;
@@ -25,6 +27,7 @@

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

/**
 * @author LiuHaoNan
@@ -39,6 +42,7 @@
    private final PlatformFileUserService platformFileUserService;
    private final AdminService adminService;
    private final ClientService clientService;
    private final Snowflake snowflake;

    /**
     * 文件上传
@@ -50,6 +54,7 @@
        PermissionToCheck();
        FileUploadDto fileUploadDto = aliOssFileService.uploadFile(file);
        PlatformFile platformFile = new PlatformFile();
        platformFile.setFileId(snowflake.nextId());
        platformFile.setFileName(fileUploadDto.getOriginName());
        platformFile.setFileSize(fileUploadDto.getFileSize());
        platformFile.setFileUrl(fileUploadDto.getFileUrl());
@@ -88,16 +93,18 @@
            throw new BusinessException("删除时参数不能为空!");
        }
        List<PlatformFile> platformFiles = listByIds(param.getFileIds());
        if (platformFiles == null && platformFiles.size() == 0) {
        if (platformFiles == null || platformFiles.size() == 0) {
            throw new BusinessException("未找到文件!");
        }
        //删除关系
        boolean remove = platformFileUserService.remove(Wrappers.lambdaQuery(PlatformFileUser.class).in(PlatformFileUser::getFileId, param.getFileIds()));
        if (remove) {
            //删除文件
            return removeByIds(param.getFileIds());

        //获取关系
        List<PlatformFileUser> list = platformFileUserService.list(Wrappers.lambdaQuery(PlatformFileUser.class).in(PlatformFileUser::getFileId, param.getFileIds()));
        if (!list.isEmpty()) {
            //删除关系
            platformFileUserService.remove(Wrappers.lambdaQuery(PlatformFileUser.class).in(PlatformFileUser::getFileId, param.getFileIds()));
        }
        return false;
        //删除文件
        return removeByIds(param.getFileIds());
    }


@@ -112,10 +119,18 @@
        List<PlatformFileListBo> platformFileListBos;
        if (AdministratorEnums.ADMIN.getCode().equals(SecurityUtils.getUserDetails().getAdministratorType())) {
            PageHelper.startPage(vo.getPageNo(), vo.getPageSize());
            platformFileListBos = baseMapper.fileLiet(keyword, null);
            platformFileListBos = baseMapper.fileList(keyword, null);
        } else {
            List<PlatformFileUser> platformFileUsers = SpringContextHolder.getBean(PlatformFileUserService.class)
                    .list(Wrappers.lambdaQuery(PlatformFileUser.class).eq(PlatformFileUser::getUserId, SecurityUtils.getUserId()));
            List<Long> fileIdList = platformFileUsers.stream().map(PlatformFileUser::getFileId).collect(Collectors.toList());

            if (fileIdList.size() == 0) {
                fileIdList.add(-1L);
            }
            PageHelper.startPage(vo.getPageNo(), vo.getPageSize());
            platformFileListBos = baseMapper.fileLiet(keyword, SecurityUtils.getUserId());
            platformFileListBos = baseMapper.fileList(keyword, fileIdList);

        }
        return platformFileListBos;
    }
@@ -146,6 +161,7 @@
     * @return
     */
    public String setPermission(PlatformFilePermissionParam platformFilePermissionParam) {
        PermissionToCheck();
        //删除旧关系
        platformFileUserService.remove(Wrappers.lambdaQuery(PlatformFileUser.class)
                .eq(PlatformFileUser::getFileId, platformFilePermissionParam.getFileId()));
@@ -173,7 +189,8 @@
     * @return
     */
    public List<AllUserInfoParam> allUserList() {
        List<Admin> adminList = adminService.list(Wrappers.lambdaQuery(Admin.class).ne(Admin::getId, 1));
        PermissionToCheck();
        List<Admin> adminList = adminService.list(Wrappers.lambdaQuery(Admin.class).ne(Admin::getId, SecurityUtils.getUserId()));
        List<Client> clientList = clientService.list();

        List<AllUserInfoParam> list = new ArrayList<>();
@@ -199,13 +216,45 @@
    }


    public List<AllUserInfoParam> permissionInfo(Long fileId) {
        PermissionToCheck();
        List<PlatformFileUser> list = platformFileUserService.list(Wrappers.lambdaQuery(PlatformFileUser.class).eq(PlatformFileUser::getFileId, fileId));
        List<Long> userIds = list.stream().map(PlatformFileUser::getUserId).collect(Collectors.toList());
        List<Admin> admins = adminService.list(Wrappers.lambdaQuery(Admin.class).in(Admin::getId, userIds));
        List<Client> clients = clientService.list(Wrappers.lambdaQuery(Client.class).in(Client::getId, userIds));


        List<AllUserInfoParam> infoList = new ArrayList<>();
        AllUserInfoParam allUserInfoParam;

        for (Admin admin : admins) {
            allUserInfoParam = new AllUserInfoParam();
            allUserInfoParam.setId(admin.getId());
            allUserInfoParam.setName(admin.getUsername());
            allUserInfoParam.setType("管理员");
            infoList.add(allUserInfoParam);
        }

        for (Client client : clients) {
            allUserInfoParam = new AllUserInfoParam();
            allUserInfoParam.setId(client.getId());
            allUserInfoParam.setName(client.getClientName());
            allUserInfoParam.setType("普通用户");
            infoList.add(allUserInfoParam);
        }

        return infoList;
    }


    /**
     * 权限校验
     * 权限超管校验
     */
    private void PermissionToCheck() {
        if (!AdministratorEnums.ADMIN.getCode().equals(SecurityUtils.getUserDetails().getAdministratorType())) {
            //超管
            //非超管
            throw new BusinessException("非超管不能操作");
        }
    }

}