com.loadcoder.cluster.clients.influxdb.InfluxDBClient Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of loadcoder-cluster Show documentation
Show all versions of loadcoder-cluster Show documentation
This project contains the core features in Loadcoder
/*******************************************************************************
* Copyright (C) 2020 Team 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 databases = listDatabases();
if (!databases.contains(dbName)) {
createDB();
}
String executionId = config.getConfiguration("LOADCODER_EXECUTION_ID");
if (executionId == null) {
executionId = DateTimeUtil.getDateTimeNowString();
}
InfluxDBTestExecution exe = createTestExecution(executionId);
log.info("Using executionId:" + executionId);
RuntimeResultConsumer influxConsumer = (res) -> exe.writeTransactions(res);
return influxConsumer;
}
public String getTestGroup() {
return testGroup;
}
public String getTestName() {
return testName;
}
public String getDatabaseName() {
return this.dbName;
}
public HttpResponse createDB() {
log.info("Creating new InfluxDB database:" + dbName);
String body = String.format("q=CREATE DATABASE %s", dbName);
return sendPost(body, QUERY_URL, Arrays.asList());
}
public List showMeasurements() {
List result = new ArrayList();
String body = String.format("q=SHOW MEASUREMENTS ON %s", dbName);
HttpResponse resp = sendPost(body, QUERY_URL, Arrays.asList());
String respBody = resp.getBody();
try {
JSONArray array = JsonPath.read(respBody, "$['results'][0]['series'][0]['values'][*][*]");
array.stream().forEach(db -> {
result.add(db.toString());
});
return result;
} catch (PathNotFoundException pnfe) {
throw new RuntimeException("There are no measurements in the InfluxDB database " + dbName
+ ". Make sure that there are results written to the database before trying to create a Grafana dashboard.");
}
}
public List listDatabases() {
List result = new ArrayList();
String body = String.format("q=SHOW DATABASES");
HttpResponse resp = sendPost(body, QUERY_URL, Arrays.asList());
String respBody = resp.getBody();
JSONArray array = JsonPath.read(respBody, "$['results'][0]['series'][0]['values'][*][0]");
array.stream().forEach(db -> {
result.add(db.toString());
});
return result;
}
public List listDistinctTransactions(String measurement) {
List result = new ArrayList();
String body = String
.format("q= show tag values on " + dbName + " from \"" + measurement + "\" with key = transaction");
HttpResponse resp = sendPost(body, QUERY_URL, Arrays.asList());
String respBody = resp.getBody();
JSONArray array = JsonPath.read(respBody, "$['results'][0]['series'][0]['values'][*][1]");
array.stream().forEach(db -> {
result.add(db.toString());
});
return result;
}
private HttpResponse writeEntries(String body) {
return sendPost(body, WRITE_URL + dbName, Arrays.asList());
}
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 final Map uniqueNanos = new HashMap();
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 HttpResponse writeTransactions(Map> transactionResults) {
String body = convertTransactionsToWriteBody(transactionResults, executionId);
HttpResponse responseCode = client.writeEntries(body);
return responseCode;
}
protected String convertTransactionsToWriteBody(
Map> transactionResults, String executionId) {
if (executionId == null || executionId.isEmpty()) {
throw new RuntimeException("ExecutionId is null or empty");
}
StringBuilder builder = new StringBuilder();
for (String key : transactionResults.keySet()) {
StringBuilder builder2 = new StringBuilder();
Integer nanoIterator = uniqueNanos.get(key);
if (nanoIterator == null) {
nanoIterator = 1000;
uniqueNanos.put(key, nanoIterator);
}
List list = transactionResults.get(key);
for (TransactionExecutionResult t : list) {
if (nanoIterator > 9998) {
nanoIterator = 1000;
}
String urlParameters = String.format(WRITE_ENTRY_BODY_TEMPLATE + (nanoIterator++), executionId, key,
t.isStatus(), t.getValue(), t.getTs());
builder2.append(urlParameters);
builder2.append("\n");
}
uniqueNanos.put(key, nanoIterator);
String transactionsString = builder2.toString();
builder.append(transactionsString);
}
return builder.toString();
}
}
/**
* Generates the code for the storeAndConsumeResultRuntime call where the
* results are sent to a InfluxDB
*
* @param originalCode the code that will be modified into the new code
* @param groupName name of the group of the test
* @param testName the name of the test
* @return returns the new generated code
*/
public static String generateCodeCallStoreAndConsumeResultRuntime(String originalCode, String groupName,
String testName) {
String result = originalCode;
String testContent = FileUtil.getResourceAsString("/cluster-codeTemplate/storeAndConsumeResultRuntime.tmp");
testContent = testContent.replace("${groupName}", groupName);
testContent = testContent.replace("${testName}", testName);
result = result.replace("${storeAndConsumeResultRuntime}", testContent);
result = result.replace("${importList}",
"import com.loadcoder.cluster.clients.docker.LoadcoderCluster;\n${importList}");
result = result.replace("${importList}",
"import com.loadcoder.cluster.clients.influxdb.InfluxDBClient;\n${importList}");
return result;
}
} © 2015 - 2025 Weber Informatics LLC | Privacy Policy