27 lines
638 B
Rust
27 lines
638 B
Rust
use redis::{Commands, RedisResult};
|
|
|
|
#[derive(Clone)]
|
|
pub struct StatsManager {
|
|
redis: redis::Client,
|
|
}
|
|
|
|
impl StatsManager {
|
|
pub fn new(url: impl Into<String>) -> RedisResult<StatsManager> {
|
|
let redis = redis::Client::open(url.into())?;
|
|
|
|
Ok(StatsManager { redis })
|
|
}
|
|
|
|
pub fn set_server_count(&self, count: usize) -> RedisResult<()> {
|
|
let mut con = self.redis.get_connection()?;
|
|
|
|
con.set("sc-bot-total-servers", count.to_string())
|
|
}
|
|
|
|
pub fn set_active_count(&self, count: usize) -> RedisResult<()> {
|
|
let mut con = self.redis.get_connection()?;
|
|
|
|
con.set("sc-bot-active-servers", count.to_string())
|
|
}
|
|
}
|