refactor: moved routes

src/lib.rs

@@ -2,6 +2,6 @@ pub mod error;
 pub mod ingress;
 pub mod rabbitmq;
 pub mod retrieval;
-pub mod routes;
+pub mod server;
 pub mod storage;
 pub mod utils;

src/server/mod.rs

@@ -0,0 +1 @@
+pub mod routes;

src/server/routes/mod.rs

@@ -1,3 +1,4 @@
 pub mod file;
 pub mod ingress;
+pub mod query;
 pub mod queue_length;

src/server/routes/query.rs

@@ -0,0 +1,17 @@
+use crate::storage::db::SurrealDbClient;
+use axum::{http::StatusCode, response::IntoResponse, Extension, Json};
+use serde::Deserialize;
+use std::sync::Arc;
+use tracing::{error, info};
+
+#[derive(Debug, Deserialize)]
+pub struct QueryInput {
+    query: String,
+}
+
+pub async fn query_handler(
+    Extension(db_client): Extension<Arc<SurrealDbClient>>,
+    Json(query): Json<QueryInput>,
+) -> impl IntoResponse {
+    info!("Received input: {:?}", query);
+}

src/server/routes/queue_length.rs

@@ -1,11 +1,14 @@
-use axum::{http::StatusCode, response::{IntoResponse, Response}};
+use axum::{
+    http::StatusCode,
+    response::{IntoResponse, Response},
+};
 use tracing::{error, info};
 
 use crate::rabbitmq::{consumer::RabbitMQConsumer, RabbitMQConfig};
 
 pub async fn queue_length_handler() -> Response {
     info!("Getting queue length");
-    
+
     // Set up RabbitMQ config
     let config = RabbitMQConfig {
         amqp_addr: "amqp://localhost".to_string(),
@@ -26,11 +29,14 @@ pub async fn queue_length_handler() -> Response {
 
             // Return the queue length with a 200 OK status
             (StatusCode::OK, queue_length.to_string()).into_response()
-        },
+        }
         Err(e) => {
             error!("Failed to create consumer: {:?}", e);
-            (StatusCode::INTERNAL_SERVER_ERROR, "Failed to connect to RabbitMQ".to_string()).into_response()
+            (
+                StatusCode::INTERNAL_SERVER_ERROR,
+                "Failed to connect to RabbitMQ".to_string(),
+            )
+                .into_response()
         }
     }
 }
-