AWS-S3上传下载

拼搏现实的明天。 2022-09-01 09:15 138阅读 0赞

@RestController
    public class FileUtils {
        // 【你的 access_key】
        static final String AWS_ACCESS_KEY = "XXX";
        // 【你的 aws_secret_key】
        static final String AWS_SECRET_KEY = " XXX";
        // 储存桶的名称
        static final String BUCKET_NAME = "XXX";
        // 设置服务器所属地区
        static final Regions REGION = Regions.CN_NORTHWEST_1;
        // 储存路径
        static final String PATH = "test/";
    
        static final BasicAWSCredentials awsCreds = new BasicAWSCredentials(AWS_ACCESS_KEY, AWS_SECRET_KEY);
    
        static final AmazonS3 s3 = AmazonS3ClientBuilder.standard()
                .withCredentials(new AWSStaticCredentialsProvider(awsCreds)).withRegion(REGION).build();
    
    
        @PostMapping("/s3")
        public  String uploadFileToS3(MultipartFile multipartFile, String type)
                throws AmazonServiceException, SdkClientException, IOException {
            if (multipartFile.isEmpty()) {
                return "文件为空";
            }
            long time = System.currentTimeMillis();
            // 拼接下文件路径，为了不会出现文件覆盖现象，加上时间戳
            String s3FilePath = PATH
                    + multipartFile.getOriginalFilename();
            ObjectMetadata metadata = new ObjectMetadata();
            metadata.setContentType(multipartFile.getContentType());
            metadata.setContentLength(multipartFile.getSize());
            // 开始上传文件
            s3.putObject(BUCKET_NAME, s3FilePath, multipartFile.getInputStream(), metadata);
            // 返回文件位置
            return s3FilePath;
        }
    
        /**
         * 用文件路径获取文件下载地址
         *
         * @param key
         * @return
         */
        @GetMapping("/s3/download")
        public ResponseEntity<byte[]> download(String key) throws IOException {
            GetObjectRequest getObjectRequest = new GetObjectRequest(BUCKET_NAME, key);
    
            S3Object s3Object =s3.getObject(getObjectRequest);
            S3ObjectInputStream objectInputStream = s3Object.getObjectContent();
            byte[] bytes = IOUtils.toByteArray(objectInputStream);
    
            HttpHeaders httpHeaders = new HttpHeaders();
            httpHeaders.setContentType(MediaType.APPLICATION_OCTET_STREAM);
            httpHeaders.setContentLength(bytes.length);
            httpHeaders.setContentDispositionFormData("attachment", key);
    
            return new ResponseEntity<>(bytes, httpHeaders, HttpStatus.OK);
        }
    }