tech.ytsaurus.client.request.OperationReq 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.function.Consumer;

import javax.annotation.Nullable;

import tech.ytsaurus.TGuid;
import tech.ytsaurus.client.rpc.RpcUtil;
import tech.ytsaurus.core.GUID;
import tech.ytsaurus.ysontree.YTreeBuilder;

abstract class OperationReq<
        TBuilder extends OperationReq.Builder,
        TRequest extends OperationReq>
        extends RequestBase {
    @Nullable
    protected final GUID operationId;
    @Nullable
    protected final String operationAlias;

    OperationReq(Builder builder) {
        super(builder);
        if (builder.operationId == null && builder.operationAlias == null) {
            throw new NullPointerException();
        }
        this.operationId = builder.operationId;
        this.operationAlias = builder.operationAlias;
    }

    void writeOperationDescriptionToProto(Consumer operationIdSetter, Consumer operationAliasSetter) {
        if (operationId != null) {
            operationIdSetter.accept(RpcUtil.toProto(operationId));
        } else {
            operationAliasSetter.accept(operationAlias);
        }
    }

    YTreeBuilder toTree(YTreeBuilder builder) {
        if (operationId != null) {
            builder.key("operation_id").value(operationId.toString());
        } else {
            builder.key("operation_alias").value(operationAlias);
        }
        return builder;
    }

    @Override
    protected void writeArgumentsLogString(StringBuilder sb) {
        if (operationId != null) {
            sb.append("Id: ").append(operationId).append("; ");
        } else {
            sb.append("Alias: ").append(operationAlias).append("; ");
        }
    }

    public abstract static class Builder<
            TBuilder extends Builder,
            TRequest extends RequestBase>
            extends RequestBase.Builder {
        @Nullable
        protected GUID operationId;
        @Nullable
        protected String operationAlias;

        Builder() {
        }

        Builder(Builder builder) {
            super(builder);
            this.operationId = builder.operationId;
            this.operationAlias = builder.operationAlias;
        }

        public TBuilder setOperationId(@Nullable GUID operationId) {
            this.operationId = operationId;
            return self();
        }

        public TBuilder setOperationAlias(@Nullable String operationAlias) {
            this.operationAlias = operationAlias;
            return self();
        }

        void writeOperationDescriptionToProto(
                Consumer operationIdSetter, Consumer operationAliasSetter) {
            if (operationId != null) {
                operationIdSetter.accept(RpcUtil.toProto(operationId));
            } else {
                operationAliasSetter.accept(operationAlias);
            }
        }

        protected YTreeBuilder toTree(YTreeBuilder builder) {
            if (operationId != null) {
                builder.key("operation_id").value(operationId.toString());
            } else {
                builder.key("operation_alias").value(operationAlias);
            }
            return builder;
        }

        @Override
        protected void writeArgumentsLogString(StringBuilder sb) {
            if (operationId != null) {
                sb.append("Id: ").append(operationId).append("; ");
            } else {
                sb.append("Alias: ").append(operationAlias).append("; ");
            }
        }
    }
}