devbox
2023面试高手
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

SpringBoot整合亚马逊S3_aws-java-sdk-sts

作者：2023面试高手 | 2024-06-12 07:26:43

赞

踩

aws-java-sdk-sts

一、参考项

AWS S3(官网): Amazon S3 - 亚马逊云科技对象存储_云存储服务-亚马逊云科技中国区域
AWS SDK for Java(官网):Setting up the AWS SDK for Java 2.x - AWS SDK for Java

 二、效果展示

三、引入Pom文件

  1. <!-- https://mvnrepository.com/artifact/com.amazonaws/aws-java-sdk-s3 -->
  2. <dependency>
  3.   <groupId>com.amazonaws</groupId>
  4.   <artifactId>aws-java-sdk-s3</artifactId>
  5.   <version>1.11.803</version>
  6. </dependency>
  7. <dependency>
  8.   <groupId>com.amazonaws</groupId>
  9.   <artifactId>aws-java-sdk-sts</artifactId>
  10.   <version>1.11.803</version>
  11. </dependency>
  12. <dependency>
  13.   <groupId>com.amazonaws</groupId>
  14.   <artifactId>aws-java-sdk-core</artifactId>
  15.   <version>1.11.803</version>
  16. </dependency>

二、定义抽象类

  1. public abstract class BaseObjectStorage {
  2.  
  3.     /**
  4.      * 上传文件
  5.      *
  6.      * @param pathAndName
  7.      * @param file
  8.      */
  9.     public abstract void upload(String pathAndName, File file);
  10.  
  11.     /**
  12.      * 授权
  13.      *
  14.      * @param pathAndName
  15.      * @param time
  16.      * @return
  17.      */
  18.     public abstract String authorize(String pathAndName, long time);
  19.  
  20.     /**
  21.      * 授权(路径全)
  22.      *
  23.      * @param pathAndName
  24.      * @param time
  25.      * @return
  26.      */
  27.     public abstract String authorizeAllName(String pathAndName, long time);
  28.  
  29.     /**
  30.      * 临时上传文件授权
  31.      *
  32.      * @param dir
  33.      * @return
  34.      */
  35.     public abstract Map<String, Object> tokens(String dir);
  36.  
  37.     /**
  38.      * 删除文件
  39.      *
  40.      * @param pathAndName
  41.      */
  42.     public abstract void deleteFile(String pathAndName);
  43.  
  44. }

