Merge pull request #757 from ViBotTeam/huntifer/add-afkcheckmodules-l…

…ogging Add afkcheck and afkmodule logging to influxDB
Thomasc33 · Feb 23, 2024 · 0f78f32 · 0f78f32
2 parents f84eacf + 6dcaf47
commit 0f78f32
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 2 deletions.
diff --git a/commands/afkCheck.js b/commands/afkCheck.js
@@ -7,6 +7,8 @@ const extensions = require(`../lib/extensions`)
 const consumablePopTemplates = require(`../data/keypop.json`);
 const popCommand = require('./pop.js');
 const AfkButton = require('../lib/afk/AfkButton');
+const { writePoint } = require('../metrics.js');
+const { Point } = require('@influxdata/influxdb-client');
 
 module.exports = {
     name: 'afk',
@@ -80,6 +82,9 @@ module.exports = {
                 console.log(afkTemplate.message())
                 continue
             }
+            writePoint(new Point('afkchecksmodules')
+                .tag('raidID', currentStoredAfkCheck.raidID)
+                .stringField('module', 'insert'))
             bot.afkModules[currentStoredAfkCheck.raidID] = new afkCheck(afkTemplate, bot, db, message, currentStoredAfkCheck.location)
             await bot.afkModules[currentStoredAfkCheck.raidID].loadBotAfkCheck(currentStoredAfkCheck)
         }
@@ -182,6 +187,9 @@ class afkCheck {
     async start() {
         if (this.phase === 0) this.phase = 1
         this.timer = new Date(Date.now() + (this.#body[this.phase].timeLimit * 1000))
+        writePoint(new Point('afkchecksmodules')
+            .tag('raidID', this.#raidID)
+            .stringField('module', 'insert'))
         this.#bot.afkModules[this.#raidID] = this
         await Promise.all([this.sendStatusMessage(), this.sendCommandsMessage(), this.sendChannelsMessage()])
         this.startTimers()
@@ -194,10 +202,17 @@ class afkCheck {
 
     saveBotAfkCheck(deleteCheck = false) {
         if (deleteCheck) {
+            writePoint(new Point('afkchecksmodules')
+                .tag('raidID', this.#raidID)
+                .stringField('check', 'delete')
+                .stringField('module', 'delete'))
             delete this.#bot.afkChecks[this.#raidID]
             delete this.#bot.afkModules[this.#raidID]
         }
         else {
+            writePoint(new Point('afkchecksmodules')
+                .tag('raidID', this.#raidID)
+                .stringField('check', 'create'))
             this.#bot.afkChecks[this.#raidID] = {
                 afkTemplateName: this.afkTemplateName,
                 message: this.#message,
@@ -229,6 +244,9 @@ class afkCheck {
                 raidChannelsMessage: this.raidChannelsMessage,
             }
         }
+        writePoint(new Point('afkchecksmodules')
+            .tag('raidID', this.#raidID)
+            .booleanField('fileWrite', true))
         fs.writeFileSync('./data/afkChecks.json', JSON.stringify(this.#bot.afkChecks, null, 4), err => { if (err) ErrorLogger.log(err, this.#bot, this.#guild) })
     }
 

diff --git a/index.js b/index.js
@@ -13,7 +13,8 @@ const ErrorLogger = require('./lib/logError');
 const botSetup = require('./botSetup.js');
 const dbSetup = require('./dbSetup.js');
 const memberHandler = require('./memberHandler.js');
-const { logWrapper } = require('./metrics.js');
+const { logWrapper, writePoint } = require('./metrics.js');
+const { Point } = require('@influxdata/influxdb-client');
 const { handleReactionRow } = require('./redis.js');
 const Modmail = require('./lib/modmail.js');
 // Specific Commands
@@ -168,6 +169,17 @@ Promise.all(botSettings.config?.guildIds.map(guildId => {
     bot.login(botSettings.key);
 });
 
+setInterval(() => {
+    for (const afkCheckId of Object.keys(bot.afkChecks)) {
+        writePoint(new Point('afkchecksmodulesinterval')
+            .stringField('afkCheck', afkCheckId));
+    }
+    for (const afkModuleId of Object.keys(bot.afkModules)) {
+        writePoint(new Point('afkchecksmodulesinterval')
+            .stringField('afkModule', afkModuleId));
+    }
+}, 60000);
+
 // ===========================================================================================================
 // Process Event Listening
 // ===========================================================================================================

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "vibot",
-  "version": "8.16.5",
+  "version": "8.16.6",
   "description": "ViBot",
   "main": "index.js",
   "dependencies": {