springboot

Java操作Zip文件、inputstream转为multipa

2021-10-14  本文已影响0人  小波同学

一、CommonsMultipartFile

<dependency>
    <groupId>commons-fileupload</groupId>
    <artifactId>commons-fileupload</artifactId>
    <version>1.3.3</version>
</dependency>
@Test
public void testOSSServiceImport(){
    File file = new File("test.png");
    DiskFileItem fileItem = (DiskFileItem) new DiskFileItemFactory().createItem("file",
            MediaType.ALL_VALUE, true, file.getName());

    try (InputStream input = new FileInputStream(file); OutputStream os = fileItem.getOutputStream()) {
        IOUtils.copy(input, os);
    } catch (Exception e) {
        throw new IllegalArgumentException("Invalid file: " + e, e);
    }

    MultipartFile multi = new CommonsMultipartFile(fileItem);

}

二、mockFile

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-test</artifactId>
</dependency>
import org.springframework.mock.web.MockMultipartFile;

MultipartFile file = new MockMultipartFile(name,name, MediaType.MULTIPART_FORM_DATA_VALUE, inputStream);

三、从网络url中操作zipFile

20211013_7935015_001_15000001903734_1634117754785_0.pdf 8043432110131215777753
20211013_7935016_008_15000001903734_1634117754402_0.pdf 8043432110131215777753

步骤:

<dependency>
    <groupId>commons-io</groupId>
    <artifactId>commons-io</artifactId>
    <version>2.11.0</version>
</dependency>

<dependency>
    <groupId>commons-fileupload</groupId>
    <artifactId>commons-fileupload</artifactId>
    <version>1.4</version>
</dependency>
public class ZipFileUtils {

    private static final Logger logger = LoggerFactory.getLogger(ZipFileUtils.class);

    /**
     * 获取网络文件流
     * @param urlStr 网络文件url
     * @return 返回输入流
     * @throws IOException IO异常
     */
    private static InputStream downLoadFromUrl(String urlStr) throws IOException {
        URL url = new URL(urlStr);
        HttpURLConnection conn = (HttpURLConnection)url.openConnection();
        //设置超时间为3秒
        conn.setConnectTimeout(3*1000);
        //防止屏蔽程序抓取而返回403错误
        conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
        //得到输入流
        return conn.getInputStream();
    }

    /**
     * 获取网络文件流中的内容,一次读一行,并将空格拆开,返回
     * @param urlStr 网络文件url
     * @return 返回pdf文件名和银行流水号映射集合
     * @throws IOException IO异常
     */
    public static Map<String,List<String>> downLoadFileContentFromUrl(String urlStr) throws IOException {
        Map<String, List<String>> resultMap = new HashMap<>();
        InputStream inputStream = null;
        BufferedReader br = null;
        try {
            //得到输入流
            inputStream = downLoadFromUrl(urlStr);
            //构造一个BufferedReader类来读取文件
            br = new BufferedReader(new InputStreamReader(inputStream, StandardCharsets.UTF_8));
            String line = null;
            //使用readLine方法,一次读一行
            while ((line = br.readLine()) != null) {
                //通过空格进行切分
                String[] split = line.split(" ");
                if(split.length >= 2){
                    List<String> pdfNameList = resultMap.get(split[1]);
                    if(CollectionUtils.isEmpty(pdfNameList)){
                        pdfNameList = new ArrayList<>();
                        pdfNameList.add(split[0]);
                        resultMap.put(split[1],pdfNameList);
                    }else{
                        pdfNameList.add(split[0]);
                    }
                }
            }
        } finally {
            //关闭流
            if(br != null){
                br.close();
            }
            if(inputStream != null){
                inputStream.close();
            }
        }
        return resultMap;
    }

    /**
     * 根据pdfName名称,从网络中获取Zip中的pdf文件InputStream
     * @param urlStr Zip文件的下载url
     * @param pdfName pdf名称
     * @return 返回输入流
     * @throws IOException IO异常
     */
    public static InputStream downLoadZipFileFromUrl(String urlStr,String pdfName) throws IOException {
        //声明字节输出数组
        ByteArrayOutputStream byteOut = null;
        ZipInputStream zipIn = null;
        InputStream inputStream = null;
        try {
            //得到输入流
            zipIn = new ZipInputStream(downLoadFromUrl(urlStr));
            ZipEntry entry = null;
            while ((entry = zipIn.getNextEntry()) != null) {
                if(pdfName.equals(entry.getName())) {
                    //初始化字节输出数组
                    byteOut = new ByteArrayOutputStream();
                    IOUtils.copy(zipIn, byteOut);
                    inputStream = new ByteArrayInputStream(byteOut.toByteArray());
                }
            }
            return inputStream;
        } finally {
            //关闭流
            if(byteOut != null){
                byteOut.close();
            }
            if(zipIn != null){
                zipIn.close();
            }
        }
    }

    /**
     * 根据pdfName名称,从网络中获取Zip中的pdf文件的MultipartFile
     * @param urlStr Zip文件的下载url
     * @param pdfName Zip文件的下载url
     * @return MultipartFile
     * @throws IOException IO异常
     */
    public static MultipartFile getMultipartFileFromUrl(String urlStr, String pdfName) throws IOException {
        String fileName = "Xxx文件.pdf";
        MultipartFile multipartFile = null;
        InputStream inputStream = downLoadZipFileFromUrl(urlStr, pdfName);
        if(inputStream != null){
            File file = new File(fileName);
            DiskFileItem fileItem = (DiskFileItem) new DiskFileItemFactory().createItem("file",
                    MediaType.ALL_VALUE, true, file.getName());

            try (OutputStream os = fileItem.getOutputStream()) {
                IOUtils.copy(inputStream, os);
            } catch (Exception e) {
                logger.error("输入流转DiskFileItem异常");
                throw new IllegalArgumentException("Invalid file: " + e, e);
            }

            multipartFile = new CommonsMultipartFile(fileItem);
        }
        return multipartFile;
    }
}

参考:
https://www.cnblogs.com/yadongliang/p/13578889.html

上一篇 下一篇

猜你喜欢

热点阅读