更新仓内抓拍

jiazx0107@163.com

2023-10-22 dfd793f14e51c48c3322f1b36f543179043bd45d

 igds-basic/src/main/java/com/ld/igds/basic/controller/FileController.java

@@ -2,16 +2,20 @@

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;

@@ -25,81 +29,145 @@
@RequestMapping("basic/file")
public class FileController {

    @Autowired
    private FileManager fileManager;
   @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 {
   /**
    * 长传 出入库图片
    *
    * @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");
      String plateNum = request.getParameter("plateNum");

        return fileManager.upLoadInoutImg(file, 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 {

    /**
     * 下载从临时目录下载，需要项目名称和当前组织
     *
     * @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 (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;
      }

        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);
         }

        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];
         fis.close();
         bis.close();
         os.close();
         os.flush();
      } catch (Exception e) {
         response.sendError(500, "文件下载出错：" + e.getMessage());

        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);
            }
         log.error("------------文件下载失败--{}", e);
         return 500;
      }

            fis.close();
            bis.close();
            os.close();
            os.flush();
        } catch (Exception e) {
            response.sendError(500, "文件下载出错：" + e.getMessage());
      return 200;
   }

            log.error("------------文件下载失败--{}", e);
            return 500;
        }
   /**
    * 文件流获取图片显示到页面--针对仓内抓拍
    *
    * @param fileName
    * @param timeStr
    *            抓拍的年月日时分秒 yyyyMMddHHmmss
    * @param response
    * @return
    * @throws IOException
    */
   @RequestMapping(value = "/view-snap-depot", method = RequestMethod.GET, produces = { "application/vnd.ms-excel;charset=UTF-8" })
   public String getDepotSnap(String fileName, String timeStr,
         HttpServletResponse response) throws IOException {

        return 200;
    }
      // 设置返回内容格式
      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();
         }
      }
   }

}