tech.ytsaurus.client.operations.SystemOperationSpecBase Maven / Gradle / Ivy
The newest version!
package tech.ytsaurus.client.operations;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import javax.annotation.Nullable;
import tech.ytsaurus.core.cypress.YPath;
import tech.ytsaurus.lang.NonNullApi;
import tech.ytsaurus.lang.NonNullFields;
import tech.ytsaurus.ysontree.YTree;
import tech.ytsaurus.ysontree.YTreeBuilder;
import tech.ytsaurus.ysontree.YTreeNode;
/**
* Immutable base class for system operation specs like merge, remote copy and sort.
*/
@NonNullApi
@NonNullFields
public class SystemOperationSpecBase {
private final List inputTables;
private final YPath outputTable;
@Nullable
private final String pool;
@Nullable
private final String title;
private final Map outputTableAttributes;
private final Map additionalSpecParameters;
protected SystemOperationSpecBase(Builder builder) {
inputTables = new ArrayList<>(builder.inputTables);
if (builder.outputTable == null) {
throw new RuntimeException("output table is not set");
}
outputTable = builder.outputTable;
pool = builder.pool;
title = builder.title;
outputTableAttributes = builder.outputTableAttributes;
additionalSpecParameters = builder.additionalSpecParameters;
}
@Override
public int hashCode() {
return super.hashCode();
}
@Override
public boolean equals(Object obj) {
if (obj == null) {
return false;
}
if (getClass() != obj.getClass()) {
return false;
}
SystemOperationSpecBase spec = (SystemOperationSpecBase) obj;
return inputTables.equals(spec.inputTables)
&& outputTable.equals(spec.outputTable)
&& Optional.ofNullable(pool).equals(Optional.ofNullable(spec.pool))
&& Optional.ofNullable(title).equals(Optional.ofNullable(spec.title))
&& outputTableAttributes.equals(spec.outputTableAttributes)
&& additionalSpecParameters.equals(spec.additionalSpecParameters);
}
public List getInputTables() {
return inputTables;
}
public YPath getOutputTable() {
return outputTable;
}
public Optional getPool() {
return Optional.ofNullable(pool);
}
public Optional getTitle() {
return Optional.ofNullable(title);
}
public Map getOutputTableAttributes() {
return outputTableAttributes;
}
public Map getAdditionalSpecParameters() {
return additionalSpecParameters;
}
/**
* Convert to yson.
*/
public YTreeBuilder toTree(YTreeBuilder mapBuilder, SpecPreparationContext context) {
return mapBuilder
.key("started_by").apply(b -> SpecUtils.startedBy(b, context))
.key("input_table_paths").value(inputTables, (b, t) -> t.toTree(b))
.key("output_table_path").apply(outputTable::toTree)
.when(pool != null, b -> b.key("pool").value(pool))
.when(title != null, b -> b.key("title").value(title))
.apply(b -> {
for (Map.Entry node : additionalSpecParameters.entrySet()) {
b.key(node.getKey()).value(node.getValue());
}
return b;
});
}
/**
* Builder of {@link SystemOperationSpecBase}.
*/
@NonNullApi
@NonNullFields
public abstract static class Builder> {
// N.B. some clients have methods taking this class as argument therefore it must be public
private List inputTables = new ArrayList<>();
@Nullable
private YPath outputTable;
@Nullable
private String pool;
@Nullable
private String title;
private Map outputTableAttributes = new HashMap<>();
private Map additionalSpecParameters = new HashMap<>();
/**
* Set input tables. It is required parameter.
*/
public T setInputTables(Collection inputTables) {
this.inputTables = new ArrayList<>(inputTables);
return self();
}
/**
* Set input tables. It is required parameter.
*/
public T setInputTables(YPath... inputTables) {
return setInputTables(Arrays.asList(inputTables));
}
/**
* Add one more input table. It is required parameter.
*/
public T addInputTable(YPath inputTable) {
this.inputTables.add(inputTable);
return self();
}
/**
* Set output table. It is required parameter.
*/
public T setOutputTable(YPath outputTable) {
this.outputTable = outputTable;
return self();
}
/**
* Set attributes for output table.
*/
public T setOutputTableAttributes(Map outputTableAttributes) {
this.outputTableAttributes = new HashMap<>(outputTableAttributes);
return self();
}
/**
* Add one more attribute for output table.
*/
public T plusOutputTableAttribute(String key, @Nullable Object value) {
return plusOutputTableAttribute(key, YTree.node(value));
}
/**
* Add one more attribute for output table.
*/
public T plusOutputTableAttribute(String key, YTreeNode value) {
this.outputTableAttributes.put(key, value);
return self();
}
/**
* Set pool in which operation should be run.
*/
public T setPool(@Nullable String pool) {
this.pool = pool;
return self();
}
/**
* Set title of operation.
*/
public T setTitle(@Nullable String title) {
this.title = title;
return self();
}
/**
* Set additional parameters for spec
*/
public T setAdditionalSpecParameters(Map additionalSpecParameters) {
this.additionalSpecParameters = new HashMap<>(additionalSpecParameters);
return self();
}
/**
* Add one more spec parameter.
*/
public T plusAdditionalSpecParameter(String key, @Nullable Object value) {
return plusAdditionalSpecParameter(key, YTree.node(value));
}
/**
* Add one more spec parameter.
*/
public T plusAdditionalSpecParameter(String key, YTreeNode value) {
this.additionalSpecParameters.put(key, value);
return self();
}
protected abstract T self();
}
}