com.sitewhere.influxdb.InfluxDbClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sitewhere-influxdb Show documentation
SiteWhere InfluxDB Integration
There is a newer version: 3.0.2
/*
 * Copyright (c) SiteWhere, LLC. All rights reserved. http://www.sitewhere.com
 *
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package com.sitewhere.influxdb;

import java.util.concurrent.TimeUnit;

import org.influxdb.InfluxDB;
import org.influxdb.InfluxDB.LogLevel;
import org.influxdb.InfluxDBFactory;
import org.influxdb.dto.Query;
import org.influxdb.dto.QueryResult;

import com.sitewhere.microservice.lifecycle.TenantEngineLifecycleComponent;
import com.sitewhere.microservice.lifecycle.parameters.StringComponentParameter;
import com.sitewhere.spi.SiteWhereException;
import com.sitewhere.spi.microservice.lifecycle.ILifecycleComponentParameter;
import com.sitewhere.spi.microservice.lifecycle.ILifecycleProgressMonitor;

/**
 * Client used for connecting to and interacting with an InfluxDB server.
 */
public class InfluxDbClient extends TenantEngineLifecycleComponent {

    /** InfluxDB configuration parameters */
    private InfluxDbConfiguration configuration;

    /** InfluxDB handle */
    private InfluxDB influx;

    /** Hostname parameter */
    private ILifecycleComponentParameter hostname;

    /** Database parameter */
    private ILifecycleComponentParameter database;

    public InfluxDbClient(InfluxDbConfiguration configuration) {
	this.configuration = configuration;
    }

    /*
     * @see com.sitewhere.server.lifecycle.LifecycleComponent#initializeParameters()
     */
    @Override
    public void initializeParameters() throws SiteWhereException {
	// Add hostname.
	this.hostname = StringComponentParameter.newBuilder(this, "Hostname").value(getConfiguration().getHostname())
		.makeRequired().build();
	getParameters().add(hostname);

	// Add database.
	this.database = StringComponentParameter.newBuilder(this, "Database").value(getConfiguration().getDatabase())
		.makeRequired().build();
	getParameters().add(database);
    }

    /*
     * @see
     * com.sitewhere.server.lifecycle.LifecycleComponent#initialize(com.sitewhere.
     * spi.server.lifecycle.ILifecycleProgressMonitor)
     */
    @Override
    public void initialize(ILifecycleProgressMonitor monitor) throws SiteWhereException {
	super.initialize(monitor);

	String connectionUrl = "http://" + getHostname().getValue() + ":" + getConfiguration().getPort();
	this.influx = InfluxDBFactory.connect(connectionUrl, getConfiguration().getUsername(),
		getConfiguration().getPassword());
	QueryResult result = getInflux()
		.query(new Query(String.format("CREATE DATABASE \"%s\";", getDatabase().getValue()), ""));
	String error = result.getError();
	if (error != null) {
	    getLogger().error(String.format("Error creating database. %s", error));
	}
	if (getConfiguration().isEnableBatch()) {
	    getInflux().enableBatch(getConfiguration().getBatchChunkSize(), getConfiguration().getBatchIntervalMs(),
		    TimeUnit.MILLISECONDS);
	}
	getInflux().setLogLevel(convertLogLevel(getConfiguration().getLogLevel()));
    }

    /**
     * Convert log level setting to expected enum value.
     * 
     * @param level
     * @return
     */
    protected LogLevel convertLogLevel(String level) {
	if ((level == null) || (level.equalsIgnoreCase("none"))) {
	    return LogLevel.NONE;
	} else if (level.equalsIgnoreCase("basic")) {
	    return LogLevel.BASIC;
	} else if (level.equalsIgnoreCase("headers")) {
	    return LogLevel.HEADERS;
	} else if (level.equalsIgnoreCase("full")) {
	    return LogLevel.FULL;
	}
	return LogLevel.NONE;
    }

    public InfluxDbConfiguration getConfiguration() {
	return configuration;
    }

    public void setConfiguration(InfluxDbConfiguration configuration) {
	this.configuration = configuration;
    }

    public InfluxDB getInflux() {
	return influx;
    }

    public void setInflux(InfluxDB influx) {
	this.influx = influx;
    }

    public ILifecycleComponentParameter getHostname() {
	return hostname;
    }

    public void setHostname(ILifecycleComponentParameter hostname) {
	this.hostname = hostname;
    }

    public ILifecycleComponentParameter getDatabase() {
	return database;
    }

    public void setDatabase(ILifecycleComponentParameter database) {
	this.database = database;
    }
}