com.loadcoder.result.clients.InfluxDBClient Maven / Gradle / Ivy

Go to download
/*******************************************************************************
 * Copyright (C) 2018 Stefan Vahlgren at Loadcoder
 * 
 * This file is part of Loadcoder.
 * 
 * Loadcoder is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * Loadcoder is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 ******************************************************************************/
package com.loadcoder.result.clients;

import java.util.Arrays;
import java.util.List;
import java.util.Map;

import com.loadcoder.result.TransactionExecutionResult;

public class InfluxDBClient extends HttpClient{

	private final String dbName;

	private final static String WRITE_URL_TEMPLATE = "%s://%s:%s/write?db=";
	private final static String QUERY_URL_TEMPLATE = "%s://%s:%s/query";
	
	private final static String WRITE_ENTRY_BODY_TEMPLATE = "%s,execution=%s value=%s %s000000";
	private final String WRITE_URL;
	private final String QUERY_URL;
	 
	/**
	 * Constructor for the InfluxDBClient
	 * @param host is the host for where the influx DB is hosted
	 * @param port is the port where the influx DB exposes its API
	 * @param https is boolean for whether or not the communication is encrypted or not.
	 * @param db is the name of the database inside the influx db, containig the results
	 */
	public InfluxDBClient(String host, int port, boolean https, String db) {
		String protocol = protocolAsString(https);
		WRITE_URL = String.format(WRITE_URL_TEMPLATE, protocol, host, port);
		QUERY_URL = String.format(QUERY_URL_TEMPLATE, protocol, host, port);
		this.dbName = db;
	}

	protected int createDB(String dbName) {
		String body = String.format("q=CREATE DATABASE %s", dbName);
		return sendPost(body, QUERY_URL, Arrays.asList());
	}

	private int writeEntries(String body) {
		return sendPost(body, WRITE_URL + dbName, Arrays.asList());
	}
	
	public InfluxDBTestExecution createTestExecution() {
		return new InfluxDBTestExecution(this);
	}

	public InfluxDBTestExecution createTestExecution(String executionId) {
		return new InfluxDBTestExecution(executionId, this);
	}
	
	
	
	/**
	 * Class that keeps the influxDB connection together with the current execution
	 */
	public static class InfluxDBTestExecution{
		final String executionId;
		final InfluxDBClient client;
		
		private InfluxDBTestExecution(String executionId, InfluxDBClient client){
			this.client = client;
			this.executionId = executionId;
		}
		
		private InfluxDBTestExecution(InfluxDBClient client){
			this.client = client;
			this.executionId = null;
		}
		
		
		/**
		 * Write a list of transaction results into the influx DB
		 * @param transactionResults that is going be be written into the influx DB
		 * @return the HTTP status code for the influx DB response
		 */
		public int writeTransactions(Map> transactionResults) {
			String body = convertTransactionsToWriteBody(transactionResults, executionId);
			int responseCode = client.writeEntries(body);
			return responseCode;
		}
		
		protected String convertTransactionsToWriteBody(Map> transactionResults, String executionId) {
			String entireBody = "";
			
			for (String key : transactionResults.keySet()) {
				List list = transactionResults.get(key);
				for (TransactionExecutionResult t : list) {
					String urlParameters = String.format(WRITE_ENTRY_BODY_TEMPLATE, key, executionId, t.getRt(), t.getTs());
					entireBody = entireBody + urlParameters + "\n";
				}
			}
			return entireBody;
		}
	}

}