package com.cloudroam.manager.impl;
|
|
import cn.hutool.core.util.IdUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.cloudroam.common.LocalUser;
|
import com.cloudroam.common.enumeration.DeletedEnum;
|
import com.cloudroam.manager.SysFileManager;
|
import com.cloudroam.model.SysFileDO;
|
import com.cloudroam.model.UserDO;
|
import com.cloudroam.module.file.FileTypeEnum;
|
import com.cloudroam.module.file.FileUtil;
|
import com.cloudroam.service.SysFileService;
|
import com.cloudroam.util.DateUtils;
|
import com.cloudroam.util.MinioUtil;
|
import org.apache.tomcat.util.http.fileupload.IOUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.InputStream;
|
import java.net.URLEncoder;
|
import java.util.Date;
|
|
@Service
|
public class SysFileManagerImpl implements SysFileManager {
|
|
@Autowired
|
private MinioUtil minioUtil;
|
|
@Autowired
|
private SysFileService sysFileService;
|
|
|
@Override
|
@Transactional
|
public SysFileDO uploadFile(MultipartFile uploadfile, String bucket, String objectName) throws Exception {
|
|
// 目录
|
String catalog=objectName;
|
String folder=objectName+DateUtils.getCurrentDateCatalog();
|
// 原来文件
|
String originalFilename=uploadfile.getOriginalFilename();
|
// 拓展名
|
String ext = FileUtil.getFileExt(originalFilename);
|
// 新文件名
|
String newFileName=FileUtil.getNewFilename(ext);
|
// 文件的MD5，如果md5一样的话，则说明是重复性文件
|
String md5=FileUtil.getFileMD5(uploadfile.getBytes());
|
|
// 不加bucket的文件路径
|
String path=folder+ newFileName;
|
|
// 文件的完整路径
|
String fullpath=bucket+"/"+folder+ newFileName;
|
|
|
String uploadObjectName=folder+ newFileName;
|
long fileSize=uploadfile.getSize();
|
|
// 不管有没有bucket都执行创建bucket命令
|
minioUtil.createBucket(bucket);
|
minioUtil.uploadFile(uploadfile.getInputStream(), bucket, path);
|
|
|
UserDO userDO= LocalUser.getLocalUser();
|
SysFileDO sysFileDO=new SysFileDO();
|
sysFileDO.setId(IdUtil.simpleUUID());
|
sysFileDO.setBucket(bucket);
|
sysFileDO.setCatalog(catalog);
|
sysFileDO.setFolder(folder);
|
sysFileDO.setPath(path);
|
sysFileDO.setFullpath(fullpath);
|
sysFileDO.setType(FileTypeEnum.REMOTE.getValue());
|
sysFileDO.setName(originalFilename);
|
sysFileDO.setExtension(ext);
|
sysFileDO.setSize(fileSize);
|
sysFileDO.setMd5(md5);
|
|
// sysFileDO.setCreateUserId(String.valueOf(userDO.getId()));
|
// sysFileDO.setCreateUserName(userDO.getNickname());
|
sysFileDO.setCreateTime(new Date());
|
|
// sysFileDO.setUpdateUserId(String.valueOf(userDO.getId()));
|
// sysFileDO.setUpdateUserName(userDO.getNickname());
|
sysFileDO.setUpdateTime(new Date());
|
|
sysFileDO.setIsDeleted(DeletedEnum.DELETED_NO.getCode());
|
|
// 信息入库
|
sysFileService.save(sysFileDO);
|
|
// 信息返回
|
|
return sysFileService.getById(sysFileDO.getId());
|
}
|
|
@Override
|
public SysFileDO getFileByFullpath(String fullpath) {
|
QueryWrapper<SysFileDO> queryWrapper=new QueryWrapper<>();
|
queryWrapper.eq("is_deleted",DeletedEnum.DELETED_NO.getCode());
|
queryWrapper.eq("fullpath",fullpath);
|
|
return sysFileService.getOne(queryWrapper);
|
}
|
|
@Override
|
public void downloadFileWithFullpath(String fullpath, HttpServletResponse response) throws Exception {
|
|
SysFileDO fileDO = getFileByFullpath(fullpath);
|
String bucket=fullpath.substring(0,fullpath.indexOf("/"));
|
String objectName=fullpath.substring(fullpath.indexOf("/"));
|
|
InputStream stream = minioUtil.download(bucket, objectName);
|
ServletOutputStream output = response.getOutputStream();
|
// response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(objectName.substring(objectName.lastIndexOf("/") + 1), "UTF-8"));
|
response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileDO.getName(), "UTF-8"));
|
response.setContentType("application/octet-stream");
|
response.setCharacterEncoding("UTF-8");
|
IOUtils.copy(stream, output);
|
}
|
}
|
