bonkmaykr/ojrc
1
0
Fork 0
mirror of https://git.ngram.ca/OpenJam/rc-servers synced 2026-08-23 23:08:52 +00:00
Code Activity

Add stats chat command

Browse Source
This commit is contained in:
NG (Graham)
2025-10-31 22:41:33 -04:00
parent 3ef1aa20d0
commit 340f571637
11 changed files with 177 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
rc_database/src/wrapper.rs
View File

@@ -3,15 +3,19 @@ use sea_orm::{ActiveModelTrait, ColumnTrait, EntityTrait, PaginatorTrait, QueryF
pub struct Database {
orm: sea_orm::DatabaseConnection,
metrics: std::sync::Arc<std::sync::Mutex<super::metrics::MetricsState>>,
}
impl Database {
pub async fn init(uri: &str) -> Result<Self, sea_orm::DbErr>{
let db = sea_orm::Database::connect(uri).await?;
let mut db = sea_orm::Database::connect(uri).await?;
let metrics_data = std::sync::Arc::new(std::sync::Mutex::new(super::metrics::MetricsState::new()));
db.set_metric_callback(super::metrics::metrics_cb(metrics_data.clone()));
//let schema_manager = SchemaManager::new(&db);
super::Migrator::up(&db, None).await?;
Ok(Self {
orm: db,
metrics: metrics_data,
})
}
@@ -423,4 +427,8 @@ impl Database {
pub async fn insert_game_event(&self, entity: crate::schema::game_event::ActiveModel) -> Result<crate::schema::game_event::Model, sea_orm::DbErr> {
entity.insert(&self.orm).await
}
pub async fn metrics(&self) -> super::DatabaseMetrics {
self.metrics.lock().unwrap().snapshot()
}
}