Add Prometheus metrics relabeling (#3734)

Co-authored-by: Guillermo Ruffino <glm.net@gmail.com>
This commit is contained in:
Jan Grewe 2022-08-30 00:55:55 +02:00 committed by GitHub
parent 2819166539
commit 84bac8356a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 117 additions and 52 deletions

View file

@ -2,16 +2,27 @@ import esphome.config_validation as cv
import esphome.codegen as cg import esphome.codegen as cg
from esphome.const import ( from esphome.const import (
CONF_ID, CONF_ID,
CONF_NAME,
CONF_INCLUDE_INTERNAL, CONF_INCLUDE_INTERNAL,
CONF_RELABEL,
) )
from esphome.components.web_server_base import CONF_WEB_SERVER_BASE_ID from esphome.components.web_server_base import CONF_WEB_SERVER_BASE_ID
from esphome.components import web_server_base from esphome.components import web_server_base
from esphome.cpp_types import EntityBase
AUTO_LOAD = ["web_server_base"] AUTO_LOAD = ["web_server_base"]
prometheus_ns = cg.esphome_ns.namespace("prometheus") prometheus_ns = cg.esphome_ns.namespace("prometheus")
PrometheusHandler = prometheus_ns.class_("PrometheusHandler", cg.Component) PrometheusHandler = prometheus_ns.class_("PrometheusHandler", cg.Component)
CUSTOMIZED_ENTITY = cv.Schema(
{
cv.Optional(CONF_ID): cv.string_strict,
cv.Optional(CONF_NAME): cv.string_strict,
},
cv.has_at_least_one_key,
)
CONFIG_SCHEMA = cv.Schema( CONFIG_SCHEMA = cv.Schema(
{ {
cv.GenerateID(): cv.declare_id(PrometheusHandler), cv.GenerateID(): cv.declare_id(PrometheusHandler),
@ -19,6 +30,11 @@ CONFIG_SCHEMA = cv.Schema(
web_server_base.WebServerBase web_server_base.WebServerBase
), ),
cv.Optional(CONF_INCLUDE_INTERNAL, default=False): cv.boolean, cv.Optional(CONF_INCLUDE_INTERNAL, default=False): cv.boolean,
cv.Optional(CONF_RELABEL, default={}): cv.Schema(
{
cv.use_id(EntityBase): CUSTOMIZED_ENTITY,
}
),
}, },
cv.only_with_arduino, cv.only_with_arduino,
).extend(cv.COMPONENT_SCHEMA) ).extend(cv.COMPONENT_SCHEMA)
@ -33,3 +49,10 @@ async def to_code(config):
await cg.register_component(var, config) await cg.register_component(var, config)
cg.add(var.set_include_internal(config[CONF_INCLUDE_INTERNAL])) cg.add(var.set_include_internal(config[CONF_INCLUDE_INTERNAL]))
for key, value in config[CONF_RELABEL].items():
entity = await cg.get_variable(key)
if CONF_ID in value:
cg.add(var.add_label_id(entity, value[CONF_ID]))
if CONF_NAME in value:
cg.add(var.add_label_name(entity, value[CONF_NAME]))

View file

