You've already forked AstralRinth
forked from didirus/AstralRinth
New Creator Notifications (#4383)
* Some new notification types * Fix error * Use existing DB models rather than inline queries * Fix template fillout * Fix ModerationThreadMessageReceived * Insert more notifications, fix some formatting * chore: query cache, clippy, fmt * chore: query cache, clippy, fmt * Use outer transactions to insert notifications instead of creating a new one * Join futures
This commit is contained in:
committed by
GitHub
parent
8149618187
commit
6da190ed01
@@ -0,0 +1,89 @@
|
||||
use crate::database::models::{DBUserId, DatabaseError};
|
||||
use chrono::{DateTime, Utc};
|
||||
|
||||
pub struct PayoutsValuesNotification {
|
||||
pub id: i32,
|
||||
pub user_id: DBUserId,
|
||||
pub date_available: DateTime<Utc>,
|
||||
}
|
||||
|
||||
impl PayoutsValuesNotification {
|
||||
pub async fn unnotified_users_with_available_payouts_with_limit(
|
||||
exec: impl sqlx::PgExecutor<'_>,
|
||||
limit: i64,
|
||||
) -> Result<Vec<PayoutsValuesNotification>, DatabaseError> {
|
||||
Ok(sqlx::query_as!(
|
||||
QueryResult,
|
||||
"
|
||||
SELECT
|
||||
id,
|
||||
user_id,
|
||||
date_available
|
||||
FROM payouts_values_notifications
|
||||
WHERE
|
||||
notified = FALSE
|
||||
AND date_available <= NOW()
|
||||
FOR UPDATE SKIP LOCKED
|
||||
LIMIT $1
|
||||
",
|
||||
limit,
|
||||
)
|
||||
.fetch_all(exec)
|
||||
.await?
|
||||
.into_iter()
|
||||
.map(Into::into)
|
||||
.collect())
|
||||
}
|
||||
|
||||
pub async fn set_notified_many(
|
||||
ids: &[i32],
|
||||
exec: impl sqlx::PgExecutor<'_>,
|
||||
) -> Result<(), DatabaseError> {
|
||||
sqlx::query!(
|
||||
"
|
||||
UPDATE payouts_values_notifications
|
||||
SET notified = TRUE
|
||||
WHERE id = ANY($1)
|
||||
",
|
||||
&ids[..],
|
||||
)
|
||||
.execute(exec)
|
||||
.await?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn synchronize_future_payout_values(
|
||||
exec: impl sqlx::PgExecutor<'_>,
|
||||
) -> Result<(), DatabaseError> {
|
||||
sqlx::query!(
|
||||
"
|
||||
INSERT INTO payouts_values_notifications (date_available, user_id, notified)
|
||||
SELECT DISTINCT date_available, user_id, false notified
|
||||
FROM payouts_values
|
||||
WHERE date_available > NOW()
|
||||
ON CONFLICT (date_available, user_id) DO NOTHING
|
||||
",
|
||||
)
|
||||
.execute(exec)
|
||||
.await?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
struct QueryResult {
|
||||
id: i32,
|
||||
user_id: i64,
|
||||
date_available: DateTime<Utc>,
|
||||
}
|
||||
|
||||
impl From<QueryResult> for PayoutsValuesNotification {
|
||||
fn from(result: QueryResult) -> Self {
|
||||
PayoutsValuesNotification {
|
||||
id: result.id,
|
||||
user_id: DBUserId(result.user_id),
|
||||
date_available: result.date_available,
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user