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

import javax.annotation.Nonnull;
import javax.annotation.Nullable;

import tech.ytsaurus.client.rpc.RpcClientRequestBuilder;
import tech.ytsaurus.rpcproxy.TReqGenerateTimestamps;

public class GenerateTimestamps extends RequestBase
        implements HighLevelRequest {
    private final int count;

    GenerateTimestamps(Builder builder) {
        super(builder);
        this.count = Objects.requireNonNull(builder.count);
    }

    public GenerateTimestamps(int count) {
        this(builder().setCount(count));
    }

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

    /**
     * Internal method: prepare request to send over network.
     */
    @Override
    public void writeTo(RpcClientRequestBuilder builder) {
        builder.body().setCount(count);
    }

    @Override
    protected void writeArgumentsLogString(@Nonnull StringBuilder sb) {
        sb.append("Count: ").append(count).append("; ");
        super.writeArgumentsLogString(sb);
    }

    public Builder toBuilder() {
        return builder()
                .setCount(count)
                .setTimeout(timeout)
                .setRequestId(requestId)
                .setUserAgent(userAgent)
                .setTraceId(traceId, traceSampled)
                .setAdditionalData(additionalData);
    }

    public static class Builder extends RequestBase.Builder {
        @Nullable
        private Integer count;

        Builder() {
        }

        Builder(Builder builder) {
            super(builder);
            count = builder.count;
        }

        public Builder setCount(int count) {
            this.count = count;
            return self();
        }

        public GenerateTimestamps build() {
            return new GenerateTimestamps(this);
        }

        @Override
        protected Builder self() {
            return this;
        }
    }
}