--- /dev/null
+#include "DaemonMetricCollector.h"
+#include "common/admin_socket_client.h"
+#include "common/debug.h"
+#include "common/hostname.h"
+#include "common/perf_counters.h"
+#include "global/global_init.h"
+#include "global/global_context.h"
+#include "common/split.h"
+#include "include/common_fwd.h"
+#include "util.h"
+
+#include <boost/json/src.hpp>
+#include <chrono>
+#include <filesystem>
+#include <iostream>
+#include <map>
+#include <memory>
+#include <regex>
+#include <string>
+#include <utility>
+
+#define dout_context g_ceph_context
+#define dout_subsys ceph_subsys_ceph_exporter
+
+using json_object = boost::json::object;
+using json_value = boost::json::value;
+using json_array = boost::json::array;
+
+void DaemonMetricCollector::request_loop(boost::asio::steady_timer &timer) {
+ timer.async_wait([&](const boost::system::error_code &e) {
+ std::cerr << e << std::endl;
+ update_sockets();
+ dump_asok_metrics();
+ auto stats_period = g_conf().get_val<int64_t>("exporter_stats_period");
+ // time to wait before sending requests again
+ timer.expires_from_now(std::chrono::seconds(stats_period));
+ request_loop(timer);
+ });
+}
+
+void DaemonMetricCollector::main() {
+ // time to wait before sending requests again
+
+ boost::asio::io_service io;
+ boost::asio::steady_timer timer{io, std::chrono::seconds(0)};
+ request_loop(timer);
+ io.run();
+}
+
+std::string DaemonMetricCollector::get_metrics() {
+ const std::lock_guard<std::mutex> lock(metrics_mutex);
+ return metrics;
+}
+
+template <class T>
+void add_metric(std::unique_ptr<MetricsBuilder> &builder, T value,
+ std::string name, std::string description, std::string mtype,
+ labels_t labels) {
+ builder->add(std::to_string(value), name, description, mtype, labels);
+}
+
+void add_double_or_int_metric(std::unique_ptr<MetricsBuilder> &builder,
+ json_value value, std::string name,
+ std::string description, std::string mtype,
+ labels_t labels) {
+ if (value.is_int64()) {
+ int64_t v = value.as_int64();
+ add_metric(builder, v, name, description, mtype, labels);
+ } else if (value.is_double()) {
+ double v = value.as_double();
+ add_metric(builder, v, name, description, mtype, labels);
+ }
+}
+
+std::string boost_string_to_std(boost::json::string js) {
+ std::string res(js.data());
+ return res;
+}
+
+std::string quote(std::string value) { return "\"" + value + "\""; }
+
+bool is_hyphen(char ch) { return ch == '-'; }
+
+void DaemonMetricCollector::dump_asok_metrics() {
+ BlockTimer timer(__FILE__, __FUNCTION__);
+
+ std::vector<std::pair<std::string, int>> daemon_pids;
+
+ bool sort = g_conf().get_val<bool>("exporter_sort_metrics");
+ if (sort) {
+ builder = std::unique_ptr<OrderedMetricsBuilder>(new OrderedMetricsBuilder());
+ } else {
+ builder = std::unique_ptr<UnorderedMetricsBuilder>(new UnorderedMetricsBuilder());
+ }
+ for (auto &[daemon_name, sock_client] : clients) {
+ bool ok;
+ sock_client.ping(&ok);
+ if (!ok) {
+ continue;
+ }
+ std::string perf_dump_response = asok_request(sock_client, "perf dump", daemon_name);
+ if (perf_dump_response.size() == 0) {
+ continue;
+ }
+ std::string perf_schema_response = asok_request(sock_client, "perf schema", daemon_name);
+ if (perf_schema_response.size() == 0) {
+ continue;
+ }
+ std::string config_show = asok_request(sock_client, "config show", daemon_name);
+ json_object pid_file_json = boost::json::parse(config_show).as_object();
+ std::string pid_path =
+ boost_string_to_std(pid_file_json["pid_file"].as_string());
+ std::string pid_str = read_file_to_string(pid_path);
+ if (!pid_path.size()) {
+ continue;
+ }
+ daemon_pids.push_back({daemon_name, std::stoi(pid_str)});
+ json_object dump = boost::json::parse(perf_dump_response).as_object();
+ json_object schema = boost::json::parse(perf_schema_response).as_object();
+ for (auto &perf : schema) {
+ auto sv = perf.key();
+ std::string perf_group = {sv.begin(), sv.end()};
+ json_object perf_group_object = perf.value().as_object();
+ for (auto &perf_counter : perf_group_object) {
+ auto sv1 = perf_counter.key();
+ std::string perf_name = {sv1.begin(), sv1.end()};
+ json_object perf_info = perf_counter.value().as_object();
+ auto prio_limit = g_conf().get_val<int64_t>("exporter_prio_limit");
+ if (perf_info["priority"].as_int64() <
+ prio_limit) {
+ continue;
+ }
+ std::string name = "ceph_" + perf_group + "_" + perf_name;
+ std::replace_if(name.begin(), name.end(), is_hyphen, '_');
+
+ // FIXME: test this, based on mgr_module perfpath_to_path_labels
+ auto labels_and_name = get_labels_and_metric_name(daemon_name, name);
+ labels_t labels = labels_and_name.first;
+ name = labels_and_name.second;
+
+ json_value perf_values = dump[perf_group].as_object()[perf_name];
+ dump_asok_metric(perf_info, perf_values, name, labels);
+ }
+ }
+ }
+ dout(10) << "Perf counters retrieved for " << clients.size() << " daemons." << dendl;
+ // get time spent on this function
+ timer.stop();
+ std::string scrap_desc("Time spent scraping and transforming perfcounters to metrics");
+ labels_t scrap_labels;
+ scrap_labels["host"] = quote(ceph_get_hostname());
+ scrap_labels["function"] = quote(__FUNCTION__);
+ add_metric(builder, timer.get_ms(), "ceph_exporter_scrape_time", scrap_desc,
+ "gauge", scrap_labels);
+
+ const std::lock_guard<std::mutex> lock(metrics_mutex);
+ get_process_metrics(daemon_pids);
+ metrics = builder->dump();
+}
+
+std::vector<std::string> read_proc_stat_file(std::string path) {
+ std::string stat = read_file_to_string(path);
+ std::vector<std::string> strings;
+ auto parts = ceph::split(stat);
+ strings.assign(parts.begin(), parts.end());
+ return strings;
+}
+
+struct pstat read_pid_stat(int pid) {
+ std::string stat_path("/proc/" + std::to_string(pid) + "/stat");
+ std::vector<std::string> stats = read_proc_stat_file(stat_path);
+ struct pstat stat;
+ stat.minflt = std::stoul(stats[9]);
+ stat.majflt = std::stoul(stats[11]);
+ stat.utime = std::stoul(stats[13]);
+ stat.stime = std::stoul(stats[14]);
+ stat.num_threads = std::stoul(stats[19]);
+ stat.start_time = std::stoul(stats[21]);
+ stat.vm_size = std::stoul(stats[22]);
+ stat.resident_size = std::stoi(stats[23]);
+ return stat;
+}
+
+void DaemonMetricCollector::get_process_metrics(std::vector<std::pair<std::string, int>> daemon_pids) {
+ std::string path("/proc");
+ std::stringstream ss;
+ for (auto &[daemon_name, pid] : daemon_pids) {
+ std::vector<std::string> uptimes = read_proc_stat_file("/proc/uptime");
+ struct pstat stat = read_pid_stat(pid);
+ int clk_tck = sysconf(_SC_CLK_TCK);
+ double start_time_seconds = stat.start_time / (double)clk_tck;
+ double user_time = stat.utime / (double)clk_tck;
+ double kernel_time = stat.stime / (double)clk_tck;
+ double total_time_seconds = user_time + kernel_time;
+ double uptime = std::stod(uptimes[0]);
+ double elapsed_time = uptime - start_time_seconds;
+ double idle_time = elapsed_time - total_time_seconds;
+ double usage = total_time_seconds * 100 / elapsed_time;
+
+ labels_t labels;
+ labels["ceph_daemon"] = quote(daemon_name);
+ add_metric(builder, stat.minflt, "ceph_exporter_minflt_total",
+ "Number of minor page faults of daemon", "counter", labels);
+ add_metric(builder, stat.majflt, "ceph_exporter_majflt_total",
+ "Number of major page faults of daemon", "counter", labels);
+ add_metric(builder, stat.num_threads, "ceph_exporter_num_threads",
+ "Number of threads used by daemon", "gauge", labels);
+ add_metric(builder, usage, "ceph_exporter_cpu_usage", "CPU usage of a daemon",
+ "gauge", labels);
+
+ std::string cpu_time_desc = "Process time in kernel/user/idle mode";
+ labels_t cpu_total_labels;
+ cpu_total_labels["ceph_daemon"] = quote(daemon_name);
+ cpu_total_labels["mode"] = quote("kernel");
+ add_metric(builder, kernel_time, "ceph_exporter_cpu_total", cpu_time_desc,
+ "counter", cpu_total_labels);
+ cpu_total_labels["mode"] = quote("user");
+ add_metric(builder, user_time, "ceph_exporter_cpu_total", cpu_time_desc,
+ "counter", cpu_total_labels);
+ cpu_total_labels["mode"] = quote("idle");
+ add_metric(builder, idle_time, "ceph_exporter_cpu_total", cpu_time_desc,
+ "counter", cpu_total_labels);
+ add_metric(builder, stat.vm_size, "ceph_exporter_vm_size", "Virtual memory used in a daemon",
+ "gauge", labels);
+ add_metric(builder, stat.resident_size, "ceph_exporter_resident_size",
+ "Resident memory in a daemon", "gauge", labels);
+ }
+}
+
+std::string DaemonMetricCollector::asok_request(AdminSocketClient &asok,
+ std::string command, std::string daemon_name) {
+ std::string request("{\"prefix\": \"" + command + "\"}");
+ std::string response;
+ std::string err = asok.do_request(request, &response);
+ if (err.length() > 0 || response.substr(0, 5) == "ERROR") {
+ dout(1) << "command " << command << "failed for daemon " << daemon_name
+ << "with error: " << err << dendl;
+ return "";
+ }
+ return response;
+}
+
+std::pair<labels_t, std::string>
+DaemonMetricCollector::get_labels_and_metric_name(std::string daemon_name,
+ std::string metric_name) {
+ std::string new_metric_name;
+ labels_t labels;
+ new_metric_name = metric_name;
+ if (daemon_name.find("rgw") != std::string::npos) {
+ std::string tmp = daemon_name.substr(16, std::string::npos);
+ std::string::size_type pos = tmp.find('.');
+ labels["instance_id"] = quote("rgw." + tmp.substr(0, pos));
+ } else {
+ labels["ceph_daemon"] = quote(daemon_name);
+ if (daemon_name.find("rbd-mirror") != std::string::npos) {
+ std::regex re("^rbd_mirror_image_([^/]+)/(?:(?:([^/]+)/"
+ ")?)(.*)\\.(replay(?:_bytes|_latency)?)$");
+ std::smatch match;
+ if (std::regex_search(daemon_name, match, re) == true) {
+ new_metric_name = "ceph_rbd_mirror_image_" + match.str(4);
+ labels["pool"] = quote(match.str(1));
+ labels["namespace"] = quote(match.str(2));
+ labels["image"] = quote(match.str(3));
+ }
+ }
+ }
+ return {labels, new_metric_name};
+}
+
+/*
+perf_values can be either a int/double or a json_object. Since
+ json_value is a wrapper of both we use that class.
+ */
+void DaemonMetricCollector::dump_asok_metric(json_object perf_info,
+ json_value perf_values,
+ std::string name,
+ labels_t labels) {
+ int64_t type = perf_info["type"].as_int64();
+ std::string metric_type =
+ boost_string_to_std(perf_info["metric_type"].as_string());
+ std::string description =
+ boost_string_to_std(perf_info["description"].as_string());
+
+ if (type & PERFCOUNTER_LONGRUNAVG) {
+ int64_t count = perf_values.as_object()["avgcount"].as_int64();
+ add_metric(builder, count, name + "_count", description, metric_type,
+ labels);
+ json_value sum_value = perf_values.as_object()["sum"];
+ add_double_or_int_metric(builder, sum_value, name + "_sum", description,
+ metric_type, labels);
+ } else if (type & PERFCOUNTER_TIME) {
+ if (perf_values.is_int64()) {
+ double value = perf_values.as_int64() / 1000000000.0f;
+ add_metric(builder, value, name, description, metric_type, labels);
+ } else if (perf_values.is_double()) {
+ double value = perf_values.as_double() / 1000000000.0f;
+ add_metric(builder, value, name, description, metric_type, labels);
+ }
+ } else {
+ add_double_or_int_metric(builder, perf_values, name, description,
+ metric_type, labels);
+ }
+}
+
+void DaemonMetricCollector::update_sockets() {
+ std::string sock_dir = g_conf().get_val<std::string>("exporter_sock_dir");
+ clients.clear();
+ std::filesystem::path sock_path = sock_dir;
+ if(!std::filesystem::is_directory(sock_path.parent_path())) {
+ dout(1) << "ERROR: No such directory exist" << sock_dir << dendl;
+ return;
+ }
+ for (const auto &entry :
+ std::filesystem::directory_iterator(sock_dir)) {
+ if (entry.path().extension() == ".asok") {
+ std::string daemon_socket_name = entry.path().filename().string();
+ std::string daemon_name =
+ daemon_socket_name.substr(0, daemon_socket_name.size() - 5);
+ if (clients.find(daemon_name) == clients.end() &&
+ !(daemon_name.find("mgr") != std::string::npos) &&
+ !(daemon_name.find("ceph-exporter") != std::string::npos)) {
+ AdminSocketClient sock(entry.path().string());
+ clients.insert({daemon_name, std::move(sock)});
+ }
+ }
+ }
+}
+
+void OrderedMetricsBuilder::add(std::string value, std::string name,
+ std::string description, std::string mtype,
+ labels_t labels) {
+
+ if (metrics.find(name) == metrics.end()) {
+ Metric metric(name, mtype, description);
+ metrics[name] = std::move(metric);
+ }
+ Metric &metric = metrics[name];
+ metric.add(labels, value);
+}
+
+std::string OrderedMetricsBuilder::dump() {
+ for (auto &[name, metric] : metrics) {
+ out += metric.dump() + "\n";
+ }
+ return out;
+}
+
+void UnorderedMetricsBuilder::add(std::string value, std::string name,
+ std::string description, std::string mtype,
+ labels_t labels) {
+
+ Metric metric(name, mtype, description);
+ metric.add(labels, value);
+ out += metric.dump() + "\n\n";
+}
+
+std::string UnorderedMetricsBuilder::dump() { return out; }
+
+void Metric::add(labels_t labels, std::string value) {
+ metric_entry entry;
+ entry.labels = labels;
+ entry.value = value;
+ entries.push_back(entry);
+}
+
+std::string Metric::dump() {
+ std::stringstream metric_ss;
+ metric_ss << "# HELP " << name << " " << description << "\n";
+ metric_ss << "# TYPE " << name << " " << mtype << "\n";
+ for (auto &entry : entries) {
+ std::stringstream labels_ss;
+ size_t i = 0;
+ for (auto &[label_name, label_value] : entry.labels) {
+ labels_ss << label_name << "=" << label_value;
+ if (i < entry.labels.size() - 1) {
+ labels_ss << ",";
+ }
+ i++;
+ }
+ metric_ss << name << "{" << labels_ss.str() << "} " << entry.value;
+ if (&entry != &entries.back()) {
+ metric_ss << "\n";
+ }
+ }
+ return metric_ss.str();
+}
+
+DaemonMetricCollector &collector_instance() {
+ static DaemonMetricCollector instance;
+ return instance;
+}
--- /dev/null
+#include "http_server.h"
+#include "common/debug.h"
+#include "common/hostname.h"
+#include "global/global_init.h"
+#include "global/global_context.h"
+#include "exporter/DaemonMetricCollector.h"
+
+#include <boost/asio.hpp>
+#include <boost/beast/core.hpp>
+#include <boost/beast/http.hpp>
+#include <boost/beast/version.hpp>
+#include <boost/thread/thread.hpp>
+#include <chrono>
+#include <cstdlib>
+#include <ctime>
+#include <iostream>
+#include <map>
+#include <memory>
+#include <string>
+
+#define dout_context g_ceph_context
+#define dout_subsys ceph_subsys_ceph_exporter
+
+namespace beast = boost::beast; // from <boost/beast.hpp>
+namespace http = beast::http; // from <boost/beast/http.hpp>
+namespace net = boost::asio; // from <boost/asio.hpp>
+using tcp = boost::asio::ip::tcp; // from <boost/asio/ip/tcp.hpp>
+
+class http_connection : public std::enable_shared_from_this<http_connection> {
+public:
+ http_connection(tcp::socket socket) : socket_(std::move(socket)) {}
+
+ // Initiate the asynchronous operations associated with the connection.
+ void start() {
+ read_request();
+ check_deadline();
+ }
+
+private:
+ tcp::socket socket_;
+ beast::flat_buffer buffer_{8192};
+ http::request<http::dynamic_body> request_;
+ http::response<http::string_body> response_;
+
+ net::steady_timer deadline_{socket_.get_executor(), std::chrono::seconds(60)};
+
+ // Asynchronously receive a complete request message.
+ void read_request() {
+ auto self = shared_from_this();
+
+ http::async_read(socket_, buffer_, request_,
+ [self](beast::error_code ec, std::size_t bytes_transferred) {
+ boost::ignore_unused(bytes_transferred);
+ if (ec) {
+ dout(1) << "ERROR: " << ec.message() << dendl;
+ return;
+ }
+ else {
+ self->process_request();
+ }
+ });
+ }
+
+ // Determine what needs to be done with the request message.
+ void process_request() {
+ response_.version(request_.version());
+ response_.keep_alive(request_.keep_alive());
+
+ switch (request_.method()) {
+ case http::verb::get:
+ response_.result(http::status::ok);
+ create_response();
+ break;
+
+ default:
+ // We return responses indicating an error if
+ // we do not recognize the request method.
+ response_.result(http::status::method_not_allowed);
+ response_.set(http::field::content_type, "text/plain");
+ std::string body("Invalid request-method '" +
+ std::string(request_.method_string()) + "'");
+ response_.body() = body;
+ break;
+ }
+
+ write_response();
+ }
+
+ // Construct a response message based on the program state.
+ void create_response() {
+ if (request_.target() == "/") {
+ response_.set(http::field::content_type, "text/html; charset=utf-8");
+ std::string body("<html>\n"
+ "<head><title>Ceph Exporter</title></head>\n"
+ "<body>\n"
+ "<h1>Ceph Exporter</h1>\n"
+ "<p><a href='/metrics'>Metrics</a></p>"
+ "</body>\n"
+ "</html>\n");
+ response_.body() = body;
+ } else if (request_.target() == "/metrics") {
+ response_.set(http::field::content_type, "text/plain; charset=utf-8");
+ DaemonMetricCollector &collector = collector_instance();
+ std::string metrics = collector.get_metrics();
+ response_.body() = metrics;
+ } else {
+ response_.result(http::status::method_not_allowed);
+ response_.set(http::field::content_type, "text/plain");
+ response_.body() = "File not found \n";
+ }
+ }
+
+ // Asynchronously transmit the response message.
+ void write_response() {
+ auto self = shared_from_this();
+
+ response_.prepare_payload();
+
+ http::async_write(socket_, response_,
+ [self](beast::error_code ec, std::size_t) {
+ self->socket_.shutdown(tcp::socket::shutdown_send, ec);
+ self->deadline_.cancel();
+ if (ec) {
+ dout(1) << "ERROR: " << ec.message() << dendl;
+ return;
+ }
+ });
+ }
+
+ // Check whether we have spent enough time on this connection.
+ void check_deadline() {
+ auto self = shared_from_this();
+
+ deadline_.async_wait([self](beast::error_code ec) {
+ if (!ec) {
+ // Close socket to cancel any outstanding operation.
+ self->socket_.close(ec);
+ }
+ });
+ }
+};
+
+// "Loop" forever accepting new connections.
+void http_server(tcp::acceptor &acceptor, tcp::socket &socket) {
+ acceptor.async_accept(socket, [&](beast::error_code ec) {
+ if (!ec)
+ std::make_shared<http_connection>(std::move(socket))->start();
+ http_server(acceptor, socket);
+ });
+}
+
+void http_server_thread_entrypoint() {
+ try {
+ std::string exporter_addr = g_conf().get_val<std::string>("exporter_addr");
+ auto const address = net::ip::make_address(exporter_addr);
+ unsigned short port = g_conf().get_val<int64_t>("exporter_http_port");
+
+ net::io_context ioc{1};
+
+ tcp::acceptor acceptor{ioc, {address, port}};
+ tcp::socket socket{ioc};
+ http_server(acceptor, socket);
+ dout(1) << "Http server running on " << exporter_addr << ":" << port << dendl;
+ ioc.run();
+ } catch (std::exception const &e) {
+ dout(1) << "Error: " << e.what() << dendl;
+ exit(EXIT_FAILURE);
+ }
+}
projects / ceph-ci.git / commitdiff