fix: dont require json content type

samply · Jul 30, 2024 · 322d396 · 322d396
1 parent cf3b8ee
commit 322d396
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -1,9 +1,5 @@
 use axum::{
-    extract::{Json, Path, Query, State},
-    http::{HeaderMap, HeaderValue},
-    response::{sse::Event, IntoResponse, Sse},
-    routing::{get, post},
-    Router
+    body::Bytes, extract::{Json, Path, Query, State}, http::{HeaderMap, HeaderValue}, response::{sse::Event, IntoResponse, Sse}, routing::{get, post}, Router
 };
 use mini_moka::sync::Cache;
 use std::{convert::Infallible, sync::Arc, time::Duration};
@@ -88,8 +84,9 @@ async fn handler_health() -> Json<HealthOutput> {
 
 async fn handle_create_beam_task(
     State(SharedState { results }): State<SharedState>,
-    Json(query): Json<LensQuery>,
+    body: Bytes,
 ) -> Result<impl IntoResponse, StatusCode> {
+    let query = serde_json::from_slice(&body).map_err(|_| StatusCode::BAD_REQUEST)?;
     #[derive(Serialize, Deserialize, Debug)]
     pub struct CqlQuery {
         pub lib: Value,