65 lines
1.8 KiB
C#
65 lines
1.8 KiB
C#
using Core;
|
|
using KubernetesWorkflow.Types;
|
|
using Utils;
|
|
|
|
namespace MetricsPlugin
|
|
{
|
|
public interface IMetricsAccess : IHasContainer
|
|
{
|
|
string TargetName { get; }
|
|
Metrics? GetAllMetrics();
|
|
MetricsSet GetMetric(string metricName);
|
|
MetricsSet GetMetric(string metricName, TimeSpan timeout);
|
|
}
|
|
|
|
public class MetricsAccess : IMetricsAccess
|
|
{
|
|
private readonly MetricsQuery query;
|
|
private readonly IMetricsScrapeTarget target;
|
|
|
|
public MetricsAccess(MetricsQuery query, IMetricsScrapeTarget target)
|
|
{
|
|
this.query = query;
|
|
this.target = target;
|
|
TargetName = target.Container.Name;
|
|
}
|
|
|
|
public string TargetName { get; }
|
|
public RunningContainer Container => query.RunningContainer;
|
|
|
|
public Metrics? GetAllMetrics()
|
|
{
|
|
return query.GetAllMetricsForNode(target);
|
|
}
|
|
|
|
public MetricsSet GetMetric(string metricName)
|
|
{
|
|
return GetMetric(metricName, TimeSpan.FromSeconds(10));
|
|
}
|
|
|
|
public MetricsSet GetMetric(string metricName, TimeSpan timeout)
|
|
{
|
|
var start = DateTime.UtcNow;
|
|
|
|
while (true)
|
|
{
|
|
var mostRecent = GetMostRecent(metricName);
|
|
if (mostRecent != null) return mostRecent;
|
|
if (DateTime.UtcNow - start > timeout)
|
|
{
|
|
throw new TimeoutException();
|
|
}
|
|
|
|
Time.Sleep(TimeSpan.FromSeconds(2));
|
|
}
|
|
}
|
|
|
|
private MetricsSet? GetMostRecent(string metricName)
|
|
{
|
|
var result = query.GetMostRecent(metricName, target);
|
|
if (result == null) return null;
|
|
return result.Sets.LastOrDefault();
|
|
}
|
|
}
|
|
}
|