itsi 0.2.15 → 0.2.16

data/crates/itsi_server/src/ruby_types/itsi_server/itsi_server_config.rs

@@ -82,7 +82,7 @@ pub struct ServerParams {
     preexisting_listeners: Option<String>,
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, Copy)]
 pub enum ItsiServerTokenPreference {
     Version,
     Name,
@@ -109,6 +109,7 @@ pub struct SocketOpts {
     pub listen_backlog: usize,
     pub nodelay: bool,
     pub recv_buffer_size: usize,
+    pub send_buffer_size: usize,
 }
 
 impl ServerParams {
@@ -248,6 +249,9 @@ impl ServerParams {
         let recv_buffer_size: usize = rb_param_hash
             .fetch::<_, Option<usize>>("recv_buffer_size")?
             .unwrap_or(262_144);
+        let send_buffer_size: usize = rb_param_hash
+            .fetch::<_, Option<usize>>("send_buffer_size")?
+            .unwrap_or(262_144);
 
         if let Some(level) = log_level {
             set_level(&level);
@@ -296,6 +300,7 @@ impl ServerParams {
             listen_backlog,
             nodelay,
             recv_buffer_size,
+            send_buffer_size,
         };
         let preexisting_listeners = rb_param_hash.delete::<_, Option<String>>("listeners")?;
 

data/crates/itsi_server/src/server/binds/listener.rs

@@ -111,7 +111,7 @@ impl TokioListener {
         Self::to_tokio_io(Stream::TcpStream(tcp_stream), None).await
     }
 
-    pub async fn spawn_state_task(&self, mut shutdown_receiver: Receiver<RunningPhase>) {
+    pub async fn spawn_acme_event_task(&self, mut shutdown_receiver: Receiver<RunningPhase>) {
         if let TokioListener::TcpTls(
             _,
             ItsiTlsAcceptor::Automatic(_acme_acceptor, state, _server_config),
@@ -414,6 +414,9 @@ fn connect_tcp_socket(addr: IpAddr, port: u16, socket_opts: &SocketOpts) -> Resu
     socket.set_reuse_port(socket_opts.reuse_port).ok();
     socket.set_nonblocking(true).ok();
     socket.set_nodelay(socket_opts.nodelay).ok();
+    socket
+        .set_send_buffer_size(socket_opts.send_buffer_size)
+        .ok();
     socket
         .set_recv_buffer_size(socket_opts.recv_buffer_size)
         .ok();

data/crates/itsi_server/src/server/http_message_types.rs

@@ -32,7 +32,7 @@ pub trait PathExt {
     fn no_trailing_slash(&self) -> &str;
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, Copy)]
 pub enum ResponseFormat {
     JSON,
     HTML,

data/crates/itsi_server/src/server/middleware_stack/middlewares/error_response/default_responses.rs

@@ -2,84 +2,84 @@ use super::{ContentSource, DefaultFormat, ErrorResponse};
 use crate::server::http_message_types::ResponseFormat;
 use bytes::Bytes;
 use http_body_util::{combinators::BoxBody, Full};
-use std::{convert::Infallible, sync::Arc};
+use std::convert::Infallible;
 
 impl DefaultFormat {
     pub fn response_for_code(&self, code: u16) -> ContentSource {
         match self {
             DefaultFormat::Plaintext => match code {
-                500 => ContentSource::Static(Arc::new("500 Internal Error".into())),
-                404 => ContentSource::Static(Arc::new("404 Not Found".into())),
-                401 => ContentSource::Static(Arc::new("401 Unauthorized".into())),
-                403 => ContentSource::Static(Arc::new("403 Forbidden".into())),
-                413 => ContentSource::Static(Arc::new("413 Payload Too Large".into())),
-                429 => ContentSource::Static(Arc::new("429 Too Many Requests".into())),
-                502 => ContentSource::Static(Arc::new("502 Bad Gateway".into())),
-                503 => ContentSource::Static(Arc::new("503 Service Unavailable".into())),
-                504 => ContentSource::Static(Arc::new("504 Gateway Timeout".into())),
-                _ => ContentSource::Static(Arc::new("Unexpected Error".into())),
+                500 => ContentSource::Static(Full::new("500 Internal Error".into())),
+                404 => ContentSource::Static(Full::new("404 Not Found".into())),
+                401 => ContentSource::Static(Full::new("401 Unauthorized".into())),
+                403 => ContentSource::Static(Full::new("403 Forbidden".into())),
+                413 => ContentSource::Static(Full::new("413 Payload Too Large".into())),
+                429 => ContentSource::Static(Full::new("429 Too Many Requests".into())),
+                502 => ContentSource::Static(Full::new("502 Bad Gateway".into())),
+                503 => ContentSource::Static(Full::new("503 Service Unavailable".into())),
+                504 => ContentSource::Static(Full::new("504 Gateway Timeout".into())),
+                _ => ContentSource::Static(Full::new("Unexpected Error".into())),
             },
             DefaultFormat::Html => match code {
-                500 => ContentSource::Static(Arc::new(
+                500 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/html/500.html").into(),
                 )),
-                404 => ContentSource::Static(Arc::new(
+                404 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/html/404.html").into(),
                 )),
-                401 => ContentSource::Static(Arc::new(
+                401 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/html/401.html").into(),
                 )),
-                403 => ContentSource::Static(Arc::new(
+                403 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/html/403.html").into(),
                 )),
-                413 => ContentSource::Static(Arc::new(
+                413 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/html/413.html").into(),
                 )),
-                429 => ContentSource::Static(Arc::new(
+                429 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/html/429.html").into(),
                 )),
-                502 => ContentSource::Static(Arc::new(
+                502 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/html/502.html").into(),
                 )),
-                503 => ContentSource::Static(Arc::new(
+                503 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/html/503.html").into(),
                 )),
-                504 => ContentSource::Static(Arc::new(
+                504 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/html/504.html").into(),
                 )),
-                _ => ContentSource::Static(Arc::new(
+                _ => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/html/500.html").into(),
                 )),
             },
             DefaultFormat::Json => match code {
-                500 => ContentSource::Static(Arc::new(
+                500 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/json/500.json").into(),
                 )),
-                404 => ContentSource::Static(Arc::new(
+                404 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/json/404.json").into(),
                 )),
-                401 => ContentSource::Static(Arc::new(
+                401 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/json/401.json").into(),
                 )),
-                403 => ContentSource::Static(Arc::new(
+                403 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/json/403.json").into(),
                 )),
-                413 => ContentSource::Static(Arc::new(
+                413 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/json/413.json").into(),
                 )),
-                429 => ContentSource::Static(Arc::new(
+                429 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/json/429.json").into(),
                 )),
-                502 => ContentSource::Static(Arc::new(
+                502 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/json/502.json").into(),
                 )),
-                503 => ContentSource::Static(Arc::new(
+                503 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/json/503.json").into(),
                 )),
-                504 => ContentSource::Static(Arc::new(
+                504 => ContentSource::Static(Full::new(
                     include_str!("../../../../default_responses/json/504.json").into(),
                 )),
-                _ => ContentSource::Static(Arc::new("Unexpected Error".into())),
+                _ => ContentSource::Static(Full::new("Unexpected Error".into())),
             },
         }
     }
@@ -94,9 +94,7 @@ impl ErrorResponse {
         };
         match source {
             ContentSource::Inline(bytes) => BoxBody::new(Full::new(Bytes::from(bytes))),
-            ContentSource::Static(text) => {
-                BoxBody::new(Full::new(Bytes::from(String::from(text.as_str()))))
-            }
+            ContentSource::Static(bytes) => BoxBody::new(bytes),
             ContentSource::File(_) => BoxBody::new(Full::new(Bytes::from("Unexpected error"))),
         }
     }

data/crates/itsi_server/src/server/middleware_stack/middlewares/error_response.rs

@@ -5,7 +5,6 @@ use http_body_util::{combinators::BoxBody, Full};
 use serde::{Deserialize, Deserializer};
 use std::convert::Infallible;
 use std::path::PathBuf;
-use std::sync::Arc;
 use tracing::warn;
 
 use crate::server::http_message_types::{HttpResponse, ResponseFormat};
@@ -20,7 +19,7 @@ pub enum ContentSource {
     File(PathBuf),
     #[serde(rename(deserialize = "static"))]
     #[serde(skip_deserializing)]
-    Static(Arc<String>),
+    Static(Full<Bytes>),
 }
 
 #[derive(Debug, Clone, Deserialize, Default)]
@@ -151,13 +150,13 @@ impl ErrorResponse {
                 return BoxBody::new(Full::new(Bytes::from(text.clone())));
             }
             Some(ContentSource::Static(text)) => {
-                return BoxBody::new(Full::new(Bytes::from(String::from(text.as_str()))));
+                return BoxBody::new(text.clone());
             }
             Some(ContentSource::File(path)) => {
                 // Convert the PathBuf to a &str (assumes valid UTF-8).
                 if let Some(path_str) = path.to_str() {
                     let response = ROOT_STATIC_FILE_SERVER
-                        .serve_single(path_str, accept.clone(), &[])
+                        .serve_single(path_str, accept, &[])
                         .await;
                     if response.status().is_success() {
                         return response.into_body();

data/crates/itsi_server/src/server/middleware_stack/middlewares/etag.rs

@@ -43,12 +43,6 @@ pub struct ETag {
     pub algorithm: HashAlgorithm,
     #[serde(default)]
     pub min_body_size: usize,
-    #[serde(default = "default_true")]
-    pub handle_if_none_match: bool,
-}
-
-fn default_true() -> bool {
-    true
 }
 
 #[async_trait]
@@ -59,14 +53,13 @@ impl MiddlewareLayer for ETag {
         context: &mut HttpRequestContext,
     ) -> Result<Either<HttpRequest, HttpResponse>> {
         // Store if-none-match header in context if present for later use in after hook
-        if self.handle_if_none_match {
-            if let Some(if_none_match) = req.headers().get(header::IF_NONE_MATCH) {
-                debug!(target: "middleware::etag", "Received If-None-Match header: {:?}", if_none_match);
-                if let Ok(etag_value) = if_none_match.to_str() {
-                    context.set_if_none_match(Some(etag_value.to_string()));
-                }
+        if let Some(if_none_match) = req.headers().get(header::IF_NONE_MATCH) {
+            debug!(target: "middleware::etag", "Received If-None-Match header: {:?}", if_none_match);
+            if let Ok(etag_value) = if_none_match.to_str() {
+                context.set_if_none_match(Some(etag_value.to_string()));
             }
         }
+
         Ok(Either::Left(req))
     }
 
@@ -85,11 +78,6 @@ impl MiddlewareLayer for ETag {
             }
         }
 
-        if resp.headers().contains_key(header::ETAG) {
-            debug!(target: "middleware::etag", "Forwarding response with existing ETag");
-            return resp;
-        }
-
         if let Some(cache_control) = resp.headers().get(header::CACHE_CONTROL) {
             if let Ok(cache_control_str) = cache_control.to_str() {
                 if cache_control_str.contains("no-store") {
@@ -155,28 +143,26 @@ impl MiddlewareLayer for ETag {
             formatted_etag
         };
 
-        if self.handle_if_none_match {
-            if let Some(if_none_match) = context.get_if_none_match() {
-                if if_none_match == etag_value || if_none_match == "*" {
-                    // Return 304 Not Modified without the body
-                    let mut not_modified = Response::new(BoxBody::new(Empty::new()));
-                    *not_modified.status_mut() = StatusCode::NOT_MODIFIED;
-                    // Copy headers we want to preserve
-                    for (name, value) in parts.headers.iter() {
-                        if matches!(
-                            name,
-                            &header::CACHE_CONTROL
-                                | &header::CONTENT_LOCATION
-                                | &header::DATE
-                                | &header::ETAG
-                                | &header::EXPIRES
-                                | &header::VARY
-                        ) {
-                            not_modified.headers_mut().insert(name, value.clone());
-                        }
+        if let Some(if_none_match) = context.get_if_none_match() {
+            if if_none_match == etag_value || if_none_match == "*" {
+                // Return 304 Not Modified without the body
+                let mut not_modified = Response::new(BoxBody::new(Empty::new()));
+                *not_modified.status_mut() = StatusCode::NOT_MODIFIED;
+                // Copy headers we want to preserve
+                for (name, value) in parts.headers.iter() {
+                    if matches!(
+                        name,
+                        &header::CACHE_CONTROL
+                            | &header::CONTENT_LOCATION
+                            | &header::DATE
+                            | &header::ETAG
+                            | &header::EXPIRES
+                            | &header::VARY
+                    ) {
+                        not_modified.headers_mut().insert(name, value.clone());
                     }
-                    return not_modified;
                 }
+                return not_modified;
             }
         }
 
@@ -190,7 +176,6 @@ impl Default for ETag {
             r#type: ETagType::Strong,
             algorithm: HashAlgorithm::Sha256,
             min_body_size: 0,
-            handle_if_none_match: true,
         }
     }
 }

data/crates/itsi_server/src/server/middleware_stack/middlewares/log_requests.rs

@@ -3,6 +3,7 @@ use either::Either;
 use itsi_tracing::*;
 use magnus::error::Result;
 use serde::Deserialize;
+use tracing::enabled;
 
 use crate::server::http_message_types::{HttpRequest, HttpResponse};
 use crate::services::itsi_http_service::HttpRequestContext;
@@ -39,18 +40,6 @@ pub enum LogMiddlewareLevel {
     Error,
 }
 
-impl LogMiddlewareLevel {
-    pub fn log(&self, message: String) {
-        match self {
-            LogMiddlewareLevel::Trace => trace!(target: "middleware::log_requests", message),
-            LogMiddlewareLevel::Debug => debug!(target: "middleware::log_requests", message),
-            LogMiddlewareLevel::Info => info!(target: "middleware::log_requests", message),
-            LogMiddlewareLevel::Warn => warn!(target: "middleware::log_requests", message),
-            LogMiddlewareLevel::Error => error!(target: "middleware::log_requests", message),
-        }
-    }
-}
-
 #[async_trait]
 impl MiddlewareLayer for LogRequests {
     async fn initialize(&self) -> Result<()> {
@@ -64,7 +53,38 @@ impl MiddlewareLayer for LogRequests {
     ) -> Result<Either<HttpRequest, HttpResponse>> {
         context.init_logging_params();
         if let Some(LogConfig { level, format }) = self.before.as_ref() {
-            level.log(format.rewrite_request(&req, context));
+            match level {
+                LogMiddlewareLevel::Trace => {
+                    if enabled!(target: "middleware::log_requests", tracing::Level::TRACE) {
+                        let message = format.rewrite_request(&req, context);
+                        trace!(target: "middleware::log_requests", message);
+                    }
+                }
+                LogMiddlewareLevel::Debug => {
+                    if enabled!(target: "middleware::log_requests", tracing::Level::DEBUG) {
+                        let message = format.rewrite_request(&req, context);
+                        debug!(target: "middleware::log_requests", message);
+                    }
+                }
+                LogMiddlewareLevel::Info => {
+                    if enabled!(target: "middleware::log_requests", tracing::Level::INFO) {
+                        let message = format.rewrite_request(&req, context);
+                        info!(target: "middleware::log_requests", message);
+                    }
+                }
+                LogMiddlewareLevel::Warn => {
+                    if enabled!(target: "middleware::log_requests", tracing::Level::WARN) {
+                        let message = format.rewrite_request(&req, context);
+                        warn!(target: "middleware::log_requests", message);
+                    }
+                }
+                LogMiddlewareLevel::Error => {
+                    if enabled!(target: "middleware::log_requests", tracing::Level::ERROR) {
+                        let message = format.rewrite_request(&req, context);
+                        error!(target: "middleware::log_requests", message);
+                    }
+                }
+            }
         }
 
         Ok(Either::Left(req))
@@ -72,7 +92,38 @@ impl MiddlewareLayer for LogRequests {
 
     async fn after(&self, resp: HttpResponse, context: &mut HttpRequestContext) -> HttpResponse {
         if let Some(LogConfig { level, format }) = self.after.as_ref() {
-            level.log(format.rewrite_response(&resp, context));
+            match level {
+                LogMiddlewareLevel::Trace => {
+                    if enabled!(target: "middleware::log_requests", tracing::Level::TRACE) {
+                        let message = format.rewrite_response(&resp, context);
+                        trace!(target: "middleware::log_requests", message);
+                    }
+                }
+                LogMiddlewareLevel::Debug => {
+                    if enabled!(target: "middleware::log_requests", tracing::Level::DEBUG) {
+                        let message = format.rewrite_response(&resp, context);
+                        debug!(target: "middleware::log_requests", message);
+                    }
+                }
+                LogMiddlewareLevel::Info => {
+                    if enabled!(target: "middleware::log_requests", tracing::Level::INFO) {
+                        let message = format.rewrite_response(&resp, context);
+                        info!(target: "middleware::log_requests", message);
+                    }
+                }
+                LogMiddlewareLevel::Warn => {
+                    if enabled!(target: "middleware::log_requests", tracing::Level::WARN) {
+                        let message = format.rewrite_response(&resp, context);
+                        warn!(target: "middleware::log_requests", message);
+                    }
+                }
+                LogMiddlewareLevel::Error => {
+                    if enabled!(target: "middleware::log_requests", tracing::Level::ERROR) {
+                        let message = format.rewrite_response(&resp, context);
+                        error!(target: "middleware::log_requests", message);
+                    }
+                }
+            }
         }
 
         resp

data/crates/itsi_server/src/server/middleware_stack/middlewares/max_body.rs

@@ -37,7 +37,7 @@ impl MiddlewareLayer for MaxBody {
     async fn after(&self, resp: HttpResponse, context: &mut HttpRequestContext) -> HttpResponse {
         if resp.status() == StatusCode::PAYLOAD_TOO_LARGE {
             self.error_response
-                .to_http_response(context.response_format().clone())
+                .to_http_response(*context.response_format())
                 .await
         } else {
             resp

data/crates/itsi_server/src/server/middleware_stack/middlewares/proxy.rs

@@ -303,7 +303,7 @@ impl MiddlewareLayer for Proxy {
         let url = self.to.rewrite_request(&req, context);
 
         let accept: ResponseFormat = req.accept().into();
-        let error_response = self.error_response.to_http_response(accept.clone()).await;
+        let error_response = self.error_response.to_http_response(accept).await;
 
         let destination = match Url::parse(&url) {
             Ok(dest) => dest,

data/crates/itsi_server/src/server/middleware_stack/middlewares/rate_limit.rs

@@ -6,6 +6,7 @@ use crate::services::rate_limiter::{
 };
 use async_trait::async_trait;
 use either::Either;
+use http::{HeaderName, HeaderValue};
 use magnus::error::Result;
 use serde::Deserialize;
 use std::collections::HashMap;
@@ -24,6 +25,8 @@ pub struct RateLimit {
     pub trusted_proxies: HashMap<String, TokenSource>,
     #[serde(default = "too_many_requests_error_response")]
     pub error_response: ErrorResponse,
+    #[serde(skip)]
+    pub limit_header_value: OnceLock<HeaderValue>,
 }
 
 fn too_many_requests_error_response() -> ErrorResponse {
@@ -38,6 +41,12 @@ pub enum RateLimitKey {
     Parameter(TokenSource),
 }
 
+static X_RATELIMIT_LIMIT: HeaderName = HeaderName::from_static("x-ratelimit-limit");
+static X_RATELIMIT_REMAINING: HeaderName = HeaderName::from_static("x-ratelimit-remaining");
+static X_RATELIMIT_RESET: HeaderName = HeaderName::from_static("x-ratelimit-reset");
+static RETRY_AFTER: HeaderName = HeaderName::from_static("retry-after");
+static ZERO_VALUE: HeaderValue = HeaderValue::from_static("0");
+
 #[async_trait]
 impl MiddlewareLayer for RateLimit {
     async fn initialize(&self) -> Result<()> {
@@ -46,6 +55,9 @@ impl MiddlewareLayer for RateLimit {
         if let Ok(limiter) = get_rate_limiter(&self.store_config).await {
             let _ = self.rate_limiter.set(limiter);
         }
+        self.limit_header_value
+            .set(self.requests.to_string().parse().unwrap())
+            .ok();
         Ok(())
     }
 
@@ -58,8 +70,7 @@ impl MiddlewareLayer for RateLimit {
         let key_value = match &self.key {
             RateLimitKey::SocketAddress => {
                 // Use the socket address from the context
-                if self.trusted_proxies.contains_key(&context.addr) {
-                    let source = self.trusted_proxies.get(&context.addr).unwrap();
+                if let Some(source) = self.trusted_proxies.get(&context.addr) {
                     source.extract_token(&req).unwrap_or(&context.addr)
                 } else {
                     &context.addr
@@ -114,18 +125,20 @@ impl MiddlewareLayer for RateLimit {
                         .error_response
                         .to_http_response(req.accept().into())
                         .await;
+                    let ttl_header_value: HeaderValue = ttl.to_string().parse().unwrap();
+                    response.headers_mut().insert(
+                        X_RATELIMIT_LIMIT.clone(),
+                        self.limit_header_value.get().unwrap().clone(),
+                    );
                     response
                         .headers_mut()
-                        .insert("X-RateLimit-Limit", limit.to_string().parse().unwrap());
-                    response
-                        .headers_mut()
-                        .insert("X-RateLimit-Remaining", "0".parse().unwrap());
+                        .insert(X_RATELIMIT_REMAINING.clone(), ZERO_VALUE.clone());
                     response
                         .headers_mut()
-                        .insert("X-RateLimit-Reset", ttl.to_string().parse().unwrap());
+                        .insert(X_RATELIMIT_RESET.clone(), ttl_header_value.clone());
                     response
                         .headers_mut()
-                        .insert("Retry-After", ttl.to_string().parse().unwrap());
+                        .insert(RETRY_AFTER.clone(), ttl_header_value);
                     Ok(Either::Right(response))
                 }
                 Err(e) => {

data/crates/itsi_server/src/server/middleware_stack/middlewares/ruby_app.rs

@@ -119,11 +119,7 @@ impl MiddlewareLayer for RubyApp {
         if self.sendfile {
             if let Some(sendfile_header) = resp.headers().get("X-Sendfile") {
                 return ROOT_STATIC_FILE_SERVER
-                    .serve_single_abs(
-                        sendfile_header.to_str().unwrap(),
-                        context.accept.clone(),
-                        &[],
-                    )
+                    .serve_single_abs(sendfile_header.to_str().unwrap(), context.accept, &[])
                     .await;
             }
         }

data/crates/itsi_server/src/server/middleware_stack/middlewares/static_response.rs

@@ -21,6 +21,10 @@ pub struct StaticResponse {
     body: Vec<u8>,
     #[serde(skip)]
     header_map: OnceLock<HeaderMap>,
+    #[serde(skip)]
+    body_bytes: OnceLock<Full<Bytes>>,
+    #[serde(skip)]
+    status_code: OnceLock<StatusCode>,
 }
 
 #[async_trait]
@@ -35,6 +39,12 @@ impl MiddlewareLayer for StaticResponse {
         self.header_map
             .set(header_map)
             .map_err(|_| ItsiError::new("Failed to set headers"))?;
+        self.body_bytes
+            .set(Full::new(Bytes::from(self.body.clone())))
+            .map_err(|_| ItsiError::new("Failed to set body bytes"))?;
+        self.status_code
+            .set(StatusCode::from_u16(self.code).unwrap_or(StatusCode::OK))
+            .map_err(|_| ItsiError::new("Failed to set status code"))?;
         Ok(())
     }
 
@@ -43,9 +53,8 @@ impl MiddlewareLayer for StaticResponse {
         _req: HttpRequest,
         _context: &mut HttpRequestContext,
     ) -> Result<Either<HttpRequest, HttpResponse>> {
-        let mut resp = Response::new(BoxBody::new(Full::new(Bytes::from(self.body.clone()))));
-        let status = StatusCode::from_u16(self.code).unwrap_or(StatusCode::OK);
-        *resp.status_mut() = status;
+        let mut resp = Response::new(BoxBody::new(self.body_bytes.get().unwrap().clone()));
+        *resp.status_mut() = *self.status_code.get().unwrap();
         *resp.headers_mut() = self.header_map.get().unwrap().clone();
 
         Ok(Either::Right(resp))

data/crates/itsi_server/src/server/process_worker.rs

@@ -42,7 +42,8 @@ impl Default for ProcessWorker {
     }
 }
 
-static CORE_IDS: LazyLock<Vec<CoreId>> = LazyLock::new(|| core_affinity::get_core_ids().unwrap());
+pub static CORE_IDS: LazyLock<Vec<CoreId>> =
+    LazyLock::new(|| core_affinity::get_core_ids().unwrap());
 
 impl ProcessWorker {
     #[instrument(skip(self, cluster_template), fields(self.worker_id = %self.worker_id))]