anthrove/fuzzysearch
4
0
Fork 0
mirror of https://github.com/Syfaro/fuzzysearch.git synced 2024-09-20 13:59:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove panic, monitor video hashing duration.

Browse Source
This commit is contained in:
Syfaro 2021-02-21 00:30:28 -05:00
parent eded80fd78
commit f94f6a35f4
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
fuzzysearch/src/handlers.rs
View File

@ -16,6 +16,11 @@ lazy_static! {
"Duration to perform an image hash operation" "Duration to perform an image hash operation"
) )
.unwrap(); .unwrap();
static ref VIDEO_HASH_DURATION: Histogram = register_histogram!(
"fuzzysearch_api_video_hash_seconds",
"Duration to perform a video hash operation"
)
.unwrap();
static ref IMAGE_URL_DOWNLOAD_DURATION: Histogram = register_histogram!( static ref IMAGE_URL_DOWNLOAD_DURATION: Histogram = register_histogram!(
"fuzzysearch_api_image_url_download_seconds", "fuzzysearch_api_image_url_download_seconds",
"Duration to download an image from a provided URL" "Duration to download an image from a provided URL"
@ -130,23 +135,25 @@ async fn hash_input(form: warp::multipart::FormData) -> (i64, img_hash::ImageHas
} }
#[tracing::instrument(skip(form))] #[tracing::instrument(skip(form))]
async fn hash_video(form: warp::multipart::FormData) -> Vec<[u8; 8]> { async fn hash_video(form: warp::multipart::FormData) -> Option<Vec<[u8; 8]>> {
use bytes::Buf; use bytes::Buf;
let bytes = get_field_bytes(form, "video").await; let bytes = get_field_bytes(form, "video").await;
let _timer = VIDEO_HASH_DURATION.start_timer();
let hashes = tokio::task::spawn_blocking(move || { let hashes = tokio::task::spawn_blocking(move || {
if infer::is_video(&bytes) { if infer::is_video(&bytes) {
fuzzysearch_common::video::extract_video_hashes(bytes.reader()).unwrap() fuzzysearch_common::video::extract_video_hashes(bytes.reader()).ok()
} else if infer::image::is_gif(&bytes) { } else if infer::image::is_gif(&bytes) {
fuzzysearch_common::video::extract_gif_hashes(bytes.reader()).unwrap() fuzzysearch_common::video::extract_gif_hashes(bytes.reader()).ok()
} else { } else {
panic!("invalid file type provided"); None
} }
}) })
.instrument(span!(tracing::Level::TRACE, "hashing video")) .instrument(span!(tracing::Level::TRACE, "hashing video"))
.await .await
.unwrap(); .unwrap();
drop(_timer);
hashes hashes
} }