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