com.axibase.tsd.network.AbstractInsertCommand Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of atsd-api-java Show documentation

The ATSD Client for Java enables Java developers to easily read and write statistics and metadata from Axibase Time-Series Database. Build reporting, analytics, and alerting solutions with minimal effort.

There is a newer version: 1.1.0

Show newest version

/*
 * 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.network;

import java.util.Collections;
import java.util.Map;

import static com.axibase.tsd.util.AtsdUtil.checkEntityIsEmpty;


public abstract class AbstractInsertCommand implements PlainCommand {
    private final String commandName;
    protected final String entityName;
    private final Long timeMillis;
    private final String isoDate;
    protected final Map tags;

    public AbstractInsertCommand(String commandName, String entityName, Long timeMillis, String isoDate,
                                 Map tags) {
        this.commandName = commandName;
        checkEntityIsEmpty(entityName);
        this.entityName = entityName;
        this.timeMillis = timeMillis;
        this.isoDate = isoDate;
        this.tags = tags == null ? Collections.emptyMap() : tags;
    }

    @Override
    public final String compose() {
        StringBuilder sb = new StringBuilder(commandName)
                .append(' ').append("e:").append(handleName(entityName));
        if (timeMillis != null) {
            sb.append(' ').append("ms:").append(timeMillis);
        }
        if (isoDate != null) {
            sb.append(' ').append("d:").append(isoDate);
        }
        appendKeysAndValues(sb, " t:", tags);
        appendValues(sb);
        return sb.append('\n').toString();
    }

    protected static void appendKeysAndValues(StringBuilder sb, String prefix, Map map) {
        for (Map.Entry tagNameAndValue : map.entrySet()) {
            sb.append(prefix)
                    .append(handleName(tagNameAndValue.getKey()))
                    .append('=')
                    .append(handleStringValue(tagNameAndValue.getValue()));
        }
    }

    protected static String handleStringValue(String value) {
        if (value == null) {
            return "null";
        } else {
            return '"' + value.replace("\"", "\"\"") + '"';
        }
    }

    protected static String handleName(String key) {
        if (key == null) {
            return "null";
        } else if (key.indexOf('"') != -1) {
            return '"' + key.replace("\"", "\"\"") + '"';
        } else if (key.indexOf('=') != -1 ) {
            return '"' + key + '"';
        } else {
            return key;
        }
    }

    protected abstract void appendValues(StringBuilder sb);
}