software.amazon.awssdk.services.sagemakergeospatial.endpoints.internal.DefaultSageMakerGeospatialEndpointProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sagemakergeospatial Show documentation
The AWS Java SDK for Sage Maker Geospatial module holds the client classes that are used for communicating with Sage Maker Geospatial.
There is a newer version: 2.29.39
Show newest version
/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance with
 * the License. A copy of the License is located at
 * 
 * http://aws.amazon.com/apache2.0
 * 
 * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
 * CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions
 * and limitations under the License.
 */

package software.amazon.awssdk.services.sagemakergeospatial.endpoints.internal;

import java.net.URI;
import java.util.concurrent.CompletableFuture;
import software.amazon.awssdk.annotations.Generated;
import software.amazon.awssdk.annotations.SdkInternalApi;
import software.amazon.awssdk.core.exception.SdkClientException;
import software.amazon.awssdk.endpoints.Endpoint;
import software.amazon.awssdk.regions.Region;
import software.amazon.awssdk.services.sagemakergeospatial.endpoints.SageMakerGeospatialEndpointParams;
import software.amazon.awssdk.services.sagemakergeospatial.endpoints.SageMakerGeospatialEndpointProvider;
import software.amazon.awssdk.utils.CompletableFutureUtils;
import software.amazon.awssdk.utils.Validate;

@Generated("software.amazon.awssdk:codegen")
@SdkInternalApi
public final class DefaultSageMakerGeospatialEndpointProvider implements SageMakerGeospatialEndpointProvider {
    @Override
    public CompletableFuture resolveEndpoint(SageMakerGeospatialEndpointParams params) {
        Validate.notNull(params.useDualStack(), "Parameter 'UseDualStack' must not be null");
        Validate.notNull(params.useFips(), "Parameter 'UseFIPS' must not be null");
        try {
            RuleResult result = endpointRule0(params, new LocalState(params.region()));
            if (result.canContinue()) {
                throw SdkClientException.create("Rule engine did not reach an error or endpoint result");
            }
            if (result.isError()) {
                String errorMsg = result.error();
                if (errorMsg.contains("Invalid ARN") && errorMsg.contains(":s3:::")) {
                    errorMsg += ". Use the bucket name instead of simple bucket ARNs in GetBucketLocationRequest.";
                }
                throw SdkClientException.create(errorMsg);
            }
            return CompletableFuture.completedFuture(result.endpoint());
        } catch (Exception error) {
            return CompletableFutureUtils.failedFuture(error);
        }
    }

    private static RuleResult endpointRule0(SageMakerGeospatialEndpointParams params, LocalState locals) {
        RuleResult result = endpointRule1(params, locals);
        if (result.isResolved()) {
            return result;
        }
        return endpointRule6(params, locals);
    }

