InputStream转MultipartFile

落日映苍穹つ 2023-10-10 13:48 22阅读 0赞

import org.apache.commons.fileupload.FileItem;
    import org.apache.commons.fileupload.FileItemFactory;
    import org.apache.commons.fileupload.disk.DiskFileItemFactory;
    import org.springframework.web.multipart.MultipartFile;
    import org.springframework.web.multipart.commons.CommonsMultipartFile;
    import org.springframework.http.MediaType;
    
    //调用放    originalFilename: 文件名  123.jpg需要带后缀
        MultipartFile multipartFile = getMultipartFile(inputStream, originalFilename);
    
     /**
         * 获取封装得MultipartFile
         *
         * @param inputStream inputStream
         * @param fileName    fileName
         * @return MultipartFile
         */
        public MultipartFile getMultipartFile(InputStream inputStream, String fileName) {
        
            FileItem fileItem = createFileItem(inputStream, fileName);
            //CommonsMultipartFile是feign对multipartFile的封装，但是要FileItem类对象
            return new CommonsMultipartFile(fileItem);
        }
    
    
        /**
         * FileItem类对象创建
         *
         * @param inputStream inputStream
         * @param fileName    fileName
         * @return FileItem
         */
        public FileItem createFileItem(InputStream inputStream, String fileName) {
        
            FileItemFactory factory = new DiskFileItemFactory(16, null);
            String textFieldName = "file";
            FileItem item = factory.createItem(textFieldName, MediaType.MULTIPART_FORM_DATA_VALUE, true, fileName);
            int bytesRead = 0;
            byte[] buffer = new byte[8192];
            OutputStream os = null;
            //使用输出流输出输入流的字节
            try {
        
                os = item.getOutputStream();
                while ((bytesRead = inputStream.read(buffer, 0, 8192)) != -1) {
        
                    os.write(buffer, 0, bytesRead);
                }
                inputStream.close();
            } catch (IOException e) {
        
                log.error("Stream copy exception", e);
                throw new IllegalArgumentException("文件上传失败");
            } finally {
        
                if (os != null) {
        
                    try {
        
                        os.close();
                    } catch (IOException e) {
        
                        log.error("Stream close exception", e);
    
                    }
                }
                if (inputStream != null) {
        
                    try {
        
                        inputStream.close();
                    } catch (IOException e) {
        
                        log.error("Stream close exception", e);
                    }
                }
            }
    
            return item;
        }