weatherstation/firmware/libraries/ESP8266_Influxdb/src/Point.cpp

/**
 * 
 * Point.cpp: Point for write into InfluxDB server
 * 
 * MIT License
 * 
 * Copyright (c) 2020 InfluxData
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
*/

#include "Point.h"
#include "util/helpers.h"

Point::Point(const String & measurement)
{
    _measurement = escapeKey(measurement, false);
    _timestamp = nullptr;
    _tsWritePrecision = WritePrecision::NoTime;
}

Point::~Point() {
    delete [] _measurement;
    delete [] _timestamp;
}

void Point::addTag(const String &name, String value) {
    if(_tags.length() > 0) {
        _tags += ',';
    }
    char *s = escapeKey(name);
    _tags += s;
    delete [] s;
    _tags += '=';
    s = escapeKey(value);
    _tags += s;
    delete [] s;
}

void Point::addField(const String &name, long long value) {
  char buff[23];
  snprintf(buff, 50, "%lld", value);
  putField(name, String(buff)+"i");
}

void Point::addField(const String &name, unsigned long long value) {
  char buff[23];
  snprintf(buff, 50, "%llu", value);
  putField(name, String(buff)+"i");
}

void Point::addField(const String &name, const char *value) { 
    putField(name, escapeValue(value)); 
}

void Point::addField(const String &name, const __FlashStringHelper *pstr) {
    addField(name, String(pstr));
}

void Point::addField(const String &name, float value, int decimalPlaces) { 
    if(!isnan(value)) putField(name, String(value, decimalPlaces)); 
}

void Point::addField(const String &name, double value, int decimalPlaces) {
    if(!isnan(value)) putField(name, String(value, decimalPlaces)); 
}

void Point::addField(const String &name, char value) { 
    addField(name, String(value).c_str()); 
}

void Point::addField(const String &name, unsigned char value) {
    putField(name, String(value)+"i"); 
}

void Point::addField(const String &name, int value) { 
    putField(name, String(value)+"i"); 
}

void Point::addField(const String &name, unsigned int value) { 
    putField(name, String(value)+"i"); 
}

void Point::addField(const String &name, long value)  { 
    putField(name, String(value)+"i"); 
}

void Point::addField(const String &name, unsigned long value) { 
    putField(name, String(value)+"i"); 
}

void Point::addField(const String &name, bool value)  { 
    putField(name, bool2string(value)); 
}

void Point::addField(const String &name, const String &value)  { 
    addField(name, value.c_str()); 
}

void Point::putField(const String &name, const String &value) {
    if(_fields.length() > 0) {
        _fields += ',';
    }
    char *s = escapeKey(name);
    _fields += s;
    delete [] s;
    _fields += '=';
    _fields += value;
}

String Point::toLineProtocol(const String &includeTags) const {
    return createLineProtocol(includeTags);
}

String Point::createLineProtocol(const String &incTags) const {
    String line;
    line.reserve(strLen(_measurement) + 1 + incTags.length() + 1 + _tags.length() + 1 + _fields.length() + 1 + strLen(_timestamp));
    line += _measurement;
    if(incTags.length()>0) {
        line += ",";
        line += incTags;
    }
    if(hasTags()) {
        line += ",";
        line += _tags;
    }
    if(hasFields()) {
        line += " ";
        line += _fields;
    }
    if(hasTime()) {
        line += " ";
        line += _timestamp;
    }
    return line;
 }

void  Point::setTime(WritePrecision precision) {
    struct timeval tv;
    gettimeofday(&tv, NULL);
    
    switch(precision) {
        case WritePrecision::NS:
            setTime(getTimeStamp(&tv,9));
            break;
        case WritePrecision::US:
            setTime(getTimeStamp(&tv,6));
            break;
        case WritePrecision::MS: 
            setTime(getTimeStamp(&tv,3));
            break;
        case WritePrecision::S:
            setTime(getTimeStamp(&tv,0));
            break;
        case WritePrecision::NoTime:
            setTime((char *)nullptr);
            break;
    }
    _tsWritePrecision = precision;
}

void  Point::setTime(unsigned long long timestamp) {
    setTime(timeStampToString(timestamp));
}

void Point::setTime(const String &timestamp) {
    setTime(cloneStr(timestamp.c_str()));
}

void Point::setTime(const char *timestamp) {
    setTime(cloneStr(timestamp));    
}

void Point::setTime(char *timestamp) {
    delete [] _timestamp;
    _timestamp = timestamp;
}

void  Point::clearFields() {
    _fields = (char *)nullptr;
    delete [] _timestamp;
    _timestamp = nullptr;
}

void Point:: clearTags() {
    _tags = (char *)nullptr;
}