Compare commits

..

30 Commits

Author SHA1 Message Date
deployer3000 fe248da39f Merge pull request 'rc' (#22) from rc into master 2025-05-20 15:01:15 +03:00
Vladislav Drozdov 9ec3ab5e1d Merge pull request 'hotfix: metric' (#21) from feature/161 into rc
test-org/prometheus-exporter/pipeline/pr-master Build succeeded
Reviewed-on: http://git.enode/deployer3000/prometheus-exporter/pulls/21
2025-05-20 14:58:57 +03:00
prplV 8db87b9fb6 hotfix: metric
test-org/prometheus-exporter/pipeline/pr-rc Build queued... Details
2025-05-20 07:58:13 -04:00
deployer3000 836759bf24 Merge pull request 'rc' (#20) from rc into master 2025-05-20 12:34:55 +03:00
Vladislav Drozdov ebba2fa008 Merge pull request 'feature/161' (#19) from feature/161 into rc
test-org/prometheus-exporter/pipeline/pr-master Build succeeded
Reviewed-on: http://git.enode/deployer3000/prometheus-exporter/pulls/19
Reviewed-by: DmitriyA <faleo1999@mail.ru>
2025-05-20 12:27:01 +03:00
prplV 0b906b10c1 fix + prefix setting
test-org/prometheus-exporter/pipeline/pr-rc This commit looks good Details
2025-05-20 04:48:15 -04:00
prplV c34a0fd176 + source_id
test-org/prometheus-exporter/pipeline/pr-rc This commit looks good Details
2025-05-19 08:05:51 -04:00
deployer3000 0e2a1fff65 Merge pull request 'rc' (#17) from rc into master 2025-05-16 17:49:44 +03:00
Vladislav Drozdov 0ca3b964e6 Merge pull request 'feature/161' (#16) from feature/161 into rc
test-org/prometheus-exporter/pipeline/pr-master Build succeeded
Reviewed-on: http://git.enode/deployer3000/prometheus-exporter/pulls/16
Reviewed-by: DmitriyA <faleo1999@mail.ru>
2025-05-16 17:47:47 +03:00
prplV 2ccf1840c0 refactor
test-org/prometheus-exporter/pipeline/pr-rc This commit looks good Details
2025-05-15 12:07:00 -04:00
prplV 1c9a38c978 multimetrics 2025-05-15 12:05:08 -04:00
prplV 8016ff80c0 + mod and device 2025-05-15 10:37:31 -04:00
deployer3000 7c77b837e4 Merge pull request 'rc' (#15) from rc into master 2025-04-10 12:10:12 +03:00
yuobrezkov c33a4432e7 Added \n
test-org/prometheus-exporter/pipeline/pr-master Build succeeded
2025-04-10 12:07:47 +03:00
yuobrezkov fcc3e5aa5a Deleted echo info 2025-04-10 12:07:30 +03:00
deployer3000 18453be2aa Merge pull request 'added fn notify' (#14) from rc into master 2025-04-10 12:01:26 +03:00
yuobrezkov 165796aa54 added fn notify
test-org/prometheus-exporter/pipeline/pr-master Build succeeded
2025-04-10 11:57:48 +03:00
deployer3000 9f71266bb9 Merge pull request 'rc' (#13) from rc into master 2025-04-10 11:45:26 +03:00
yuobrezkov ca09d2c44c changed main to master
test-org/prometheus-exporter/pipeline/pr-master Build started... Details
2025-04-10 11:40:21 +03:00
yuobrezkov 85c7422735 auto versioning added 2025-04-10 11:37:02 +03:00
deployer3000 141d0332da Merge pull request 'rc' (#12) from rc into master 2025-04-08 12:44:23 +03:00
YurijO 586c664f09 Merge pull request 'feature/1184' (#11) from feature/1184 into rc
test-org/prometheus-exporter/pipeline/pr-master Build started... Details
Reviewed-on: http://git.enode/deployer3000/prometheus-exporter/pulls/11
Reviewed-by: DmitriyA <faleo1999@mail.ru>
Reviewed-by: YurijO <ya@ya.ru>
2025-04-08 12:30:39 +03:00
prplV 5842c93ecf main log logic rework 2025-04-08 04:44:41 -04:00
prplV 9055142073 SIGSEGV (Address boundary error) bug fixed! 2025-04-08 03:55:25 -04:00
prplV c558d8bcc7 - comments 2025-04-08 03:54:52 -04:00
prplV ab1ed5a57c US changes 2025-03-28 05:13:16 -04:00
YurijO f9c90b82f9 Merge pull request 'readme update merging' (#10) from feature/readme into rc
Reviewed-on: http://git.enode/deployer3000/prometheus-exporter/pulls/10
2025-03-20 16:29:51 +03:00
prplV ac218ea845 tracing and logging added 2025-03-20 05:27:07 -04:00
deployer3000 0842f344cf Merge pull request 'rc' (#9) from rc into master 2025-03-07 12:21:08 +03:00
deployer3000 d5f0f470d6 Merge pull request 'rc' (#7) from rc into master 2025-03-04 16:22:53 +03:00
6 changed files with 341 additions and 215 deletions

View File

@ -1,3 +1,13 @@
# port binding for prometheus-exporter
# default value = 9100
PROMETHEUS_EXPORTER_PORT = 9100
# prefix for metric naming
# default value = "zvks"
PROMETHEUS_EXPORTER_PREFIX = "zvks"
# setting up max level of logging
# default - INFO
# values : WARN, ERROR, INFO, DEBUG, TRACE
PROMETHEUS_EXPORTER_LOG_LEVEL = "TRACE"

56
Jenkinsfile vendored
View File

@ -1,3 +1,24 @@
def notify(
String context,
String giteaUser,
String giteaPass,
String repositoryUrl,
String repositoryName,
String commitHash,
String buildStatus
) {
def status = buildStatus == 'success' ? 'success' : 'failure'
def description = buildStatus == 'success' ? 'Build succeeded' : 'Build failed'
sh """
curl -X POST \
-u "${giteaUser}:${giteaPass}" \
-H "Content-Type: application/json" \
-d '{"context":"${context}","state": "${status}", "description": "${description}"}' \
${repositoryUrl}deployer3000/${repositoryName}/statuses/${commitHash}
"""
}
pipeline {
agent any
environment {
@ -13,7 +34,22 @@ pipeline {
}
steps {
script {
env.IMAGE_TAG = sh(script: "git describe --tags --abbrev=0", returnStdout: true).trim()
def hasTags = sh(script: "git tag -l | wc -l", returnStdout: true).trim().toInteger() > 0
def lastVersion = "0.0.0"
if (hasTags) {
lastVersion = sh(script: "git describe --tags --abbrev=0", returnStdout: true).trim()
}
echo "Last version: ${lastVersion}"
def (major, minor, patch) = lastVersion.tokenize('.')
def newVersion = "${major}.${minor}.${patch.toInteger() + 1}"
echo "New version: ${newVersion}"
env.IMAGE_TAG = newVersion
env.NEW_VERSION = newVersion
}
}
}
@ -58,14 +94,30 @@ pipeline {
echo "Attempting to merge PR ${env.CHANGE_ID} into master..."
withCredentials([usernamePassword(credentialsId: 'gitea_creds', usernameVariable: 'GITEA_USER', passwordVariable: 'GITEA_PASS')]) {
def prId = env.CHANGE_ID
sh """
curl -X POST \
-u "${GITEA_USER}:${GITEA_PASS}" \
-H "Content-Type: application/json" \
-d '{"do":"merge"}' \
http://git.entcor/api/v1/repos/deployer3000/prometheus-exporter/pulls/${prId}/merge
http://git.entcor/api/v1/repos/deployer3000/${env.IMAGE_NAME}/pulls/${prId}/merge
"""
def commitHash = sh(script: "git rev-parse HEAD~1", returnStdout: true).trim() // необходим для корректного отображения статусов
echo "PR ${prId} merged successfully into master!"
sleep(time: 15, unit: 'SECONDS')
sh "git checkout master && git pull origin master"
sh """
curl -v -X POST -u "${GITEA_USER}:${GITEA_PASS}" \
-H "Content-Type: application/json" \
-d '{"tag_name": "${env.NEW_VERSION}", "name": "Release ${env.NEW_VERSION}", "target_commitish": "master"}' \
"${env.GITEA_REPOSITORY_URL}deployer3000/${env.IMAGE_NAME}/releases"
"""
echo "New release succeeded!"
def context = "test-org/${env.IMAGE_NAME}/pipeline/pr-${env.CHANGE_TARGET}"
notify(context, GITEA_USER, GITEA_PASS, env.GITEA_REPOSITORY_URL, env.IMAGE_NAME, commitHash, "success")
}
}
}

View File

@ -4,11 +4,66 @@ use axum::{
http
};
use crate::structs::v3::PrometheusMetrics;
use prometheus::{ Encoder, Gauge, Registry, TextEncoder};
use std::sync::{ Arc, MutexGuard };
use prometheus::{ core::Collector, Encoder, GaugeVec, Registry, TextEncoder, proto::Metric};
use std::{collections::HashMap, fmt::Display, sync::{ Arc, MutexGuard }};
use crate::AppState;
use tracing::{ debug, error, info, warn };
use tracing::{ debug, error, info, warn, trace };
use crate::metrics::{MetricsProcesser, MetricsValueType};
use prometheus::opts;
#[derive(Clone)]
struct CloneableCollector(Arc<dyn Collector>);
impl CloneableCollector {
fn from_boxed(collector: Box<dyn Collector>) -> Self {
CloneableCollector(Arc::from(collector))
}
fn get_collector(&self) -> Box<dyn Collector> {
Box::new(self.clone())
}
fn to_gauge_vec(&self) -> GaugeVec {
let mut fam = self.0.collect()[0].clone();
let metric_name = fam.take_name();
let metric_help = fam.take_help();
let lables = fam.get_metric().iter()
.enumerate()
.filter(|(id, _)| id == &0)
.map(|(_, metric)| metric.get_label())
.flat_map(|a| a.iter())
.map(|a| a.get_name())
.collect::<Vec<&str>>();
let gv = GaugeVec::new(opts!(&metric_name, &metric_help), &lables).unwrap();
for metric in fam.get_metric() {
// gather values
let vals = metric.get_label()
.iter()
.map(|pair| pair.get_value())
.collect::<Vec<&str>>();
gv.with_label_values(&vals).set(metric.get_gauge().get_value());
}
gv
}
}
impl Display for CloneableCollector {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> Result<(), std::fmt::Error> {
self.0.collect();
write!(f, "{:#?}", self.0.collect())
}
}
impl Collector for CloneableCollector {
fn desc(&self) -> Vec<&prometheus::core::Desc> {
self.0.desc()
}
fn collect(&self) -> Vec<prometheus::proto::MetricFamily> {
self.0.collect()
}
}
/// An `Update` endpoint
///
@ -18,31 +73,47 @@ use crate::metrics::{MetricsProcesser, MetricsValueType};
/// # Usage
///
/// ``` bash
/// curl -X POST -d '"id" : ...' 'http::/localhost:9100/update'
/// curl -X POST -d '...' 'http://127.0.0.1:9100/update' -d ...
/// ```
///
pub async fn update_metrics(
State(state): State<Arc<AppState>>,
Json(request) : Json<PrometheusMetrics<'_>>
) -> impl IntoResponse {
info!("post on /update");
trace!("post on /update");
let service = &request.service_name;
let endpoint = &request.endpoint_name;
let mut metrics = Vec::new();
for i in request.metrics {
debug!("Processing metric: {:?}", &i);
let metric_name = format!("{}_{}_{}", service, endpoint, &i.id);
let metric_name = format!("{}_{}", service, &i.id);
match MetricsProcesser::get_type_of_value(&i) {
MetricsValueType::Array |
MetricsValueType::TaggedArray => {
trace!("processing an array of metrics");
// ...
},
MetricsValueType::Number => {
trace!("processing a number type of metric");
let gauge = MetricsProcesser::gauge_from_number(
&i,
&metric_name,
&i.desc.clone().unwrap_or_else(|| std::borrow::Cow::Borrowed(&i.id))
&i.desc.clone().unwrap_or_else(|| std::borrow::Cow::Borrowed(&i.id)),
&state.prefix
);
metrics.push(gauge);
},
MetricsValueType::ArrayOfStrings => {
trace!("processing an array of strings");
warn!("String arrays are unsupported, ignoring ...");
},
_ => {
trace!("processing unrecognized type of metric");
warn!("Unrecognized metric type was supplied, ignoring ...");
}
}
}
for gauge in metrics {
if let Some(gauge) = gauge {
match state.registry.lock() {
Err(er) => {
@ -50,25 +121,18 @@ pub async fn update_metrics(
},
Ok(registry) => {
// todo: error handler
let _ = update_or_insert_metric(
if let Err(er) = update_or_insert_metric(
gauge,
registry,
&metric_name
);
},
}
) {
error!("Update or insert metric crushed: {}", er);
return (http::StatusCode::INTERNAL_SERVER_ERROR, er.to_string())
}
},
MetricsValueType::ArrayOfStrings => {
warn!("String arrays are unsupported, ignoring ...");
},
_ => {
warn!("Unrecognized metric type was supplied, ignoring ...");
}
}
}
(http::StatusCode::ACCEPTED, "Ok")
(http::StatusCode::ACCEPTED, String::from("Ok"))
}
/// An `Metrics` endpoint
@ -79,17 +143,21 @@ pub async fn update_metrics(
/// # Usage
///
/// ``` bash
/// curl -X GET 'http::/localhost:9100/metrics'
/// curl -X GET 'http://127.0.0.1:9100/metrics'
/// ```
///
pub async fn metrics_handler(State(state): State<Arc<AppState>>) -> String {
trace!("get on /metrics");
let registry = state.registry.lock();
debug!("registry mutex lock is {}", registry.is_ok());
return match registry {
Ok(registry) => {
let encoder = TextEncoder::new();
let mut buffer = Vec::new();
let metric_families = registry.gather();
debug!("vec of metric families - {:?}", &metric_families);
encoder.encode(&metric_families, &mut buffer).unwrap();
String::from_utf8(buffer).unwrap()
},
@ -105,75 +173,126 @@ pub async fn metrics_handler(State(state): State<Arc<AppState>>) -> String {
/// `Registry`
///
pub fn update_or_insert_metric<'a>(
metric: Gauge,
metric: Box<dyn Collector>,
registry: MutexGuard<'a, Registry>,
metric_name: &str
) -> anyhow::Result<()> {
use prometheus::Error;
// let mut counter = 0;
match registry.register(Box::new(metric.clone())) {
Ok(_) => {
info!("Metric `{}` was registered!", metric_name);
},
Err(er) => {
// update or throw away
match er {
Error::AlreadyReg => {
trace!("fn update_or_insert_metric is running");
let prod = CloneableCollector::from_boxed(metric);
match registry.unregister(Box::new(metric.clone())) {
Ok(_) => {
if let Err(er) = registry.register(Box::new(metric)) {
warn!("Cannot update metric `{}`", metric_name);
return Err(anyhow::Error::msg(
format!("Cannot update metric `{}` due to {}", metric_name, er)
))
let family = registry.gather();
let new_metric = prod.get_collector().collect();
let new_metric_family = &new_metric[0];
let new_metric_name = new_metric_family.get_name();
if let Some(mut fam) = family.into_iter()
.find(|fam| fam.get_name() == new_metric_name) {
trace!("found exising metric. processing update ...");
debug!("found metric : {:?}", &fam);
let metric_name = fam.take_name();
let metric_help = fam.take_help();
let lables = fam.get_metric().iter()
.enumerate()
.filter(|(id, _)| id == &0)
.map(|(_, metric)| metric.get_label())
.flat_map(|a| a.iter())
.map(|a| a.get_name())
.collect::<Vec<&str>>();
let gv = GaugeVec::new(opts!(&metric_name, &metric_help), &lables)?;
for metric in fam.get_metric() {
let vals = metric.get_label()
.iter()
.map(|pair| pair.get_value())
.collect::<Vec<&str>>();
gv.with_label_values(&vals).set(metric.get_gauge().get_value());
}
trace!("recreated metric as gauge vec {:?}", gv.collect());
let new_vec = prod.to_gauge_vec().compare_with_old(&gv);
registry.unregister(Box::new(gv))?;
info!("Unregistered old metric!");
registry.register(Box::new(new_vec))?;
info!("Metric `{}` was re-registered!", new_metric_name);
} else {
info!("OK on metric `{}` update", metric_name);
}
},
match registry.register(prod.get_collector()) {
Ok(_) => info!("Metric `{}` was registered!", new_metric_name),
Err(er) => {
error!("Cannot unregister metric `{}` due to {}", metric_name, er);
error!("Cannot register new metric `{}` due to {}", new_metric_name, er);
return Err(anyhow::Error::msg(
format!("Cannot unregister metric `{}` due to {}", metric_name, er)
format!("Cannot register new metric `{}` due to {}", new_metric_name, er)
))
},
}
// use prometheus::opts;
// use prometheus::GaugeVec;
// let vec = GaugeVec::new(opts!("test", "test_help"), &["label"]).unwrap();
// // vec.with_label_values(&["default"]).set(42.0);
// if registry.unregister(Box::new(vec)).is_err() {
// debug!("unregister failed");
// };
// let vec = GaugeVec::new(opts!("test1", "test_help1"), &["label"]).unwrap();
// vec.with_label_values(&["goood!"]).set(412.0);
// let _ = registry.register(Box::new(vec));
// registry
// .gather()
// .iter_mut()
// .filter(|target| target.get_name() == metric_name.trim())
// .for_each(|family| {
// // let prev: &mut GaugeVec = family.mut_metric()[0].mut_gauge();
// // GaugeVec::
// // info!("Metric `{}` was updated, new value - {}", metric_name, new);
// });
},
_ => {
error!("Cannot register new metric `{}` due to {}", metric_name, er);
return Err(anyhow::Error::msg(
format!("Cannot register new metric `{}` due to {}", metric_name, er)
))
}
}
},
}
Ok(())
// registry.gather()
// .iter()
// .filter(|fam| fam.get_name().)
}
trait CompareGaugeVec {
fn compare_with_old(self, old: &GaugeVec) -> Self;
}
impl CompareGaugeVec for GaugeVec {
fn compare_with_old(mut self, old: &GaugeVec) -> Self {
// old.
let old_fam = old.collect();
let new = self.collect();
'outer: for old_fam in old_fam {
for metric in old_fam.get_metric() {
// labels for current old version of metric in vec
let lables = get_hashmap_lables_from_metric(metric);
let value = metric.get_gauge().get_value();
for new in &new {
for new_metric in new.get_metric() {
let new_lables = get_hashmap_lables_from_metric(new_metric);
if lables.len() != new_lables.len() {
error!("Trying to save invalid metric type. Reseting changes ...");
self = old.clone();
break 'outer;
} else {
match (lables.get("status"), new_lables.get("status")) {
(Some(&status), Some(_)) => {
match (
(lables.get("device"), lables.get("source_id")),
(new_lables.get("device"), new_lables.get("source_id")),
) {
((Some(&device), Some(&source_id)),
(Some(&new_device), Some(&new_source_id))) => {
/* */
if device != new_device || source_id != new_source_id {
self.with_label_values(&[device, source_id, status]).set(value);
}
},
((Some(&device), None),
(Some(&new_device), None)) => {
/* */
if device != new_device {
self.with_label_values(&[device, status]).set(value);
}
},
_ => { /* DEAD END */},
}
},
_ => { /* DEAD END */ },
}
}
}
}
}
}
self
}
}
fn get_hashmap_lables_from_metric(metric: &Metric) -> HashMap<&str, &str> {
metric.get_label()
.iter()
.map(|pair| (pair.get_name(), pair.get_value()))
.collect::<HashMap<&str, &str>>()
}

View File

@ -11,7 +11,7 @@ use axum::{
routing::{get, post},
Router};
use prometheus::Registry;
use std::sync::{Arc, Mutex};
use std::{collections::HashMap, str::FromStr, sync::{Arc, Mutex}};
use endpoints::*;
use tokio::net::TcpListener;
use tracing::info;
@ -23,16 +23,30 @@ use dotenv::dotenv;
///
struct AppState {
registry : Mutex<Registry>,
prefix : String,
}
#[tokio::main]
async fn main() -> anyhow::Result<()> {
dotenv().ok();
let log_level = std::env::var("PROMETHEUS_EXPORTER_LOG_LEVEL")
.unwrap_or_else(|_| "INFO".to_owned());
tracing_subscriber::fmt()
.with_max_level(tracing::Level::DEBUG)
.with_max_level(tracing::Level::from_str(&log_level).unwrap_or_else(|_| tracing::Level::INFO))
.with_writer(std::io::stdout)
.with_span_events(tracing_subscriber::fmt::format::FmtSpan::NEW)
.with_line_number(false)
.with_target(false)
.with_file(false)
.compact()
.init();
info!("Loading env vars from .env if exists ...");
dotenv().ok();
info!("Logger was created and configurated, dotenv vars were loaded (if exist)");
let prefix = std::env::var("PROMETHEUS_EXPORTER_PREFIX")
.unwrap_or_else(|_| "zvks".to_owned());
info!("Initializing local Prometehus metrics registry ...");
let registry = Registry::new();
@ -40,6 +54,7 @@ async fn main() -> anyhow::Result<()> {
info!("Initializing shared state for Prometheus Exporter web-server ...");
let state = Arc::new(AppState {
registry : Mutex::new(registry),
prefix : prefix,
});
@ -49,10 +64,12 @@ async fn main() -> anyhow::Result<()> {
.route("/update", post(update_metrics))
.with_state(state.clone());
let bind_address = format!("0.0.0.0:{}", std::env::var("PROMETHEUS_EXPORTER_PORT").unwrap_or_else(|_| "9100".to_owned()));
let listener = TcpListener::bind(bind_address).await.unwrap();
let port = std::env::var("PROMETHEUS_EXPORTER_PORT")
.unwrap_or_else(|_| "9100".to_owned());
let bind_address = format!("0.0.0.0:{}", &port);
let listener = TcpListener::bind(bind_address).await?;
info!("Serving on ...:9100");
info!("Serving on ...:{}", &port);
axum::serve(listener, app).await?;
Ok(())
}

View File

@ -2,8 +2,8 @@ use crate::structs::v3::MetricOutput;
use serde_json::{Map, Value};
use prometheus::Gauge;
use tracing::error;
use prometheus::Opts;
use prometheus::GaugeVec;
use prometheus::{opts, GaugeVec, core::Collector};
use tracing::{debug, trace};
#[derive(Debug)]
pub enum MetricsValueType {
@ -18,20 +18,25 @@ pub struct MetricsProcesser;
impl MetricsProcesser {
pub fn get_type_of_value(metrics: &MetricOutput) -> MetricsValueType {
trace!("defining metric type");
if Self::is_number(metrics) {
debug!("processing Number");
return MetricsValueType::Number;
}
else if Self::is_array(metrics) {
if Self::is_tagged_array(metrics) {
debug!("processing TaggedArray");
return MetricsValueType::TaggedArray;
}
if Self::is_array_of_string_values(metrics) {
debug!("processing ArrayOfStrings");
return MetricsValueType::ArrayOfStrings;
}
debug!("processing Array");
return MetricsValueType::Array;
}
debug!("processing undefined type");
MetricsValueType::None
}
@ -39,19 +44,39 @@ impl MetricsProcesser {
pub fn gauge_from_number(
metric: &MetricOutput,
metric_name: &str,
metric_desc: &str
) -> Option<Gauge> {
metric_desc: &str,
prefix: &str
) -> Option<Box<dyn Collector>> {
trace!("fn gauge_from_number is running");
if let Some(status) = metric.status {
if let Some(device) = metric.device {
if let Some(source_id) = &metric.source_id {
let vec = GaugeVec::new(opts!(format!("{}_{}", prefix, metric.name), metric_desc), &["status", "device", "source_id"]).unwrap();
vec.with_label_values(&[&status.to_string(), &device.to_string(), &source_id.to_string()]).set(metric.value.as_f64().unwrap_or_else(|| 0.0));
debug!("processed metric: {:?}", &vec);
return Some(Box::new(vec));
}
else {
let vec = GaugeVec::new(opts!(format!("{}_{}", prefix, metric.name), metric_desc), &["status", "device"]).unwrap();
vec.with_label_values(&[&status.to_string(), &device.to_string()]).set(metric.value.as_f64().unwrap_or_else(|| 0.0));
debug!("processed metric: {:?}", &vec);
return Some(Box::new(vec));
}
} else {
let vec = GaugeVec::new(opts!(format!("{}_{}", prefix, metric.name), metric_desc), &["status"]).unwrap();
vec.with_label_values(&[&status.to_string()]).set(metric.value.as_f64().unwrap_or_else(|| 0.0));
debug!("processed metric: {:?}", &vec);
return Some(Box::new(vec));
}
}
let gauge = Gauge::new(
metric_name,
format!("{}_{}", prefix, metric.name),
metric_desc
);
match gauge {
Ok(gauge) => {
// let value = metric.value.as_number().unwrap_or({
// error!("Cannot convert {} metric value to f64 type. Value was set to 0.0", &metric_name);
// });
// let value = value.as_f64()
let val = match metric.value.as_number() {
Some(val) => {
val.as_f64().unwrap_or_else(||
@ -64,97 +89,13 @@ impl MetricsProcesser {
},
};
gauge.set(val);
return Some(gauge);
debug!("processed metric: {:?}", &gauge);
return Some(Box::new(gauge));
},
Err(er) => error!("Cannot create Gauge metric {} due to {}", &metric_name, er),
}
None
}
pub fn gauge_from_map_metrics(
map: &Map<String, Value>,
service: &str,
endpoint: &str
) -> Option<Gauge> {
let map = map.clone();
let help: String = map.keys()
.enumerate()
.map(|(idx, key)| {
if idx == 1 {
return key.to_owned();
}
"".to_owned()
})
.collect();
let name = format!("{}_{}_{}", service, endpoint, &help);
if map.len() > 1 {
// tagged
if map.len() > 2 {
error!("Cannot create Gauge {}. It can be only 1 tag", &name);
} else {
let mut label_name = String::new();
let mut label_value = String::new();
let mut metric_value = 0.0;
map.iter()
.enumerate()
.for_each(|(idx, (key, value))|{
if idx == 0 {
label_name = key.to_owned();
label_value = value.as_str()
.unwrap_or("")
.to_owned();
} else {
metric_value = value.as_f64().unwrap_or(0.0)
}
});
let opts = Opts::new(&name, &help);
let gauge_vec = GaugeVec::new(opts, &[&label_name]);
match gauge_vec {
Ok(vec) => {
// vec.get_metric_with_label_values(vals)
match vec.get_metric_with_label_values(&[&label_value]) {
Ok(metric) => {
metric.set(metric_value); // Устанавливаем значение метрики
return Some(metric.clone()); // Возвращаем `Gauge`
},
Err(er) => {
error!("Cannot create Gauge {} due to {}", &name, er);
},
}
},
Err(er) => error!("Cannot create Gauge {} due to {}", &name, er),
}
}
} else {
// not-tagged
let metric = Gauge::new(&name, &help);
match metric {
Ok(gauge) => {
let mut value = 0.0;
map.values()
.map(|val| val.clone().as_f64())
.for_each(|val| {
value = val.unwrap_or(0.0)
});
gauge.set(value);
return Some(gauge);
},
Err(er) => {
error!("Cannot create Gauge {} due to {}", &name, er);
}
}
}
None
}
pub fn get_value_as_vec_map(metrics: &MetricOutput) -> Vec<Map<String, Value>>{
let mut vec: Vec<Map<String, Value>> = Vec::new();
let arr = metrics.value.as_array().unwrap();
arr.iter()
.for_each(|a| {
vec.push(serde_json::from_value(a.clone()).unwrap());
});
vec
}
pub fn is_array_of_string_values(metrics: &MetricOutput) -> bool {
let arr = metrics.value.clone();
let arr = arr.as_array().unwrap();
@ -164,9 +105,6 @@ impl MetricsProcesser {
map.values()
.all(|val| val.is_string())
}
// fn is_valid(metrics: &PrometheusMetrics) -> bool {
// false
// }
fn is_array(metrics: &MetricOutput) -> bool {
metrics.value.is_array()
}
@ -180,18 +118,3 @@ impl MetricsProcesser {
.value.is_number()
}
}
trait IsTaggedArray {
fn is_tagged_array(&self) -> bool;
}
impl IsTaggedArray for Value {
fn is_tagged_array(&self) -> bool {
if let Some(arr) = self.as_array() {
return arr[0].get("tag_name").is_some();
}
false
}
}

View File

@ -11,18 +11,23 @@ pub mod v3 {
#[derive(Serialize, Deserialize, Debug, Clone)]
pub struct MetricOutput<'a> {
pub id : String,
pub name : String,
#[serde(rename = "type")]
json_type : String,
addr : String,
pub value : Value,
#[serde(rename = "description")]
pub desc : Option<Cow<'a, String>>,
pub status: Option<isize>,
pub device: Option<isize>,
#[serde(rename = "source")]
pub source_id: Option<String>,
}
#[derive(Serialize, Deserialize, Debug)]
pub struct PrometheusMetrics<'a> {
pub service_name: String,
pub endpoint_name: String,
// pub endpoint_name: String,
pub metrics: Vec<MetricOutput<'a>>,
}
}