1. package com.hikvision.idatafusion.psbddatachange.common.util;
    3. import com.hikvision.idatafusion.exception.BusinessException;
    4. import com.hikvision.idatafusion.psbddatachange.common.constant.PsbddatachangeErrorCode;
    5. import lombok.extern.slf4j.Slf4j;
    6. import org.apache.commons.lang3.StringUtils;
    7. import org.slf4j.Logger;
    8. import org.slf4j.LoggerFactory;
    9. import org.springframework.core.io.ClassPathResource;
    11. import javax.servlet.http.HttpServletResponse;
    12. import java.io.*;
    13. import java.net.URLEncoder;
    14. import java.nio.charset.StandardCharsets;
    16. /**
    17.  * 文件工具类
    18.  *
    19.  * @author 韩仁松
    20.  * @since businessV1.0.0
    21.  */
    23. @Slf4j
    24. public class FileUtil {
    26.     private final static Logger logger = LoggerFactory.getLogger(FileUtil.class);
    28.     private final static int DEFAULT_BUFFER_SIZE = 1024;
    29.     private final static String LEFT_SLASH = "/";
    32.     /**
    33.      * 读取UTF-8资源文件内容
    34.      *
    35.      * @param name  资源文件的路径
    36.      * @param clazz 加载资源文件的class
    37.      * @return 返回读取的资源文件内容，异常时返回null
    38.      */
    39.     public static String getUTF8ResourceAsString(String name, Class clazz) {
    40.         return getResourceAsString(name, clazz, StandardCharsets.UTF_8.name());
    41.     }
    43.     /**
    44.      * 读取资源文件内容
    45.      *
    46.      * @param name        资源文件的路径
    47.      * @param clazz       加载资源文件的class
    48.      * @param charsetName 加载文件的字符集
    49.      * @return 返回读取的资源文件内容，异常时返回null
    50.      */
    51.     private static String getResourceAsString(String name, Class clazz, String charsetName) {
    52.         if (!name.startsWith(LEFT_SLASH)) {
    53.             name = LEFT_SLASH + name;
    54.         }
    55.         try (InputStream in = clazz.getResourceAsStream(name)) {
    56.             return streamToString(in, charsetName);
    57.         } catch (IOException e) {
    58.             logger.error("load resource file error: {}, and load class: {}", name, clazz.getName());
    59.             return null;
    60.         }
    61.     }
    62.     /**
    63.      * InputStream转化为String
    64.      *
    65.      * @param in          InputStream数据流
    66.      * @param charsetName 加载文件的字符集
    67.      * @return 返回转化后的字符串
    68.      * @throws IOException 读取异常转抛
    69.      */
    70.     private static String streamToString(InputStream in, String charsetName) throws IOException {
    71.         try (ByteArrayOutputStream result = new ByteArrayOutputStream()) {
    72.             byte[] buffer = new byte[DEFAULT_BUFFER_SIZE];
    73.             int length;
    74.             while ((length = in.read(buffer)) != -1) {
    75.                 result.write(buffer, 0, length);
    76.             }
    77.             return result.toString(charsetName);
    78.         } catch (IOException e) {
    79.             throw e;
    80.         }
    81.     }
    85.     /**
    86.      * 读取资源文件内容
    87.      *
    88.      * @param name        资源文件的路径
    89.      * @return 返回读取的资源文件内容，异常时返回null
    90.      */
    91.     public static String getResourceAsStringByPath(String name) {
    92.         if (!name.startsWith(LEFT_SLASH)) {
    93.             name = LEFT_SLASH + name;
    94.         }
    95.         ClassPathResource resource = new ClassPathResource(name);
    96.         if(!resource.exists()){
    97.             resource = new ClassPathResource("/config" + name);
    98.         }
    99.         try (InputStream in = resource.getInputStream()) {
    100.             return streamToString(in, StandardCharsets.UTF_8.name());
    101.         } catch (IOException e) {
    102.             log.error("load resource file error: {},", name);
    103.             return null;
    104.         }
    105.     }
    107.     public static void inputStreamToFile(InputStream in, File file){
    108.         try {
    109.             OutputStream os = new FileOutputStream(file);
    110.             int bytes = 0;
    111.             byte[] buffer = new byte[1024];
    112.             while ((bytes = in.read(buffer, 0, buffer.length)) != -1){
    113.                 os.write(buffer, 0 ,bytes);
    114.             }
    115.             os.close();
    116.             in.close();
    117.         } catch (Exception e) {
    118.             e.printStackTrace();
    119.         }
    120.     }
    122.     public static boolean download(HttpServletResponse response, File file) {
    123.         String fileName = file.getName();
    124.         try {
    125.             fileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8.name());
    126.         } catch (UnsupportedEncodingException e) {
    127.             //log.error(PsbddatachangeLogHelper.message(PsbddatachangeErrorCode.FILE_READ_ERROR, "file name encoder error:"+ fileName), e);
    128.         }
    129.         response.setContentType("application/octet-stream");
    130.         response.setHeader("Content-Disposition", String.format("attachment;filename=%s", fileName));
    131.         response.setContentLength((int) file.length());
    132.         response.addHeader("Content-Length", String.valueOf(file.length()));
    134.         InputStream is = null;
    135.         OutputStream out = null;
    136.         byte[] bytes = new byte[2048];
    137.         int count = 0;
    138.         try {
    139.             is = new FileInputStream(file);
    140.             out = response.getOutputStream();
    141.             while ((count = is.read(bytes)) > 0) {
    142.                 out.write(bytes, 0, count);
    143.             }
    144.             return true;
    145.         } catch (IOException e) {
    146.             //log.error(PsbddatachangeLogHelper.message(PsbddatachangeErrorCode.FILE_WRITE_ERROR, "download file error:"+ file.getAbsolutePath()), e);
    147.             return false;
    148.         } finally {
    149.             try {
    150.                 if (is != null) {
    151.                     is.close();
    152.                 }
    153.                 if (out != null) {
    154.                     out.close();
    155.                 }
    156.                 if(file!=null && file.exists()) {
    157.                     file.delete();
    158.                 }
    159.             } catch (IOException e) {
    160.                 //log.error(PsbddatachangeLogHelper.message(PsbddatachangeErrorCode.A_FILE_CLOSE_ERROR, "i/o close failed"), e);
    161.             }
    162.         }
    163.     }
    165.     public static boolean download(HttpServletResponse response, InputStream is, String fileName) {
    166.         try {
    167.             fileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8.name());
    168.         } catch (UnsupportedEncodingException e) {
    170.         }
    171.         response.setContentType("application/octet-stream");
    172.         response.setHeader("Content-Disposition", String.format("attachment;filename=%s", fileName));
    174.         OutputStream out = null;
    175.         byte[] bytes = new byte[2048];
    176.         int count = 0;
    177.         try {
    178.             out = response.getOutputStream();
    179.             while ((count = is.read(bytes)) > 0) {
    180.                 out.write(bytes, 0, count);
    181.             }
    182.             out.flush();
    183.             return true;
    184.         } catch (IOException e) {
    185.             log.error(PsbddatachangeErrorCode.FILE_WRITE_ERROR, "download file error:"+fileName, e);
    186.             return false;
    187.         } finally {
    188.             try {
    189.                 if (is != null) {
    190.                     is.close();
    191.                 }
    192.                 if (out != null) {
    193.                     out.close();
    194.                 }
    195.             } catch (IOException e) {
    196.                 log.error(PsbddatachangeErrorCode.A_FILE_CLOSE_ERROR, "i/o close failed", e);
    197.             }
    198.         }
    199.     }
    201.     /**
    202.      * dirName  只有下载文件到本地的时候才有用
    203.      *
    204.      * @param response
    205.      */
    206.     public static boolean downloadV2(HttpServletResponse response, InputStream is, String fileName) {
    207.         try {
    208.             fileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8.name());
    209.         } catch (UnsupportedEncodingException e) {
    210.             //throw new BusinessException(PsbddatachangeErrorCode.IO_ERROR, "download_template_failed", e);
    211.         }
    212.         response.setContentType("application/octet-stream");
    213.         response.setHeader("Content-Disposition", String.format("attachment;filename=%s", fileName));
    215.         OutputStream out = null;
    216.         byte[] bytes = new byte[2048];
    217.         int count = 0;
    218.         try {
    219.             out = response.getOutputStream();
    220.             while ((count = is.read(bytes)) > 0) {
    221.                 out.write(bytes, 0, count);
    222.             }
    223.             out.flush();
    224.             return true;
    225.         } catch (IOException e) {
    226.             //throw new BusinessException(PsbddatachangeErrorCode.IO_ERROR, "download_template_failed", e);
    227.         } finally {
    228.             try {
    229.                 if (is != null) {
    230.                     is.close();
    231.                 }
    232.                 if (out != null) {
    233.                     out.close();
    234.                 }
    235.             } catch (IOException e) {
    236.                 //throw new BusinessException(PsbddatachangeErrorCode.A_FILE_CLOSE_ERROR, "download_template_failed", e);
    237.             }
    238.         }
    239.         return true;
    240.     }
    242.     public static String getResourceAsStringByClassPathResource(String path){
    243.         return getResourceAsStringByClassPathResource(path, StandardCharsets.UTF_8.name());
    244.     }
    246.     public static String getResourceAsStringByClassPathResource(String path,String charsetName){
    247.         if(StringUtils.isEmpty(path)) {
    248.             log.info(PsbddatachangeLogHelper.message("path is null..."));
    249.             return null;
    250.         }
    251.         if(StringUtils.isEmpty(charsetName)) {
    252.             log.info(PsbddatachangeLogHelper.message("charsetName is null..."));
    253.             return null;
    254.         }
    255.         ClassPathResource resource = new ClassPathResource(path);
    256.         if(!resource.exists()) {
    257.             log.info(PsbddatachangeLogHelper.message("resource not exists..."));
    258.             return null;
    259.         }
    260.         try {
    261.             return getStringByStream(resource.getInputStream(),charsetName);
    262.         } catch (IOException e) {
    263.             log.error(PsbddatachangeLogHelper.message("load resource file error: {} "), path);
    264.             return null;
    265.         }
    266.     }
    268.     public static String getStringByStream(InputStream in, String charsetName) throws IOException {
    269.         ByteArrayOutputStream result = new ByteArrayOutputStream();
    270.         try{
    271.             byte[] buffer = new byte[1024];
    272.             int length;
    273.             while ((length = in.read(buffer)) != -1) {
    274.                 result.write(buffer, 0, length);
    275.             }
    276.             return result.toString(charsetName);
    277.         } catch (IOException e) {
    278.             throw e;
    279.         }finally {
    280.             if(in!=null) {
    281.                 in.close();
    282.             }
    283.             if(result!=null) {
    284.                 result.close();
    285.             }
    286.         }
    287.     }
    289.     public static boolean download(HttpServletResponse response, File file, String fileName, boolean isDelete) {
    290.         try {
    291.             fileName = URLEncoder.encode(fileName, "UTF-8");
    292.         } catch (UnsupportedEncodingException e) {
    294.             log.error(PsbddatachangeLogHelper.message("file name encoder error:" + fileName), e);
    295.         }
    296.         response.setContentType("application/octet-stream;charset=utf-8");
    297.         response.setHeader("Content-Disposition", String.format("attachment;filename=%s", fileName));
    298.         response.setContentLength((int) file.length());
    299.         response.addHeader("Content-Length", String.valueOf(file.length()));
    301.         InputStream is = null;
    302.         OutputStream out = null;
    303.         byte[] bytes = new byte[2048];
    304.         int count = 0;
    305.         try {
    306.             is = new FileInputStream(file);
    307.             out = response.getOutputStream();
    308.             while ((count = is.read(bytes)) > 0) {
    309.                 out.write(bytes, 0, count);
    310.             }
    311.             out.flush();
    312. //            return true;
    313.         } catch (IOException e) {
    314.             log.error(PsbddatachangeLogHelper.message("download file error:" + file.getAbsolutePath()), e);
    315.             return false;
    316.         } finally {
    317.             try {
    318.                 if (is != null) {
    319.                     is.close();
    320.                 }
    321.                 if (out != null) {
    322.                     out.close();
    323.                 }
    324.                 if (file != null && file.exists() && isDelete) {
    325.                     file.delete();
    326.                 }
    327.             } catch (IOException e) {
    328.                 log.error(PsbddatachangeLogHelper.message("i/o close failed"), e);
    329.             }
    330.         }
    331.         return true;
    332.     }
    334.     public static boolean download(HttpServletResponse response, File file, boolean isDelete) {
    335.         return download(response, file, file.getName(), isDelete);
    336.     }
    337.     /**
    338.      * 获取文件名称后缀
    339.      *
    340.      * @param fileName 文件名
    341.      * @return String
    342.      */
    343.     public static String getSuffix(String fileName) {
    344.         if (fileName.contains(".")) {
    345.             return fileName.substring(fileName.lastIndexOf(".") + 1);
    346.         } else {
    347.             return "";
    348.         }
    349.     }
    351.     public static void downloadFileInClassPath(HttpServletResponse response, String fileNameInClassPath, String outFileName) {
    353.         ClassPathResource resource = new ClassPathResource("template/" + fileNameInClassPath);
    354.         try {
    355.             downloadV2(response, resource.getInputStream(),outFileName);
    356.         } catch (IOException e) {
    357.             throw new BusinessException(PsbddatachangeErrorCode.FILE_DOWNLOAD_ERROR, e);
    358.         }
    359.     }
    360. }