Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
/*
* SPDX-License-Identifier: Apache-2.0
* Copyright Blazebit
*/
package com.blazebit.query.connector.aws.s3;

import com.blazebit.query.connector.aws.base.AwsWrapper;
import software.amazon.awssdk.services.s3.model.LifecycleRule;

/**
* @author Donghwi Kim
* @since 1.0.0
*/
public class AwsLifeCycleRule extends AwsWrapper<LifecycleRule> {
public AwsLifeCycleRule(String accountId, String region, String resourceId, LifecycleRule payload) {
super( accountId, region, resourceId, payload );
}

@Override
public LifecycleRule getPayload() {
return super.getPayload();
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ public Set<? extends DataFetcher<?>> resolveSchemaObjects(ConfigurationProvider
return Set.of(
BucketDataFetcher.INSTANCE,
BucketAclFetcher.INSTANCE,
LifecycleRuleFetcher.INSTANCE,
LoggingEnabledFetcher.INSTANCE,
ObjectLockConfigurationFetcher.INSTANCE,
PolicyStatusFetcher.INSTANCE,
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,90 @@
/*
* SPDX-License-Identifier: Apache-2.0
* Copyright Blazebit
*/
package com.blazebit.query.connector.aws.s3;

import com.blazebit.query.connector.aws.base.AwsConnectorConfig;
import com.blazebit.query.connector.aws.base.AwsConventionContext;
import com.blazebit.query.connector.base.DataFormats;
import com.blazebit.query.spi.DataFetchContext;
import com.blazebit.query.spi.DataFetcher;
import com.blazebit.query.spi.DataFetcherException;
import com.blazebit.query.spi.DataFormat;
import software.amazon.awssdk.http.SdkHttpClient;
import software.amazon.awssdk.regions.Region;
import software.amazon.awssdk.services.s3.S3Client;
import software.amazon.awssdk.services.s3.S3ClientBuilder;
import software.amazon.awssdk.services.s3.model.Bucket;
import software.amazon.awssdk.services.s3.model.GetBucketLifecycleConfigurationRequest;
import software.amazon.awssdk.services.s3.model.LifecycleRule;
import software.amazon.awssdk.services.s3.model.S3Exception;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

/**
* @author Donghwi Kim
* @since 1.0.0
*/
public class LifecycleRuleFetcher implements DataFetcher<AwsLifeCycleRule>, Serializable {

public static final LifecycleRuleFetcher INSTANCE = new LifecycleRuleFetcher();

private LifecycleRuleFetcher() {
}

@Override
public List<AwsLifeCycleRule> fetch(DataFetchContext context) {
try {
List<AwsConnectorConfig.Account> accounts = AwsConnectorConfig.ACCOUNT.getAll( context );
SdkHttpClient sdkHttpClient = AwsConnectorConfig.HTTP_CLIENT.find( context );
List<AwsLifeCycleRule> list = new ArrayList<>();
for ( AwsConnectorConfig.Account account : accounts ) {
for ( Region region : account.getRegions() ) {
S3ClientBuilder s3ClientBuilder = S3Client.builder()
.region( region )
.credentialsProvider( account.getCredentialsProvider() );
if ( sdkHttpClient != null ) {
s3ClientBuilder.httpClient( sdkHttpClient );
}
try (S3Client client = s3ClientBuilder.build()) {
for ( Bucket bucket : client.listBuckets().buckets() ) {
try {
var bucketLifecycleConfiguration = client.getBucketLifecycleConfiguration(
GetBucketLifecycleConfigurationRequest.builder().bucket( bucket.name() )
.build() );
for ( LifecycleRule lifecycleRule : bucketLifecycleConfiguration.rules() ) {
list.add( new AwsLifeCycleRule(
account.getAccountId(),
region.id(),
bucket.name(),
lifecycleRule
) );
}
}
catch (S3Exception e) {
if ( Objects.equals( e.awsErrorDetails().errorCode(),
"NoSuchLifecycleConfiguration" ) ) {
continue;
}
throw e;
}
}
}
}
}
return list;
}
catch (Exception e) {
throw new DataFetcherException( "Could not fetch life cycle rule list", e );
}
}

@Override
public DataFormat getDataFormat() {
return DataFormats.componentMethodConvention( AwsLifeCycleRule.class, AwsConventionContext.INSTANCE );
}
}
8 changes: 8 additions & 0 deletions examples/app/src/main/java/com/blazebit/query/app/Main.java
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@
import com.blazebit.query.connector.aws.s3.AwsLoggingEnabled;
import com.blazebit.query.connector.aws.s3.AwsObjectLockConfiguration;
import com.blazebit.query.connector.aws.s3.AwsPolicyStatus;
import com.blazebit.query.connector.aws.s3.AwsLifeCycleRule;
import com.blazebit.query.connector.aws.s3.AwsPublicAccessBlockConfiguration;
import com.blazebit.query.connector.aws.s3.AwsBucket;
import com.blazebit.query.connector.aws.s3.AwsServerSideEncryptionRule;
Expand Down Expand Up @@ -268,6 +269,7 @@ public static void main(String[] args) throws Exception {
// S3
queryContextBuilder.registerSchemaObjectAlias( AwsBucket.class, "AwsBucket" );
queryContextBuilder.registerSchemaObjectAlias( AwsBucketAcl.class, "AwsBucketAcl" );
queryContextBuilder.registerSchemaObjectAlias( AwsLifeCycleRule.class, "AwsLifeCycleRule" );
queryContextBuilder.registerSchemaObjectAlias( AwsLoggingEnabled.class, "AwsLoggingEnabled" );
queryContextBuilder.registerSchemaObjectAlias( AwsObjectLockConfiguration.class, "AwsObjectLockConfiguration" );
queryContextBuilder.registerSchemaObjectAlias( AwsPolicyStatus.class, "AwsPolicyStatus" );
Expand Down Expand Up @@ -521,6 +523,12 @@ private static void testAws(QuerySession session) {
System.out.println("AwsPolicyStatus");
print(awsPolicyStatusResult);

TypedQuery<Object[]> awsLifeCycleRuleQuery = session.createQuery(
"select f.* from AwsLifeCycleRule f" );
List<Object[]> awsLifeCycleRuleResult = awsLifeCycleRuleQuery.getResultList();
System.out.println("AwsLifeCycleRule");
print(awsLifeCycleRuleResult);

TypedQuery<Object[]> awsPublicAccessBlockConfigurationQuery = session.createQuery(
"select f.* from AwsPublicAccessBlockConfiguration f" );
List<Object[]> awsPublicAccessBlockConfigurationResult = awsPublicAccessBlockConfigurationQuery.getResultList();
Expand Down