FileUtils.java 1.42 KB
package org.nafmii.utils;

import lombok.extern.slf4j.Slf4j;
import org.springframework.util.StringUtils;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;

@Slf4j
public class FileUtils {

    public static void downLoadFile(InputStream inputStream, String fileName){
        if (inputStream == null || StringUtils.isEmpty(fileName)){
            return;
        }
        try {
            HttpServletResponse response = RequestUtils.getResponse();
            response.reset();
            response.setContentType("multipart/form-data");
            response.setHeader("Content-Disposition", "attachment;filename=" + new String(fileName.getBytes(), "ISO-8859-1"));
            ServletOutputStream outputStream = response.getOutputStream();
            byte[] buff = new byte[1024];
            int length = 0;
            while ((length = inputStream.read(buff)) != -1) {
                outputStream.write(buff, 0, length);
            }
            if (outputStream != null) {
                outputStream.flush();
                outputStream.close();
            }
        } catch (Exception e) {
            log.error(e.getMessage());
        } finally {
            try {
                inputStream.close();
            } catch (IOException e) {
                log.error("关闭资源出错" + e.getMessage());
            }
        }
    }
}