software.amazon.awssdk.services.sagemaker.model.DesiredWeightAndCapacity Maven / Gradle / Ivy
/*
* 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.Consumer;
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;
/**
*
* Specifies weight and capacity values for a production variant.
*
*/
@Generated("software.amazon.awssdk:codegen")
public final class DesiredWeightAndCapacity implements SdkPojo, Serializable,
ToCopyableBuilder {
private static final SdkField VARIANT_NAME_FIELD = SdkField. builder(MarshallingType.STRING)
.memberName("VariantName").getter(getter(DesiredWeightAndCapacity::variantName)).setter(setter(Builder::variantName))
.traits(LocationTrait.builder().location(MarshallLocation.PAYLOAD).locationName("VariantName").build()).build();
private static final SdkField DESIRED_WEIGHT_FIELD = SdkField. builder(MarshallingType.FLOAT)
.memberName("DesiredWeight").getter(getter(DesiredWeightAndCapacity::desiredWeight))
.setter(setter(Builder::desiredWeight))
.traits(LocationTrait.builder().location(MarshallLocation.PAYLOAD).locationName("DesiredWeight").build()).build();
private static final SdkField DESIRED_INSTANCE_COUNT_FIELD = SdkField. builder(MarshallingType.INTEGER)
.memberName("DesiredInstanceCount").getter(getter(DesiredWeightAndCapacity::desiredInstanceCount))
.setter(setter(Builder::desiredInstanceCount))
.traits(LocationTrait.builder().location(MarshallLocation.PAYLOAD).locationName("DesiredInstanceCount").build())
.build();
private static final SdkField SERVERLESS_UPDATE_CONFIG_FIELD = SdkField
. builder(MarshallingType.SDK_POJO).memberName("ServerlessUpdateConfig")
.getter(getter(DesiredWeightAndCapacity::serverlessUpdateConfig)).setter(setter(Builder::serverlessUpdateConfig))
.constructor(ProductionVariantServerlessUpdateConfig::builder)
.traits(LocationTrait.builder().location(MarshallLocation.PAYLOAD).locationName("ServerlessUpdateConfig").build())
.build();
private static final List> SDK_FIELDS = Collections.unmodifiableList(Arrays.asList(VARIANT_NAME_FIELD,
DESIRED_WEIGHT_FIELD, DESIRED_INSTANCE_COUNT_FIELD, SERVERLESS_UPDATE_CONFIG_FIELD));
private static final long serialVersionUID = 1L;
private final String variantName;
private final Float desiredWeight;
private final Integer desiredInstanceCount;
private final ProductionVariantServerlessUpdateConfig serverlessUpdateConfig;
private DesiredWeightAndCapacity(BuilderImpl builder) {
this.variantName = builder.variantName;
this.desiredWeight = builder.desiredWeight;
this.desiredInstanceCount = builder.desiredInstanceCount;
this.serverlessUpdateConfig = builder.serverlessUpdateConfig;
}
/**
*
* The name of the variant to update.
*
*
* @return The name of the variant to update.
*/
public final String variantName() {
return variantName;
}
/**
*
* The variant's weight.
*
*
* @return The variant's weight.
*/
public final Float desiredWeight() {
return desiredWeight;
}
/**
*
* The variant's capacity.
*
*
* @return The variant's capacity.
*/
public final Integer desiredInstanceCount() {
return desiredInstanceCount;
}
/**
*
* Specifies the serverless update concurrency configuration for an endpoint variant.
*
*
* @return Specifies the serverless update concurrency configuration for an endpoint variant.
*/
public final ProductionVariantServerlessUpdateConfig serverlessUpdateConfig() {
return serverlessUpdateConfig;
}
@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(variantName());
hashCode = 31 * hashCode + Objects.hashCode(desiredWeight());
hashCode = 31 * hashCode + Objects.hashCode(desiredInstanceCount());
hashCode = 31 * hashCode + Objects.hashCode(serverlessUpdateConfig());
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 DesiredWeightAndCapacity)) {
return false;
}
DesiredWeightAndCapacity other = (DesiredWeightAndCapacity) obj;
return Objects.equals(variantName(), other.variantName()) && Objects.equals(desiredWeight(), other.desiredWeight())
&& Objects.equals(desiredInstanceCount(), other.desiredInstanceCount())
&& Objects.equals(serverlessUpdateConfig(), other.serverlessUpdateConfig());
}
/**
* 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("DesiredWeightAndCapacity").add("VariantName", variantName())
.add("DesiredWeight", desiredWeight()).add("DesiredInstanceCount", desiredInstanceCount())
.add("ServerlessUpdateConfig", serverlessUpdateConfig()).build();
}
public final Optional getValueForField(String fieldName, Class clazz) {
switch (fieldName) {
case "VariantName":
return Optional.ofNullable(clazz.cast(variantName()));
case "DesiredWeight":
return Optional.ofNullable(clazz.cast(desiredWeight()));
case "DesiredInstanceCount":
return Optional.ofNullable(clazz.cast(desiredInstanceCount()));
case "ServerlessUpdateConfig":
return Optional.ofNullable(clazz.cast(serverlessUpdateConfig()));
default:
return Optional.empty();
}
}
@Override
public final List> sdkFields() {
return SDK_FIELDS;
}
private static Function getter(Function g) {
return obj -> g.apply((DesiredWeightAndCapacity) obj);
}
private static BiConsumer setter(BiConsumer s) {
return (obj, val) -> s.accept((Builder) obj, val);
}
public interface Builder extends SdkPojo, CopyableBuilder {
/**
*
* The name of the variant to update.
*
*
* @param variantName
* The name of the variant to update.
* @return Returns a reference to this object so that method calls can be chained together.
*/
Builder variantName(String variantName);
/**
*
* The variant's weight.
*
*
* @param desiredWeight
* The variant's weight.
* @return Returns a reference to this object so that method calls can be chained together.
*/
Builder desiredWeight(Float desiredWeight);
/**
*
* The variant's capacity.
*
*
* @param desiredInstanceCount
* The variant's capacity.
* @return Returns a reference to this object so that method calls can be chained together.
*/
Builder desiredInstanceCount(Integer desiredInstanceCount);
/**
*
* Specifies the serverless update concurrency configuration for an endpoint variant.
*
*
* @param serverlessUpdateConfig
* Specifies the serverless update concurrency configuration for an endpoint variant.
* @return Returns a reference to this object so that method calls can be chained together.
*/
Builder serverlessUpdateConfig(ProductionVariantServerlessUpdateConfig serverlessUpdateConfig);
/**
*
* Specifies the serverless update concurrency configuration for an endpoint variant.
*
* This is a convenience method that creates an instance of the
* {@link ProductionVariantServerlessUpdateConfig.Builder} avoiding the need to create one manually via
* {@link ProductionVariantServerlessUpdateConfig#builder()}.
*
*
* When the {@link Consumer} completes, {@link ProductionVariantServerlessUpdateConfig.Builder#build()} is
* called immediately and its result is passed to
* {@link #serverlessUpdateConfig(ProductionVariantServerlessUpdateConfig)}.
*
* @param serverlessUpdateConfig
* a consumer that will call methods on {@link ProductionVariantServerlessUpdateConfig.Builder}
* @return Returns a reference to this object so that method calls can be chained together.
* @see #serverlessUpdateConfig(ProductionVariantServerlessUpdateConfig)
*/
default Builder serverlessUpdateConfig(Consumer serverlessUpdateConfig) {
return serverlessUpdateConfig(ProductionVariantServerlessUpdateConfig.builder().applyMutation(serverlessUpdateConfig)
.build());
}
}
static final class BuilderImpl implements Builder {
private String variantName;
private Float desiredWeight;
private Integer desiredInstanceCount;
private ProductionVariantServerlessUpdateConfig serverlessUpdateConfig;
private BuilderImpl() {
}
private BuilderImpl(DesiredWeightAndCapacity model) {
variantName(model.variantName);
desiredWeight(model.desiredWeight);
desiredInstanceCount(model.desiredInstanceCount);
serverlessUpdateConfig(model.serverlessUpdateConfig);
}
public final String getVariantName() {
return variantName;
}
public final void setVariantName(String variantName) {
this.variantName = variantName;
}
@Override
public final Builder variantName(String variantName) {
this.variantName = variantName;
return this;
}
public final Float getDesiredWeight() {
return desiredWeight;
}
public final void setDesiredWeight(Float desiredWeight) {
this.desiredWeight = desiredWeight;
}
@Override
public final Builder desiredWeight(Float desiredWeight) {
this.desiredWeight = desiredWeight;
return this;
}
public final Integer getDesiredInstanceCount() {
return desiredInstanceCount;
}
public final void setDesiredInstanceCount(Integer desiredInstanceCount) {
this.desiredInstanceCount = desiredInstanceCount;
}
@Override
public final Builder desiredInstanceCount(Integer desiredInstanceCount) {
this.desiredInstanceCount = desiredInstanceCount;
return this;
}
public final ProductionVariantServerlessUpdateConfig.Builder getServerlessUpdateConfig() {
return serverlessUpdateConfig != null ? serverlessUpdateConfig.toBuilder() : null;
}
public final void setServerlessUpdateConfig(ProductionVariantServerlessUpdateConfig.BuilderImpl serverlessUpdateConfig) {
this.serverlessUpdateConfig = serverlessUpdateConfig != null ? serverlessUpdateConfig.build() : null;
}
@Override
public final Builder serverlessUpdateConfig(ProductionVariantServerlessUpdateConfig serverlessUpdateConfig) {
this.serverlessUpdateConfig = serverlessUpdateConfig;
return this;
}
@Override
public DesiredWeightAndCapacity build() {
return new DesiredWeightAndCapacity(this);
}
@Override
public List> sdkFields() {
return SDK_FIELDS;
}
}
}