tech.ytsaurus.client.rows.UnversionedRowsetDeserializer 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.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;

import javax.annotation.Nonnull;


import tech.ytsaurus.core.tables.TableSchema;

public class UnversionedRowsetDeserializer
        extends UnversionedRowDeserializer
        implements WireRowsetDeserializer {
    private final TableSchema schema;
    private List rows = Collections.emptyList();

    public UnversionedRowsetDeserializer(TableSchema schema) {
        this.schema = Objects.requireNonNull(schema);
    }

    @Override
    public void setRowCount(int rowCount) {
        this.rows = new ArrayList<>(rowCount);
    }

    @Override
    @Nonnull
    public UnversionedRow onCompleteRow() {
        UnversionedRow row = super.onCompleteRow();
        this.rows.add(row);
        return row;
    }

    @Override
    public UnversionedRow onNullRow() {
        UnversionedRow row = super.onNullRow();
        this.rows.add(row);
        return row;
    }

    public UnversionedRowset getRowset() {
        return new UnversionedRowset(schema, this.rows);
    }
}