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 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 url;
}
|
