com.evrythng.thng.resource.model.store.BatchPopulatingTask Maven / Gradle / Ivy
Show all versions of thng-resource-model Show documentation
/*
* (c) Copyright 2015 EVRYTHNG Ltd London / Zurich
* www.evrythng.com
*/
package com.evrythng.thng.resource.model.store;
import com.fasterxml.jackson.annotation.JsonIgnore;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.function.ToIntFunction;
import java.util.stream.Collectors;
public class BatchPopulatingTask extends TaskOnBatch {
private static final long serialVersionUID = 2514640549756857875L;
// product in here
private InputParameters inputParameters;
public static final String FIELD_INPUT_PARAMETERS = "inputParameters";
private OutputParameters outputParameters;
public static final String FIELD_OUTPUT_PARAMETERS = "outputParameters";
private Progress progress;
public static final String FIELD_PROGRESS = "progress";
public BatchPopulatingTask() {
setType(Type.POPULATING);
}
/**
* Container class that holds the sub-tasks of the current task.
*
* The tasks are held as a map of {@link String} => {@link Contribution} instances with the sub-task ID as the key.
*/
public static final class Progress {
public static final String FIELD_CONTRIBUTIONS = "contributions";
private Map contributions;
private Integer totalAmount;
public Progress() {
contributions = new HashMap<>();
}
@JsonIgnore
public Integer getThngsCount() {
return contributions.values().stream().map(new Function() {
@Override
public Integer apply(final Contribution contribution) {
return contribution.getThngs();
}
}).collect(Collectors.summingInt(new ToIntFunction() {
@Override
public int applyAsInt(final Integer value) {
return value;
}
}));
}
@JsonIgnore
public Integer getUrlBindingsCount() {
return contributions.values().stream().map(new Function() {
@Override
public Integer apply(final Contribution contribution) {
return contribution.getUrlBindings();
}
}).collect(Collectors.summingInt(new ToIntFunction() {
@Override
public int applyAsInt(final Integer value) {
return value;
}
}));
}
public Integer getTotalAmount() {
return totalAmount;
}
public void setTotalAmount(final Integer totalAmount) {
this.totalAmount = totalAmount;
}
public Map getContributions() {
return contributions;
}
public void setContributions(final Map contributions) {
this.contributions = contributions;
}
@JsonIgnore
public boolean isComplete() {
return getThngsCount() >= totalAmount && getUrlBindingsCount() >= totalAmount;
}
/**
* A {@link Contribution} represents a sub-task of {@link BatchPopulatingTask}.
*/
public static final class Contribution {
public static final String FIELD_URL_BINDINGS = "urlBindings";
private int urlBindings;
public static final String FIELD_THNGS = "thngs";
private int thngs;
@JsonIgnore
public void addCreatedThngsCount(final int count) {
thngs += count;
}
@JsonIgnore
public void addCreatedUrlBindingsCount(final Integer count) {
urlBindings += count;
}
/**
* Alias for {@link #getThngs()}
*
* @return the number of THNGS created by this sub-task
*
* @see #getThngs()
* @deprecated since 1.21
*/
@JsonIgnore
@Deprecated
public Integer getThngsCount() {
return thngs;
}
/**
* Alias for {@link #getUrlBindings()}
*
* @return the number of URL Bindings created by this sub-task
*
* @see #getUrlBindings()
* @deprecated since 1.21
*/
@JsonIgnore
@Deprecated
public Integer getUrlBindingsCount() {
return urlBindings;
}
/**
* @return the number of THNGS created by this sub-task
*/
public Integer getThngs() {
return thngs;
}
/**
* @return the number of URL Bindings created by this sub-task
*/
public Integer getUrlBindings() {
return urlBindings;
}
/**
* Sets the number of THNGS created by this sub-task
*
* @param thngs the number of THNGS
*/
public void setThngs(final int thngs) {
this.thngs = thngs;
}
/**
* Sets the number of URL Bindings created by this sub-task
*
* @param urlBindings the number of URL Bindings
*/
public void setUrlBindings(final int urlBindings) {
this.urlBindings = urlBindings;
}
}
}
public InputParameters getInputParameters() {
return inputParameters;
}
public void setInputParameters(final InputParameters inputParameters) {
this.inputParameters = inputParameters;
}
public Progress getProgress() {
return progress;
}
public void setProgress(final Progress progress) {
this.progress = progress;
}
public OutputParameters getOutputParameters() {
return outputParameters;
}
public void setOutputParameters(final OutputParameters outputParameters) {
this.outputParameters = outputParameters;
}
public static interface InputParameters {
String FIELD_TYPE = "type";
public static final String FIELD_SHORT_DOMAIN = "shortDomain";
public static final String FIELD_GENERATE_THNGS = "generateThngs";
public static final String FIELD_GENERATE_REDIRECTIONS = "generateRedirections";
public static final String FIELD_DEFAULT_REDIRECT_URL = "defaultRedirectUrl";
public static final String FIELD_THNG_TEMPLATE = "thngTemplate";
Type getType();
void setType(final Type type);
public static enum Type {
FIXED_AMOUNT, FILE_BASED, LIST_BASED
}
Boolean getGenerateThngs();
Boolean getGenerateRedirections();
String getShortDomain();
String getDefaultRedirectUrl();
ThngTemplate getThngTemplate();
void setShortDomain(String shortDomain);
void setDefaultRedirectUrl(String defaultRedirectUrl);
void setThngTemplate(ThngTemplate thngTemplate);
}
public static interface OutputParameters {
String FIELD_TYPE = "type";
String FIELD_COLUMNS = "columns";
Type getType();
void setType(final Type type);
public static enum Type {
CSV(Format.CSV);
private final Format format;
private Type(final Format format) {
this.format = format;
}
public Format getFormat() {
return format;
}
}
public static enum Format {
CSV("csv", "text/csv");
private final String extension;
private final String mimeType;
private Format(final String extension, final String mimeType) {
this.extension = extension;
this.mimeType = mimeType;
}
public String getExtension() {
return extension;
}
public String getMimeType() {
return mimeType;
}
}
public static enum Column {
SHORT_ID("shortId"), THNG("thng");
private final String name;
private Column(final String name) {
this.name = name;
}
public String getName() {
return name;
}
}
List getColumns();
}
public static final class Result extends BaseTaskResult {
private Integer totalCount;
private String location;
private OutputParameters.Format format;
private Map headers;
public Result() {
setType(Type.POPULATING);
}
public Integer getTotalCount() {
return totalCount;
}
public void setTotalCount(final Integer totalCount) {
this.totalCount = totalCount;
}
public String getLocation() {
return location;
}
public void setLocation(final String location) {
this.location = location;
}
public OutputParameters.Format getFormat() {
return format;
}
public void setFormat(final OutputParameters.Format format) {
this.format = format;
}
public Map getHeaders() {
return headers;
}
public void setHeaders(final Map headers) {
this.headers = headers;
}
}
}