You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.3 KiB
Rust
51 lines
1.3 KiB
Rust
use std::time::SystemTime;
|
|
|
|
use diesel::dsl::count;
|
|
use diesel::insert_into;
|
|
use diesel::prelude::*;
|
|
use tokio_diesel::*;
|
|
|
|
use crate::error::DatabaseResult;
|
|
use crate::models::*;
|
|
use crate::schema::*;
|
|
use crate::Database;
|
|
|
|
impl Database {
|
|
/// Adds a command statistic to the database
|
|
pub async fn add_statistic(
|
|
&self,
|
|
version: &str,
|
|
command: &str,
|
|
executed_at: SystemTime,
|
|
success: bool,
|
|
error_msg: Option<String>,
|
|
) -> DatabaseResult<()> {
|
|
use statistics::dsl;
|
|
log::trace!("Adding statistic to database");
|
|
insert_into(dsl::statistics)
|
|
.values(StatisticInsert {
|
|
version: version.to_string(),
|
|
command: command.to_string(),
|
|
executed_at,
|
|
success,
|
|
error_msg,
|
|
})
|
|
.execute_async(&self.pool)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Returns the total number of commands executed
|
|
pub async fn get_total_commands_statistic(&self) -> DatabaseResult<u64> {
|
|
use statistics::dsl;
|
|
log::trace!("Querying total number of commands");
|
|
let total_count: i64 = dsl::statistics
|
|
.select(count(dsl::id))
|
|
.first_async::<i64>(&self.pool)
|
|
.await?;
|
|
|
|
Ok(total_count as u64)
|
|
}
|
|
}
|