tech.ytsaurus.client.request.PreparedModifyRowRequest 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.request;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

import javax.annotation.Nullable;

import tech.ytsaurus.client.SerializationResolver;
import tech.ytsaurus.client.rpc.Compression;
import tech.ytsaurus.client.rpc.RpcClientRequestBuilder;
import tech.ytsaurus.client.rpc.RpcCompression;
import tech.ytsaurus.client.rpc.RpcUtil;
import tech.ytsaurus.core.tables.TableSchema;
import tech.ytsaurus.rpcproxy.ERowModificationType;
import tech.ytsaurus.rpcproxy.TReqModifyRows;

/**
 * Immutable row modification request that contains serialized and compressed rowset
 * 
 * It is useful when same request is performed multiple times (e.g. to different clusters)
 * and saves CPU resources used on serialization and compressing.
 * 

 * It should be built using {@link PreparableModifyRowsRequest#prepare(RpcCompression, SerializationResolver)}
 * 
 * Compression used in this request have to match with compression of the client otherwise exception will be thrown
 * when trying to execute this request.
 */
public class PreparedModifyRowRequest
        extends AbstractModifyRowsRequest {
    private final Compression codecId;
    private final List compressedAttachments;

    public PreparedModifyRowRequest(BuilderBase builder) {
        super(builder);
        this.codecId = Objects.requireNonNull(builder.codecId);
        this.compressedAttachments = new ArrayList<>(Objects.requireNonNull(builder.compressedAttachments));
    }

    PreparedModifyRowRequest(
            String path,
            TableSchema schema,
            List rowModificationTypes,
            Compression codecId,
            List compressedAttachments
    ) {
        this(builder()
                .setPath(path)
                .setSchema(schema)
                .setRowModificationTypes(rowModificationTypes)
                .setCodecId(codecId)
                .setCompressedAttachments(compressedAttachments));
    }

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

    @Override
    public void serializeRowsetTo(RpcClientRequestBuilder builder) {
        builder.setCompressedAttachments(codecId, compressedAttachments);
    }

    @Override
    public void convertValues(SerializationResolver serializationResolver) {
    }

    @Override
    public Builder toBuilder() {
        return builder()
                .setCodecId(codecId)
                .setCompressedAttachments(compressedAttachments)
                .setPath(path)
                .setSchema(schema)
                .setRequireSyncReplica(requireSyncReplica)
                .setRowModificationTypes(rowModificationTypes);
    }

    public static class Builder extends BuilderBase {
        @Override
        protected Builder self() {
            return this;
        }

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

    public abstract static class BuilderBase>
            extends AbstractModifyRowsRequest.Builder {
        @Nullable
        private Compression codecId;
        @Nullable
        private List compressedAttachments;

        public TBuilder setCodecId(Compression codecId) {
            this.codecId = codecId;
            return self();
        }

        public TBuilder setCompressedAttachments(List compressedAttachments) {
            this.compressedAttachments = compressedAttachments;
            return self();
        }
    }
}

abstract class PreparableModifyRowsRequest<
        TBuilder extends PreparableModifyRowsRequest.Builder,
        TRequest extends PreparableModifyRowsRequest>
        extends AbstractModifyRowsRequest {
    PreparableModifyRowsRequest(Builder builder) {
        super(builder);
    }

    @Override
    public void serializeRowsetTo(RpcClientRequestBuilder builder) {
        serializeRowsetTo(builder.attachments());
    }

    abstract void serializeRowsetTo(List attachments);

    /**
     * Serialize and compress rowset.
     */
    public PreparedModifyRowRequest prepare(
            RpcCompression rpcCompression, SerializationResolver serializationResolver) {
        convertValues(serializationResolver);
        List attachments = new ArrayList<>();
        serializeRowsetTo(attachments);
        Compression codecId = rpcCompression.getRequestCodecId().orElse(Compression.fromValue(0));
        List preparedMessage = RpcUtil.createCompressedAttachments(attachments, codecId);

        return new PreparedModifyRowRequest(path, schema, rowModificationTypes, codecId, preparedMessage);
    }

    public abstract static class Builder<
            TBuilder extends Builder,
            TRequest extends AbstractModifyRowsRequest>
            extends AbstractModifyRowsRequest.Builder {
    }
}