eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-13 09:10:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(stats): query http gateway, wait until gateway up to collect discord stats

Browse source
This commit is contained in:
alyssa 2024-09-27 18:53:04 +09:00
parent e4ed354536
commit 9ff824c37b
7 changed files with 130 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

18
services/gateway/src/cache_api.rs
View file

@ -5,12 +5,15 @@ use axum::{
routing::get,
Router,
};
use serde_json::to_string;
use serde_json::{json, to_string};
use tracing::{error, info};
use twilight_model::guild::Permissions;
use twilight_model::id::Id;
use crate::discord::cache::{dm_channel, DiscordCache, DM_PERMISSIONS};
use crate::discord::{
cache::{dm_channel, DiscordCache, DM_PERMISSIONS},
gateway::cluster_config,
};
use std::sync::Arc;
fn status_code(code: StatusCode, body: String) -> Response {
@ -156,6 +159,17 @@ pub async fn run_server(cache: Arc<DiscordCache>) -> anyhow::Result<()> {
})
)
.route("/stats", get(|State(cache): State<Arc<DiscordCache>>| async move {
let cluster = cluster_config();
let has_been_up = cache.2.read().await.len() as u32 == if cluster.total_shards > 16 {16} else {cluster.total_shards};
let stats = json!({
"guild_count": cache.0.stats().guilds(),
"channel_count": cache.0.stats().channels(),
"up": has_been_up,
});
status_code(StatusCode::FOUND, to_string(&stats).unwrap())
}))
.layer(axum::middleware::from_fn(crate::logger::logger))
.with_state(cache);