阿里云oss 图片上传工具类

青旅半醒 2022-01-07 10:19 471阅读 0赞

package com.opencloud.common.utils;
    
    import com.aliyun.oss.ClientException;
    import com.aliyun.oss.HttpMethod;
    import com.aliyun.oss.OSSClient;
    import com.aliyun.oss.OSSException;
    import com.aliyun.oss.model.*;
    import org.apache.commons.lang3.StringUtils;
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.beans.factory.annotation.Value;
    import org.springframework.web.multipart.MultipartFile;
    
    import javax.imageio.ImageIO;
    import java.awt.*;
    import java.io.ByteArrayInputStream;
    import java.io.File;
    import java.io.FileOutputStream;
    import java.io.IOException;
    import java.net.URL;
    import java.text.SimpleDateFormat;
    import java.util.ArrayList;
    import java.util.Date;
    import java.util.List;
    
    /**
     * @author linfan.sun
     * @date 2019/6/2616:10
     */
    public class AliyunOSSUtil {
        private static final Logger logger = LoggerFactory.getLogger(AliyunOSSUtil.class);
        private static String FILE_URL;
    //    private static String bucketName = null;
    //    private static String endpoint = null;
    //    private static String accessKeyId = null;
    //    private static String accessKeySecret = null;
    
        @Value("${aliyun.oss.bucketName}")
        private static String bucketName;
        @Value("${aliyun.oss.endpoint}")
        private static String endpoint;
        @Value("${aliyun.oss.accessKeyId}")
        private static String accessKeyId;
        @Value("${aliyun.oss.accessKeySecret}")
        private static String accessKeySecret;
    
        /**
         * 上传文件。
         *
         * @param file 需要上传的文件路径
         * @return 如果上传的文件是图片的话，会返回图片的"URL"，如果非图片的话会返回"非图片，不可预览。文件路径为：+文件路径"
         */
        public static String upLoad(File file, String fileHost, String suffix) {
    
            // 默认值为：true
            boolean isImage = true;
            // 判断所要上传的图片是否是图片，图片可以预览，其他文件不提供通过URL预览
            try {
                Image image = ImageIO.read(file);
                isImage = image == null ? false : true;
            } catch (IOException e) {
                e.printStackTrace();
            }
    
            logger.info("------OSS文件上传开始--------" + file.getName());
    
            // 文件名格式
            SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddhhmmss");
            String dateString = sdf.format(new Date()) + "." + suffix; // 20180322010634.jpg
    
            // 判断文件
            if (file == null) {
                return null;
            }
            // 创建OSSClient实例。
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            try {
                // 判断容器是否存在,不存在就创建
                if (!ossClient.doesBucketExist(bucketName)) {
                    ossClient.createBucket(bucketName);
                    CreateBucketRequest createBucketRequest = new CreateBucketRequest(bucketName);
                    createBucketRequest.setCannedACL(CannedAccessControlList.PublicRead);
                    ossClient.createBucket(createBucketRequest);
                }
                // 设置文件路径和名称
    //            String fileUrl = fileHost + "/" + (dateString + "/" + UUID.randomUUID().toString().replace("-", "") + "-" + file.getName());
                String fileUrl = fileHost + "/" + dateString;
                if (isImage) {//如果是图片，则图片的URL为：....
                    FILE_URL = "https://" + bucketName + "." + endpoint + "/" + fileUrl;
                } else {
                    FILE_URL = fileUrl;
                    logger.info("非图片,不可预览。文件路径为：" + fileUrl);
                }
    
                // 上传文件
                PutObjectResult result = ossClient.putObject(new PutObjectRequest(bucketName, fileUrl, file));
                // 设置权限(公开读)
                ossClient.setBucketAcl(bucketName, CannedAccessControlList.PublicRead);
                if (result != null) {
                    logger.info("------OSS文件上传成功------" + fileUrl);
                }
            } catch (OSSException oe) {
                logger.error(oe.getMessage());
            } catch (ClientException ce) {
                logger.error(ce.getErrorMessage());
            } finally {
                if (ossClient != null) {
                    ossClient.shutdown();
                }
            }
            return FILE_URL;
        }
    
        /**
         * 通过文件名下载文件
         *
         * @param objectName    要下载的文件名
         * @param localFileName 本地要创建的文件名
         */
        public static void downloadFile(String objectName, String localFileName) {
    
            // 创建OSSClient实例。
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            // 下载OSS文件到本地文件。如果指定的本地文件存在会覆盖，不存在则新建。
            ossClient.getObject(new GetObjectRequest(bucketName, objectName), new File(localFileName));
            // 关闭OSSClient。
            ossClient.shutdown();
        }
    
        /**
         * 删除文件
         * objectName key 地址
         *
         * @param filePath
         */
        public static Boolean delFile(String filePath) {
            logger.info("删除开始，objectName=" + filePath);
            // 创建OSSClient实例。
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
    
            // 删除Object.
            boolean exist = ossClient.doesObjectExist(bucketName, filePath);
            if (!exist) {
                logger.error("文件不存在,filePath={}", filePath);
                return false;
            }
            logger.info("删除文件,filePath={}", filePath);
            ossClient.deleteObject(bucketName, filePath);
            ossClient.shutdown();
            return true;
        }
    
        /**
         * 批量删除
         *
         * @param keys
         */
        public static Boolean delFileList(java.util.List<String> keys) {
            // 创建OSSClient实例。
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            try {
                // 删除文件。
                DeleteObjectsResult deleteObjectsResult = ossClient.deleteObjects(new DeleteObjectsRequest(bucketName).withKeys(keys));
                java.util.List<String> deletedObjects = deleteObjectsResult.getDeletedObjects();
            } catch (Exception e) {
                e.printStackTrace();
                return false;
            } finally {
                ossClient.shutdown();
            }
            return true;
    
        }
    
        /**
         * 获取文件夹
         *
         * @param fileName
         * @return
         */
        public static java.util.List<String> fileFolder(String fileName) {
            // 创建OSSClient实例。
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            // 构造ListObjectsRequest请求。
            ListObjectsRequest listObjectsRequest = new ListObjectsRequest(bucketName);
            // 设置正斜线（/）为文件夹的分隔符。
            listObjectsRequest.setDelimiter("/");
            // 设置prefix参数来获取fun目录下的所有文件。
            if (StringUtils.isNotBlank(fileName)) {
                listObjectsRequest.setPrefix(fileName + "/");
            }
            // 列出文件
            ObjectListing listing = ossClient.listObjects(listObjectsRequest);
            // 遍历所有commonPrefix
            java.util.List<String> list = new ArrayList<>();
            for (String commonPrefix : listing.getCommonPrefixes()) {
                String newCommonPrefix = commonPrefix.substring(0, commonPrefix.length() - 1);
                String[] s = newCommonPrefix.split("/");
                list.add(s[1]);
            }
            // 关闭OSSClient
            ossClient.shutdown();
            return list;
        }
    
        /**
         * 列举文件下所有的文件url信息
         */
        public static java.util.List<String> listFile(String fileHost) {
            // 创建OSSClient实例。
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            // 构造ListObjectsRequest请求
            ListObjectsRequest listObjectsRequest = new ListObjectsRequest(bucketName);
    
            // 设置prefix参数来获取fun目录下的所有文件。
            listObjectsRequest.setPrefix(fileHost + "/");
            // 列出文件。
            ObjectListing listing = ossClient.listObjects(listObjectsRequest);
            // 遍历所有文件。
            List<String> list = new ArrayList<>();
            for (int i = 0; i < listing.getObjectSummaries().size(); i++) {
                if (i == 0) {
                    continue;
                }
                FILE_URL = "https://" + bucketName + "." + endpoint + "/" + listing.getObjectSummaries().get(i).getKey();
                list.add(FILE_URL);
            }
            // 关闭OSSClient。
            ossClient.shutdown();
            return list;
        }
    
        /**
         * 获得url链接
         *
         * @param objectName
         * @return
         */
        public static String getUrl(String objectName) {
            // 创建OSSClient实例。
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            // 设置权限(公开读)
            ossClient.setBucketAcl(bucketName, CannedAccessControlList.PublicRead);
            // 设置图片处理样式。
    //        String style = "image/resize,m_fixed,w_100,h_100/rotate,90";
            Date expiration = new Date(new Date().getTime() + 3600l * 1000 * 24 * 365 * 100);
            GeneratePresignedUrlRequest req = new GeneratePresignedUrlRequest(bucketName, objectName, HttpMethod.GET);
            req.setExpiration(expiration);
    //        req.setProcess(style);
            URL signedUrl = ossClient.generatePresignedUrl(req);
            // 关闭OSSClient。
            logger.info("------OSS文件文件信息--------" + signedUrl.toString());
            ossClient.shutdown();
            if (signedUrl != null) {
                return signedUrl.toString();
            }
            return null;
        }
    
        // 获取文 MultipartFile 文件后缀名工具
        public static String getSuffix(MultipartFile fileupload) {
            String originalFilename = fileupload.getOriginalFilename();
            String suffix = originalFilename.substring(originalFilename.lastIndexOf(".") + 1);
            System.out.println(suffix);
            return suffix;
        }
    
        /**
         * 创建文件夹
         *
         * @param folder
         * @return
         */
        public static String createFolder(String folder) {
            // 创建OSSClient实例。
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            // 文件夹名
            final String keySuffixWithSlash = folder;
            // 判断文件夹是否存在，不存在则创建
            if (!ossClient.doesObjectExist(bucketName, keySuffixWithSlash)) {
                // 创建文件夹
                ossClient.putObject(bucketName, keySuffixWithSlash, new ByteArrayInputStream(new byte[0]));
                logger.info("创建文件夹成功");
                // 得到文件夹名
                OSSObject object = ossClient.getObject(bucketName, keySuffixWithSlash);
                String fileDir = object.getKey();
                ossClient.shutdown();
                return fileDir;
            }
    
            return keySuffixWithSlash;
        }
    
        public static String uploadFile(MultipartFile file, String fileHost) {
            logger.info("文件上传");
            String suffix = getSuffix(file);
            String filename = file.getOriginalFilename();
            try {
                if (file != null) {
                    if (!"".equals(filename.trim())) {
                        File newFile = new File(filename);
                        FileOutputStream os = new FileOutputStream(newFile);
                        os.write(file.getBytes());
                        os.close();
                        file.transferTo(newFile);
                        // 上传到OSS
                        String uploadUrl = upLoad(newFile, fileHost, suffix);
                        return uploadUrl;
                    }
                }
            } catch (Exception ex) {
                ex.printStackTrace();
            }
            return null;
        }
    
        public static void main(String[] args) {
    //        createFolder("test/");
    //        getUrl("product/class/26065946787446951.jpg");
            listFile("document");
    //        fileFolder(null);
    //        fileFolder("document");
    //        String bucketUrl = "https://jiangpin.oss-cn-shenzhen.aliyuncs.com/other/20190509023515.jpg";
    //        String[] s = bucketUrl.split(".com");
    //        String path = s[1];
    //        path = path.substring(1, path.length());
    //        System.out.println(path);
        }
    }