software.amazon.awssdk.services.sagemaker.model.ParallelismConfiguration 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.31.2
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.HashMap;
import java.util.List;
import java.util.Map;
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;

/**
 * 
 * Configuration that controls the parallelism of the pipeline. By default, the parallelism configuration specified
 * applies to all executions of the pipeline unless overridden.
 * 
 */
@Generated("software.amazon.awssdk:codegen")
public final class ParallelismConfiguration implements SdkPojo, Serializable,
        ToCopyableBuilder {
    private static final SdkField MAX_PARALLEL_EXECUTION_STEPS_FIELD = SdkField
            . builder(MarshallingType.INTEGER).memberName("MaxParallelExecutionSteps")
            .getter(getter(ParallelismConfiguration::maxParallelExecutionSteps))
            .setter(setter(Builder::maxParallelExecutionSteps))
            .traits(LocationTrait.builder().location(MarshallLocation.PAYLOAD).locationName("MaxParallelExecutionSteps").build())
            .build();

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

    private static final Map> SDK_NAME_TO_FIELD = memberNameToFieldInitializer();

    private static final long serialVersionUID = 1L;

    private final Integer maxParallelExecutionSteps;

    private ParallelismConfiguration(BuilderImpl builder) {
        this.maxParallelExecutionSteps = builder.maxParallelExecutionSteps;
    }

    /**
     * 
     * The max number of steps that can be executed in parallel.
     * 
     * 
     * @return The max number of steps that can be executed in parallel.
     */
    public final Integer maxParallelExecutionSteps() {
        return maxParallelExecutionSteps;
    }

    @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(maxParallelExecutionSteps());
        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 ParallelismConfiguration)) {
            return false;
        }
        ParallelismConfiguration other = (ParallelismConfiguration) obj;
        return Objects.equals(maxParallelExecutionSteps(), other.maxParallelExecutionSteps());
    }

    /**
     * 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("ParallelismConfiguration").add("MaxParallelExecutionSteps", maxParallelExecutionSteps()).build();
    }

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

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

    @Override
    public final Map> sdkFieldNameToField() {
        return SDK_NAME_TO_FIELD;
    }

    private static Map> memberNameToFieldInitializer() {
        Map> map = new HashMap<>();
        map.put("MaxParallelExecutionSteps", MAX_PARALLEL_EXECUTION_STEPS_FIELD);
        return Collections.unmodifiableMap(map);
    }

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

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

    public interface Builder extends SdkPojo, CopyableBuilder {
        /**
         * 
         * The max number of steps that can be executed in parallel.
         * 
         * 
         * @param maxParallelExecutionSteps
         *        The max number of steps that can be executed in parallel.
         * @return Returns a reference to this object so that method calls can be chained together.
         */
        Builder maxParallelExecutionSteps(Integer maxParallelExecutionSteps);
    }

    static final class BuilderImpl implements Builder {
        private Integer maxParallelExecutionSteps;

        private BuilderImpl() {
        }

        private BuilderImpl(ParallelismConfiguration model) {
            maxParallelExecutionSteps(model.maxParallelExecutionSteps);
        }

        public final Integer getMaxParallelExecutionSteps() {
            return maxParallelExecutionSteps;
        }

        public final void setMaxParallelExecutionSteps(Integer maxParallelExecutionSteps) {
            this.maxParallelExecutionSteps = maxParallelExecutionSteps;
        }

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

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

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

        @Override
        public Map> sdkFieldNameToField() {
            return SDK_NAME_TO_FIELD;
        }
    }
}