software.amazon.awssdk.services.athena.model.QueryExecutionContext Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of athena Show documentation
The AWS Java SDK for Amazon Athena module holds the client classes that are used for communicating with Amazon Athena 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.athena.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;

/**
 * 
 * The database and data catalog context in which the query execution occurs.
 * 
 */
@Generated("software.amazon.awssdk:codegen")
public final class QueryExecutionContext implements SdkPojo, Serializable,
        ToCopyableBuilder {
    private static final SdkField DATABASE_FIELD = SdkField. builder(MarshallingType.STRING)
            .getter(getter(QueryExecutionContext::database)).setter(setter(Builder::database))
            .traits(LocationTrait.builder().location(MarshallLocation.PAYLOAD).locationName("Database").build()).build();

    private static final SdkField CATALOG_FIELD = SdkField. builder(MarshallingType.STRING)
            .getter(getter(QueryExecutionContext::catalog)).setter(setter(Builder::catalog))
            .traits(LocationTrait.builder().location(MarshallLocation.PAYLOAD).locationName("Catalog").build()).build();

    private static final List> SDK_FIELDS = Collections
            .unmodifiableList(Arrays.asList(DATABASE_FIELD, CATALOG_FIELD));

    private static final long serialVersionUID = 1L;

    private final String database;

    private final String catalog;

    private QueryExecutionContext(BuilderImpl builder) {
        this.database = builder.database;
        this.catalog = builder.catalog;
    }

    /**
     * 
     * The name of the database used in the query execution.
     * 
     * 
     * @return The name of the database used in the query execution.
     */
    public String database() {
        return database;
    }

    /**
     * 
     * The name of the data catalog used in the query execution.
     * 
     * 
     * @return The name of the data catalog used in the query execution.
     */
    public String catalog() {
        return catalog;
    }

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

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

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

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

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

    @Override
    public boolean equalsBySdkFields(Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj == null) {
            return false;
        }
        if (!(obj instanceof QueryExecutionContext)) {
            return false;
        }
        QueryExecutionContext other = (QueryExecutionContext) obj;
        return Objects.equals(database(), other.database()) && Objects.equals(catalog(), other.catalog());
    }

    /**
     * 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 String toString() {
        return ToString.builder("QueryExecutionContext").add("Database", database()).add("Catalog", catalog()).build();
    }

    public  Optional getValueForField(String fieldName, Class clazz) {
        switch (fieldName) {
        case "Database":
            return Optional.ofNullable(clazz.cast(database()));
        case "Catalog":
            return Optional.ofNullable(clazz.cast(catalog()));
        default:
            return Optional.empty();
        }
    }

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

    private static  Function getter(Function g) {
        return obj -> g.apply((QueryExecutionContext) 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 database used in the query execution.
         * 
         * 
         * @param database
         *        The name of the database used in the query execution.
         * @return Returns a reference to this object so that method calls can be chained together.
         */
        Builder database(String database);

        /**
         * 
         * The name of the data catalog used in the query execution.
         * 
         * 
         * @param catalog
         *        The name of the data catalog used in the query execution.
         * @return Returns a reference to this object so that method calls can be chained together.
         */
        Builder catalog(String catalog);
    }

    static final class BuilderImpl implements Builder {
        private String database;

        private String catalog;

        private BuilderImpl() {
        }

        private BuilderImpl(QueryExecutionContext model) {
            database(model.database);
            catalog(model.catalog);
        }

        public final String getDatabase() {
            return database;
        }

        @Override
        public final Builder database(String database) {
            this.database = database;
            return this;
        }

        public final void setDatabase(String database) {
            this.database = database;
        }

        public final String getCatalog() {
            return catalog;
        }

        @Override
        public final Builder catalog(String catalog) {
            this.catalog = catalog;
            return this;
        }

        public final void setCatalog(String catalog) {
            this.catalog = catalog;
        }

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

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