Add more counters

This commit is contained in:
jude 2023-10-22 09:55:36 +01:00
parent 605bc37db6
commit 48b50f783d
3 changed files with 16 additions and 4 deletions

View File

@ -1,6 +1,8 @@
use poise::serenity_prelude::{Attachment, GuildId, RoleId};
use tokio::fs::File;
#[cfg(feature = "metrics")]
use crate::metrics::{DELETE_COUNTER, UPLOAD_COUNTER};
use crate::{
cmds::autocomplete_sound,
consts::{MAX_SOUNDS, PATREON_GUILD, PATREON_ROLE},
@ -21,6 +23,9 @@ pub async fn upload_new_sound(
#[description = "Name to upload sound to"] name: String,
#[description = "Sound file (max. 2MB)"] file: Attachment,
) -> Result<(), Error> {
#[cfg(feature = "metrics")]
UPLOAD_COUNTER.inc();
ctx.defer().await?;
fn is_numeric(s: &String) -> bool {
@ -110,6 +115,9 @@ pub async fn delete_sound(
#[autocomplete = "autocomplete_sound"]
name: String,
) -> Result<(), Error> {
#[cfg(feature = "metrics")]
DELETE_COUNTER.inc();
let pool = ctx.data().database.clone();
let uid = ctx.author().id.0;

View File

@ -25,13 +25,13 @@ pub async fn play(
#[channel_types("Voice")]
channel: Option<GuildChannel>,
) -> Result<(), Error> {
#[cfg(feature = "metrics")]
PLAY_COUNTER.inc();
ctx.defer().await?;
let guild = ctx.guild().unwrap();
#[cfg(feature = "metrics")]
PLAY_COUNTER.inc();
ctx.say(
play_from_query(
&ctx.serenity_context(),

View File

@ -8,7 +8,11 @@ use prometheus::{register_int_counter, IntCounter, Registry};
lazy_static! {
static ref REGISTRY: Registry = Registry::new();
pub static ref PLAY_COUNTER: IntCounter =
register_int_counter!("play", "Number of calls to /play").unwrap();
register_int_counter!("play_cmd", "Number of calls to /play").unwrap();
pub static ref UPLOAD_COUNTER: IntCounter =
register_int_counter!("upload_cmd", "Number of calls to /upload").unwrap();
pub static ref DELETE_COUNTER: IntCounter =
register_int_counter!("delete_cmd", "Number of calls to /delete").unwrap();
}
pub fn init_metrics() {