【云存储】oss文件上传工具类

亦凉 2022-02-24 01:14 452阅读 0赞

知识在于点滴积累。

将文件或图片上传至oss的工具类：

package com.xcauto.proxy.external.signature.utils;
    
    import com.aliyun.oss.OSSClient;
    import com.aliyun.oss.OSSException;
    import com.aliyun.oss.model.OSSObject;
    import com.aliyun.oss.model.ObjectMetadata;
    import com.aliyun.oss.model.PutObjectResult;
    import org.springframework.beans.factory.annotation.Value;
    import org.springframework.stereotype.Component;
    import org.springframework.util.StringUtils;
    import org.springframework.web.multipart.MultipartFile;
    
    import java.io.*;
    import java.net.URL;
    import java.util.Date;
    import java.util.Random;
    
    /**
     * 阿里云 OSS工具类
     *
     * @author hellen
     * @version 1.0
     */
    @Component
    public class OSSClientUtil {
    	//参数在yaml配置
        @Value("${oss.aliyun.endpoint}")
        private String endpoint;
    
        @Value("${oss.aliyun.accessKeyId}")
        private String accessKeyId;
    
        @Value("${oss.aliyun.accessKeySecret}")
        private String accessKeySecret;
    
        @Value("${oss.aliyun.bucketName}")
        private String bucketName;
        // 文件存储目录
        private String filedir = "";
    
        /**
         * 上传图片
         *
         * @param url
         * @throws Exception
         */
        public String uploadImg2Oss(String url) throws Exception {
            File fileOnServer = new File(url);
            FileInputStream fin;
            String serverUrl = null;
            try {
                fin = new FileInputStream(fileOnServer);
                String[] split = url.split("/");
                serverUrl = this.uploadFile2OSS(fin, split[split.length - 1]);
            } catch (FileNotFoundException e) {
                throw new Exception("图片上传失败");
            }
            return serverUrl;
        }
    
        public String uploadImg2Oss(MultipartFile file) throws Exception {
            if (file.getSize() > 10 * 1024 * 1024) {
                throw new Exception("上传图片大小不能超过10M！");
            }
            String serverUrl = null;
            String originalFilename = file.getOriginalFilename();
            String substring = originalFilename.substring(originalFilename.lastIndexOf(".")).toLowerCase();
            Random random = new Random();
            String name = random.nextInt(10000) + System.currentTimeMillis() + substring;
            try {
                InputStream inputStream = file.getInputStream();
                serverUrl = this.uploadFile2OSS(inputStream, name);
                return serverUrl;
            } catch (Exception e) {
                e.printStackTrace();
                throw new Exception("图片上传失败");
            }
        }
    
        public String uploadFile2Oss(String filePath) throws Exception {
            File file = new File(filePath);
            if (file.exists() && file.isFile()) {
                String serverUrl = null;
                String originalFilename = file.getName();
                String substring = originalFilename.substring(originalFilename.lastIndexOf(".")).toLowerCase();
                Random random = new Random();
                String name = random.nextInt(10000) + System.currentTimeMillis() + substring;
                InputStream inputStream = null;
                try {
                    inputStream = new FileInputStream(file);
                    serverUrl = this.uploadFile2OSS(inputStream, name);
                    return serverUrl;
                } catch (Exception e) {
                    e.printStackTrace();
                    throw new Exception("图片上传失败");
                } finally {
                    try {
                        if (inputStream != null) {
                            inputStream.close();
                        }
                    } catch (Exception e) {
    
                    }
                }
            }
            return null;
        }
    
        /**
         * 获得图片路径
         *
         * @param fileUrl
         * @return
         */
        private String getImgUrl(String fileUrl, OSSClient ossClient) {
            System.out.println(fileUrl);
            if (!StringUtils.isEmpty(fileUrl)) {
                String[] split = fileUrl.split("/");
                return this.getUrl(this.filedir + split[split.length - 1], ossClient);
            }
            return null;
        }
    
        /**
         * 上传到OSS服务器 如果同名文件会覆盖服务器上的
         *
         * @param instream
         *            文件流
         * @param fileName
         *            文件名称 包括后缀名
         * @return 出错返回"" ,唯一MD5数字签名
         */
        public String uploadFile2OSS(InputStream instream, String fileName) {
            String url = "";
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            try {
                // 创建上传Object的Metadata
                ObjectMetadata objectMetadata = new ObjectMetadata();
                objectMetadata.setContentLength(instream.available());
                objectMetadata.setCacheControl("no-cache");
                objectMetadata.setHeader("Pragma", "no-cache");
                objectMetadata.setContentType(getcontentType(fileName.substring(fileName.lastIndexOf("."))));
                objectMetadata.setContentDisposition("inline;filename=" + fileName);
                // 上传文件
                PutObjectResult putResult = ossClient.putObject(bucketName, filedir + fileName, instream, objectMetadata);
                if (putResult != null) {
                    url = getUrl(fileName, ossClient);
                }
            } catch (IOException e) {
    
            } finally {
                try {
                    if (instream != null) {
                        instream.close();
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
                ossClient.shutdown();
            }
            return url;
        }
    
        /**
         * Description: 判断OSS服务文件上传时文件的contentType
         *
         * @param
         *
         * @return String
         */
        public static String getcontentType(String filenameExtension) {
            if ("bmp".equalsIgnoreCase(filenameExtension)) {
                return "image/bmp";
            }
            if ("gif".equalsIgnoreCase(filenameExtension)) {
                return "image/gif";
            }
            if ("jpeg".equalsIgnoreCase(filenameExtension) || "jpg".equalsIgnoreCase(filenameExtension)
                    || "png".equalsIgnoreCase(filenameExtension)) {
                return "image/jpeg";
            }
            if ("html".equalsIgnoreCase(filenameExtension)) {
                return "text/html";
            }
            if ("txt".equalsIgnoreCase(filenameExtension)) {
                return "text/plain";
            }
            if ("vsd".equalsIgnoreCase(filenameExtension)) {
                return "application/vnd.visio";
            }
            if ("pptx".equalsIgnoreCase(filenameExtension) || "ppt".equalsIgnoreCase(filenameExtension)) {
                return "application/vnd.ms-powerpoint";
            }
            if ("docx".equalsIgnoreCase(filenameExtension) || "doc".equalsIgnoreCase(filenameExtension)) {
                return "application/msword";
            }
            if ("xml".equalsIgnoreCase(filenameExtension)) {
                return "text/xml";
            }
            return "image/jpeg";
        }
    
        /**
         * 获得url链接
         *
         * @param key
         * @return
         */
        private String getUrl(String key, OSSClient ossClient) {
            boolean flag = false;
            if (ossClient == null) {
                flag = true;
                ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            }
            try {
                // 设置URL过期时间为1000年 3600l* 1000*24*365*1000
                Date expiration = new Date(System.currentTimeMillis() + 3600L * 1000 * 24 * 365 * 1000);
                // 生成URL
                URL url = ossClient.generatePresignedUrl(bucketName, key, expiration);
                if (url != null) {
                    return url.toString();
                }
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                if (true) {
                    ossClient.shutdown();
                }
            }
            return null;
        }
    
        /**
         * 删除文件
         */
        public void delFileOSS(String fileName){
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            if(fileName != null && fileName.length() > 0) {
                ossClient.deleteObject(bucketName, fileName);
            }
        }
    
        /**
         * 文件是否存在
         * @param key
         * @return
         */
        public boolean doesExist(String key){
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            boolean exist = ossClient.doesObjectExist(bucketName, key);
            return exist;
        }
    
        /**
         * 获取文件
         * @param key
         * @return
         * @throws IOException
         */
        public byte[] getFile(String key) throws IOException {
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            OSSObject object = null;
            try {
                object = ossClient.getObject(bucketName, key);
                if (object != null) {
                    object.close();
                }
            } catch (OSSException e) {
                // OSS在查找不到某对象时，会抛出ErrorCode为“NoSuchKey”的OSSException，而不是返回null
                if (e.getErrorCode().contains("NoSuchKey")) {
                    System.out.println("找不到OSS文件：" + key);
                }else {
                    e.printStackTrace();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
            InputStream inputStream = object.getObjectContent();
            ByteArrayOutputStream swapStream = new ByteArrayOutputStream();
            byte[] buff = new byte[100];
            int rc = 0;
            while ((rc = inputStream.read(buff, 0, 100)) > 0) {
                swapStream.write(buff, 0, rc);
            }
            byte[] in2b = swapStream.toByteArray();
            return in2b;
        }
    
    }

OSS的API文档：  
[http://doc.oss.aliyuncs.com/\#\_Toc336676734][http_doc.oss.aliyuncs.com_Toc336676734]

总结：  
redis，fastdfs，oss使用方法：引入jar，util类，存储，可以发现他们使用方式上有共同点，关键是了解他们奇妙的内部机制。

[http_doc.oss.aliyuncs.com_Toc336676734]: http://doc.oss.aliyuncs.com/#_Toc336676734