robin/rss-webhook-trigger
1
0
Fork 0
mirror of https://codeberg.org/icewind/rss-webhook-trigger.git synced 2026-06-03 09:54:18 +02:00
Code Issues Projects Releases Wiki Activity

cleanup not-modified handling

Browse source
This commit is contained in:
Robin Appelman 2024-12-26 15:28:21 +01:00
commit 3b430e3ab1
3 changed files with 44 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

33
src/fetcher.rs
View file

@ -29,6 +29,10 @@ impl FetchPlanInput {
FetchPlanInput::WithCache { headers } => headers,
}
}
pub fn is_retry(&self) -> bool {
matches!(self, FetchPlanInput::Retry { .. })
}
}
#[derive(Default, Debug)]
@ -126,6 +130,9 @@ pub enum FetchResponse<T, E> {
headers: CacheHeaders,
response: T,
},
NotModified {
headers: CacheHeaders,
},
Error {
headers: CacheHeaders,
error: E,
@ -138,20 +145,24 @@ impl<T, E> FetchResponse<T, E> {
match self {
FetchResponse::Retry { time, headers } => FetchPlanInput::Retry { time, headers },
FetchResponse::Ok { headers, .. } => FetchPlanInput::WithCache { headers },
FetchResponse::NotModified { headers, .. } => FetchPlanInput::WithCache { headers },
FetchResponse::Error { headers, .. } => FetchPlanInput::WithCache { headers },
}
}
pub fn into_result(self) -> Result<(Option<T>, FetchPlanInput), (E, FetchPlanInput)> {
pub fn into_result(self) -> (Result<Option<T>, E>, FetchPlanInput) {
match self {
FetchResponse::Retry { time, headers } => {
Ok((None, FetchPlanInput::Retry { time, headers }))
(Ok(None), FetchPlanInput::Retry { time, headers })
}
FetchResponse::Ok { headers, response } => {
Ok((Some(response), FetchPlanInput::WithCache { headers }))
(Ok(Some(response)), FetchPlanInput::WithCache { headers })
}
FetchResponse::NotModified { headers } => {
(Ok(None), FetchPlanInput::WithCache { headers })
}
FetchResponse::Error { headers, error } => {
Err((error, FetchPlanInput::WithCache { headers }))
(Err(error), FetchPlanInput::WithCache { headers })
}
}
}
@ -174,6 +185,10 @@ impl<E> FetchResponse<Response, E> {
time: Instant::now() + after + ONE_SEC,
headers: cache_header,
}
} else if response.status() == StatusCode::NOT_MODIFIED {
FetchResponse::NotModified {
headers: cache_header,
}
} else {
FetchResponse::Ok {
headers: cache_header,
@ -227,6 +242,7 @@ impl<T, E> FetchResponse<T, E> {
headers,
response: f(response).await,
},
FetchResponse::NotModified { headers } => FetchResponse::NotModified { headers },
FetchResponse::Error { error, headers } => FetchResponse::Error { error, headers },
}
}
@ -236,7 +252,8 @@ impl<T, E> FetchResponse<T, E> {
{
match self {
FetchResponse::Retry { time, headers } => FetchResponse::Retry { time, headers },
FetchResponse::Ok { headers, response } => FetchResponse::Ok { headers, response },
FetchResponse::Ok { response, headers } => FetchResponse::Ok { response, headers },
FetchResponse::NotModified { headers } => FetchResponse::NotModified { headers },
FetchResponse::Error { error, headers } => FetchResponse::Error {
error: f(error),
headers,
@ -257,7 +274,11 @@ impl<T, E> FetchResponse<Result<T, E>, E> {
headers,
response: Err(error),
} => FetchResponse::Error { error, headers },
FetchResponse::Error { error, headers } => FetchResponse::Error { error, headers },
FetchResponse::NotModified { headers } => FetchResponse::NotModified { headers },
FetchResponse::Error { error, headers } => FetchResponse::Error {
error,
headers,
},
}
}
}

5
src/hub.rs
View file

@ -1,6 +1,6 @@
use crate::error::HubError;
use crate::fetcher::{CacheHeaders, FetchResponse};
use reqwest::{Client, StatusCode};
use reqwest::{Client};
use reqwest::header::{HeaderValue, USER_AGENT};
use serde::Deserialize;
use time::OffsetDateTime;
@ -28,9 +28,6 @@ pub async fn tags(
.map_err(HubError::Network)
.check_status_code(HubError::ClientError, HubError::ServerError)
.map(|response| async {
if response.status() == StatusCode::NOT_MODIFIED {
return Ok(Vec::new());
}
response
.text()
.await

42
src/main.rs
View file

@ -7,7 +7,7 @@ use crate::config::{Config, FeedConfig};
use crate::error::{FetchError, FetchFeedError, HubError, ParseFeedError};
use crate::fetcher::{next_fetch, CacheHeaders, FetchPlan, FetchResponse};
use main_error::MainResult;
use reqwest::{Client, Response, StatusCode};
use reqwest::{Client, Response};
use std::collections::hash_map::DefaultHasher;
use std::collections::HashMap;
use std::future::ready;
@ -121,40 +121,35 @@ impl FeedFetcher {
let plan = self.fetch_plans.remove(feed).unwrap_or_default();
let fetch_result = self.get_feed_key(feed, &plan.headers).await;
let new_key = match fetch_result.into_result() {
Ok((new_key, new_plan)) => {
let (result, new_plan) = fetch_result.into_result();
let is_retry = new_plan.is_retry();
self.fetch_plans.insert(feed.into(), next_fetch(self.base_interval, Some(new_plan)));
new_key
}
Err((err, new_plan)) => {
self.fetch_plans.insert(feed.into(), next_fetch(self.base_interval, Some(new_plan)));
return Err(err);
}
};
let new_key = result?;
Ok(match (self.cache.get_mut(feed), new_key) {
(Some(cached), Some(Some(new_key))) => {
(Some(cached), Some(new_key)) => {
debug!(cached, new_key, "checked existing feed");
if new_key != *cached {
*cached = new_key;
info!("feed updated");
true
} else {
false
}
}
(None, Some(Some(new_key))) => {
(None, Some(new_key)) => {
debug!(feed, "new feed");
self.cache.insert(feed.into(), new_key);
// don't trigger the actions on start
false
}
(_, Some(None)) => {
debug!("not modified response");
(_, None) if is_retry => {
warn!("rate limited by server");
false
}
(_, None) => {
warn!("rate limited by server");
debug!("not modified");
false
}
})
@ -165,21 +160,18 @@ impl FeedFetcher {
&self,
feed: &str,
cache_headers: &CacheHeaders,
) -> FetchResponse<Option<u64>, FetchError> {
) -> FetchResponse<u64, FetchError> {
if let Some(hub) = feed.strip_prefix("docker-hub://") {
if let Some((user, repo)) = hub.split_once('/') {
hub::tags(&self.client, user, repo, cache_headers)
.await
.map(|tags| {
if tags.is_empty() {
return ready(None);
}
let mut hasher = DefaultHasher::new();
for tag in tags {
tag.id.hash(&mut hasher);
tag.last_updated.hash(&mut hasher);
}
ready(Some(hasher.finish()))
ready(hasher.finish())
}).await
.map_err(FetchError::Hub)
} else {
@ -200,7 +192,7 @@ impl FeedFetcher {
&self,
feed: &str,
cache_headers: &CacheHeaders,
) -> FetchResponse<Option<u64>, FetchFeedError> {
) -> FetchResponse<u64, FetchFeedError> {
let response = self
.client
.get(feed)
@ -219,11 +211,7 @@ impl FeedFetcher {
}
}
async fn parse_rss_response(response: Response) -> Result<Option<u64>, FetchFeedError> {
if response.status() == StatusCode::NOT_MODIFIED {
return Ok(None);
}
async fn parse_rss_response(response: Response) -> Result<u64, FetchFeedError> {
let content = response.text().await?;
let channel = Feed::from_str(&content).map_err(ParseFeedError::Parse)?;
@ -249,5 +237,5 @@ async fn parse_rss_response(response: Response) -> Result<Option<u64>, FetchFeed
}
}
Ok(Some(hasher.finish()))
Ok(hasher.finish())
}