Track messages by channel.

1 files changed, 14 insertions, 3 deletions

diff --git a/bot/cogs/metrics.py b/bot/cogs/metrics.py
index ce58f763f..dc86f8e82 100644
--- a/bot/cogs/metrics.py
+++ b/bot/cogs/metrics.py
@@ -1,8 +1,8 @@
 from collections import defaultdict
 
-from discord import Member
+from discord import Member, Message
 from discord.ext.commands import Cog
-from prometheus_client import Gauge
+from prometheus_client import Counter, Gauge
 
 from bot.bot import Bot
 
@@ -16,10 +16,15 @@ class Metrics(Cog):
         self.bot = bot
 
         self.guild_members = Gauge(
-            name=f'{self.PREFIX}server_members',
+            name=f'{self.PREFIX}guild_members',
             documentation="Total members by status.",
             labelnames=('guild_id', 'status')
         )
+        self.guild_messages = Counter(
+            name=f'{self.PREFIX}guild_messages',
+            documentation="Guild messages by channel.",
+            labelnames=('channel_id', 'channel_name')
+        )
 
     @Cog.listener()
     async def on_ready(self) -> None:
@@ -49,6 +54,12 @@ class Metrics(Cog):
             self.guild_members.labels(guild_id=after.guild.id, status=str(before.status)).dec()
             self.guild_members.labels(guild_id=after.guild.id, status=str(after.status)).inc()
 
+    @Cog.listener()
+    async def on_message(self, message: Message) -> None:
+        self.guild_messages.labels(
+            channel_id=message.channel.id, channel_name=message.channel.name
+        ).inc()
+
 
 def setup(bot: Bot) -> None:
     """Load the Metrics cog."""