根据图片url获取图片信息并返回MultipartFile

Posted 2021-06-23 G_whang

在图片上传的过程中，如果是三方接口对接，在传输图片的过程中，其第三方很可能会给一个图片的基本信息，比如：文件名称，类型，url等等，这个时候我们需要接收到他的数据展示到我们的业务平台上面，但是，业务平台展示图片的时候需要从文件管理系统中取数据，这个时候就需要我们在接收数据的时候就根据其url将其拉取下来，并传输至文件系统中

代码如下：

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItem;
import org.apache.tomcat.util.http.fileupload.IOUtils;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.commons.CommonsMultipartFile;

import java.io.*;
import java.net.URI;
import java.net.URL;
import java.nio.file.Files;

/**
 * 根据图片url 获取图片信息
 */
@Slf4j
public class ImageTest {


    public static void main(String[] args) throws Exception {

        System.out.println(getMultipartFile("https://img2.baidu.com/it/u=3206689113,2237998950&fm=26&fmt=auto&gp=0.jpg"));


    }

    /**
     * 根据url 拉取文件
     * @param url
     * @return
     * @throws Exception
     */
    public static File getFile(String url) throws Exception {
        //对本地文件命名
        String fileName = url.substring(url.lastIndexOf("."),url.length());
        File file = null;

        URL urlfile;
        InputStream inStream = null;
        OutputStream os = null;
        try {
            file = File.createTempFile("fwj_url", fileName);
            //下载
            urlfile = new URL(url);
            inStream = urlfile.openStream();
            os = new FileOutputStream(file);

            int bytesRead = 0;
            byte[] buffer = new byte[8192];
            while ((bytesRead = inStream.read(buffer, 0, 8192)) != -1) {
                os.write(buffer, 0, bytesRead);
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (null != os) {
                    os.close();
                }
                if (null != inStream) {
                    inStream.close();
                }

            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        return file;
    }

    /**
     * 根据url获取图片并转换为multipartFile类型
     * @param url
     * @return
     */
    private static MultipartFile getMultipartFile(String url) {
        FileInputStream inputStream = null;
        OutputStream outputStream = null;
        try {
            File file = getFile(url);
            System.out.println(file.toPath());
            FileItem fileItem = new DiskFileItem("formFieldName",//form表单文件控件的名字随便起
                    Files.probeContentType(file.toPath()),//文件类型
                    false, //是否是表单字段
                    file.getName(),//原始文件名
                    (int) file.length(),//Interger的最大值可以存储两部1G的电影
                    file.getParentFile());//文件会在哪个目录创建
            //为DiskFileItem的OutputStream赋值
            inputStream = new FileInputStream(file);
            outputStream = fileItem.getOutputStream();
            IOUtils.copy(inputStream, outputStream);
            return new CommonsMultipartFile(fileItem);
        } catch (Exception e) {
            log.error("文件类型转换失败" + e.getMessage());
            return null;
        } finally {
            try {
                if (null != inputStream) {
                    inputStream.close();
                }

                if (null != outputStream) {
                    outputStream.close();
                }
            } catch (IOException e) {
                log.error(">>文件流关闭失败" + e.getMessage());
            }
        }

    }
}