Add Aws S3 bucket versioning fetcher (#75)

Author: donghwikim2

connector/aws/s3/src/main/java/com/blazebit/query/connector/aws/s3/AwsBucketVersioning.java

@@ -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.GetBucketVersioningResponse;
+
+/**
+ * @author Donghwi Kim
+ * @since 1.0.0
+ */
+public class AwsBucketVersioning extends AwsWrapper<GetBucketVersioningResponse> {
+	public AwsBucketVersioning(String accountId, String region, String resourceId, GetBucketVersioningResponse payload) {
+		super( accountId, region, resourceId, payload );
+	}
+
+	@Override
+	public GetBucketVersioningResponse getPayload() {
+		return super.getPayload();
+	}
+}

connector/aws/s3/src/main/java/com/blazebit/query/connector/aws/s3/AwsS3SchemaProvider.java

@@ -22,6 +22,7 @@ public Set<? extends DataFetcher<?>> resolveSchemaObjects(ConfigurationProvider
 		return Set.of(
 				BucketDataFetcher.INSTANCE,
 				BucketAclFetcher.INSTANCE,
+				BucketVersioningFetcher.INSTANCE,
 				BucketPolicyFetcher.INSTANCE,
 				LifecycleRuleFetcher.INSTANCE,
 				LoggingEnabledFetcher.INSTANCE,

connector/aws/s3/src/main/java/com/blazebit/query/connector/aws/s3/BucketVersioningFetcher.java

@@ -0,0 +1,76 @@
+/*
+ * 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.GetBucketVersioningRequest;
+
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * @author Donghwi Kim
+ * @since 1.0.0
+ */
+public class BucketVersioningFetcher implements DataFetcher<AwsBucketVersioning>, Serializable {
+
+	public static final BucketVersioningFetcher INSTANCE = new BucketVersioningFetcher();
+
+	private BucketVersioningFetcher() {
+	}
+
+	@Override
+	public List<AwsBucketVersioning> fetch(DataFetchContext context) {
+		try {
+			List<AwsConnectorConfig.Account> accounts = AwsConnectorConfig.ACCOUNT.getAll( context );
+			SdkHttpClient sdkHttpClient = AwsConnectorConfig.HTTP_CLIENT.find( context );
+			List<AwsBucketVersioning> 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() ) {
+							var bucketVersioning = client.getBucketVersioning(
+									GetBucketVersioningRequest.builder().bucket( bucket.name() )
+											.build() );
+							list.add( new AwsBucketVersioning(
+									account.getAccountId(),
+									region.id(),
+									bucket.name(),
+									bucketVersioning
+							) );
+						}
+					}
+				}
+			}
+			return list;
+		}
+		catch (Exception e) {
+			throw new DataFetcherException( "Could not fetch bucket versioning list", e );
+		}
+	}
+
+	@Override
+	public DataFormat getDataFormat() {
+		return DataFormats.componentMethodConvention( AwsBucketVersioning.class, AwsConventionContext.INSTANCE );
+	}
+}

examples/app/src/main/java/com/blazebit/query/app/Main.java

@@ -45,6 +45,7 @@
 import com.blazebit.query.connector.aws.route53.AwsHealthCheck;
 import com.blazebit.query.connector.aws.route53.AwsHostedZone;
 import com.blazebit.query.connector.aws.s3.AwsBucketAcl;
+import com.blazebit.query.connector.aws.s3.AwsBucketVersioning;
 import com.blazebit.query.connector.aws.s3.AwsBucketPolicy;
 import com.blazebit.query.connector.aws.s3.AwsLoggingEnabled;
 import com.blazebit.query.connector.aws.s3.AwsObjectLockConfiguration;
@@ -282,6 +283,7 @@ public static void main(String[] args) throws Exception {
 			// S3
 			queryContextBuilder.registerSchemaObjectAlias( AwsBucket.class, "AwsBucket" );
 			queryContextBuilder.registerSchemaObjectAlias( AwsBucketAcl.class, "AwsBucketAcl" );
+			queryContextBuilder.registerSchemaObjectAlias( AwsBucketVersioning.class, "AwsBucketVersioning" );
 			queryContextBuilder.registerSchemaObjectAlias( AwsBucketPolicy.class, "AwsBucketPolicy" );
 			queryContextBuilder.registerSchemaObjectAlias( AwsLifeCycleRule.class, "AwsLifeCycleRule" );
 			queryContextBuilder.registerSchemaObjectAlias( AwsLoggingEnabled.class, "AwsLoggingEnabled" );
@@ -552,7 +554,13 @@ private static void testAws(QuerySession session) {
 		System.out.println("AwsBucketAcl");
 		print(awsBucketAclResult);
 
-		TypedQuery<Object[]> awsBucketPolicyQuery = session.createQuery(
+		TypedQuery<Object[]> awsBucketVersioningQuery = session.createQuery(
+				"select f.* from AwsBucketVersioning f" );
+		List<Object[]> awsBucketVersioningResult = awsBucketVersioningQuery.getResultList();
+		System.out.println("AwsBucketVersioning");
+		print(awsBucketVersioningResult);
+
+	TypedQuery<Object[]> awsBucketPolicyQuery = session.createQuery(
 				"select f.* from AwsBucketPolicy f" );
 		List<Object[]> awsBucketPolicyResult = awsBucketPolicyQuery.getResultList();
 		System.out.println("AwsBucketPolicy");