Add basic InfluxDB reporter

2019-07-16 23:34:22 +02:00
parent 2d58705e85
commit 02b41413b3
7 changed files with 125 additions and 6 deletions
--- a/PluralKit.Bot/Services/PeriodicStatCollector.cs
+++ b/PluralKit.Bot/Services/PeriodicStatCollector.cs
@@ -0,0 +1,57 @@
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using App.Metrics;
+using Discord;
+using Discord.WebSocket;
+using PluralKit.Core;
+
+namespace PluralKit.Bot
+{
+    public class PeriodicStatCollector
+    {
+        private DiscordShardedClient _client;
+        private IMetrics _metrics;
+
+        private SystemStore _systems;
+        private MemberStore _members;
+        private SwitchStore _switches;
+        private MessageStore _messages;
+
+        public PeriodicStatCollector(IDiscordClient client, IMetrics metrics, SystemStore systems, MemberStore members, SwitchStore switches, MessageStore messages)
+        {
+            _client = (DiscordShardedClient) client;
+            _metrics = metrics;
+            _systems = systems;
+            _members = members;
+            _switches = switches;
+            _messages = messages;
+        }
+
+        public async Task CollectStats()
+        {
+            // Aggregate guild/channel stats
+            _metrics.Measure.Gauge.SetValue(BotMetrics.Guilds, _client.Guilds.Count);
+            _metrics.Measure.Gauge.SetValue(BotMetrics.Channels, _client.Guilds.Sum(g => g.TextChannels.Count));
+
+            // Aggregate member stats
+            var usersKnown = new HashSet<ulong>();
+            var usersOnline = new HashSet<ulong>();
+            foreach (var guild in _client.Guilds)
+            foreach (var user in guild.Users)
+            {
+                usersKnown.Add(user.Id);
+                if (user.Status == UserStatus.Online) usersOnline.Add(user.Id);
+            }
+
+            _metrics.Measure.Gauge.SetValue(BotMetrics.MembersTotal, usersKnown.Count);
+            _metrics.Measure.Gauge.SetValue(BotMetrics.MembersOnline, usersOnline.Count);
+            
+            // Aggregate DB stats
+            _metrics.Measure.Gauge.SetValue(CoreMetrics.SystemCount, await _systems.Count());
+            _metrics.Measure.Gauge.SetValue(CoreMetrics.MemberCount, await _members.Count());
+            _metrics.Measure.Gauge.SetValue(CoreMetrics.SwitchCount, await _switches.Count());
+            _metrics.Measure.Gauge.SetValue(CoreMetrics.MessageCount, await _messages.Count());
+        }
+    }
+}