#

luxiaotao1123

2022-03-01 5e4bef66a37f1bbdf48d845dda518fd8cc54ef37

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
package com.zy.sc.common.service;
 
import com.aliyun.oss.OSSClient;
import com.aliyun.oss.model.OSSObject;
import com.aliyun.oss.model.ObjectMetadata;
import com.core.common.SnowflakeIdWorker;
import com.core.exception.CoolException;
import lombok.Data;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
 
import java.io.InputStream;
import java.net.URL;
import java.security.SecureRandom;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Objects;
 
/**
 * 阿里云OSS服务类
 * Created by vincent on 2019-11-26
 */
@Data
@Service("ossService")
public class OssService {
 
    private static Logger log = LoggerFactory.getLogger(OssService.class);
 
    private static final String OSS_ENDPOINT = "http://oss-cn-hangzhou.aliyuncs.com";
 
    @Value("${aliyun.oss.id}")
    private String accessKeyId;
 
    @Value("${aliyun.oss.secret}")
    private String accessKeySecret;
 
    @Value("${aliyun.oss.bucket}")
    private String bucket;
 
    @Value("${aliyun.oss.endpoint}")
    private String endpoint;
 
    @Value("${aliyun.oss.prefix}")
    private String prefix;
 
    @Autowired
    private SnowflakeIdWorker snowflakeIdWorker;
 
    /**
     * 上传OSS文件
     * @return the url
     */
    public String upload(InputStream inputStream, String fileName){
        OSSClient ossClient = new OSSClient(OSS_ENDPOINT, accessKeyId, accessKeySecret);
        try {
            if (!ossClient.doesBucketExist(bucket)) {
                ossClient.createBucket(bucket);
                log.info("您的Bucket不存在，创建Bucket：{}",bucket);
            }
            String suffix = Objects.requireNonNull(fileName).substring(fileName.lastIndexOf(".") + 1);
            String name = String.valueOf(snowflakeIdWorker.nextId()).concat(".").concat(suffix);
            ObjectMetadata objectMetadata = new ObjectMetadata();
            ossClient.putObject(bucket, name, inputStream, objectMetadata);
            Date expiration = new Date(new Date().getTime() + 3600L * 1000 * 24 * 365 * 100);
            URL url = ossClient.generatePresignedUrl(bucket, name, expiration);
            log.info("OSS文件上传成功: {}", name);
//            return String.valueOf(url);
            return prefix+name;
        } catch (Exception e){
            e.printStackTrace();
        } finally {
            ossClient.shutdown();
        }
        throw new CoolException("500-上传失败");
    }
 
    /**
     * 下载OSS文件
     */
    public void download(String name) {
        OSSClient ossClient = new OSSClient(OSS_ENDPOINT, accessKeyId, accessKeySecret);
        try {
            OSSObject ossObject = ossClient.getObject(bucket, name);
            InputStream inputStream = ossObject.getObjectContent();
            // do transfer
            inputStream.close();
            log.info("OSS文件下载成功: {}", name);
        } catch (Exception e){
            e.printStackTrace();
        } finally {
            ossClient.shutdown();
        }
    }
 
    /**
     * 删除OSS文件
     */
    public boolean delete(String name) {
        OSSClient ossClient = new OSSClient(OSS_ENDPOINT, accessKeyId, accessKeySecret);
        try {
            ossClient.deleteObject(bucket, name);
        } catch (Exception e){
            e.printStackTrace();
            return false;
        } finally {
            ossClient.shutdown();
        }
        return true;
    }
 
    public String uploadFile(InputStream inputStream, String fileName, Long len, String suffix){
        OSSClient ossClient = new OSSClient(OSS_ENDPOINT, accessKeyId, accessKeySecret);
        try {
            if (!ossClient.doesBucketExist(bucket)) {
                ossClient.createBucket(bucket);
                log.info("您的Bucket不存在，创建Bucket：{}", bucket);
            }
 
            SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
 
            // 生成上传文件名
            String finalFileName = System.currentTimeMillis() + "" + new SecureRandom().nextInt(0x0400) + "." + suffix;
            String objectName = sdf.format(new Date()) + "/" + finalFileName;
 
            // 添加 ContentType
            ObjectMetadata objectMetadata = new ObjectMetadata();
            objectMetadata.setContentLength(len);
            ossClient.putObject(bucket, objectName, inputStream, objectMetadata);
 
            // 设置URL过期时间为100年。
            Date expiration = new Date(new Date().getTime() + 3600L * 1000 * 24 * 365 * 100);
 
            // 生成以GET方法访问的签名URL，访客可以直接通过浏览器访问相关内容。
            URL url = ossClient.generatePresignedUrl(bucket, objectName, expiration);
 
            log.info("OSS文件上传成功: {}", fileName);
//            return String.valueOf(url);
            return prefix + objectName;
        } catch (Exception e){
            e.printStackTrace();
        } finally {
            ossClient.shutdown();
        }
        throw new CoolException("500-上传失败");
    }
 
}