tech.ytsaurus.client.rows.VersionedRowDeserializer 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;

public class VersionedRowDeserializer
        extends VersionedValueDeserializer
        implements WireVersionedRowDeserializer {
    private final List keyColumnSchema;
    private List writeTimestamps = Collections.emptyList();

    private List deleteTimestamps = Collections.emptyList();

    private List keys = Collections.emptyList();
    private List values = Collections.emptyList();

    private final UnversionedValueDeserializer keyDeserializer;
    private final VersionedValueDeserializer valueDeserializer;

    public VersionedRowDeserializer(List keyColumnSchema) {
        this.keyColumnSchema = Objects.requireNonNull(keyColumnSchema);

        this.keyDeserializer = new UnversionedValueDeserializer() {
            @Override
            public UnversionedValue build() {
                final UnversionedValue value = super.build();
                keys.add(value);
                return value;
            }
        };
        this.valueDeserializer = new VersionedValueDeserializer() {
            @Override
            public VersionedValue build() {
                final VersionedValue value = super.build();
                values.add(value);
                return value;
            }
        };
    }


    @Override
    public WireValueDeserializer keys(int keyCount) {
        this.keys = new ArrayList<>(keyCount);
        return this.keyDeserializer;
    }

    @Override
    public WireValueDeserializer values(int valueCount) {
        this.values = new ArrayList<>(valueCount);
        return this.valueDeserializer;
    }

    @Override
    public VersionedRow onCompleteRow() {
        return new VersionedRow(writeTimestamps, deleteTimestamps, keys, values);
    }

    @Override
    public void onWriteTimestamps(List timestamps) {
        this.writeTimestamps = timestamps;
    }

    @Override
    public void onDeleteTimestamps(List timestamps) {
        this.deleteTimestamps = timestamps;
    }

    @Override
    public List getKeyColumnSchema() {
        return keyColumnSchema;
    }

}