using Core; using KubernetesWorkflow.Types; using Logging; using System.Globalization; namespace MetricsPlugin { public class MetricsQuery { private readonly IHttp http; private readonly ILog log; public MetricsQuery(IPluginTools tools, RunningContainer runningContainer) { RunningContainer = runningContainer; log = tools.GetLog(); http = tools.CreateHttp(RunningContainer.GetAddress(log, PrometheusContainerRecipe.PortTag), "api/v1"); } public RunningContainer RunningContainer { get; } public Metrics? GetMostRecent(string metricName, IMetricsScrapeTarget target) { var response = GetLastOverTime(metricName, GetInstanceStringForNode(target)); if (response == null) return null; var result = new Metrics { Sets = response.data.result.Select(r => { return new MetricsSet { Instance = r.metric.instance, Values = MapSingleValue(r.value) }; }).ToArray() }; Log(target, metricName, result); return result; } public Metrics? GetMetrics(string metricName) { var response = GetAll(metricName); if (response == null) return null; var result = MapResponseToMetrics(response); Log(metricName, result); return result; } public Metrics? GetAllMetricsForNode(IMetricsScrapeTarget target) { var response = http.HttpGetJson($"query?query={GetInstanceStringForNode(target)}{GetQueryTimeRange()}"); if (response.status != "success") return null; var result = MapResponseToMetrics(response); Log(target, result); return result; } private PrometheusQueryResponse? GetLastOverTime(string metricName, string instanceString) { var response = http.HttpGetJson($"query?query=last_over_time({metricName}{instanceString}{GetQueryTimeRange()})"); if (response.status != "success") return null; return response; } private PrometheusQueryResponse? GetAll(string metricName) { var response = http.HttpGetJson($"query?query={metricName}{GetQueryTimeRange()}"); if (response.status != "success") return null; return response; } private Metrics MapResponseToMetrics(PrometheusQueryResponse response) { return new Metrics { Sets = response.data.result.Select(r => { return new MetricsSet { Name = r.metric.__name__, Instance = r.metric.instance, Values = MapMultipleValues(r.values) }; }).ToArray() }; } private MetricsSetValue[] MapSingleValue(object[] value) { if (value != null && value.Length > 0) { return new[] { MapValue(value) }; } return Array.Empty(); } private MetricsSetValue[] MapMultipleValues(object[][] values) { if (values != null && values.Length > 0) { return values.Select(v => MapValue(v)).ToArray(); } return Array.Empty(); } private MetricsSetValue MapValue(object[] value) { if (value.Length != 2) throw new InvalidOperationException("Expected value to be [double, string]."); return new MetricsSetValue { Timestamp = ToTimestamp(value[0]), Value = ToValue(value[1]) }; } private string GetInstanceNameForNode(IMetricsScrapeTarget target) { return ScrapeTargetHelper.FormatTarget(log, target); } private string GetInstanceStringForNode(IMetricsScrapeTarget target) { return "{instance=\"" + GetInstanceNameForNode(target) + "\"}"; } private string GetQueryTimeRange() { return "[12h]"; } private double ToValue(object v) { return Convert.ToDouble(v, CultureInfo.InvariantCulture); } private DateTime ToTimestamp(object v) { var unixSeconds = ToValue(v); return new DateTime(1970, 1, 1, 0, 0, 0, DateTimeKind.Utc).AddSeconds(unixSeconds); } private void Log(IMetricsScrapeTarget target, string metricName, Metrics result) { Log($"{target.Container.Name} '{metricName}' = {result}"); } private void Log(string metricName, Metrics result) { Log($"'{metricName}' = {result}"); } private void Log(IMetricsScrapeTarget target, Metrics result) { Log($"{target.Container.Name} => {result}"); } private void Log(string msg) { log.Log(msg); } } public class Metrics { public MetricsSet[] Sets { get; set; } = Array.Empty(); public override string ToString() { return "[" + string.Join(',', Sets.Select(s => s.ToString())) + "]"; } } public class MetricsSet { public string Name { get; set; } = string.Empty; public string Instance { get; set; } = string.Empty; public MetricsSetValue[] Values { get; set; } = Array.Empty(); public override string ToString() { return $"{Name} ({Instance}) : {{{string.Join(",", Values.Select(v => v.ToString()))}}}"; } } public class MetricsSetValue { public DateTime Timestamp { get; set; } public double Value { get; set; } public override string ToString() { return $"<{Timestamp.ToString("o")}={Value}>"; } } public class PrometheusQueryResponse { public string status { get; set; } = string.Empty; public PrometheusQueryResponseData data { get; set; } = new(); } public class PrometheusQueryResponseData { public string resultType { get; set; } = string.Empty; public PrometheusQueryResponseDataResultEntry[] result { get; set; } = Array.Empty(); } public class PrometheusQueryResponseDataResultEntry { public ResultEntryMetric metric { get; set; } = new(); public object[] value { get; set; } = Array.Empty(); public object[][] values { get; set; } = Array.Empty(); } public class ResultEntryMetric { public string __name__ { get; set; } = string.Empty; public string instance { get; set; } = string.Empty; public string job { get; set; } = string.Empty; } public class PrometheusAllNamesResponse { public string status { get; set; } = string.Empty; public string[] data { get; set; } = Array.Empty(); } }