mirror of
https://github.com/n08i40k/schedule-parser-rusted.git
synced 2025-12-06 09:47:50 +03:00
Улучшенное отображение ошибок при обновлении ссылки расписания.
This commit is contained in:
@@ -60,20 +60,20 @@ pub async fn update_download_url(
|
|||||||
}
|
}
|
||||||
},
|
},
|
||||||
Err(error) => {
|
Err(error) => {
|
||||||
if let FetchError::Unknown(error) = error {
|
if let FetchError::Unknown(error) = &error {
|
||||||
sentry::capture_error(&error);
|
sentry::capture_error(&error);
|
||||||
}
|
}
|
||||||
|
|
||||||
ErrorCode::DownloadFailed.into_response()
|
ErrorCode::DownloadFailed(error).into_response()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Err(error) => {
|
Err(error) => {
|
||||||
if let FetchError::Unknown(error) = error {
|
if let FetchError::Unknown(error) = &error {
|
||||||
sentry::capture_error(&error);
|
sentry::capture_error(&error);
|
||||||
}
|
}
|
||||||
|
|
||||||
ErrorCode::FetchFailed.into_response()
|
ErrorCode::FetchFailed(error).into_response()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -85,6 +85,7 @@ mod schema {
|
|||||||
use derive_more::Display;
|
use derive_more::Display;
|
||||||
use serde::{Deserialize, Serialize, Serializer};
|
use serde::{Deserialize, Serialize, Serializer};
|
||||||
use utoipa::ToSchema;
|
use utoipa::ToSchema;
|
||||||
|
use crate::xls_downloader::interface::FetchError;
|
||||||
|
|
||||||
pub type ServiceResponse = crate::routes::schema::Response<CacheStatus, ErrorCode>;
|
pub type ServiceResponse = crate::routes::schema::Response<CacheStatus, ErrorCode>;
|
||||||
|
|
||||||
@@ -103,12 +104,12 @@ mod schema {
|
|||||||
NonWhitelistedHost,
|
NonWhitelistedHost,
|
||||||
|
|
||||||
/// Failed to retrieve file metadata.
|
/// Failed to retrieve file metadata.
|
||||||
#[display("Unable to retrieve metadata from the specified URL.")]
|
#[display("Unable to retrieve metadata from the specified URL: {_0}")]
|
||||||
FetchFailed,
|
FetchFailed(FetchError),
|
||||||
|
|
||||||
/// Failed to download the file.
|
/// Failed to download the file.
|
||||||
#[display("Unable to retrieve data from the specified URL.")]
|
#[display("Unable to retrieve data from the specified URL: {_0}")]
|
||||||
DownloadFailed,
|
DownloadFailed(FetchError),
|
||||||
|
|
||||||
/// The link leads to an outdated schedule.
|
/// The link leads to an outdated schedule.
|
||||||
///
|
///
|
||||||
@@ -129,8 +130,8 @@ mod schema {
|
|||||||
{
|
{
|
||||||
match self {
|
match self {
|
||||||
ErrorCode::NonWhitelistedHost => serializer.serialize_str("NON_WHITELISTED_HOST"),
|
ErrorCode::NonWhitelistedHost => serializer.serialize_str("NON_WHITELISTED_HOST"),
|
||||||
ErrorCode::FetchFailed => serializer.serialize_str("FETCH_FAILED"),
|
ErrorCode::FetchFailed(_) => serializer.serialize_str("FETCH_FAILED"),
|
||||||
ErrorCode::DownloadFailed => serializer.serialize_str("DOWNLOAD_FAILED"),
|
ErrorCode::DownloadFailed(_) => serializer.serialize_str("DOWNLOAD_FAILED"),
|
||||||
ErrorCode::OutdatedSchedule => serializer.serialize_str("OUTDATED_SCHEDULE"),
|
ErrorCode::OutdatedSchedule => serializer.serialize_str("OUTDATED_SCHEDULE"),
|
||||||
ErrorCode::InvalidSchedule(_) => serializer.serialize_str("INVALID_SCHEDULE"),
|
ErrorCode::InvalidSchedule(_) => serializer.serialize_str("INVALID_SCHEDULE"),
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
use crate::xls_downloader::interface::{FetchError, FetchOk, FetchResult, XLSDownloader};
|
use crate::xls_downloader::interface::{FetchError, FetchOk, FetchResult, XLSDownloader};
|
||||||
use chrono::{DateTime, Utc};
|
use chrono::{DateTime, Utc};
|
||||||
use std::env;
|
use std::env;
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
pub struct BasicXlsDownloader {
|
pub struct BasicXlsDownloader {
|
||||||
pub url: Option<String>,
|
pub url: Option<String>,
|
||||||
@@ -22,7 +23,7 @@ async fn fetch_specified(url: &String, user_agent: &String, head: bool) -> Fetch
|
|||||||
match response {
|
match response {
|
||||||
Ok(r) => {
|
Ok(r) => {
|
||||||
if r.status().as_u16() != 200 {
|
if r.status().as_u16() != 200 {
|
||||||
return Err(FetchError::BadStatusCode);
|
return Err(FetchError::BadStatusCode(r.status().as_u16()));
|
||||||
}
|
}
|
||||||
|
|
||||||
let headers = r.headers();
|
let headers = r.headers();
|
||||||
@@ -32,11 +33,18 @@ async fn fetch_specified(url: &String, user_agent: &String, head: bool) -> Fetch
|
|||||||
let last_modified = headers.get("last-modified");
|
let last_modified = headers.get("last-modified");
|
||||||
let date = headers.get("date");
|
let date = headers.get("date");
|
||||||
|
|
||||||
if content_type.is_none() || etag.is_none() || last_modified.is_none() || date.is_none()
|
if content_type.is_none() {
|
||||||
{
|
Err(FetchError::BadHeaders("Content-Type".to_string()))
|
||||||
Err(FetchError::BadHeaders)
|
} else if etag.is_none() {
|
||||||
|
Err(FetchError::BadHeaders("ETag".to_string()))
|
||||||
|
} else if last_modified.is_none() {
|
||||||
|
Err(FetchError::BadHeaders("Last-Modified".to_string()))
|
||||||
|
} else if date.is_none() {
|
||||||
|
Err(FetchError::BadHeaders("Date".to_string()))
|
||||||
} else if content_type.unwrap() != "application/vnd.ms-excel" {
|
} else if content_type.unwrap() != "application/vnd.ms-excel" {
|
||||||
Err(FetchError::BadContentType)
|
Err(FetchError::BadContentType(
|
||||||
|
content_type.unwrap().to_str().unwrap().to_string(),
|
||||||
|
))
|
||||||
} else {
|
} else {
|
||||||
let etag = etag.unwrap().to_str().unwrap().to_string();
|
let etag = etag.unwrap().to_str().unwrap().to_string();
|
||||||
let last_modified =
|
let last_modified =
|
||||||
@@ -51,7 +59,7 @@ async fn fetch_specified(url: &String, user_agent: &String, head: bool) -> Fetch
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Err(e) => Err(FetchError::Unknown(e)),
|
Err(error) => Err(FetchError::Unknown(Arc::new(error))),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -86,7 +94,7 @@ impl XLSDownloader for BasicXlsDownloader {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::xls_downloader::basic_impl::{BasicXlsDownloader, fetch_specified};
|
use crate::xls_downloader::basic_impl::{fetch_specified, BasicXlsDownloader};
|
||||||
use crate::xls_downloader::interface::{FetchError, XLSDownloader};
|
use crate::xls_downloader::interface::{FetchError, XLSDownloader};
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
@@ -116,14 +124,10 @@ mod tests {
|
|||||||
assert!(results[0].is_err());
|
assert!(results[0].is_err());
|
||||||
assert!(results[1].is_err());
|
assert!(results[1].is_err());
|
||||||
|
|
||||||
assert_eq!(
|
let expected_error = FetchError::BadStatusCode(404);
|
||||||
*results[0].as_ref().err().unwrap(),
|
|
||||||
FetchError::BadStatusCode
|
assert_eq!(*results[0].as_ref().err().unwrap(), expected_error);
|
||||||
);
|
assert_eq!(*results[1].as_ref().err().unwrap(), expected_error);
|
||||||
assert_eq!(
|
|
||||||
*results[1].as_ref().err().unwrap(),
|
|
||||||
FetchError::BadStatusCode
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
@@ -139,8 +143,10 @@ mod tests {
|
|||||||
assert!(results[0].is_err());
|
assert!(results[0].is_err());
|
||||||
assert!(results[1].is_err());
|
assert!(results[1].is_err());
|
||||||
|
|
||||||
assert_eq!(*results[0].as_ref().err().unwrap(), FetchError::BadHeaders);
|
let expected_error = FetchError::BadHeaders("ETag".to_string());
|
||||||
assert_eq!(*results[1].as_ref().err().unwrap(), FetchError::BadHeaders);
|
|
||||||
|
assert_eq!(*results[0].as_ref().err().unwrap(), expected_error);
|
||||||
|
assert_eq!(*results[1].as_ref().err().unwrap(), expected_error);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
@@ -155,15 +161,6 @@ mod tests {
|
|||||||
|
|
||||||
assert!(results[0].is_err());
|
assert!(results[0].is_err());
|
||||||
assert!(results[1].is_err());
|
assert!(results[1].is_err());
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
*results[0].as_ref().err().unwrap(),
|
|
||||||
FetchError::BadContentType
|
|
||||||
);
|
|
||||||
assert_eq!(
|
|
||||||
*results[1].as_ref().err().unwrap(),
|
|
||||||
FetchError::BadContentType
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
|
|||||||
@@ -1,23 +1,32 @@
|
|||||||
use chrono::{DateTime, Utc};
|
use chrono::{DateTime, Utc};
|
||||||
|
use derive_more::Display;
|
||||||
use std::mem::discriminant;
|
use std::mem::discriminant;
|
||||||
|
use std::sync::Arc;
|
||||||
|
use utoipa::ToSchema;
|
||||||
|
|
||||||
/// XLS data retrieval errors.
|
/// XLS data retrieval errors.
|
||||||
#[derive(Debug)]
|
#[derive(Clone, Debug, ToSchema, Display)]
|
||||||
pub enum FetchError {
|
pub enum FetchError {
|
||||||
/// File url is not set.
|
/// File url is not set.
|
||||||
|
#[display("The link to the timetable was not provided earlier.")]
|
||||||
NoUrlProvided,
|
NoUrlProvided,
|
||||||
|
|
||||||
/// Unknown error.
|
/// Unknown error.
|
||||||
Unknown(reqwest::Error),
|
#[display("An unknown error occurred while downloading the file.")]
|
||||||
|
#[schema(value_type = String)]
|
||||||
|
Unknown(Arc<reqwest::Error>),
|
||||||
|
|
||||||
/// Server returned a status code different from 200.
|
/// Server returned a status code different from 200.
|
||||||
BadStatusCode,
|
#[display("Server returned a status code {_0}.")]
|
||||||
|
BadStatusCode(u16),
|
||||||
|
|
||||||
/// The url leads to a file of a different type.
|
/// The url leads to a file of a different type.
|
||||||
BadContentType,
|
#[display("The link leads to a file of type '{_0}'.")]
|
||||||
|
BadContentType(String),
|
||||||
|
|
||||||
/// Server doesn't return expected headers.
|
/// Server doesn't return expected headers.
|
||||||
BadHeaders,
|
#[display("Server doesn't return expected header(s) '{_0}'.")]
|
||||||
|
BadHeaders(String),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl PartialEq for FetchError {
|
impl PartialEq for FetchError {
|
||||||
|
|||||||
Reference in New Issue
Block a user