forked from Eauldane/SnowcloakServer
Initial
This commit is contained in:
@@ -0,0 +1,84 @@
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Prometheus;
|
||||
|
||||
namespace MareSynchronosShared.Metrics;
|
||||
|
||||
public class MareMetrics
|
||||
{
|
||||
public MareMetrics(ILogger<MareMetrics> logger, List<string> countersToServe, List<string> gaugesToServe)
|
||||
{
|
||||
logger.LogInformation("Initializing MareMetrics");
|
||||
foreach (var counter in countersToServe)
|
||||
{
|
||||
logger.LogInformation($"Creating Metric for Counter {counter}");
|
||||
counters.Add(counter, Prometheus.Metrics.CreateCounter(counter, counter));
|
||||
}
|
||||
|
||||
foreach (var gauge in gaugesToServe)
|
||||
{
|
||||
logger.LogInformation($"Creating Metric for Counter {gauge}");
|
||||
if (!string.Equals(gauge, MetricsAPI.GaugeConnections, StringComparison.OrdinalIgnoreCase))
|
||||
gauges.Add(gauge, Prometheus.Metrics.CreateGauge(gauge, gauge));
|
||||
else
|
||||
gauges.Add(gauge, Prometheus.Metrics.CreateGauge(gauge, gauge, new[] { "continent" }));
|
||||
}
|
||||
}
|
||||
|
||||
private readonly Dictionary<string, Counter> counters = new(StringComparer.Ordinal);
|
||||
|
||||
private readonly Dictionary<string, Gauge> gauges = new(StringComparer.Ordinal);
|
||||
|
||||
public void IncGaugeWithLabels(string gaugeName, double value = 1.0, params string[] labels)
|
||||
{
|
||||
if (gauges.TryGetValue(gaugeName, out Gauge gauge))
|
||||
{
|
||||
lock (gauge)
|
||||
gauge.WithLabels(labels).Inc(value);
|
||||
}
|
||||
}
|
||||
|
||||
public void DecGaugeWithLabels(string gaugeName, double value = 1.0, params string[] labels)
|
||||
{
|
||||
if (gauges.TryGetValue(gaugeName, out Gauge gauge))
|
||||
{
|
||||
lock (gauge)
|
||||
gauge.WithLabels(labels).Dec(value);
|
||||
}
|
||||
}
|
||||
|
||||
public void SetGaugeTo(string gaugeName, double value)
|
||||
{
|
||||
if (gauges.TryGetValue(gaugeName, out Gauge gauge))
|
||||
{
|
||||
lock (gauge)
|
||||
gauge.Set(value);
|
||||
}
|
||||
}
|
||||
|
||||
public void IncGauge(string gaugeName, double value = 1.0)
|
||||
{
|
||||
if (gauges.TryGetValue(gaugeName, out Gauge gauge))
|
||||
{
|
||||
lock (gauge)
|
||||
gauge.Inc(value);
|
||||
}
|
||||
}
|
||||
|
||||
public void DecGauge(string gaugeName, double value = 1.0)
|
||||
{
|
||||
if (gauges.TryGetValue(gaugeName, out Gauge gauge))
|
||||
{
|
||||
lock (gauge)
|
||||
gauge.Dec(value);
|
||||
}
|
||||
}
|
||||
|
||||
public void IncCounter(string counterName, double value = 1.0)
|
||||
{
|
||||
if (counters.TryGetValue(counterName, out Counter counter))
|
||||
{
|
||||
lock (counter)
|
||||
counter.Inc(value);
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user