server: server struct

This commit is contained in:
nym21
2025-03-06 14:52:26 +01:00
parent d2ca6f1d46
commit f1851b304c
10 changed files with 125 additions and 112 deletions

View File

@@ -3,9 +3,9 @@ use axum::{Router, routing::get};
use super::AppState;
mod explorer;
mod vecs;
mod query;
pub use vecs::DTS;
pub use query::DTS;
pub trait ApiRoutes {
fn add_api_routes(self) -> Self;
@@ -13,6 +13,6 @@ pub trait ApiRoutes {
impl ApiRoutes for Router<AppState> {
fn add_api_routes(self) -> Self {
self.route("/api/vecs", get(vecs::handler))
self.route("/api/query", get(query::handler))
}
}

View File

@@ -7,8 +7,6 @@ use axum::{
response::{IntoResponse, Response},
};
use brk_query::{Format, Index, Output, Params};
use color_eyre::eyre::eyre;
use serde_json::Value;
use crate::{log_result, traits::HeaderMapExtended};
@@ -26,17 +24,15 @@ pub async fn handler(
) -> Response {
let instant = Instant::now();
let path = uri.path();
match req_to_response_res(headers, query, app_state) {
Ok(response) => {
log_result(response.status(), path, instant);
log_result(response.status(), &uri, instant);
response
}
Err(error) => {
let mut response =
(StatusCode::INTERNAL_SERVER_ERROR, error.to_string()).into_response();
log_result(response.status(), path, instant);
log_result(response.status(), &uri, instant);
response.headers_mut().insert_cors();
response
}