From c27ad3dcb8b67a6c10a99c98b9bd6dd1f08b7967 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@63.com>
Date: 星期二, 01 三月 2022 10:43:15 +0800
Subject: [PATCH] #
---
src/main/java/com/zy/sc/common/service/OssService.java | 43 +++++++------------------------------------
1 files changed, 7 insertions(+), 36 deletions(-)
diff --git a/src/main/java/com/zy/sc/common/service/OssService.java b/src/main/java/com/zy/sc/common/service/OssService.java
index ebda706..81b2a34 100644
--- a/src/main/java/com/zy/sc/common/service/OssService.java
+++ b/src/main/java/com/zy/sc/common/service/OssService.java
@@ -6,30 +6,26 @@
import com.core.common.SnowflakeIdWorker;
import com.core.exception.CoolException;
import lombok.Data;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
+import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
-import java.io.File;
-import java.io.FileInputStream;
-import java.io.FileNotFoundException;
import java.io.InputStream;
import java.net.URL;
import java.security.SecureRandom;
import java.text.SimpleDateFormat;
import java.util.Date;
+import java.util.Objects;
/**
* 闃块噷浜慜SS鏈嶅姟绫�
* Created by vincent on 2019-11-26
*/
@Data
+@Slf4j
@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";
@@ -55,21 +51,16 @@
* 涓婁紶OSS鏂囦欢
* @return the url
*/
- public String upload(InputStream inputStream, String suffix){
- if(suffix.trim().equals("file")){
- suffix="jpg";
- }
+ public String upload(InputStream inputStream, String fileName){
OSSClient ossClient = new OSSClient(OSS_ENDPOINT, accessKeyId, accessKeySecret);
- String name = "";
try {
if (!ossClient.doesBucketExist(bucket)) {
ossClient.createBucket(bucket);
log.info("鎮ㄧ殑Bucket涓嶅瓨鍦紝鍒涘缓Bucket锛歿}",bucket);
}
- name = String.valueOf(snowflakeIdWorker.nextId()).concat(".").concat(suffix);
- // 娣诲姞 ContentType
+ String suffix = Objects.requireNonNull(fileName).substring(fileName.lastIndexOf(".") + 1);
+ String name = String.valueOf(snowflakeIdWorker.nextId()).concat(".").concat(suffix);
ObjectMetadata objectMetadata = new ObjectMetadata();
- objectMetadata.setContentType("image/jpg");
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);
@@ -77,15 +68,12 @@
// return String.valueOf(url);
return prefix+name;
} catch (Exception e){
+ log.error("OSS鏂囦欢涓婁紶澶辫触锛佹枃浠跺悕{}", fileName);
e.printStackTrace();
} finally {
ossClient.shutdown();
}
throw new CoolException("500-涓婁紶澶辫触");
- }
-
- public String upload(InputStream inputStream) {
- return upload(inputStream, ".jpg");
}
/**
@@ -120,23 +108,6 @@
ossClient.shutdown();
}
return true;
- }
-
- public static void main(String[] args) {
- File file = new File("E:\\tmp\\v2-fbbb97b977b5cebc66dc3cefab0ac981_r.jpg");
- try {
- InputStream in = new FileInputStream(file);
- OssService ossService = new OssService();
- ossService.setAccessKeyId("LTAI4GDzr6ioSHuRw2mk22ug");
- ossService.setAccessKeySecret("84CHL7tF21LbU1qpaP0jn9mIAZP9bv");
- ossService.setBucket("tjdt");
- ossService.setEndpoint("http://oss-cn-hangzhou.aliyuncs.com");
- ossService.setSnowflakeIdWorker(new SnowflakeIdWorker());
- System.out.println(ossService.upload(in));
-
- } catch (FileNotFoundException e) {
- e.printStackTrace();
- }
}
public String uploadFile(InputStream inputStream, String fileName, Long len, String suffix){
--
Gitblit v1.9.1