mirror of
https://github.com/matrix-construct/construct
synced 2024-12-26 07:23:53 +01:00
123 lines
2.5 KiB
C++
123 lines
2.5 KiB
C++
// Matrix Construct
|
|
//
|
|
// Copyright (C) Matrix Construct Developers, Authors & Contributors
|
|
// Copyright (C) 2016-2018 Jason Volk <jason@zemos.net>
|
|
//
|
|
// Permission to use, copy, modify, and/or distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice is present in all copies. The
|
|
// full license for this software is available in the LICENSE file.
|
|
|
|
namespace ircd::stats
|
|
{
|
|
static const_buffer print_item(stats::item<void> &item, const mutable_buffer &, const time_t &);
|
|
static void each_item(resource::response::chunked &, stats::item<void> &, window_buffer &, const time_t &);
|
|
static resource::response get_stats(client &, const resource::request &);
|
|
|
|
extern resource::method method_get;
|
|
extern resource stats_resource;
|
|
}
|
|
|
|
ircd::mapi::header
|
|
IRCD_MODULE
|
|
{
|
|
"Prometheus Metrics"
|
|
};
|
|
|
|
decltype(ircd::stats::stats_resource)
|
|
ircd::stats::stats_resource
|
|
{
|
|
"/stats",
|
|
{
|
|
"Prometheus Metrics"
|
|
}
|
|
};
|
|
|
|
decltype(ircd::stats::method_get)
|
|
ircd::stats::method_get
|
|
{
|
|
stats_resource, "GET", get_stats
|
|
};
|
|
|
|
ircd::resource::response
|
|
ircd::stats::get_stats(client &client,
|
|
const resource::request &request)
|
|
{
|
|
resource::response::chunked response
|
|
{
|
|
client, http::OK, "text/plain"
|
|
};
|
|
|
|
window_buffer buf
|
|
{
|
|
response.buf
|
|
};
|
|
|
|
const time_t ts
|
|
{
|
|
ircd::time<milliseconds>()
|
|
};
|
|
|
|
for(const auto &item : items)
|
|
each_item(response, *item, buf, ts);
|
|
|
|
const auto flushed
|
|
{
|
|
response.flush(buf.completed())
|
|
};
|
|
|
|
buf.shift(size(flushed));
|
|
assert(!buf.consumed());
|
|
return response;
|
|
}
|
|
|
|
void
|
|
ircd::stats::each_item(resource::response::chunked &response,
|
|
stats::item<void> &item,
|
|
window_buffer &buf,
|
|
const time_t &ts)
|
|
{
|
|
buf([&item, &ts]
|
|
(const mutable_buffer &buf)
|
|
{
|
|
return print_item(item, buf, ts);
|
|
});
|
|
|
|
if(buf.remaining() >= 1_KiB)
|
|
return;
|
|
|
|
const auto completed
|
|
{
|
|
buf.completed()
|
|
};
|
|
|
|
const auto flushed
|
|
{
|
|
response.flush(completed)
|
|
};
|
|
|
|
buf.shift(size(flushed));
|
|
}
|
|
|
|
ircd::const_buffer
|
|
ircd::stats::print_item(stats::item<void> &item,
|
|
const mutable_buffer &buf,
|
|
const time_t &ts)
|
|
{
|
|
char name[2][128], val[64];
|
|
const string_view _name
|
|
{
|
|
replace(name[0], item.name, '.', '_')
|
|
};
|
|
|
|
return string_view
|
|
{
|
|
data(buf), size_t(::snprintf
|
|
(
|
|
data(buf), size(buf), "%s %s %lu\n",
|
|
data(strlcpy(name[1], _name)),
|
|
data(string(val, item)),
|
|
ts
|
|
))
|
|
};
|
|
}
|