software.amazon.awssdk.services.sagemaker.model.AsyncInferenceClientConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sagemaker Show documentation
The AWS Java SDK for Amazon SageMaker module holds the client classes that are used for communicating with Amazon SageMaker Service
There is a newer version: 2.28.4
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.sagemaker.model;

import java.io.Serializable;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.function.BiConsumer;
import java.util.function.Function;
import software.amazon.awssdk.annotations.Generated;
import software.amazon.awssdk.core.SdkField;
import software.amazon.awssdk.core.SdkPojo;
import software.amazon.awssdk.core.protocol.MarshallLocation;
import software.amazon.awssdk.core.protocol.MarshallingType;
import software.amazon.awssdk.core.traits.LocationTrait;
import software.amazon.awssdk.utils.ToString;
import software.amazon.awssdk.utils.builder.CopyableBuilder;
import software.amazon.awssdk.utils.builder.ToCopyableBuilder;

/**
 * 
 * Configures the behavior of the client used by SageMaker to interact with the model container during asynchronous
 * inference.
 * 
 */
@Generated("software.amazon.awssdk:codegen")
public final class AsyncInferenceClientConfig implements SdkPojo, Serializable,
        ToCopyableBuilder {
    private static final SdkField MAX_CONCURRENT_INVOCATIONS_PER_INSTANCE_FIELD = SdkField
            . builder(MarshallingType.INTEGER)
            .memberName("MaxConcurrentInvocationsPerInstance")
            .getter(getter(AsyncInferenceClientConfig::maxConcurrentInvocationsPerInstance))
            .setter(setter(Builder::maxConcurrentInvocationsPerInstance))
            .traits(LocationTrait.builder().location(MarshallLocation.PAYLOAD)
                    .locationName("MaxConcurrentInvocationsPerInstance").build()).build();

    private static final List> SDK_FIELDS = Collections.unmodifiableList(Arrays
            .asList(MAX_CONCURRENT_INVOCATIONS_PER_INSTANCE_FIELD));

    private static final long serialVersionUID = 1L;

    private final Integer maxConcurrentInvocationsPerInstance;

    private AsyncInferenceClientConfig(BuilderImpl builder) {
        this.maxConcurrentInvocationsPerInstance = builder.maxConcurrentInvocationsPerInstance;
    }

    /**
     * 
     * The maximum number of concurrent requests sent by the SageMaker client to the model container. If no value is
     * provided, SageMaker chooses an optimal value.
     * 
     * 
     * @return The maximum number of concurrent requests sent by the SageMaker client to the model container. If no
     *         value is provided, SageMaker chooses an optimal value.
     */
    public final Integer maxConcurrentInvocationsPerInstance() {
        return maxConcurrentInvocationsPerInstance;
    }

    @Override
    public Builder toBuilder() {
        return new BuilderImpl(this);
    }

    public static Builder builder() {
        return new BuilderImpl();
    }

    public static Class serializableBuilderClass() {
        return BuilderImpl.class;
    }

    @Override
    public final int hashCode() {
        int hashCode = 1;
        hashCode = 31 * hashCode + Objects.hashCode(maxConcurrentInvocationsPerInstance());
        return hashCode;
    }

    @Override
    public final boolean equals(Object obj) {
        return equalsBySdkFields(obj);
    }

    @Override
    public final boolean equalsBySdkFields(Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj == null) {
            return false;
        }
        if (!(obj instanceof AsyncInferenceClientConfig)) {
            return false;
        }
        AsyncInferenceClientConfig other = (AsyncInferenceClientConfig) obj;
        return Objects.equals(maxConcurrentInvocationsPerInstance(), other.maxConcurrentInvocationsPerInstance());
    }

    /**
     * Returns a string representation of this object. This is useful for testing and debugging. Sensitive data will be
     * redacted from this string using a placeholder value.
     */
    @Override
    public final String toString() {
        return ToString.builder("AsyncInferenceClientConfig")
                .add("MaxConcurrentInvocationsPerInstance", maxConcurrentInvocationsPerInstance()).build();
    }

    public final  Optional getValueForField(String fieldName, Class clazz) {
        switch (fieldName) {
        case "MaxConcurrentInvocationsPerInstance":
            return Optional.ofNullable(clazz.cast(maxConcurrentInvocationsPerInstance()));
        default:
            return Optional.empty();
        }
    }

    @Override
    public final List> sdkFields() {
        return SDK_FIELDS;
    }

    private static  Function getter(Function g) {
        return obj -> g.apply((AsyncInferenceClientConfig) obj);
    }

    private static  BiConsumer setter(BiConsumer s) {
        return (obj, val) -> s.accept((Builder) obj, val);
    }

    public interface Builder extends SdkPojo, CopyableBuilder {
        /**
         * 
         * The maximum number of concurrent requests sent by the SageMaker client to the model container. If no value is
         * provided, SageMaker chooses an optimal value.
         * 
         * 
         * @param maxConcurrentInvocationsPerInstance
         *        The maximum number of concurrent requests sent by the SageMaker client to the model container. If no
         *        value is provided, SageMaker chooses an optimal value.
         * @return Returns a reference to this object so that method calls can be chained together.
         */
        Builder maxConcurrentInvocationsPerInstance(Integer maxConcurrentInvocationsPerInstance);
    }

    static final class BuilderImpl implements Builder {
        private Integer maxConcurrentInvocationsPerInstance;

        private BuilderImpl() {
        }

        private BuilderImpl(AsyncInferenceClientConfig model) {
            maxConcurrentInvocationsPerInstance(model.maxConcurrentInvocationsPerInstance);
        }

        public final Integer getMaxConcurrentInvocationsPerInstance() {
            return maxConcurrentInvocationsPerInstance;
        }

        public final void setMaxConcurrentInvocationsPerInstance(Integer maxConcurrentInvocationsPerInstance) {
            this.maxConcurrentInvocationsPerInstance = maxConcurrentInvocationsPerInstance;
        }

        @Override
        public final Builder maxConcurrentInvocationsPerInstance(Integer maxConcurrentInvocationsPerInstance) {
            this.maxConcurrentInvocationsPerInstance = maxConcurrentInvocationsPerInstance;
            return this;
        }

        @Override
        public AsyncInferenceClientConfig build() {
            return new AsyncInferenceClientConfig(this);
        }

        @Override
        public List> sdkFields() {
            return SDK_FIELDS;
        }
    }
}