rs-zero 0.2.7

Rust-first microservice framework inspired by go-zero engineering practices
Documentation 
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
57
58
59
60
61
62
63
64
65
66
67
68

use std::{collections::BTreeMap, fmt::Write};

use super::{DurationMetricValue, escape_label, write_duration_metric};

/// Low-cardinality labels recorded for each gRPC unary call.
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
pub struct RpcMetricLabels {
    /// gRPC service name.
    pub service: String,
    /// gRPC method name.
    pub method: String,
    /// Tonic status code name.
    pub code: String,
}

impl RpcMetricLabels {
    /// Creates a label set for one gRPC call.
    pub fn new(
        service: impl Into<String>,
        method: impl Into<String>,
        code: impl Into<String>,
    ) -> Self {
        Self {
            service: service.into(),
            method: method.into(),
            code: code.into(),
        }
    }
}

pub(crate) fn render(
    output: &mut String,
    metrics: &BTreeMap<RpcMetricLabels, DurationMetricValue>,
) {
    output.push_str("# HELP rs_zero_rpc_requests_total Total number of gRPC requests.\n");
    output.push_str("# TYPE rs_zero_rpc_requests_total counter\n");
    for (labels, value) in metrics {
        write!(output, "rs_zero_rpc_requests_total{{").ok();
        write_labels(output, labels, None);
        writeln!(output, "}} {}", value.count).ok();
    }

    output.push_str("# HELP rs_zero_rpc_request_duration_seconds gRPC request duration.\n");
    output.push_str("# TYPE rs_zero_rpc_request_duration_seconds histogram\n");
    for (labels, value) in metrics {
        write_duration_metric(
            output,
            "rs_zero_rpc_request_duration_seconds",
            labels,
            value,
            write_labels,
        );
    }
}

fn write_labels(output: &mut String, labels: &RpcMetricLabels, le: Option<&str>) {
    write!(
        output,
        "service=\"{}\",method=\"{}\",code=\"{}\"",
        escape_label(&labels.service),
        escape_label(&labels.method),
        escape_label(&labels.code)
    )
    .ok();
    if let Some(le) = le {
        write!(output, ",le=\"{}\"", escape_label(le)).ok();
    }
}