-
Notifications
You must be signed in to change notification settings - Fork 1
/
S3RepositoryImpl.java
100 lines (80 loc) · 3.23 KB
/
S3RepositoryImpl.java
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
package robertocrespo.net.aws.repository;
import com.amazonaws.services.s3.AmazonS3;
import com.amazonaws.services.s3.model.CopyObjectRequest;
import com.amazonaws.services.s3.model.PutObjectRequest;
import com.amazonaws.services.s3.model.S3Object;
import com.amazonaws.services.s3.model.S3ObjectInputStream;
import com.amazonaws.services.s3.model.S3ObjectSummary;
import com.amazonaws.util.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import robertocrespo.net.aws.model.Asset;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;
import java.io.File;
import java.io.IOException;
import java.util.List;
import java.util.stream.Collectors;
@Repository
public class S3RepositoryImpl implements S3Repository{
private AmazonS3 s3Client;
@Autowired
public S3RepositoryImpl(AmazonS3 s3Client) {
this.s3Client = s3Client;
}
private static final Logger log = LoggerFactory.getLogger(S3RepositoryImpl.class);
@Override
public List<Asset> listObjectsInBucket(String bucket) {
List<Asset> items =
s3Client.listObjectsV2(bucket).getObjectSummaries().stream()
.parallel()
.map(S3ObjectSummary::getKey)
.map(key -> mapS3ToObject(bucket, key))
.collect(Collectors.toList());
log.info("Found " + items.size() + " objects in the bucket " + bucket);
return items;
}
private Asset mapS3ToObject(String bucket, String key) {
return Asset.builder()
.name(s3Client.getObjectMetadata(bucket, key).getUserMetaDataOf("name"))
.key(key)
.url(s3Client.getUrl(bucket, key))
.build();
}
@Override
public S3ObjectInputStream getObject(String bucketName, String fileName) throws IOException {
if (!s3Client.doesBucketExistV2(bucketName)) {
log.error("No Bucket Found");
return null;
}
S3Object s3object = s3Client.getObject(bucketName, fileName);
return s3object.getObjectContent();
}
@Override
public byte[] downloadFile(String bucketName, String fileName) {
S3Object s3Object = s3Client.getObject(bucketName, fileName);
S3ObjectInputStream inputStream = s3Object.getObjectContent();
try {
byte[] content = IOUtils.toByteArray(inputStream);
return content;
} catch (IOException e) {
e.printStackTrace();
}
return null;
}
@Override
public void moveObject(String bucketName, String fileKey, String destinationFileKey) {
CopyObjectRequest copyObjRequest = new CopyObjectRequest(bucketName, fileKey, bucketName, destinationFileKey);
s3Client.copyObject(copyObjRequest);
deleteObject(bucketName, fileKey);
}
@Override
public void deleteObject (String bucketName, String fileKey) {
s3Client.deleteObject(bucketName, fileKey);
}
public String uploadFile(String bucketName, String fileName, File fileObj) {
s3Client.putObject(new PutObjectRequest(bucketName, fileName, fileObj));
fileObj.delete();
return "File uploaded : " + fileName;
}
}