Browse Source

无jira任务 fix: 测试连接后释放资源;list接口过滤自身目录

release/10.0
Afly 2 years ago
parent
commit
7d4a1f54e3
  1. 7
      src/main/java/com/fanruan/fs/s3/repository/core/S3RepositoryFactory.java
  2. 14
      src/main/java/com/fanruan/fs/s3/repository/core/S3ResourceRepository.java

7
src/main/java/com/fanruan/fs/s3/repository/core/S3RepositoryFactory.java

@ -40,6 +40,7 @@ public class S3RepositoryFactory extends ConfigRepositoryFactory<S3Config> {
@Override @Override
public boolean verifyConfig(S3Config config) { public boolean verifyConfig(S3Config config) {
AmazonS3 s3 = null;
try { try {
BasicAWSCredentials credentials = new BasicAWSCredentials(config.getAccessKeyId(), config.getPassword()); BasicAWSCredentials credentials = new BasicAWSCredentials(config.getAccessKeyId(), config.getPassword());
AmazonS3ClientBuilder amazonS3ClientBuilder = AmazonS3ClientBuilder.standard() AmazonS3ClientBuilder amazonS3ClientBuilder = AmazonS3ClientBuilder.standard()
@ -57,11 +58,15 @@ public class S3RepositoryFactory extends ConfigRepositoryFactory<S3Config> {
clientConfiguration.setProtocol(Protocol.HTTP); clientConfiguration.setProtocol(Protocol.HTTP);
} }
amazonS3ClientBuilder = amazonS3ClientBuilder.withClientConfiguration(clientConfiguration); amazonS3ClientBuilder = amazonS3ClientBuilder.withClientConfiguration(clientConfiguration);
AmazonS3 s3 = amazonS3ClientBuilder.build(); s3 = amazonS3ClientBuilder.build();
s3.listObjects(config.getBucket()); s3.listObjects(config.getBucket());
} catch (Exception e) { } catch (Exception e) {
LogKit.error(e.getMessage(), e); LogKit.error(e.getMessage(), e);
return false; return false;
} finally {
if (s3 != null) {
s3.shutdown();
}
} }
return true; return true;
} }

14
src/main/java/com/fanruan/fs/s3/repository/core/S3ResourceRepository.java

@ -99,15 +99,15 @@ public class S3ResourceRepository extends BaseResourceRepository {
ListObjectsRequest listObjectsRequest = new ListObjectsRequest().withBucketName(bucket) ListObjectsRequest listObjectsRequest = new ListObjectsRequest().withBucketName(bucket)
.withPrefix(dir).withDelimiter(DELIMITER); .withPrefix(dir).withDelimiter(DELIMITER);
ObjectListing objectListing = s3.listObjects(listObjectsRequest); ObjectListing objectListing = s3.listObjects(listObjectsRequest);
collectFileEntry(result, objectListing); collectFileEntry(dir, result, objectListing);
while (objectListing.isTruncated()) { while (objectListing.isTruncated()) {
objectListing = s3.listNextBatchOfObjects(objectListing); objectListing = s3.listNextBatchOfObjects(objectListing);
collectFileEntry(result, objectListing); collectFileEntry(dir, result, objectListing);
} }
return result.toArray(new FineFileEntry[0]); return result.toArray(new FineFileEntry[0]);
} }
private void collectFileEntry(List<FineFileEntry> result, ObjectListing objectListing) { private void collectFileEntry(String dir, List<FineFileEntry> result, ObjectListing objectListing) {
for (S3ObjectSummary summary : objectListing.getObjectSummaries()) { for (S3ObjectSummary summary : objectListing.getObjectSummaries()) {
String key = summary.getKey(); String key = summary.getKey();
if (!key.endsWith(DELIMITER)) { if (!key.endsWith(DELIMITER)) {
@ -115,11 +115,13 @@ public class S3ResourceRepository extends BaseResourceRepository {
} }
} }
for (String prefix : objectListing.getCommonPrefixes()) { for (String prefix : objectListing.getCommonPrefixes()) {
if (StringUtils.isNotEmpty(prefix.substring(dir.length()).replaceAll(DELIMITER, StringUtils.EMPTY))) {
FineFileEntry entry = new FineFileEntry(prefix); FineFileEntry entry = new FineFileEntry(prefix);
entry.setDirectory(true); entry.setDirectory(true);
result.add(entry); result.add(entry);
} }
} }
}
@Override @Override
public URL getResource(String path) { public URL getResource(String path) {
@ -319,7 +321,11 @@ public class S3ResourceRepository extends BaseResourceRepository {
result.add(key.substring(key.lastIndexOf(DELIMITER) + 1)); result.add(key.substring(key.lastIndexOf(DELIMITER) + 1));
} }
for (String prefix : objectListing.getCommonPrefixes()) { for (String prefix : objectListing.getCommonPrefixes()) {
result.add(prefix.substring(prefix.lastIndexOf(DELIMITER) + 1)); if (StringUtils.isNotEmpty(prefix.substring(dir.length()).replaceAll(DELIMITER, StringUtils.EMPTY))) {
String[] arr = prefix.split(DELIMITER);
String name = arr[arr.length - 1] + DELIMITER;
result.add(name);
}
} }
} }

Loading…
Cancel
Save