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.
52 lines
1.6 KiB
Rust
52 lines
1.6 KiB
Rust
use sea_orm::ActiveValue::Set;
|
|
use std::time::SystemTime;
|
|
|
|
use crate::entity::ephemeral_messages;
|
|
use crate::error::DatabaseResult;
|
|
use sea_orm::prelude::*;
|
|
|
|
impl super::BotDatabase {
|
|
/// Adds a command statistic to the database
|
|
#[tracing::instrument(level = "debug", skip(self))]
|
|
pub async fn add_ephemeral_message(
|
|
&self,
|
|
channel_id: u64,
|
|
message_id: u64,
|
|
timeout: SystemTime,
|
|
) -> DatabaseResult<()> {
|
|
let model = ephemeral_messages::ActiveModel {
|
|
channel_id: Set(channel_id as i64),
|
|
message_id: Set(message_id as i64),
|
|
timeout: Set(DateTimeLocal::from(timeout).into()),
|
|
..Default::default()
|
|
};
|
|
model.insert(&self.db).await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Returns a vec of all ephemeral messages
|
|
#[tracing::instrument(level = "debug", skip(self))]
|
|
pub async fn get_ephemeral_messages(&self) -> DatabaseResult<Vec<ephemeral_messages::Model>> {
|
|
let messages = ephemeral_messages::Entity::find().all(&self.db).await?;
|
|
|
|
Ok(messages)
|
|
}
|
|
|
|
/// Deletes a single ephemeral message
|
|
#[tracing::instrument(level = "debug", skip(self))]
|
|
pub async fn delete_ephemeral_message(
|
|
&self,
|
|
channel_id: i64,
|
|
message_id: i64,
|
|
) -> DatabaseResult<()> {
|
|
ephemeral_messages::Entity::delete_many()
|
|
.filter(ephemeral_messages::Column::ChannelId.eq(channel_id))
|
|
.filter(ephemeral_messages::Column::MessageId.eq(message_id))
|
|
.exec(&self.db)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
}
|