searchengines-web/MeiliSearch
searchengines-web/MeiliSearch
1
0
Fork 0
mirror of https://github.com/meilisearch/MeiliSearch synced 2024-11-30 00:34:26 +01:00
Code Issues Releases Wiki Activity

error change and tokio file use change

Browse Source
This commit is contained in:
jiangbo212 2022-12-07 16:20:36 +08:00
parent 6766712840
commit 35f3dd68b6
2 changed files with 13 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
meilisearch-http/src/error.rs
View File

@ -95,8 +95,8 @@ pub enum PayloadError {
MalformedPayload(serde_json::error::Error),
#[error("A json payload is missing.")]
MissingPayload,
#[error("Exception when accepting a playload to a temporary file")]
ReceivePayloadErr,
#[error("Error while writing the playload to disk: `{0}`.")]
ReceivePayloadErr(Box<dyn std::error::Error + Send + Sync + 'static>),
}
impl ErrorCode for PayloadError {
@ -128,7 +128,7 @@ impl ErrorCode for PayloadError {
},
PayloadError::MissingPayload => Code::MissingPayload,
PayloadError::MalformedPayload(_) => Code::MalformedPayload,
PayloadError::ReceivePayloadErr => Code::Internal,
PayloadError::ReceivePayloadErr(_) => Code::Internal,
}
}
}

24
meilisearch-http/src/routes/indexes/documents.rs
View File

@ -4,7 +4,7 @@ use actix_web::{web, HttpMessage, HttpRequest, HttpResponse};
use bstr::ByteSlice;
use futures::StreamExt;
use index_scheduler::IndexScheduler;
use log::{debug, error};
use log::debug;
use meilisearch_types::document_formats::{read_csv, read_json, read_ndjson, PayloadType};
use meilisearch_types::error::ResponseError;
use meilisearch_types::heed::RoTxn;
@ -233,19 +233,17 @@ async fn document_addition(
let temp_file = match NamedTempFile::new() {
Ok(temp_file) => temp_file,
Err(e) => {
error!("create a temporary file error: {}", e);
return Err(MeilisearchHttpError::Payload(ReceivePayloadErr));
return Err(MeilisearchHttpError::Payload(ReceivePayloadErr(Box::new(e))));
}
};
debug!("temp file path: {:?}", temp_file.as_ref());
let buffer_file = match temp_file.reopen() {
Ok(buffer_file) => buffer_file,
let mut buffer = match File::create(&temp_file.as_ref()).await {
Ok(buffer) => buffer,
Err(e) => {
error!("reopen payload temporary file error: {}", e);
return Err(MeilisearchHttpError::Payload(ReceivePayloadErr));
return Err(MeilisearchHttpError::Payload(ReceivePayloadErr(Box::new(e))));
}
};
let mut buffer = File::from_std(buffer_file);
let mut buffer_write_size: usize = 0;
while let Some(bytes) = body.next().await {
let byte = &bytes?;
@ -257,16 +255,14 @@ async fn document_addition(
match buffer.write_all(byte).await {
Ok(()) => buffer_write_size += 1,
Err(e) => {
error!("bufWriter write error: {}", e);
return Err(MeilisearchHttpError::Payload(ReceivePayloadErr));
return Err(MeilisearchHttpError::Payload(ReceivePayloadErr(Box::new(e))));
}
};
}
if let Err(e) = buffer.flush().await {
error!("bufWriter flush error: {}", e);
return Err(MeilisearchHttpError::Payload(ReceivePayloadErr));
};
return Err(MeilisearchHttpError::Payload(ReceivePayloadErr(Box::new(e))));
}
if buffer_write_size == 0 {
return Err(MeilisearchHttpError::MissingPayload(format));