io.prestosql.jdbc.Row Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of presto-jdbc Show documentation
There is a newer version: 350
/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License 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 io.prestosql.jdbc;

import io.prestosql.jdbc.$internal.javax.annotation.Nullable;

import java.util.List;
import java.util.Objects;
import java.util.Optional;

import static io.prestosql.jdbc.$internal.guava.collect.ImmutableList.toImmutableList;
import static java.util.Objects.requireNonNull;

// A public facade for Row from presto-client
public final class Row
{
    private final io.prestosql.jdbc.$internal.client.Row row;

    Row(io.prestosql.jdbc.$internal.client.Row row)
    {
        this.row = requireNonNull(row, "row is null");
    }

    public List getFields()
    {
        return row.getFields().stream()
                .map(RowField::new)
                .collect(toImmutableList());
    }

    @Override
    public boolean equals(Object o)
    {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }
        Row other = (Row) o;
        return Objects.equals(row, other.row);
    }

    @Override
    public int hashCode()
    {
        return row.hashCode();
    }

    @Override
    public String toString()
    {
        return row.toString();
    }

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

    // A public facade for Row.Builder from presto-client
    public static final class Builder
    {
        private final io.prestosql.jdbc.$internal.client.Row.Builder builder = io.prestosql.jdbc.$internal.client.Row.builder();

        private Builder() {}

        public Builder addField(String name, @Nullable Object value)
        {
            builder.addField(name, value);
            return this;
        }

        public Builder addUnnamedField(@Nullable Object value)
        {
            builder.addUnnamedField(value);
            return this;
        }

        Builder addField(Optional name, @Nullable Object value)
        {
            requireNonNull(name, "name is null");
            if (name.isPresent()) {
                return addField(name.get(), value);
            }
            return addUnnamedField(value);
        }

        public Row build()
        {
            return new Row(builder.build());
        }
    }
}