summaryrefslogtreecommitdiffstats
path: root/lib/perfdata/influxdbwriter.cpp
blob: 30240f7f41c1b54ba95995329c433cf2a6282d5e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */

#include "perfdata/influxdbwriter.hpp"
#include "perfdata/influxdbwriter-ti.cpp"
#include "base/base64.hpp"
#include "remote/url.hpp"
#include "base/configtype.hpp"
#include "base/perfdatavalue.hpp"
#include "base/statsfunction.hpp"
#include <boost/beast/http/message.hpp>
#include <boost/beast/http/string_body.hpp>
#include <utility>

using namespace icinga;

REGISTER_TYPE(InfluxdbWriter);

REGISTER_STATSFUNCTION(InfluxdbWriter, &InfluxdbWriter::StatsFunc);

void InfluxdbWriter::StatsFunc(const Dictionary::Ptr& status, const Array::Ptr& perfdata)
{
	InfluxdbCommonWriter::StatsFunc<InfluxdbWriter>(status, perfdata);
}

boost::beast::http::request<boost::beast::http::string_body> InfluxdbWriter::AssembleRequest(String body)
{
	auto request (AssembleBaseRequest(std::move(body)));
	Dictionary::Ptr basicAuth = GetBasicAuth();

	if (basicAuth) {
		request.set(
			boost::beast::http::field::authorization,
			"Basic " + Base64::Encode(basicAuth->Get("username") + ":" + basicAuth->Get("password"))
		);
	}

	return std::move(request);
}

Url::Ptr InfluxdbWriter::AssembleUrl()
{
	auto url (AssembleBaseUrl());

	std::vector<String> path;
	path.emplace_back("write");
	url->SetPath(path);

	url->AddQueryElement("db", GetDatabase());

	if (!GetUsername().IsEmpty())
		url->AddQueryElement("u", GetUsername());
	if (!GetPassword().IsEmpty())
		url->AddQueryElement("p", GetPassword());

	return std::move(url);
}