src/main/java/org/influxdb/dto/BatchPoints.java
package org.influxdb.dto;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.TreeMap;
import java.util.concurrent.TimeUnit;
import org.influxdb.InfluxDB.ConsistencyLevel;
/**
* {Purpose of This Type}.
*
* {Other Notes Relating to This Type (Optional)}
*
* @author stefan
*
*/
public class BatchPoints {
private String database;
private String retentionPolicy;
private Map<String, String> tags;
private List<Point> points;
private ConsistencyLevel consistency;
private TimeUnit precision;
BatchPoints() {
// Only visible in the Builder
}
/**
* Create a new BatchPoints build to create a new BatchPoints in a fluent manner.
*
* @return the Builder to be able to add further Builder calls.
*/
public static Builder builder() {
return new Builder(null);
}
/**
* Create a new BatchPoints build to create a new BatchPoints in a fluent manner.
*
* @param database
* the name of the Database
* @return the Builder to be able to add further Builder calls.
*/
public static Builder database(final String database) {
return new Builder(database);
}
/**
* The Builder to create a new BatchPoints instance.
*/
public static final class Builder {
private final String database;
private String retentionPolicy;
private final Map<String, String> tags = new TreeMap<>();
private final List<Point> points = new ArrayList<>();
private ConsistencyLevel consistency;
private TimeUnit precision;
/**
* @param database
*/
Builder(final String database) {
this.database = database;
}
/**
* The retentionPolicy to use.
*
* @param policy the retentionPolicy to use
* @return the Builder instance
*/
public Builder retentionPolicy(final String policy) {
this.retentionPolicy = policy;
return this;
}
/**
* Add a tag to this set of points.
*
* @param tagName
* the tag name
* @param value
* the tag value
* @return the Builder instance.
*/
public Builder tag(final String tagName, final String value) {
this.tags.put(tagName, value);
return this;
}
/**
* Add a Point to this set of points.
*
* @param pointToAdd the Point to add
* @return the Builder instance
*/
public Builder point(final Point pointToAdd) {
this.points.add(pointToAdd);
return this;
}
/**
* Add a set of Points to this set of points.
*
* @param pointsToAdd the Points to add
* @return the Builder instance
*/
public Builder points(final Point... pointsToAdd) {
this.points.addAll(Arrays.asList(pointsToAdd));
return this;
}
/**
* Add a set of Points to this set of points.
*
* @param pointsToAdd the Points to add
* @return the Builder instance
*/
public Builder points(final Collection<Point> pointsToAdd) {
this.points.addAll(pointsToAdd);
return this;
}
/**
* Set the ConsistencyLevel to use. If not given it defaults to {@link ConsistencyLevel#ONE}
*
* @param consistencyLevel the ConsistencyLevel
* @return the Builder instance
*/
public Builder consistency(final ConsistencyLevel consistencyLevel) {
this.consistency = consistencyLevel;
return this;
}
/**
* Set the time precision to use for the whole batch. If unspecified, will default to {@link TimeUnit#NANOSECONDS}
* @param precision the precision of the points
* @return the Builder instance
*/
public Builder precision(final TimeUnit precision) {
this.precision = precision;
return this;
}
/**
* Create a new BatchPoints instance.
*
* @return the created BatchPoints.
*/
public BatchPoints build() {
BatchPoints batchPoints = new BatchPoints();
batchPoints.setDatabase(this.database);
for (Point point : this.points) {
point.getTags().putAll(this.tags);
}
batchPoints.setPoints(this.points);
batchPoints.setRetentionPolicy(this.retentionPolicy);
batchPoints.setTags(this.tags);
if (null == this.consistency) {
this.consistency = ConsistencyLevel.ONE;
}
batchPoints.setConsistency(this.consistency);
if (null == this.precision) {
this.precision = TimeUnit.NANOSECONDS;
}
batchPoints.setPrecision(this.precision);
return batchPoints;
}
}
/**
* @return the database
*/
public String getDatabase() {
return this.database;
}
/**
* @param database
* the database to set
*/
void setDatabase(final String database) {
this.database = database;
}
/**
* @return the retentionPolicy
*/
public String getRetentionPolicy() {
return this.retentionPolicy;
}
/**
* @param retentionPolicy
* the retentionPolicy to set
*/
void setRetentionPolicy(final String retentionPolicy) {
this.retentionPolicy = retentionPolicy;
}
/**
* @return the points
*/
public List<Point> getPoints() {
return this.points;
}
/**
* @param points
* the points to set
*/
void setPoints(final List<Point> points) {
this.points = points;
}
/**
* @return the time precision unit
*/
public TimeUnit getPrecision() {
return precision;
}
/**
* @param precision the time precision to set for the batch points
*/
void setPrecision(final TimeUnit precision) {
this.precision = precision;
}
/**
* Add a single Point to these batches.
*
* @param point the Point to add
* @return this Instance to be able to daisy chain calls.
*/
public BatchPoints point(final Point point) {
point.getTags().putAll(this.tags);
this.points.add(point);
return this;
}
/**
* @return the tags
*/
public Map<String, String> getTags() {
return this.tags;
}
/**
* @param tags
* the tags to set
*/
void setTags(final Map<String, String> tags) {
this.tags = tags;
}
/**
* @return the consistency
*/
public ConsistencyLevel getConsistency() {
return this.consistency;
}
/**
* @param consistency
* the consistency to set
*/
void setConsistency(final ConsistencyLevel consistency) {
this.consistency = consistency;
}
@Override
public boolean equals(final Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
BatchPoints that = (BatchPoints) o;
return Objects.equals(database, that.database)
&& Objects.equals(retentionPolicy, that.retentionPolicy)
&& Objects.equals(tags, that.tags)
&& Objects.equals(points, that.points)
&& consistency == that.consistency
&& precision == that.precision;
}
@Override
public int hashCode() {
return Objects.hash(database, retentionPolicy, tags, points, consistency, precision);
}
/**
* {@inheritDoc}
*/
@Override
public String toString() {
StringBuilder builder = new StringBuilder();
builder.append("BatchPoints [database=")
.append(this.database)
.append(", retentionPolicy=")
.append(this.retentionPolicy)
.append(", consistency=")
.append(this.consistency)
.append(", tags=")
.append(this.tags)
.append(", precision=")
.append(this.precision)
.append(", points=")
.append(this.points)
.append("]");
return builder.toString();
}
// measurement[,tag=value,tag2=value2...] field=value[,field2=value2...] [unixnano]
/**
* calculate the lineprotocol for all Points.
*
* @return the String with newLines.
*/
public String lineProtocol() {
StringBuilder sb = new StringBuilder();
for (Point point : this.points) {
sb.append(point.lineProtocol(this.precision)).append("\n");
}
return sb.toString();
}
/**
* Test whether is possible to merge two BatchPoints objects.
*
* @param that batch point to merge in
* @return true if the batch points can be sent in a single HTTP request write
*/
public boolean isMergeAbleWith(final BatchPoints that) {
return Objects.equals(database, that.database)
&& Objects.equals(retentionPolicy, that.retentionPolicy)
&& Objects.equals(tags, that.tags)
&& consistency == that.consistency;
}
/**
* Merge two BatchPoints objects.
*
* @param that batch point to merge in
* @return true if the batch points have been merged into this BatchPoints instance. Return false otherwise.
*/
public boolean mergeIn(final BatchPoints that) {
boolean mergeAble = isMergeAbleWith(that);
if (mergeAble) {
this.points.addAll(that.points);
}
return mergeAble;
}
}