Consul Diagnostics (#1828)

asynkron · Oct 19, 2022 · 3aa0ca3 · 3aa0ca3
1 parent 0532178
commit 3aa0ca3
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/Proto.Cluster.Consul/ConsulProvider.cs b/src/Proto.Cluster.Consul/ConsulProvider.cs
@@ -12,6 +12,7 @@
 using JetBrains.Annotations;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
+using Proto.Diagnostics;
 
 namespace Proto.Cluster.Consul;
 //TLDR;
@@ -49,6 +50,26 @@ private readonly TimeSpan
     private int _port;
     private bool _shutdown;
 
+    public async Task<DiagnosticsEntry[]> GetDiagnostics()
+    {
+        try
+        {
+            var statuses = await _client.Health.Service(_consulServiceName, null, false, new QueryOptions
+                {
+
+                }
+                , _cluster.System.Shutdown
+            );
+
+            var health = new DiagnosticsEntry("ConsulProvider", "Services", statuses.Response);
+            return new[] { health };
+        }
+        catch (Exception x)
+        {
+            return new[] { new DiagnosticsEntry("ConsulProvider", "Exception", x.ToString() ) };
+        }
+    }
+
     public ConsulProvider(ConsulProviderConfig config) : this(config, _ => { })
     {
     }