tech.ytsaurus.client.rpc.LazyResponse 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.rpc;

import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

import javax.annotation.Nullable;

import com.google.protobuf.MessageLite;
import com.google.protobuf.Parser;
import tech.ytsaurus.lang.NonNullApi;
import tech.ytsaurus.lang.NonNullFields;
import tech.ytsaurus.rpc.TResponseHeader;


/**
 * Реализация RpcClientResponse с ленивой десериализацией тела ответа при первом обращении
 */
@NonNullApi
@NonNullFields
public class LazyResponse implements RpcClientResponse {
    private final Parser parser;
    @Nullable
    private byte[] bodyData;
    @Nullable
    private ResponseType bodyMessage;
    private final List attachments;
    private final RpcClient sender;
    @Nullable
    private final Compression compression;

    public LazyResponse(
            Parser parser,
            byte[] body,
            List attachments,
            RpcClient sender,
            @Nullable TResponseHeader responseHeader
    ) {
        this.parser = parser;
        this.bodyData = body;
        this.sender = Objects.requireNonNull(sender);

        if (responseHeader != null && responseHeader.hasCodec()) {
            this.compression = Compression.fromValue(responseHeader.getCodec());
        } else {
            this.compression = null;
        }

        if (compression != null) {
            Codec codec = Codec.codecFor(compression);
            this.attachments = attachments.stream().map(codec::decompress).collect(Collectors.toList());
        } else {
            this.attachments = attachments;
        }
    }

    @Override
    public synchronized ResponseType body() {
        if (bodyMessage == null) {
            bodyMessage = compression != null
                    ? RpcUtil.parseMessageBodyWithCompression(bodyData, parser, compression)
                    : RpcUtil.parseMessageBodyWithEnvelope(bodyData, parser);
            bodyData = null;
        }
        return bodyMessage;
    }

    @Override
    public List attachments() {
        return attachments;
    }

    @Override
    public RpcClient sender() {
        return sender;
    }
}