tech.ytsaurus.client.rows.UnversionedRow Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ytsaurus-client Show documentation
YTsaurus client
The newest version!
package tech.ytsaurus.client.rows;

import java.util.Collections;
import java.util.List;
import java.util.Objects;

import tech.ytsaurus.core.tables.TableSchema;
import tech.ytsaurus.yson.YsonConsumer;
import tech.ytsaurus.ysontree.YTree;
import tech.ytsaurus.ysontree.YTreeBuilder;
import tech.ytsaurus.ysontree.YTreeMapNode;


public class UnversionedRow {
    private final List values;

    public UnversionedRow(List values) {
        this.values = Objects.requireNonNull(values);
    }

    public List getValues() {
        return Collections.unmodifiableList(values);
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (!(o instanceof UnversionedRow)) {
            return false;
        }

        UnversionedRow that = (UnversionedRow) o;

        return values.equals(that.values);
    }

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

    @Override
    public String toString() {
        return "UnversionedRow{" +
                "values=" + values +
                '}';
    }

    public void writeTo(YsonConsumer consumer, TableSchema schema) {
        writeTo(consumer, schema, false);
    }

    public void writeTo(YsonConsumer consumer, TableSchema schema, boolean ignoreSystemColumns) {
        consumer.onBeginMap();
        for (UnversionedValue value : values) {
            int index = value.getId();
            String name = schema.getColumnName(index);

            if (ignoreSystemColumns && name.startsWith("$")) {
                continue;
            }

            consumer.onKeyedItem(name);
            value.writeTo(consumer);
        }
        consumer.onEndMap();
    }

    public YTreeMapNode toYTreeMap(TableSchema schema) {
        return toYTreeMap(schema, false);
    }

    public YTreeMapNode toYTreeMap(TableSchema schema, boolean ignoreSystemColumns) {
        YTreeBuilder builder = YTree.builder();
        writeTo(builder, schema, ignoreSystemColumns);
        return builder.build().mapNode();
    }
}