三、AWS实现类

  1. package cn.xhh.core.objectstorage;
  2.  
  3. import com.amazonaws.AmazonClientException;
  4. import com.amazonaws.AmazonServiceException;
  5. import com.amazonaws.auth.AWSStaticCredentialsProvider;
  6. import com.amazonaws.auth.BasicAWSCredentials;
  7. import com.amazonaws.client.builder.AwsClientBuilder.EndpointConfiguration;
  8. import com.amazonaws.services.s3.AmazonS3;
  9.  
  10. import com.amazonaws.services.s3.AmazonS3ClientBuilder;
  11. import com.amazonaws.services.s3.model.CannedAccessControlList;
  12. import com.amazonaws.services.s3.model.PutObjectRequest;
  13. import com.amazonaws.services.securitytoken.AWSSecurityTokenService;
  14. import com.amazonaws.services.securitytoken.AWSSecurityTokenServiceAsyncClientBuilder;
  15. import com.amazonaws.services.securitytoken.model.Credentials;
  16. import com.amazonaws.services.securitytoken.model.GetFederationTokenRequest;
  17. import com.amazonaws.services.securitytoken.model.GetFederationTokenResult;
  18. import com.google.common.collect.Maps;
  19. import lombok.Data;
  20. import lombok.extern.slf4j.Slf4j;
  21. import org.springframework.beans.factory.annotation.Autowired;
  22. import org.springframework.boot.context.properties.ConfigurationProperties;
  23. import org.springframework.stereotype.Component;
  24.  
  25. import java.io.File;
  26. import java.net.URL;
  27. import java.util.Date;
  28. import java.util.Map;
  29.  
  30. /**
  31.  * s3cloud上传文件
  32.  */
  33. @Component
  34. @Slf4j
  35. public class S3ObjectStorage extends BaseObjectStorage {
  36.  
  37.  
  38.     @Data
  39.     @Component
  40.     @ConfigurationProperties(prefix = "s3")
  41.     public static class OssInfo {
  42.         private String host;
  43.         private String endpoint;
  44.         private String accessKeyId;
  45.         private String accessKeySecret;
  46.         private String bucketName;
  47.         private String rootDirectory;
  48.         private String stsEndpoint;
  49.         private String region;
  50.     }
  51.  
  52.     @Autowired
  53.     private OssInfo ossInfo;
  54.  
  55.     @Override
  56.     public void upload(String pathAndName, File file) {
  57.         AWSStaticCredentialsProvider credential = new AWSStaticCredentialsProvider(new BasicAWSCredentials(ossInfo.accessKeyId, ossInfo.accessKeySecret));
  58.         EndpointConfiguration endpointConfiguration = new EndpointConfiguration(ossInfo.endpoint, null);
  59.         AmazonS3 s3 = AmazonS3ClientBuilder.standard().withCredentials(credential).withEndpointConfiguration(endpointConfiguration).build();
  60.         try {
  61.             String bucketPath = ossInfo.bucketName + "/" + ossInfo.rootDirectory;
  62.             s3.putObject(new PutObjectRequest(bucketPath, pathAndName, file)
  63.                     .withCannedAcl(CannedAccessControlList.PublicRead));
  64.             log.info("===s3===上传文件记录：成功");
  65.         } catch (AmazonServiceException ase) {
  66.             log.error("===s3===文件上传服务端异常:", ase);
  67.         } catch (AmazonClientException ace) {
  68.             log.error("===s3===文件上传客户端异常:", ace);
  69.         } finally {
  70.             s3.shutdown();
  71.         }
  72.     }
  73.  
  74.     @Override
  75.     public String authorize(String pathAndName, long time) {
  76.         AWSStaticCredentialsProvider credential = new AWSStaticCredentialsProvider(new BasicAWSCredentials(ossInfo.accessKeyId, ossInfo.accessKeySecret));
  77.         EndpointConfiguration endpointConfiguration = new EndpointConfiguration(ossInfo.endpoint, null);
  78.         AmazonS3 s3 = AmazonS3ClientBuilder.standard().withCredentials(credential).withEndpointConfiguration(endpointConfiguration).build();
  79.         try {
  80.             Date expiration = new Date(System.currentTimeMillis() + time);
  81.             URL url = s3.generatePresignedUrl(ossInfo.bucketName, ossInfo.rootDirectory + "/" + pathAndName, expiration);
  82.             String resultUrl = url.toString();
  83.             log.info("===s3===文件上传客户端返回url：{}", resultUrl);
  84.             resultUrl = resultUrl.substring(0, resultUrl.indexOf("?"));
  85.             resultUrl = resultUrl.replaceAll(ossInfo.host, ossInfo.endpoint);
  86.             log.info("===s3===文件上传客户端返回url：{}", resultUrl);
  87.             return resultUrl;
  88.         } finally {
  89.             s3.shutdown();
  90.         }
  91.     }
  92.  
  93.     @Override
  94.     public String authorizeAllName(String pathAndName, long time) {
  95.         AWSStaticCredentialsProvider credential = new AWSStaticCredentialsProvider(new BasicAWSCredentials(ossInfo.accessKeyId, ossInfo.accessKeySecret));
  96.         EndpointConfiguration endpointConfiguration = new EndpointConfiguration(ossInfo.endpoint, null);
  97.         AmazonS3 s3 = AmazonS3ClientBuilder.standard().withCredentials(credential).withEndpointConfiguration(endpointConfiguration).build();
  98.         try {
  99.             Date expiration = new Date(System.currentTimeMillis() + time);
  100.             URL url = s3.generatePresignedUrl(ossInfo.bucketName, pathAndName, expiration);
  101.             String resultUrl = url.toString();
  102.             resultUrl = resultUrl.replaceAll(ossInfo.host, ossInfo.endpoint);
  103.             log.info("===s3==========authorizeAllName，S3文件上传客户端返回url：{}", resultUrl);
  104.             return resultUrl;
  105.         } finally {
  106.             s3.shutdown();
  107.         }
  108.     }
  109.  
  110.     @Override
  111.     public Map<String, Object> tokens(String dir) {
  112.         Map<String, Object> result = null;
  113.         AWSSecurityTokenService stsClient = null;
  114.         try {
  115.             result = Maps.newHashMap();
  116.             AWSStaticCredentialsProvider credential = new AWSStaticCredentialsProvider(new BasicAWSCredentials(ossInfo.accessKeyId, ossInfo.accessKeySecret));
  117.             EndpointConfiguration endpointConfiguration = new EndpointConfiguration(ossInfo.stsEndpoint, null);
  118.             stsClient = AWSSecurityTokenServiceAsyncClientBuilder.standard().withCredentials(credential)
  119.                     .withEndpointConfiguration(endpointConfiguration).build();
  120.             GetFederationTokenRequest request = new GetFederationTokenRequest().withName("Bob")
  121.                     .withPolicy("{\"Version\":\"2012-10-17\",\"Statement\":[{\"Sid\":\"Sid1\",\"Effect\":\"Allow\",\"Action\":[\"s3:*\"],\"Resource\":[\"*\"]}]}")
  122.                     .withDurationSeconds(3600);
  123.             GetFederationTokenResult response = stsClient.getFederationToken(request);
  124.             Credentials tempCredentials = response.getCredentials();
  125.  
  126.  /*
  127.             // TODO 备份获取Token
  128.             stsClient = AWSSecurityTokenServiceClientBuilder.standard().withCredentials(new AWSStaticCredentialsProvider(new BasicAWSCredentials(ossInfo.accessKeyId, ossInfo.accessKeySecret))).withRegion(ossInfo.region).build();
  129.             //获取sessionToken实体
  130.             GetSessionTokenRequest getSessionTokenRequest = new GetSessionTokenRequest().withDurationSeconds(3000);
  131.             //创建请求
  132.             Credentials tempCredentials = stsClient.getSessionToken(getSessionTokenRequest).getCredentials();
  133. */
  134.  
  135.             result.put("storeType", "s3");
  136.             result.put("accessKeyId", tempCredentials.getAccessKeyId());
  137.             result.put("sessionToken", tempCredentials.getSessionToken());
  138.             result.put("secretKey", tempCredentials.getSecretAccessKey());
  139.             result.put("expire", tempCredentials.getExpiration());
  140.             result.put("dir", dir);
  141.             result.put("bucketName", ossInfo.bucketName);
  142.             result.put("region", ossInfo.region);
  143.             result.put("host", "https://" + ossInfo.endpoint + "/" + ossInfo.bucketName);
  144.             log.info("===s3===上传文件记录：accessKeyId：{}，sessionToken：{}", tempCredentials.getAccessKeyId(), tempCredentials.getSessionToken());
  145.         } catch (Exception e) {
  146.             e.printStackTrace();
  147.         } finally {
  148.             if (null != stsClient) {
  149.                 stsClient.shutdown();
  150.             }
  151.         }
  152.         return result;
  153.  
  154.     }
  155.  
  156.     @Override
  157.     public void deleteFile(String pathAndName) {
  158.         AWSStaticCredentialsProvider credential = new AWSStaticCredentialsProvider(new BasicAWSCredentials(ossInfo.accessKeyId, ossInfo.accessKeySecret));
  159.         EndpointConfiguration endpointConfiguration = new EndpointConfiguration(ossInfo.endpoint, null);
  160.         AmazonS3 s3 = AmazonS3ClientBuilder.standard().withCredentials(credential).withEndpointConfiguration(endpointConfiguration).build();
  161.         try {
  162.             s3.deleteObject(ossInfo.bucketName, ossInfo.bucketName + pathAndName);
  163.         } finally {
  164.             s3.shutdown();
  165.         }
  166.     }
  167.  
  168.  
  169. }

四、application配置文件

  1. objectstorage.type: s3
  2. s3:
  3.   endpoint: s3.us-east-1.amazonaws.com
  4.   access-key-id: 您的公钥AKIAXZXXXX2GMAJVNUS
  5.   access-key-secret: 您的秘钥CGNF3NQl4d0zvDuGEGuBsW9OS
  6.   bucket-name: xhh-test-bucket
  7.   root-directory: xhh/export
  8.   region: us-east-1
  9.

声明：本文内容由网友自发贡献，转载请注明出处：【wpsshop】
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号