Mirrors/coco-app
1
0
Fork 0
mirror of https://github.com/infinilabs/coco-app.git synced 2025-12-29 00:24:46 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor: refactoring to check status code before parse the response (#121)

Browse Source
This commit is contained in:
Medcl
2025-02-06 18:03:40 +08:00
committed by GitHub
parent bd81a62df1
commit b85dd178f5
1 changed files with 25 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src-tauri/src/server/search.rs
View File

@@ -163,21 +163,36 @@ pub async fn query_coco_servers<R: Runtime>(
match res_response {
Ok(response) => {
let status_code = response.status().as_u16();
match parse_search_results_with_score(response).await {
Ok(documents) => {
total_hits += documents.len(); // No need for `&` here, as `len` is `usize`
for (doc, score) in documents {
let score = score.unwrap_or(0.0) as f64;
docs_collector.push(server_id.clone(), doc, score);
// Check if the status code indicates a successful request (2xx)
if status_code >= 200 && status_code < 400 {
// Parse the response only if the status code is success
match parse_search_results_with_score(response).await {
Ok(documents) => {
total_hits += documents.len(); // No need for `&` here, as `len` is `usize`
for (doc, score) in documents {
let score = score.unwrap_or(0.0) as f64;
docs_collector.push(server_id.clone(), doc, score);
}
}
Err(err) => {
failed_requests.push(create_failed_request(
&server_id, &coco_servers, &err.to_string(), status_code,
));
}
}
Err(err) => {
failed_requests.push(create_failed_request(&server_id, &coco_servers, &err.to_string(), status_code));
}
} else {
// If status code is not successful, log the failure
failed_requests.push(create_failed_request(
&server_id, &coco_servers, "Unsuccessful response", status_code,
));
}
}
Err(err) => {
failed_requests.push(create_failed_request(&server_id,&coco_servers, &err.to_string(), 0));
// Handle the error from the future itself
failed_requests.push(create_failed_request(
&server_id, &coco_servers, &err.to_string(), 0,
));
}
}
}