AddSeriesCommand.java

/*
 * Copyright 2016 Axibase Corporation or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 * https://www.axibase.com/atsd/axibase-apache-2.0.pdf
 *
 * or in the "license" file accompanying this file. This file is distributed
 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */
package com.axibase.tsd.model.data.command;

import com.axibase.tsd.model.data.series.Sample;
import com.axibase.tsd.util.AtsdUtil;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;

import java.util.*;


@JsonInclude(JsonInclude.Include.NON_NULL)
public class AddSeriesCommand {
    @JsonProperty("entity")
    private String entityName;
    @JsonProperty("metric")
    private String metricName;
    private Map<String, String> tags;
    private List<Sample> data;

    public AddSeriesCommand() {
    }

    public AddSeriesCommand(String entityName, String metricName, String... tagNamesAndValues) {
        this.entityName = entityName;
        this.metricName = metricName;
        this.tags = AtsdUtil.toMap(tagNamesAndValues);
    }

    public static AddSeriesCommand createSingle(String entityName, String metricName, long time, double value, String... tagNamesAndValues) {
        AddSeriesCommand addSeriesCommand = new AddSeriesCommand(entityName, metricName, tagNamesAndValues);
        addSeriesCommand.addSeries(new Sample(time, value));
        return addSeriesCommand;
    }

    public String getEntityName() {
        return entityName;
    }

    public String getMetricName() {
        return metricName;
    }

    public Map<String, String> getTags() {
        return tags;
    }

    public List<Sample> getData() {
        return data;
    }

    public AddSeriesCommand addSeries(Sample sample) {
        return addSeries(Collections.singleton(sample));
    }

    public AddSeriesCommand addSeries(Collection<Sample> samples) {
        if (data == null) {
            data = new ArrayList<Sample>();
        }
        data.addAll(samples);
        return this;
    }

    public AddSeriesCommand addSeries(Sample... series) {
        return addSeries(Arrays.asList(series));
    }

    public AddSeriesCommand setTags(Map<String, String> tags) {
        this.tags = tags;
        return this;
    }

    public AddSeriesCommand setEntityName(String entityName) {
        this.entityName = entityName;
        return this;
    }

    public AddSeriesCommand setMetricName(String metricName) {
        this.metricName = metricName;
        return this;
    }
}