refactor: moved knowledge of subscription change time to store

main
Jef Roosens 2025-03-15 20:20:38 +01:00
parent 8a9744c4a9
commit 330877c8c5
No known key found for this signature in database
GPG Key ID: 21FD3D77D56BAF49
6 changed files with 72 additions and 25 deletions

View File

@ -1,5 +1,6 @@
use std::collections::HashSet;
use chrono::DateTime;
use diesel::prelude::*;
use super::SqliteRepository;
@ -8,24 +9,39 @@ use crate::{
gpodder,
};
impl From<(String, i64)> for gpodder::Subscription {
fn from((url, ts): (String, i64)) -> Self {
Self {
url,
time_changed: DateTime::from_timestamp(ts, 0).unwrap(),
}
}
}
impl gpodder::SubscriptionRepository for SqliteRepository {
fn subscriptions_for_user(
&self,
user: &gpodder::User,
) -> Result<Vec<String>, gpodder::AuthErr> {
) -> Result<Vec<gpodder::Subscription>, gpodder::AuthErr> {
Ok(device_subscriptions::table
.inner_join(devices::table)
.filter(devices::user_id.eq(user.id))
.select(device_subscriptions::podcast_url)
.select((
device_subscriptions::podcast_url,
device_subscriptions::time_changed,
))
.distinct()
.get_results(&mut self.pool.get()?)?)
.get_results::<(String, i64)>(&mut self.pool.get()?)?
.into_iter()
.map(Into::into)
.collect())
}
fn subscriptions_for_device(
&self,
user: &gpodder::User,
device_id: &str,
) -> Result<Vec<String>, gpodder::AuthErr> {
) -> Result<Vec<gpodder::Subscription>, gpodder::AuthErr> {
Ok(device_subscriptions::table
.inner_join(devices::table)
.filter(
@ -33,8 +49,14 @@ impl gpodder::SubscriptionRepository for SqliteRepository {
.eq(user.id)
.and(devices::device_id.eq(device_id)),
)
.select(device_subscriptions::podcast_url)
.get_results(&mut self.pool.get()?)?)
.select((
device_subscriptions::podcast_url,
device_subscriptions::time_changed,
))
.get_results::<(String, i64)>(&mut self.pool.get()?)?
.into_iter()
.map(Into::into)
.collect())
}
fn set_subscriptions_for_device(
@ -230,7 +252,7 @@ impl gpodder::SubscriptionRepository for SqliteRepository {
user: &gpodder::User,
device_id: &str,
since: chrono::DateTime<chrono::Utc>,
) -> Result<(chrono::DateTime<chrono::Utc>, Vec<String>, Vec<String>), gpodder::AuthErr> {
) -> Result<(Vec<gpodder::Subscription>, Vec<gpodder::Subscription>), gpodder::AuthErr> {
let since = since.timestamp();
let (mut added, mut removed) = (Vec::new(), Vec::new());
@ -245,20 +267,22 @@ impl gpodder::SubscriptionRepository for SqliteRepository {
)
.select(db::DeviceSubscription::as_select());
let mut max_time: chrono::DateTime<chrono::Utc> = chrono::DateTime::<chrono::Utc>::MIN_UTC;
for sub in query.load_iter(&mut self.pool.get()?)? {
let sub = sub?;
if sub.deleted {
removed.push(sub.podcast_url);
removed.push(gpodder::Subscription {
url: sub.podcast_url,
time_changed: DateTime::from_timestamp(sub.time_changed, 0).unwrap(),
});
} else {
added.push(sub.podcast_url);
added.push(gpodder::Subscription {
url: sub.podcast_url,
time_changed: DateTime::from_timestamp(sub.time_changed, 0).unwrap(),
});
}
max_time = max_time.max(chrono::DateTime::from_timestamp(sub.time_changed, 0).unwrap());
}
Ok((max_time, added, removed))
Ok((added, removed))
}
}

View File

@ -74,10 +74,10 @@ pub trait SubscriptionRepository {
&self,
user: &User,
device_id: &str,
) -> Result<Vec<String>, AuthErr>;
) -> Result<Vec<models::Subscription>, AuthErr>;
/// Return all subscriptions for a given user
fn subscriptions_for_user(&self, user: &User) -> Result<Vec<String>, AuthErr>;
fn subscriptions_for_user(&self, user: &User) -> Result<Vec<models::Subscription>, AuthErr>;
/// Replace the list of subscriptions for a device with the given list
fn set_subscriptions_for_device(
@ -104,7 +104,7 @@ pub trait SubscriptionRepository {
user: &User,
device_id: &str,
since: DateTime<Utc>,
) -> Result<(DateTime<Utc>, Vec<String>, Vec<String>), AuthErr>;
) -> Result<(Vec<Subscription>, Vec<Subscription>), AuthErr>;
}
pub trait EpisodeActionRepository {

View File

@ -52,3 +52,8 @@ pub struct Session {
pub last_seen: DateTime<Utc>,
pub user: User,
}
pub struct Subscription {
pub url: String,
pub time_changed: DateTime<Utc>,
}

View File

@ -86,11 +86,14 @@ impl GpodderRepository {
&self,
user: &models::User,
device_id: &str,
) -> Result<Vec<String>, AuthErr> {
) -> Result<Vec<models::Subscription>, AuthErr> {
self.store.subscriptions_for_device(user, device_id)
}
pub fn subscriptions_for_user(&self, user: &models::User) -> Result<Vec<String>, AuthErr> {
pub fn subscriptions_for_user(
&self,
user: &models::User,
) -> Result<Vec<models::Subscription>, AuthErr> {
self.store.subscriptions_for_user(user)
}
@ -128,11 +131,26 @@ impl GpodderRepository {
user: &models::User,
device_id: &str,
since: DateTime<Utc>,
) -> Result<(DateTime<Utc>, Vec<String>, Vec<String>), AuthErr> {
let (max_time_changed, added, removed) = self
) -> Result<
(
DateTime<Utc>,
Vec<models::Subscription>,
Vec<models::Subscription>,
),
AuthErr,
> {
let now = chrono::Utc::now();
let (added, removed) = self
.store
.subscription_updates_for_device(user, device_id, since)?;
let max_time_changed = added
.iter()
.chain(removed.iter())
.map(|s| s.time_changed)
.max()
.unwrap_or(now);
Ok((max_time_changed + TimeDelta::seconds(1), added, removed))
}

View File

@ -85,8 +85,8 @@ pub async fn get_subscription_changes(
.unwrap()
.map(|(next_time_changed, add, remove)| {
Json(SubscriptionDeltaResponse {
add,
remove,
add: add.into_iter().map(|s| s.url).collect(),
remove: remove.into_iter().map(|s| s.url).collect(),
timestamp: next_time_changed.timestamp(),
})
})?)

View File

@ -37,7 +37,7 @@ pub async fn get_device_subscriptions(
tokio::task::spawn_blocking(move || ctx.store.subscriptions_for_device(&user, &id))
.await
.unwrap()
.map(Json)?,
.map(|subs| Json(subs.into_iter().map(|s| s.url).collect()))?,
)
}
@ -54,7 +54,7 @@ pub async fn get_user_subscriptions(
tokio::task::spawn_blocking(move || ctx.store.subscriptions_for_user(&user))
.await
.unwrap()
.map(Json)?,
.map(|subs| Json(subs.into_iter().map(|s| s.url).collect()))?,
)
}