0
0
mirror of https://github.com/neon-mmd/websurfx.git synced 2024-11-21 21:48:21 -05:00

🧹 chore: make clippy happy (#244)

This commit is contained in:
neon_arch 2023-09-17 19:56:48 +03:00
parent 1726ccc8ce
commit d33129c4c9
4 changed files with 44 additions and 41 deletions

56
src/cache/cacher.rs vendored
View File

@ -10,7 +10,7 @@ use tokio::sync::Mutex;
use crate::{config::parser::Config, models::aggregation_models::SearchResults}; use crate::{config::parser::Config, models::aggregation_models::SearchResults};
use super::error::PoolError; use super::error::CacheError;
#[cfg(feature = "redis-cache")] #[cfg(feature = "redis-cache")]
use super::redis_cacher::RedisCache; use super::redis_cacher::RedisCache;
@ -42,25 +42,27 @@ impl Cache {
/// It returns a newly initialized variant based on the feature enabled by the user. /// It returns a newly initialized variant based on the feature enabled by the user.
pub async fn build(_config: &Config) -> Self { pub async fn build(_config: &Config) -> Self {
#[cfg(all(feature = "redis-cache", feature = "memory-cache"))] #[cfg(all(feature = "redis-cache", feature = "memory-cache"))]
log::info!("Using a hybrid cache"); {
#[cfg(all(feature = "redis-cache", feature = "memory-cache"))] log::info!("Using a hybrid cache");
return Cache::new_hybrid( Cache::new_hybrid(
RedisCache::new(&_config.redis_url, 5) RedisCache::new(&_config.redis_url, 5)
.await .await
.expect("Redis cache configured"), .expect("Redis cache configured"),
); )
}
#[cfg(all(feature = "redis-cache", not(feature = "memory-cache")))] #[cfg(all(feature = "redis-cache", not(feature = "memory-cache")))]
log::info!("Listening redis server on {}", &_config.redis_url); {
#[cfg(all(feature = "redis-cache", not(feature = "memory-cache")))] log::info!("Listening redis server on {}", &_config.redis_url);
return Cache::new( Cache::new(
RedisCache::new(&_config.redis_url, 5) RedisCache::new(&_config.redis_url, 5)
.await .await
.expect("Redis cache configured"), .expect("Redis cache configured"),
); )
}
#[cfg(all(feature = "memory-cache", not(feature = "redis-cache")))] #[cfg(all(feature = "memory-cache", not(feature = "redis-cache")))]
{ {
log::info!("Using an in-memory cache"); log::info!("Using an in-memory cache");
return Cache::new_in_memory(); Cache::new_in_memory()
} }
#[cfg(not(any(feature = "memory-cache", feature = "redis-cache")))] #[cfg(not(any(feature = "memory-cache", feature = "redis-cache")))]
{ {
@ -131,27 +133,27 @@ impl Cache {
/// ///
/// Returns the `SearchResults` from the cache if the program executes normally otherwise /// Returns the `SearchResults` from the cache if the program executes normally otherwise
/// returns a `CacheError` if the results cannot be retrieved from the cache. /// returns a `CacheError` if the results cannot be retrieved from the cache.
pub async fn cached_json(&mut self, _url: &str) -> Result<SearchResults, Report<PoolError>> { pub async fn cached_json(&mut self, _url: &str) -> Result<SearchResults, Report<CacheError>> {
match self { match self {
Cache::Disabled => Err(Report::new(PoolError::MissingValue)), Cache::Disabled => Err(Report::new(CacheError::MissingValue)),
#[cfg(all(feature = "redis-cache", not(feature = "memory-cache")))] #[cfg(all(feature = "redis-cache", not(feature = "memory-cache")))]
Cache::Redis(redis_cache) => { Cache::Redis(redis_cache) => {
let json = redis_cache.cached_json(_url).await?; let json = redis_cache.cached_json(_url).await?;
Ok(serde_json::from_str::<SearchResults>(&json) Ok(serde_json::from_str::<SearchResults>(&json)
.map_err(|_| PoolError::SerializationError)?) .map_err(|_| CacheError::SerializationError)?)
} }
#[cfg(all(feature = "memory-cache", not(feature = "redis-cache")))] #[cfg(all(feature = "memory-cache", not(feature = "redis-cache")))]
Cache::InMemory(in_memory) => match in_memory.get(&_url.to_string()) { Cache::InMemory(in_memory) => match in_memory.get(&_url.to_string()) {
Some(res) => Ok(res), Some(res) => Ok(res),
None => Err(Report::new(PoolError::MissingValue)), None => Err(Report::new(CacheError::MissingValue)),
}, },
#[cfg(all(feature = "redis-cache", feature = "memory-cache"))] #[cfg(all(feature = "redis-cache", feature = "memory-cache"))]
Cache::Hybrid(redis_cache, in_memory) => match redis_cache.cached_json(_url).await { Cache::Hybrid(redis_cache, in_memory) => match redis_cache.cached_json(_url).await {
Ok(res) => Ok(serde_json::from_str::<SearchResults>(&res) Ok(res) => Ok(serde_json::from_str::<SearchResults>(&res)
.map_err(|_| PoolError::SerializationError)?), .map_err(|_| CacheError::SerializationError)?),
Err(_) => match in_memory.get(&_url.to_string()) { Err(_) => match in_memory.get(&_url.to_string()) {
Some(res) => Ok(res), Some(res) => Ok(res),
None => Err(Report::new(PoolError::MissingValue)), None => Err(Report::new(CacheError::MissingValue)),
}, },
}, },
} }
@ -174,13 +176,13 @@ impl Cache {
&mut self, &mut self,
_search_results: &SearchResults, _search_results: &SearchResults,
_url: &str, _url: &str,
) -> Result<(), Report<PoolError>> { ) -> Result<(), Report<CacheError>> {
match self { match self {
Cache::Disabled => Ok(()), Cache::Disabled => Ok(()),
#[cfg(all(feature = "redis-cache", not(feature = "memory-cache")))] #[cfg(all(feature = "redis-cache", not(feature = "memory-cache")))]
Cache::Redis(redis_cache) => { Cache::Redis(redis_cache) => {
let json = serde_json::to_string(_search_results) let json = serde_json::to_string(_search_results)
.map_err(|_| PoolError::SerializationError)?; .map_err(|_| CacheError::SerializationError)?;
redis_cache.cache_results(&json, _url).await redis_cache.cache_results(&json, _url).await
} }
#[cfg(all(feature = "memory-cache", not(feature = "redis-cache")))] #[cfg(all(feature = "memory-cache", not(feature = "redis-cache")))]
@ -191,7 +193,7 @@ impl Cache {
#[cfg(all(feature = "memory-cache", feature = "redis-cache"))] #[cfg(all(feature = "memory-cache", feature = "redis-cache"))]
Cache::Hybrid(redis_cache, cache) => { Cache::Hybrid(redis_cache, cache) => {
let json = serde_json::to_string(_search_results) let json = serde_json::to_string(_search_results)
.map_err(|_| PoolError::SerializationError)?; .map_err(|_| CacheError::SerializationError)?;
match redis_cache.cache_results(&json, _url).await { match redis_cache.cache_results(&json, _url).await {
Ok(_) => Ok(()), Ok(_) => Ok(()),
Err(_) => { Err(_) => {
@ -235,7 +237,7 @@ impl SharedCache {
/// ///
/// Returns a `SearchResults` struct containing the search results from the cache if nothing /// Returns a `SearchResults` struct containing the search results from the cache if nothing
/// goes wrong otherwise returns a `CacheError`. /// goes wrong otherwise returns a `CacheError`.
pub async fn cached_json(&self, url: &str) -> Result<SearchResults, Report<PoolError>> { pub async fn cached_json(&self, url: &str) -> Result<SearchResults, Report<CacheError>> {
let mut mut_cache = self.cache.lock().await; let mut mut_cache = self.cache.lock().await;
mut_cache.cached_json(url).await mut_cache.cached_json(url).await
} }
@ -258,7 +260,7 @@ impl SharedCache {
&self, &self,
search_results: &SearchResults, search_results: &SearchResults,
url: &str, url: &str,
) -> Result<(), Report<PoolError>> { ) -> Result<(), Report<CacheError>> {
let mut mut_cache = self.cache.lock().await; let mut mut_cache = self.cache.lock().await;
mut_cache.cache_results(search_results, url).await mut_cache.cache_results(search_results, url).await
} }

14
src/cache/error.rs vendored
View File

@ -7,7 +7,7 @@ use redis::RedisError;
/// A custom error type used for handling redis async pool associated errors. /// A custom error type used for handling redis async pool associated errors.
#[derive(Debug)] #[derive(Debug)]
pub enum PoolError { pub enum CacheError {
/// This variant handles all errors related to `RedisError`, /// This variant handles all errors related to `RedisError`,
#[cfg(feature = "redis-cache")] #[cfg(feature = "redis-cache")]
RedisError(RedisError), RedisError(RedisError),
@ -20,31 +20,31 @@ pub enum PoolError {
MissingValue, MissingValue,
} }
impl fmt::Display for PoolError { impl fmt::Display for CacheError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self { match self {
#[cfg(feature = "redis-cache")] #[cfg(feature = "redis-cache")]
PoolError::RedisError(redis_error) => { CacheError::RedisError(redis_error) => {
if let Some(detail) = redis_error.detail() { if let Some(detail) = redis_error.detail() {
write!(f, "{}", detail) write!(f, "{}", detail)
} else { } else {
write!(f, "") write!(f, "")
} }
} }
PoolError::PoolExhaustionWithConnectionDropError => { CacheError::PoolExhaustionWithConnectionDropError => {
write!( write!(
f, f,
"Error all connections from the pool dropped with connection error" "Error all connections from the pool dropped with connection error"
) )
} }
PoolError::MissingValue => { CacheError::MissingValue => {
write!(f, "The value is missing from the cache") write!(f, "The value is missing from the cache")
} }
PoolError::SerializationError => { CacheError::SerializationError => {
write!(f, "Unable to serialize, deserialize from the cache") write!(f, "Unable to serialize, deserialize from the cache")
} }
} }
} }
} }
impl error_stack::Context for PoolError {} impl error_stack::Context for CacheError {}

View File

@ -6,7 +6,7 @@ use futures::future::try_join_all;
use md5::compute; use md5::compute;
use redis::{aio::ConnectionManager, AsyncCommands, Client, RedisError}; use redis::{aio::ConnectionManager, AsyncCommands, Client, RedisError};
use super::error::PoolError; use super::error::CacheError;
/// A named struct which stores the redis Connection url address to which the client will /// A named struct which stores the redis Connection url address to which the client will
/// connect to. /// connect to.
@ -72,7 +72,7 @@ impl RedisCache {
/// ///
/// Returns the results as a String from the cache on success otherwise returns a `CacheError` /// Returns the results as a String from the cache on success otherwise returns a `CacheError`
/// on a failure. /// on a failure.
pub async fn cached_json(&mut self, url: &str) -> Result<String, Report<PoolError>> { pub async fn cached_json(&mut self, url: &str) -> Result<String, Report<CacheError>> {
self.current_connection = Default::default(); self.current_connection = Default::default();
let hashed_url_string: &str = &self.hash_url(url); let hashed_url_string: &str = &self.hash_url(url);
@ -95,7 +95,7 @@ impl RedisCache {
self.current_connection += 1; self.current_connection += 1;
if self.current_connection == self.pool_size { if self.current_connection == self.pool_size {
return Err(Report::new( return Err(Report::new(
PoolError::PoolExhaustionWithConnectionDropError, CacheError::PoolExhaustionWithConnectionDropError,
)); ));
} }
result = self.connection_pool[self.current_connection as usize] result = self.connection_pool[self.current_connection as usize]
@ -103,7 +103,7 @@ impl RedisCache {
.await; .await;
continue; continue;
} }
false => return Err(Report::new(PoolError::RedisError(error))), false => return Err(Report::new(CacheError::RedisError(error))),
}, },
Ok(res) => return Ok(res), Ok(res) => return Ok(res),
} }
@ -127,7 +127,7 @@ impl RedisCache {
&mut self, &mut self,
json_results: &str, json_results: &str,
url: &str, url: &str,
) -> Result<(), Report<PoolError>> { ) -> Result<(), Report<CacheError>> {
self.current_connection = Default::default(); self.current_connection = Default::default();
let hashed_url_string: &str = &self.hash_url(url); let hashed_url_string: &str = &self.hash_url(url);
@ -150,7 +150,7 @@ impl RedisCache {
self.current_connection += 1; self.current_connection += 1;
if self.current_connection == self.pool_size { if self.current_connection == self.pool_size {
return Err(Report::new( return Err(Report::new(
PoolError::PoolExhaustionWithConnectionDropError, CacheError::PoolExhaustionWithConnectionDropError,
)); ));
} }
result = self.connection_pool[self.current_connection as usize] result = self.connection_pool[self.current_connection as usize]
@ -158,7 +158,7 @@ impl RedisCache {
.await; .await;
continue; continue;
} }
false => return Err(Report::new(PoolError::RedisError(error))), false => return Err(Report::new(CacheError::RedisError(error))),
}, },
Ok(_) => return Ok(()), Ok(_) => return Ok(()),
} }

View File

@ -12,6 +12,7 @@ fn spawn_app() -> String {
let server = run( let server = run(
listener, listener,
config, config,
#[cfg(all(feature = "memory-cache", not(feature = "redis-cache")))]
websurfx::cache::cacher::Cache::new_in_memory(), websurfx::cache::cacher::Cache::new_in_memory(),
) )
.expect("Failed to bind address"); .expect("Failed to bind address");