JAVA写了一个七牛云上传文件图片的小工具

为什么80%的码农都做不了架构师？>>>   

上传图片只能上传png,jpg,jpeg等格式的，静态代码块加载配置文件。

因为文件名必须经过简单加密名字上传，所以需要获取图片格式，对originalName进行处理。

 

依赖：自定义ImageType枚举类。

功能：

public static String uploadToQiNiu(InputStream inputStream,String key)   //上传内容到七牛云，不做任何修饰

public static String uploadFileToQiNiu(InputStream inputStream,String key) //同上，修改含义，上传的是“文件”

public static String uploadImageToQiNiu(MultipartFile file)   //传入一个MultipartFile图片，上传到七牛云并返回图片链接

private static ImageType getImgType(String originalImageName)    //内部方法，用于获取图片类型，比如PNG JPG JPEG，因为涉及到对文件名进行MD5加密，必须保留文件后缀，该后缀仅对图片生效。

private static String getTail(String fileOriginName)    //内部方法，获取尾巴,如 .png   .jpg   .jpeg

 

代码如下：

package xyz.ruankun.machinemother.util.constant;

/**
 * 图片都有什么格式呢？！常用的就这几个格式吧？位图不算了
 */
public enum ImageType {
    JPG,JPEG,PNG,GIF
}

 

package xyz.ruankun.machinemother.util;

import com.google.gson.Gson;
import com.qiniu.common.Zone;
import com.qiniu.http.Response;
import com.qiniu.storage.Configuration;
import com.qiniu.storage.UploadManager;
import com.qiniu.storage.model.DefaultPutRet;
import com.qiniu.util.Auth;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.multipart.MultipartFile;
import xyz.ruankun.machinemother.util.constant.ImageType;

import java.io.IOException;
import java.io.InputStream;
import java.util.Date;
import java.util.Properties;

/**
 * 对象存储工具
 */
public class QiNiuFileUtil {

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

    private static String accessKey;
    private static String secretKey;
    private static String dns;

    /**
     * 加载七牛云的配置文件
     */
    static{
        Properties properties = new Properties();
        InputStream inputStream = QiNiuFileUtil.class.getClassLoader()
                .getResourceAsStream("qiniuyun.properties");
        try {
            properties.load(inputStream);
            accessKey = properties.getProperty("accessKey");
            secretKey = properties.getProperty("secretKey");
            dns = properties.getProperty("dns");
        } catch (IOException e) {
            logger.error("qiniu cloud load properties[name:qiniuyun.properties] in classpath failed.");
            e.printStackTrace();
        }
    }

    /**
     *
     * 上传文件到七牛云的工具
     * @param inputStream
     * @return
     */
    public static String uploadToQiNiu(InputStream inputStream,String key){

        if(inputStream == null){
            logger.info("传入的图片二进制流不存在，忽略上传过程");
            return null;
        }
        //zone0表示华东区域，该bucket在华东
        Configuration configuration = new Configuration(Zone.zone0());
        UploadManager uploadManager = new UploadManager(configuration);
        String bucket = "public";
        Auth auth = Auth.create(accessKey, secretKey);
        String upToken = auth.uploadToken(bucket);
        //System.out.println(upToken);
        try {
            Response response = uploadManager.put(inputStream,key,upToken,null,null);
            DefaultPutRet putRet = new Gson().fromJson(response.bodyString(), DefaultPutRet.class);
            System.out.println(putRet.key);
            System.out.println(putRet.hash);
            return getDns() + "/" + putRet.key;
        } catch (IOException e) {
            e.printStackTrace();
        }
        logger.info("请求时发生了错误信息");
        return null;
    }

    /**
     * 传文件
     * @param inputStream
     * @param key
     * @return 文件路径
     */
    public static String uploadFileToQiNiu(InputStream inputStream,String key){
        return  uploadToQiNiu(inputStream,key);
    }

    /**
     *  MD5 KEY
     * @param file
     * @return
     */
    public static String uploadImageToQiNiu(MultipartFile file){
        InputStream inputStream = null;
        try {
            inputStream = file.getInputStream();
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
        String tail = getTail(file.getOriginalFilename());
        return uploadToQiNiu(inputStream,MD5Util.md5(new Date().toString()) + tail);
    }

    public static String getDns(){
        return dns;
    }

    /**
     * 判断图片的格式，只支持jpg jpeg png gif
     * @param originalImageName
     * @return
     */
    private static ImageType getImgType(String originalImageName){
        if (originalImageName.toLowerCase().indexOf(".jpg") > 0)
            return ImageType.JPG;
        if (originalImageName.toLowerCase().indexOf(".jpeg") > 0)
            return ImageType.JPEG;
        if (originalImageName.toLowerCase().indexOf(".png") > 0)
            return ImageType.PNG;
        if (originalImageName.toLowerCase().indexOf(".gif") > 0)
            return ImageType.GIF;
        return null;
    }

    /**
     * 从其中拿到图片后缀
     * @param fileOriginName 文件名
     * @return
     */
    private static String getTail(String fileOriginName){
        ImageType type = getImgType(fileOriginName);
        String tail = null;
        switch (type){
            case GIF:
                tail = ".gif";
                break;
            case JPEG:
                tail = ".jpeg";
                break;
            case JPG:
                tail = ".jpg";
                break;
            case PNG:
                tail = ".png";
                break;
        }
        return tail;
    }
}

 

转载于:https://my.oschina.net/qkmc/blog/3059817