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;
|
import com.sandu.common.file.FileUploadDto;
|
import com.sandu.common.file.impl.AliOssFileServiceImpl;
|
import com.sandu.common.object.BaseConditionVO;
|
import com.sandu.common.service.impl.BaseServiceImpl;
|
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;
|
import com.sandu.ximon.dao.mapper.PlatformFileMapper;
|
import lombok.AllArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
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
|
* @date 2022/9/14
|
*/
|
@Slf4j
|
@Service
|
@AllArgsConstructor
|
public class PlatformFileService extends BaseServiceImpl<PlatformFileMapper, PlatformFile> {
|
|
private final AliOssFileServiceImpl aliOssFileService;
|
private final PlatformFileUserService platformFileUserService;
|
private final AdminService adminService;
|
private final ClientService clientService;
|
private final Snowflake snowflake;
|
|
/**
|
* 文件上传
|
*
|
* @param file
|
* @return
|
*/
|
public boolean addProgramFile(MultipartFile file) {
|
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());
|
log.error("超管进行文件上传[ 文件名:" + fileUploadDto.getFileName() + ", 文件大小: " + fileUploadDto.getFileSize() + ", url: " + fileUploadDto.getFileUrl() + "]");
|
return save(platformFile);
|
}
|
|
/**
|
* 修改文件名称
|
*
|
* @param param
|
* @return
|
*/
|
public boolean updateFileName(EditFileParam param) {
|
PermissionToCheck();
|
if (param.getFileId() == null || param.getFileName().isEmpty()) {
|
throw new BusinessException("编辑时参数不能为空!");
|
}
|
PlatformFile byId = getById(param.getFileId());
|
if (byId == null) {
|
throw new BusinessException("未找到文件!");
|
}
|
byId.setFileName(param.getFileName());
|
return updateById(byId);
|
}
|
|
/**
|
* 删除文件
|
*
|
* @param param
|
* @return
|
*/
|
public boolean deleteFile(EditFileParam param) {
|
PermissionToCheck();
|
if (param.getFileIds() == null || param.getFileIds().size() == 0) {
|
throw new BusinessException("删除时参数不能为空!");
|
}
|
List<PlatformFile> platformFiles = listByIds(param.getFileIds());
|
if (platformFiles == null && platformFiles.size() == 0) {
|
throw new BusinessException("未找到文件!");
|
}
|
|
//获取关系
|
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 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);
|
} else {
|
PageHelper.startPage(vo.getPageNo(), vo.getPageSize());
|
platformFileListBos = baseMapper.fileLiet(keyword, SecurityUtils.getUserId());
|
}
|
return platformFileListBos;
|
}
|
|
/**
|
* 下载文件
|
*
|
* @param fileId
|
* @return
|
*/
|
public String downFile(Long fileId) {
|
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) {
|
throw new BusinessException("抱歉,你无权下载 !");
|
}
|
}
|
PlatformFile byId = getById(fileId);
|
return byId.getFileUrl();
|
}
|
|
|
/**
|
* 设置文件浏览权限
|
*
|
* @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("非超管不能操作");
|
}
|
}
|
|
}
|