    private static RuleResult endpointRule1(SageMakerGeospatialEndpointParams params, LocalState locals) {
        if (params.endpoint() != null) {
            RuleResult result = endpointRule2(params, locals);
            if (result.isResolved()) {
                return result;
            }
            return endpointRule3(params, locals);
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule2(SageMakerGeospatialEndpointParams params, LocalState locals) {
        if (params.useFips()) {
            return RuleResult.error("Invalid Configuration: FIPS and custom endpoint are not supported");
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule3(SageMakerGeospatialEndpointParams params, LocalState locals) {
        RuleResult result = endpointRule4(params, locals);
        if (result.isResolved()) {
            return result;
        }
        return endpointRule5(params, locals);
    }

    private static RuleResult endpointRule4(SageMakerGeospatialEndpointParams params, LocalState locals) {
        if (params.useDualStack()) {
            return RuleResult.error("Invalid Configuration: Dualstack and custom endpoint are not supported");
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule5(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return RuleResult.endpoint(Endpoint.builder().url(URI.create(params.endpoint())).build());
    }

    private static RuleResult endpointRule6(SageMakerGeospatialEndpointParams params, LocalState locals) {
        RuleResult result = endpointRule7(params, locals);
        if (result.isResolved()) {
            return result;
        }
        return endpointRule26(params, locals);
    }

    private static RuleResult endpointRule7(SageMakerGeospatialEndpointParams params, LocalState locals) {
        if (locals.region() != null) {
            return endpointRule8(params, locals);
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule8(SageMakerGeospatialEndpointParams params, LocalState locals) {
        RulePartition partitionResult = null;
        if ((partitionResult = RulesFunctions.awsPartition(locals.region())) != null) {
            locals = locals.toBuilder().partitionResult(partitionResult).build();
            RuleResult result = endpointRule9(params, locals);
            if (result.isResolved()) {
                return result;
            }
            result = endpointRule14(params, locals);
            if (result.isResolved()) {
                return result;
            }
            result = endpointRule19(params, locals);
            if (result.isResolved()) {
                return result;
            }
            return endpointRule24(params, locals);
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule9(SageMakerGeospatialEndpointParams params, LocalState locals) {
        if (params.useFips() && params.useDualStack()) {
            RuleResult result = endpointRule10(params, locals);
            if (result.isResolved()) {
                return result;
            }
            return endpointRule13(params, locals);
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule10(SageMakerGeospatialEndpointParams params, LocalState locals) {
        if (locals.partitionResult().supportsFIPS() && locals.partitionResult().supportsDualStack()) {
            return endpointRule11(params, locals);
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule11(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return endpointRule12(params, locals);
    }

    private static RuleResult endpointRule12(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return RuleResult.endpoint(Endpoint
                .builder()
                .url(URI.create("https://sagemaker-geospatial-fips." + locals.region() + "."
                        + locals.partitionResult().dualStackDnsSuffix())).build());
    }

    private static RuleResult endpointRule13(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return RuleResult.error("FIPS and DualStack are enabled, but this partition does not support one or both");
    }

    private static RuleResult endpointRule14(SageMakerGeospatialEndpointParams params, LocalState locals) {
        if (params.useFips()) {
            RuleResult result = endpointRule15(params, locals);
            if (result.isResolved()) {
                return result;
            }
            return endpointRule18(params, locals);
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule15(SageMakerGeospatialEndpointParams params, LocalState locals) {
        if (locals.partitionResult().supportsFIPS()) {
            return endpointRule16(params, locals);
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule16(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return endpointRule17(params, locals);
    }

    private static RuleResult endpointRule17(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return RuleResult.endpoint(Endpoint
                .builder()
                .url(URI.create("https://sagemaker-geospatial-fips." + locals.region() + "."
                        + locals.partitionResult().dnsSuffix())).build());
    }

    private static RuleResult endpointRule18(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return RuleResult.error("FIPS is enabled but this partition does not support FIPS");
    }

    private static RuleResult endpointRule19(SageMakerGeospatialEndpointParams params, LocalState locals) {
        if (params.useDualStack()) {
            RuleResult result = endpointRule20(params, locals);
            if (result.isResolved()) {
                return result;
            }
            return endpointRule23(params, locals);
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule20(SageMakerGeospatialEndpointParams params, LocalState locals) {
        if (locals.partitionResult().supportsDualStack()) {
            return endpointRule21(params, locals);
        }
        return RuleResult.carryOn();
    }

    private static RuleResult endpointRule21(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return endpointRule22(params, locals);
    }

    private static RuleResult endpointRule22(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return RuleResult.endpoint(Endpoint
                .builder()
                .url(URI.create("https://sagemaker-geospatial." + locals.region() + "."
                        + locals.partitionResult().dualStackDnsSuffix())).build());
    }

    private static RuleResult endpointRule23(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return RuleResult.error("DualStack is enabled but this partition does not support DualStack");
    }

    private static RuleResult endpointRule24(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return endpointRule25(params, locals);
    }

    private static RuleResult endpointRule25(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return RuleResult.endpoint(Endpoint.builder()
                .url(URI.create("https://sagemaker-geospatial." + locals.region() + "." + locals.partitionResult().dnsSuffix()))
                .build());
    }

    private static RuleResult endpointRule26(SageMakerGeospatialEndpointParams params, LocalState locals) {
        return RuleResult.error("Invalid Configuration: Missing Region");
    }

    @Override
    public boolean equals(Object rhs) {
        return rhs != null && getClass().equals(rhs.getClass());
    }

    @Override
    public int hashCode() {
        return getClass().hashCode();
    }

    private static final class LocalState {
        private final String region;

        private final RulePartition partitionResult;

        LocalState() {
            this.region = null;
            this.partitionResult = null;
        }

        LocalState(Region region) {
            if (region != null) {
                this.region = region.id();
            } else {
                this.region = null;
            }
            this.partitionResult = null;
        }

        LocalState(LocalStateBuilder builder) {
            this.region = builder.region;
            this.partitionResult = builder.partitionResult;
        }

        public String region() {
            return this.region;
        }

        public RulePartition partitionResult() {
            return this.partitionResult;
        }

        public LocalStateBuilder toBuilder() {
            return new LocalStateBuilder(this);
        }
    }

    private static final class LocalStateBuilder {
        private String region;

        private RulePartition partitionResult;

        LocalStateBuilder() {
            this.region = null;
            this.partitionResult = null;
        }

        LocalStateBuilder(LocalState locals) {
            this.region = locals.region;
            this.partitionResult = locals.partitionResult;
        }

        public LocalStateBuilder region(String value) {
            this.region = value;
            return this;
        }

        public LocalStateBuilder partitionResult(RulePartition value) {
            this.partitionResult = value;
            return this;
        }

        LocalState build() {
            return new LocalState(this);
        }
    }
}