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.RequestMethod;
|
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;
|
}
|
|
/**
|
* 文件流获取图片显示到页面--针对仓内抓拍
|
*
|
* @param fileName
|
* @param timeStr
|
* 抓拍的年月日时分秒 yyyyMMdd
|
* @param response
|
* @return
|
* @throws IOException
|
*/
|
@RequestMapping(value = "/get-snap-depot", method = RequestMethod.GET, produces = { "application/vnd.ms-excel;charset=UTF-8" })
|
public String getDepotSnap(String fileName, String timeStr,
|
HttpServletResponse response) throws IOException {
|
|
// 设置返回内容格式
|
response.setContentType("image/jpeg/jpg/png/gif/bmp/tiff/svg");
|
|
String filePath;
|
// 创建一个输入流
|
InputStream in = null;
|
// 创建输出流
|
OutputStream os = null;
|
try {
|
if (null == fileName || null == timeStr) {
|
filePath = fileManager.getFailImg();
|
}else{
|
filePath = fileManager.getSnapFilePath(fileName, timeStr);
|
|
File file = new File(filePath);
|
|
if(file.exists()){
|
log.debug("----抓拍图片地址-------{}", filePath);
|
}else{
|
filePath = fileManager.getFailImg();
|
log.debug("----未获取到图纸，默认地址-------{}", filePath);
|
}
|
}
|
|
// 用该文件创建一个输入流
|
in = new FileInputStream(filePath);
|
// 创建输出流
|
os = response.getOutputStream();
|
byte[] b = new byte[1024];
|
while (in.read(b) != -1) {
|
os.write(b);
|
}
|
in.close();
|
os.close();
|
|
return null;
|
|
} catch (Exception e) {
|
e.printStackTrace();
|
return null;
|
} finally {
|
if (null != in) {
|
in.close();
|
}
|
if (null != os) {
|
os.close();
|
}
|
}
|
}
|
|
}
|