@ -54,6 +54,16 @@ void PrometheusHandler::handleRequest(AsyncWebServerRequest *req) {
req->send(stream); req->send(stream);
} }
std::string PrometheusHandler::relabel_id_(EntityBase *obj) {
auto item = relabel_map_id_.find(obj);
return item == relabel_map_id_.end() ? obj->get_object_id() : item->second;
}
std::string PrometheusHandler::relabel_name_(EntityBase *obj) {
auto item = relabel_map_name_.find(obj);
return item == relabel_map_name_.end() ? obj->get_name() : item->second;
}
// Type-specific implementation // Type-specific implementation
#ifdef USE_SENSOR #ifdef USE_SENSOR
void PrometheusHandler::sensor_type_(AsyncResponseStream *stream) { void PrometheusHandler::sensor_type_(AsyncResponseStream *stream) {
@ -66,15 +76,15 @@ void PrometheusHandler::sensor_row_(AsyncResponseStream *stream, sensor::Sensor
if (!std::isnan(obj->state)) { if (!std::isnan(obj->state)) {
// We have a valid value, output this value // We have a valid value, output this value
stream->print(F("esphome_sensor_failed{id=\"")); stream->print(F("esphome_sensor_failed{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} 0\n")); stream->print(F("\"} 0\n"));
// Data itself // Data itself
stream->print(F("esphome_sensor_value{id=\"")); stream->print(F("esphome_sensor_value{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\",unit=\"")); stream->print(F("\",unit=\""));
stream->print(obj->get_unit_of_measurement().c_str()); stream->print(obj->get_unit_of_measurement().c_str());
stream->print(F("\"} ")); stream->print(F("\"} "));
@ -83,9 +93,9 @@ void PrometheusHandler::sensor_row_(AsyncResponseStream *stream, sensor::Sensor
} else { } else {
// Invalid state // Invalid state
stream->print(F("esphome_sensor_failed{id=\"")); stream->print(F("esphome_sensor_failed{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} 1\n")); stream->print(F("\"} 1\n"));
} }
} }
@ -103,24 +113,24 @@ void PrometheusHandler::binary_sensor_row_(AsyncResponseStream *stream, binary_s
if (obj->has_state()) { if (obj->has_state()) {
// We have a valid value, output this value // We have a valid value, output this value
stream->print(F("esphome_binary_sensor_failed{id=\"")); stream->print(F("esphome_binary_sensor_failed{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} 0\n")); stream->print(F("\"} 0\n"));
// Data itself // Data itself
stream->print(F("esphome_binary_sensor_value{id=\"")); stream->print(F("esphome_binary_sensor_value{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} ")); stream->print(F("\"} "));
stream->print(obj->state); stream->print(obj->state);
stream->print('\n'); stream->print('\n');
} else { } else {
// Invalid state // Invalid state
stream->print(F("esphome_binary_sensor_failed{id=\"")); stream->print(F("esphome_binary_sensor_failed{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} 1\n")); stream->print(F("\"} 1\n"));
} }
} }
@ -137,24 +147,24 @@ void PrometheusHandler::fan_row_(AsyncResponseStream *stream, fan::Fan *obj) {
if (obj->is_internal() && !this->include_internal_) if (obj->is_internal() && !this->include_internal_)
return; return;
stream->print(F("esphome_fan_failed{id=\"")); stream->print(F("esphome_fan_failed{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} 0\n")); stream->print(F("\"} 0\n"));
// Data itself // Data itself
stream->print(F("esphome_fan_value{id=\"")); stream->print(F("esphome_fan_value{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} ")); stream->print(F("\"} "));
stream->print(obj->state); stream->print(obj->state);
stream->print('\n'); stream->print('\n');
// Speed if available // Speed if available
if (obj->get_traits().supports_speed()) { if (obj->get_traits().supports_speed()) {
stream->print(F("esphome_fan_speed{id=\"")); stream->print(F("esphome_fan_speed{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} ")); stream->print(F("\"} "));
stream->print(obj->speed); stream->print(obj->speed);
stream->print('\n'); stream->print('\n');
@ -162,9 +172,9 @@ void PrometheusHandler::fan_row_(AsyncResponseStream *stream, fan::Fan *obj) {
// Oscillation if available // Oscillation if available
if (obj->get_traits().supports_oscillation()) { if (obj->get_traits().supports_oscillation()) {
stream->print(F("esphome_fan_oscillation{id=\"")); stream->print(F("esphome_fan_oscillation{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} ")); stream->print(F("\"} "));
stream->print(obj->oscillating); stream->print(obj->oscillating);
stream->print('\n'); stream->print('\n');
@ -183,9 +193,9 @@ void PrometheusHandler::light_row_(AsyncResponseStream *stream, light::LightStat
return; return;
// State // State
stream->print(F("esphome_light_state{id=\"")); stream->print(F("esphome_light_state{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} ")); stream->print(F("\"} "));
stream->print(obj->remote_values.is_on()); stream->print(obj->remote_values.is_on());
stream->print(F("\n")); stream->print(F("\n"));
@ -195,37 +205,37 @@ void PrometheusHandler::light_row_(AsyncResponseStream *stream, light::LightStat
color.as_brightness(&brightness); color.as_brightness(&brightness);
color.as_rgbw(&r, &g, &b, &w); color.as_rgbw(&r, &g, &b, &w);
stream->print(F("esphome_light_color{id=\"")); stream->print(F("esphome_light_color{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\",channel=\"brightness\"} ")); stream->print(F("\",channel=\"brightness\"} "));
stream->print(brightness); stream->print(brightness);
stream->print(F("\n")); stream->print(F("\n"));
stream->print(F("esphome_light_color{id=\"")); stream->print(F("esphome_light_color{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\",channel=\"r\"} ")); stream->print(F("\",channel=\"r\"} "));
stream->print(r); stream->print(r);
stream->print(F("\n")); stream->print(F("\n"));
stream->print(F("esphome_light_color{id=\"")); stream->print(F("esphome_light_color{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\",channel=\"g\"} ")); stream->print(F("\",channel=\"g\"} "));
stream->print(g); stream->print(g);
stream->print(F("\n")); stream->print(F("\n"));
stream->print(F("esphome_light_color{id=\"")); stream->print(F("esphome_light_color{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\",channel=\"b\"} ")); stream->print(F("\",channel=\"b\"} "));
stream->print(b); stream->print(b);
stream->print(F("\n")); stream->print(F("\n"));
stream->print(F("esphome_light_color{id=\"")); stream->print(F("esphome_light_color{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\",channel=\"w\"} ")); stream->print(F("\",channel=\"w\"} "));
stream->print(w); stream->print(w);
stream->print(F("\n")); stream->print(F("\n"));
@ -233,15 +243,15 @@ void PrometheusHandler::light_row_(AsyncResponseStream *stream, light::LightStat
std::string effect = obj->get_effect_name(); std::string effect = obj->get_effect_name();
if (effect == "None") { if (effect == "None") {
stream->print(F("esphome_light_effect_active{id=\"")); stream->print(F("esphome_light_effect_active{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\",effect=\"None\"} 0\n")); stream->print(F("\",effect=\"None\"} 0\n"));
} else { } else {
stream->print(F("esphome_light_effect_active{id=\"")); stream->print(F("esphome_light_effect_active{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\",effect=\"")); stream->print(F("\",effect=\""));
stream->print(effect.c_str()); stream->print(effect.c_str());
stream->print(F("\"} 1\n")); stream->print(F("\"} 1\n"));
@ -260,23 +270,23 @@ void PrometheusHandler::cover_row_(AsyncResponseStream *stream, cover::Cover *ob
if (!std::isnan(obj->position)) { if (!std::isnan(obj->position)) {
// We have a valid value, output this value // We have a valid value, output this value
stream->print(F("esphome_cover_failed{id=\"")); stream->print(F("esphome_cover_failed{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} 0\n")); stream->print(F("\"} 0\n"));
// Data itself // Data itself
stream->print(F("esphome_cover_value{id=\"")); stream->print(F("esphome_cover_value{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} ")); stream->print(F("\"} "));
stream->print(obj->position); stream->print(obj->position);
stream->print('\n'); stream->print('\n');
if (obj->get_traits().get_supports_tilt()) { if (obj->get_traits().get_supports_tilt()) {
stream->print(F("esphome_cover_tilt{id=\"")); stream->print(F("esphome_cover_tilt{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} ")); stream->print(F("\"} "));
stream->print(obj->tilt); stream->print(obj->tilt);
stream->print('\n'); stream->print('\n');
@ -284,9 +294,9 @@ void PrometheusHandler::cover_row_(AsyncResponseStream *stream, cover::Cover *ob
} else { } else {
// Invalid state // Invalid state
stream->print(F("esphome_cover_failed{id=\"")); stream->print(F("esphome_cover_failed{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} 1\n")); stream->print(F("\"} 1\n"));
} }
} }
@ -301,15 +311,15 @@ void PrometheusHandler::switch_row_(AsyncResponseStream *stream, switch_::Switch
if (obj->is_internal() && !this->include_internal_) if (obj->is_internal() && !this->include_internal_)
return; return;
stream->print(F("esphome_switch_failed{id=\"")); stream->print(F("esphome_switch_failed{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} 0\n")); stream->print(F("\"} 0\n"));
// Data itself // Data itself
stream->print(F("esphome_switch_value{id=\"")); stream->print(F("esphome_switch_value{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} ")); stream->print(F("\"} "));
stream->print(obj->state); stream->print(obj->state);
stream->print('\n'); stream->print('\n');
@ -325,15 +335,15 @@ void PrometheusHandler::lock_row_(AsyncResponseStream *stream, lock::Lock *obj)
if (obj->is_internal() && !this->include_internal_) if (obj->is_internal() && !this->include_internal_)
return; return;
stream->print(F("esphome_lock_failed{id=\"")); stream->print(F("esphome_lock_failed{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} 0\n")); stream->print(F("\"} 0\n"));
// Data itself // Data itself
stream->print(F("esphome_lock_value{id=\"")); stream->print(F("esphome_lock_value{id=\""));
stream->print(obj->get_object_id().c_str()); stream->print(relabel_id_(obj).c_str());
stream->print(F("\",name=\"")); stream->print(F("\",name=\""));
stream->print(obj->get_name().c_str()); stream->print(relabel_name_(obj).c_str());
stream->print(F("\"} ")); stream->print(F("\"} "));
stream->print(obj->state); stream->print(obj->state);
stream->print('\n'); stream->print('\n');

View file

@ -2,6 +2,9 @@
#ifdef USE_ARDUINO #ifdef USE_ARDUINO
#include <map>
#include <utility>
#include "esphome/components/web_server_base/web_server_base.h" #include "esphome/components/web_server_base/web_server_base.h"
#include "esphome/core/controller.h" #include "esphome/core/controller.h"
#include "esphome/core/component.h" #include "esphome/core/component.h"
@ -20,6 +23,20 @@ class PrometheusHandler : public AsyncWebHandler, public Component {
*/ */
void set_include_internal(bool include_internal) { include_internal_ = include_internal; } void set_include_internal(bool include_internal) { include_internal_ = include_internal; }
/** Add the value for an entity's "id" label.
*
* @param obj The entity for which to set the "id" label
* @param value The value for the "id" label
*/
void add_label_id(EntityBase *obj, const std::string &value) { relabel_map_id_.insert({obj, value}); }
/** Add the value for an entity's "name" label.
*
* @param obj The entity for which to set the "name" label
* @param value The value for the "name" label
*/
void add_label_name(EntityBase *obj, const std::string &value) { relabel_map_name_.insert({obj, value}); }
bool canHandle(AsyncWebServerRequest *request) override { bool canHandle(AsyncWebServerRequest *request) override {
if (request->method() == HTTP_GET) { if (request->method() == HTTP_GET) {
if (request->url() == "/metrics") if (request->url() == "/metrics")
@ -41,6 +58,9 @@ class PrometheusHandler : public AsyncWebHandler, public Component {
} }
protected: protected:
std::string relabel_id_(EntityBase *obj);
std::string relabel_name_(EntityBase *obj);
#ifdef USE_SENSOR #ifdef USE_SENSOR
/// Return the type for prometheus /// Return the type for prometheus
void sensor_type_(AsyncResponseStream *stream); void sensor_type_(AsyncResponseStream *stream);
@ -92,6 +112,8 @@ class PrometheusHandler : public AsyncWebHandler, public Component {
web_server_base::WebServerBase *base_; web_server_base::WebServerBase *base_;
bool include_internal_{false}; bool include_internal_{false};
std::map<EntityBase *, std::string> relabel_map_id_;
std::map<EntityBase *, std::string> relabel_map_name_;
}; };
} // namespace prometheus } // namespace prometheus

View file

@ -244,6 +244,8 @@ def iter_ids(config, path=None):
yield from iter_ids(item, path + [i]) yield from iter_ids(item, path + [i])
elif isinstance(config, dict): elif isinstance(config, dict):
for key, value in config.items(): for key, value in config.items():
if isinstance(key, core.ID):
yield key, path
yield from iter_ids(value, path + [key]) yield from iter_ids(value, path + [key])

View file

@ -565,6 +565,7 @@ CONF_REF = "ref"
CONF_REFERENCE_RESISTANCE = "reference_resistance" CONF_REFERENCE_RESISTANCE = "reference_resistance"
CONF_REFERENCE_TEMPERATURE = "reference_temperature" CONF_REFERENCE_TEMPERATURE = "reference_temperature"
CONF_REFRESH = "refresh" CONF_REFRESH = "refresh"
CONF_RELABEL = "relabel"
CONF_REPEAT = "repeat" CONF_REPEAT = "repeat"
CONF_REPOSITORY = "repository" CONF_REPOSITORY = "repository"
CONF_RESET_DURATION = "reset_duration" CONF_RESET_DURATION = "reset_duration"

View file

@ -667,3 +667,10 @@ media_player:
- media_player.volume_up: - media_player.volume_up:
- media_player.volume_down: - media_player.volume_down:
- media_player.volume_set: 50% - media_player.volume_set: 50%
prometheus:
include_internal: true
relabel:
ha_hello_world:
id: hellow_world
name: "Hello World"