package org.jeecg.modules.doc.controller;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.io.IOUtils;
|
import org.apache.commons.lang3.StringUtils;
|
import org.apache.shiro.SecurityUtils;
|
import org.jeecg.common.api.vo.Result;
|
import org.jeecg.common.constant.CommonConstant;
|
import org.jeecg.common.system.util.JwtUtil;
|
import org.jeecg.common.system.vo.LoginUser;
|
import org.jeecg.common.util.TokenUtils;
|
import org.jeecg.modules.doc.api.JeecgSystemClient;
|
import org.jeecg.modules.doc.component.DocFileDealComp;
|
import org.jeecg.modules.doc.component.Downloader;
|
import org.jeecg.modules.doc.component.LocalStorageDownloader;
|
import org.jeecg.modules.doc.constant.Constant;
|
import org.jeecg.modules.doc.dto.BatchDownloadFileDTO;
|
import org.jeecg.modules.doc.dto.DownloadFileDTO;
|
import org.jeecg.modules.doc.dto.PreviewDTO;
|
import org.jeecg.modules.doc.dto.UploadFileDTO;
|
import org.jeecg.modules.doc.entity.*;
|
import org.jeecg.modules.doc.service.*;
|
import org.jeecg.modules.doc.util.MimeUtils;
|
import org.jeecg.modules.doc.util.UFOPUtils;
|
import org.jeecg.modules.doc.vo.*;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.Cookie;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import javax.servlet.http.Part;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.io.OutputStream;
|
import java.io.UnsupportedEncodingException;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
@Slf4j
|
@RestController
|
@RequestMapping("/fileTransfer")
|
public class FileTransferController {
|
|
@Autowired
|
private IFiletransferService filetransferService;
|
|
@Autowired
|
private DocFileDealComp docFileDealComp;
|
@Autowired
|
private IDocFilePathService filePathService;
|
|
@Autowired
|
private IDocFileService docFileService;
|
|
@Autowired
|
private JeecgSystemClient jeecgSystemClient;
|
|
@Autowired
|
private IUserStorageFlowService userStorageFlowService;
|
|
@Autowired
|
private IDocOperationHisService operationHisService;
|
|
@ApiOperation( value = "极速上传", notes = "校验文件MD5,存在则跳过上传,否则调用post上传接口")
|
@GetMapping("uploadfile")
|
public Result<UploadFileVo> uploadFileSpeed(UploadFileDTO uploadFileDto) {
|
LoginUser loginUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
|
UploadFileVo uploadFileVo = filetransferService.uploadFileSpeed(uploadFileDto);
|
operationHisService.saveHis(loginUser, uploadFileDto, Constant.OPERATION_UPLOAD);
|
return Result.OK(uploadFileVo);
|
}
|
|
|
@ApiOperation( value = "文件上传", notes = "真正的上传文件接口")
|
@PostMapping("/uploadfile")
|
public Result<UploadFileVo> uploadFile(HttpServletRequest request, UploadFileDTO uploadFileDto) {
|
filetransferService.uploadFile(request, uploadFileDto);
|
LoginUser loginUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
|
// response.setHeader("Access-Control-Allow-Origin", "*");
|
UploadFileVo uploadFileVo = new UploadFileVo();
|
// operationHisService.saveHis(loginUser, uploadFileDto, Constant.OPERATION_UPLOAD);
|
return Result.OK(uploadFileVo);
|
}
|
|
@PostMapping("/encryptUpload")
|
public DocFile encryptUpload(@RequestBody MultipartFile file, String identifier) throws IOException {
|
|
return docFileService.saveFile(file, identifier);
|
}
|
|
@GetMapping("/encryptDownload")
|
public void encryptDownload(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, String userFileId, String userToken) {
|
log.info("加密下载————————————————————————————————userFileId:"+userFileId);
|
DocFilePath userFile = filePathService.getById(userFileId);
|
String fileName = userFile.getFileName() + "." + userFile.getExtendName();
|
try {
|
fileName = new String(fileName.getBytes("utf-8"), "ISO-8859-1");
|
} catch (UnsupportedEncodingException e) {
|
e.printStackTrace();
|
}
|
DownloadFileDTO dto = new DownloadFileDTO();
|
dto.setUserFileId(userFileId);
|
dto.setUserToken(userToken);
|
httpServletResponse.addHeader("Content-Disposition", "attachment;fileName=" + fileName);// 设置文件名
|
|
filetransferService.downloadFile(httpServletResponse, dto);
|
}
|
|
@ApiOperation("下载文件")
|
@GetMapping("/downloadfile")
|
public void downloadFile(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, DownloadFileDTO downloadFileDTO) {
|
Result<Map<String, String>> byToken = jeecgSystemClient.getUserSectionInfoByToken(downloadFileDTO.getUserToken());
|
|
httpServletResponse.setContentType("application/force-download");// 设置强制下载不打开
|
DocFilePath userFile = filePathService.getById(downloadFileDTO.getUserFileId());
|
String fileName = "";
|
if (userFile.getIsDir() == Constant.IS_DIR) {
|
fileName = userFile.getFileName() + ".zip";
|
} else {
|
fileName = userFile.getFileName() + "." + userFile.getExtendName();
|
|
}
|
try {
|
fileName = new String(fileName.getBytes("utf-8"), "ISO-8859-1");
|
} catch (UnsupportedEncodingException e) {
|
e.printStackTrace();
|
}
|
|
httpServletResponse.addHeader("Content-Disposition", "attachment;fileName=" + fileName);// 设置文件名
|
operationHisService.saveHis(byToken.getResult(), userFile, Constant.OPERATION_DOWNLOAD);
|
filetransferService.downloadFile(httpServletResponse, downloadFileDTO);
|
}
|
|
|
@ApiOperation(value = "批量下载文件", notes = "批量下载文件", tags = {"filetransfer"})
|
@GetMapping(value = "/batchDownloadFile")
|
@ResponseBody
|
public void batchDownloadFile(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, BatchDownloadFileDTO batchDownloadFileDTO) {
|
Result<Map<String, String>> byToken = jeecgSystemClient.getUserSectionInfoByToken(batchDownloadFileDTO.getUserToken());
|
String files = batchDownloadFileDTO.getUserFileIds();
|
String[] userFileIdStrs = files.split(",");
|
List<String> userFileIds = new ArrayList<>();
|
for(String userFileId : userFileIdStrs) {
|
DocFilePath userFile = filePathService.getById(userFileId);
|
if (userFile.getIsDir() == Constant.IS_NOT_DIR) {
|
userFileIds.add(userFileId);
|
} else {
|
QiwenFile qiwenFile = new QiwenFile(userFile.getFilePath(), userFile.getFileName(), true);
|
List<DocFilePath> userFileList = filePathService.selectUserFileByLikeRightFilePath(qiwenFile.getPath());
|
List<String> userFileIds1 = userFileList.stream().map(DocFilePath::getPathId).collect(Collectors.toList());
|
userFileIds.add(userFile.getPathId());
|
userFileIds.addAll(userFileIds1);
|
}
|
operationHisService.saveHis(byToken.getResult(), userFile, Constant.OPERATION_DOWNLOAD);
|
}
|
DocFilePath userFile = filePathService.getById(userFileIdStrs[0]);
|
httpServletResponse.setContentType("application/force-download");// 设置强制下载不打开
|
Date date = new Date();
|
String fileName = String.valueOf(date.getTime());
|
httpServletResponse.addHeader("Content-Disposition", "attachment;fileName=" + fileName + ".zip");// 设置文件名
|
filetransferService.downloadUserFileList(httpServletResponse, userFile.getFilePath()+QiwenFile.separator, fileName, userFileIds, batchDownloadFileDTO.getUserToken());
|
}
|
|
|
@ApiOperation(value="预览文件", notes="用于文件预览", tags = {"fileTransfer"})
|
@GetMapping("/preview")
|
public void preview(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, PreviewDTO previewDTO) throws IOException {
|
|
// if (previewDTO.getPlatform() != null && previewDTO.getPlatform() == 2) {
|
// filetransferService.previewPictureFile(httpServletResponse, previewDTO);
|
// return ;
|
// }
|
// String token = "";
|
// if (StringUtils.isNotEmpty(previewDTO.getToken())) {
|
// token = previewDTO.getToken();
|
// } else {
|
// Cookie[] cookieArr = httpServletRequest.getCookies();
|
// if (cookieArr != null) {
|
// for (Cookie cookie : cookieArr) {
|
// if ("token".equals(cookie.getName())) {
|
// token = cookie.getValue();
|
// }
|
// }
|
// }
|
// }
|
|
DocFilePath userFile = filePathService.getById(previewDTO.getUserFileId());
|
// boolean authResult = docFileDealComp.checkAuthDownloadAndPreview(previewDTO.getShareBatchNum(),
|
// previewDTO.getExtractionCode(),
|
// token,
|
// previewDTO.getUserFileId(),
|
// previewDTO.getPlatform());
|
//
|
// if (!authResult) {
|
// log.error("没有权限预览！！！");
|
// return;
|
// }
|
|
String fileName = userFile.getFileName() + "." + userFile.getExtendName();
|
try {
|
fileName = new String(fileName.getBytes("utf-8"), "ISO-8859-1");
|
} catch (UnsupportedEncodingException e) {
|
e.printStackTrace();
|
}
|
|
httpServletResponse.addHeader("Content-Disposition", "fileName=" + fileName);// 设置文件名
|
String mime = MimeUtils.getMime(userFile.getExtendName());
|
httpServletResponse.setHeader("Content-Type", mime);
|
|
DocFile fileBean = docFileService.getById(userFile.getFileId());
|
if ((UFOPUtils.isVideoFile(userFile.getExtendName()) || "mp3".equalsIgnoreCase(userFile.getExtendName()) || "flac".equalsIgnoreCase(userFile.getExtendName()))
|
&& !"true".equals(previewDTO.getIsMin())) {
|
//获取从那个字节开始读取文件
|
String rangeString = httpServletRequest.getHeader("Range");
|
int start = 0;
|
if (StringUtils.isNotBlank(rangeString)) {
|
start = Integer.valueOf(rangeString.substring(rangeString.indexOf("=") + 1, rangeString.indexOf("-")));
|
}
|
|
Downloader downloader = new LocalStorageDownloader();
|
DownloadFile downloadFile = new DownloadFile();
|
downloadFile.setFileUrl(fileBean.getFileUrl());
|
Range range = new Range();
|
range.setStart(start);
|
range.setLength(1024 * 1024 * 1);
|
downloadFile.setRange(range);
|
InputStream inputStream = downloader.getInputStream(downloadFile);
|
|
OutputStream outputStream = httpServletResponse.getOutputStream();
|
try {
|
|
//返回码需要为206，代表只处理了部分请求，响应了部分数据
|
httpServletResponse.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);
|
// 每次请求只返回1MB的视频流
|
|
httpServletResponse.setHeader("Accept-Ranges", "bytes");
|
//设置此次相应返回的数据范围
|
httpServletResponse.setHeader("Content-Range", "bytes " + start + "-" + (fileBean.getFileSize() - 1) + "/" + fileBean.getFileSize());
|
IOUtils.copy(inputStream, outputStream);
|
|
|
} finally {
|
IOUtils.closeQuietly(inputStream);
|
IOUtils.closeQuietly(outputStream);
|
if (downloadFile.getOssClient() != null) {
|
downloadFile.getOssClient().shutdown();
|
}
|
}
|
|
} else {
|
filetransferService.previewFile(httpServletResponse, previewDTO);
|
}
|
|
}
|
|
|
}
|
