changes

liuhaonan

2022-09-16 23f5a0a8ae0612faa8b3bfc87ac6d0f32b4936aa

 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;
@@ -20,12 +21,12 @@
import com.sandu.ximon.dao.mapper.PlatformFileMapper;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections.ArrayStack;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

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

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

    /**
     * 文件上传
@@ -51,7 +53,8 @@
        PermissionToCheck();
        FileUploadDto fileUploadDto = aliOssFileService.uploadFile(file);
        PlatformFile platformFile = new PlatformFile();
        platformFile.setFileName(fileUploadDto.getFileName());
        platformFile.setFileId(snowflake.nextId());
        platformFile.setFileName(fileUploadDto.getOriginName());
        platformFile.setFileSize(fileUploadDto.getFileSize());
        platformFile.setFileUrl(fileUploadDto.getFileUrl());
        log.error("超管进行文件上传[ 文件名:" + fileUploadDto.getFileName() + ", 文件大小: " + fileUploadDto.getFileSize() + ", url: " + fileUploadDto.getFileUrl() + "]");
@@ -92,7 +95,15 @@
        if (platformFiles == null && platformFiles.size() == 0) {
            throw new BusinessException("未找到文件!");
        }
        return removeByIds(platformFiles);

        //获取关系
        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 removeByIds(param.getFileIds());
    }


@@ -103,7 +114,7 @@
     * @param keyword
     * @return
     */
    public Object fileList(BaseConditionVO vo, String keyword) {
    public List<PlatformFileListBo> fileList(BaseConditionVO vo, String keyword) {
        List<PlatformFileListBo> platformFileListBos;
        if (AdministratorEnums.ADMIN.getCode().equals(SecurityUtils.getUserDetails().getAdministratorType())) {
            PageHelper.startPage(vo.getPageNo(), vo.getPageSize());
@@ -141,6 +152,7 @@
     * @return
     */
    public String setPermission(PlatformFilePermissionParam platformFilePermissionParam) {
        PermissionToCheck();
        //删除旧关系
        platformFileUserService.remove(Wrappers.lambdaQuery(PlatformFileUser.class)
                .eq(PlatformFileUser::getFileId, platformFilePermissionParam.getFileId()));
@@ -164,10 +176,12 @@

    /**
     * 返回所有用户
     *
     * @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<>();
@@ -193,13 +207,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("非超管不能操作");
        }
    }

}