package com.ld.igds.basic.controller;
|
|
import com.ld.igds.basic.manager.FileManager;
|
import com.ld.igds.data.PageResponse;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.RestController;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.net.URLEncoder;
|
|
/**
|
* 文件上传接口
|
*
|
* @author
|
*/
|
@Slf4j
|
@RestController
|
@RequestMapping("basic/file")
|
public class FileController {
|
|
@Autowired
|
private FileManager fileManager;
|
|
/**
|
* 长传 出入库图片
|
*
|
* @param file
|
* @param request
|
* @return
|
* @throws IOException
|
*/
|
@RequestMapping("/update-file")
|
public PageResponse<String> updateAreationPos(
|
@RequestParam(value = "file", required = true) MultipartFile file,
|
HttpServletRequest request) throws IOException {
|
|
String plateNum = request.getParameter("plateNum");
|
|
return fileManager.upLoadInoutImg(file, plateNum);
|
}
|
|
|
/**
|
* 下载从临时目录下载，需要项目名称和当前组织
|
*
|
* @param request
|
* @return
|
* @throws IOException
|
*/
|
@RequestMapping("/download-temp")
|
public int downloadTemp(
|
@RequestParam(value = "fileName", required = false) String fileName,
|
@RequestParam(value = "companyId", required = false) String companyId,
|
HttpServletRequest request, HttpServletResponse response) throws IOException {
|
|
if (StringUtils.isEmpty(fileName) || StringUtils.isEmpty(companyId)) {
|
response.sendError(404, "缺少下载参数条件，无法执行下载。");
|
return 404;
|
}
|
File file = fileManager.getTempFilePath(fileName, companyId);
|
|
if (!file.exists()) {
|
response.sendError(404, "没有获取到需要下载的文件，可能已被删除");
|
return 404;
|
}
|
|
|
response.setContentType("application/octet-stream");
|
response.setHeader("content-type", "application/octet-stream");
|
response.setHeader("Content-Length", String.valueOf(file.length()));
|
response.setHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode(fileName, "utf8"));
|
byte[] buffer = new byte[1024];
|
|
OutputStream os;
|
try (FileInputStream fis = new FileInputStream(file);
|
BufferedInputStream bis = new BufferedInputStream(fis)) {
|
os = response.getOutputStream();
|
int i = bis.read(buffer);
|
while (i != -1) {
|
os.write(buffer);
|
i = bis.read(buffer);
|
}
|
|
fis.close();
|
bis.close();
|
os.close();
|
os.flush();
|
} catch (Exception e) {
|
response.sendError(500, "文件下载出错：" + e.getMessage());
|
|
log.error("------------文件下载失败--{}", e);
|
return 500;
|
}
|
|
return 200;
|
}
|
|
}
|
