changes

liuhaonan

2022-10-21 f214709cf1d896e7c9743f206b7a708f020e5322

 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,9 +8,14 @@
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;
import com.sandu.ximon.admin.security.SecurityUtils;
import com.sandu.ximon.dao.bo.PlatformFileListBo;
import com.sandu.ximon.dao.domain.Admin;
import com.sandu.ximon.dao.domain.Client;
import com.sandu.ximon.dao.domain.PlatformFile;
import com.sandu.ximon.dao.domain.PlatformFileUser;
import com.sandu.ximon.dao.enums.AdministratorEnums;
@@ -19,7 +25,9 @@
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
@@ -32,6 +40,9 @@

    private final AliOssFileServiceImpl aliOssFileService;
    private final PlatformFileUserService platformFileUserService;
    private final AdminService adminService;
    private final ClientService clientService;
    private final Snowflake snowflake;

    /**
     * 文件上传
@@ -43,7 +54,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() + "]");
@@ -52,6 +64,7 @@

    /**
     * 修改文件名称
     *
     * @param param
     * @return
     */
@@ -70,6 +83,7 @@

    /**
     * 删除文件
     *
     * @param param
     * @return
     */
@@ -79,38 +93,56 @@
            throw new BusinessException("删除时参数不能为空!");
        }
        List<PlatformFile> platformFiles = listByIds(param.getFileIds());
        if (platformFiles == null && platformFiles.size() == 0) {
        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());
    }


    /**
     * 平台文件资源列表
     *
     * @param vo
     * @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());
            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;
    }

    /**
     * 下载文件
     *
     * @param fileId
     * @return
     */
    public String downFile(Long fileId) {
        if(!AdministratorEnums.ADMIN.getCode().equals(SecurityUtils.getUserDetails().getAdministratorType())){
        if (!AdministratorEnums.ADMIN.getCode().equals(SecurityUtils.getUserDetails().getAdministratorType())) {
            PlatformFileUser one = platformFileUserService.getOne(Wrappers.lambdaQuery(PlatformFileUser.class)
                    .eq(PlatformFileUser::getUserId, SecurityUtils.getUserId()).eq(PlatformFileUser::getFileId, fileId));
            if (one == null) {
@@ -122,16 +154,107 @@
    }


//    public boolean  setPermission(){}
    /**
     * 设置文件浏览权限
     *
     * @param platformFilePermissionParam
     * @return
     */
    public String setPermission(PlatformFilePermissionParam platformFilePermissionParam) {
        PermissionToCheck();
        //删除旧关系
        platformFileUserService.remove(Wrappers.lambdaQuery(PlatformFileUser.class)
                .eq(PlatformFileUser::getFileId, platformFilePermissionParam.getFileId()));

        //添加新关系
        List<PlatformFileUser> list = new ArrayList<>();
        PlatformFileUser bean;
        for (Long userId : platformFilePermissionParam.getUserIdList()) {
            bean = new PlatformFileUser();
            bean.setFileId(platformFilePermissionParam.getFileId());
            bean.setUserId(userId);
            list.add(bean);
        }

        if (platformFileUserService.saveBatch(list)) {
            return "关系存储成功！";
        } else {
            throw new BusinessException("关系存储失败！");
        }
    }

    /**
     * 返回所有用户
     *
     * @return
     */
    public List<AllUserInfoParam> allUserList() {
        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<>();
        AllUserInfoParam allUserInfoParam;

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

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

        return list;
    }


    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("非超管不能操作");
        }
    }

}