software.amazon.awssdk.services.bedrockagent.model.GetDataSourceRequest Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of bedrockagent Show documentation
Show all versions of bedrockagent Show documentation
The AWS Java SDK for Bedrock Agent module holds the client classes that are used for
communicating with Bedrock Agent.
/*
* 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.bedrockagent.model;
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.awscore.AwsRequestOverrideConfiguration;
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;
/**
*/
@Generated("software.amazon.awssdk:codegen")
public final class GetDataSourceRequest extends BedrockAgentRequest implements
ToCopyableBuilder {
private static final SdkField DATA_SOURCE_ID_FIELD = SdkField. builder(MarshallingType.STRING)
.memberName("dataSourceId").getter(getter(GetDataSourceRequest::dataSourceId)).setter(setter(Builder::dataSourceId))
.traits(LocationTrait.builder().location(MarshallLocation.PATH).locationName("dataSourceId").build()).build();
private static final SdkField KNOWLEDGE_BASE_ID_FIELD = SdkField. builder(MarshallingType.STRING)
.memberName("knowledgeBaseId").getter(getter(GetDataSourceRequest::knowledgeBaseId))
.setter(setter(Builder::knowledgeBaseId))
.traits(LocationTrait.builder().location(MarshallLocation.PATH).locationName("knowledgeBaseId").build()).build();
private static final List> SDK_FIELDS = Collections.unmodifiableList(Arrays.asList(DATA_SOURCE_ID_FIELD,
KNOWLEDGE_BASE_ID_FIELD));
private final String dataSourceId;
private final String knowledgeBaseId;
private GetDataSourceRequest(BuilderImpl builder) {
super(builder);
this.dataSourceId = builder.dataSourceId;
this.knowledgeBaseId = builder.knowledgeBaseId;
}
/**
*
* The unique identifier of the data source.
*
*
* @return The unique identifier of the data source.
*/
public final String dataSourceId() {
return dataSourceId;
}
/**
*
* The unique identifier of the knowledge base that the data source was added to.
*
*
* @return The unique identifier of the knowledge base that the data source was added to.
*/
public final String knowledgeBaseId() {
return knowledgeBaseId;
}
@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 + super.hashCode();
hashCode = 31 * hashCode + Objects.hashCode(dataSourceId());
hashCode = 31 * hashCode + Objects.hashCode(knowledgeBaseId());
return hashCode;
}
@Override
public final boolean equals(Object obj) {
return super.equals(obj) && equalsBySdkFields(obj);
}
@Override
public final boolean equalsBySdkFields(Object obj) {
if (this == obj) {
return true;
}
if (obj == null) {
return false;
}
if (!(obj instanceof GetDataSourceRequest)) {
return false;
}
GetDataSourceRequest other = (GetDataSourceRequest) obj;
return Objects.equals(dataSourceId(), other.dataSourceId()) && Objects.equals(knowledgeBaseId(), other.knowledgeBaseId());
}
/**
* 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("GetDataSourceRequest").add("DataSourceId", dataSourceId())
.add("KnowledgeBaseId", knowledgeBaseId()).build();
}
public final Optional getValueForField(String fieldName, Class clazz) {
switch (fieldName) {
case "dataSourceId":
return Optional.ofNullable(clazz.cast(dataSourceId()));
case "knowledgeBaseId":
return Optional.ofNullable(clazz.cast(knowledgeBaseId()));
default:
return Optional.empty();
}
}
@Override
public final List> sdkFields() {
return SDK_FIELDS;
}
private static Function getter(Function g) {
return obj -> g.apply((GetDataSourceRequest) obj);
}
private static BiConsumer setter(BiConsumer s) {
return (obj, val) -> s.accept((Builder) obj, val);
}
public interface Builder extends BedrockAgentRequest.Builder, SdkPojo, CopyableBuilder {
/**
*
* The unique identifier of the data source.
*
*
* @param dataSourceId
* The unique identifier of the data source.
* @return Returns a reference to this object so that method calls can be chained together.
*/
Builder dataSourceId(String dataSourceId);
/**
*
* The unique identifier of the knowledge base that the data source was added to.
*
*
* @param knowledgeBaseId
* The unique identifier of the knowledge base that the data source was added to.
* @return Returns a reference to this object so that method calls can be chained together.
*/
Builder knowledgeBaseId(String knowledgeBaseId);
@Override
Builder overrideConfiguration(AwsRequestOverrideConfiguration overrideConfiguration);
@Override
Builder overrideConfiguration(Consumer builderConsumer);
}
static final class BuilderImpl extends BedrockAgentRequest.BuilderImpl implements Builder {
private String dataSourceId;
private String knowledgeBaseId;
private BuilderImpl() {
}
private BuilderImpl(GetDataSourceRequest model) {
super(model);
dataSourceId(model.dataSourceId);
knowledgeBaseId(model.knowledgeBaseId);
}
public final String getDataSourceId() {
return dataSourceId;
}
public final void setDataSourceId(String dataSourceId) {
this.dataSourceId = dataSourceId;
}
@Override
public final Builder dataSourceId(String dataSourceId) {
this.dataSourceId = dataSourceId;
return this;
}
public final String getKnowledgeBaseId() {
return knowledgeBaseId;
}
public final void setKnowledgeBaseId(String knowledgeBaseId) {
this.knowledgeBaseId = knowledgeBaseId;
}
@Override
public final Builder knowledgeBaseId(String knowledgeBaseId) {
this.knowledgeBaseId = knowledgeBaseId;
return this;
}
@Override
public Builder overrideConfiguration(AwsRequestOverrideConfiguration overrideConfiguration) {
super.overrideConfiguration(overrideConfiguration);
return this;
}
@Override
public Builder overrideConfiguration(Consumer builderConsumer) {
super.overrideConfiguration(builderConsumer);
return this;
}
@Override
public GetDataSourceRequest build() {
return new GetDataSourceRequest(this);
}
@Override
public List> sdkFields() {
return SDK_FIELDS;
}
}
}