Start9/rpc-toolkit
2
0
Fork 0
mirror of https://github.com/Start9Labs/rpc-toolkit.git synced 2026-03-26 02:11:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

expose DynMiddleware

Browse Source
This commit is contained in:
Aiden McClelland
2025-12-17 15:50:19 -07:00
parent 2dd2832e04
commit 40de465a76
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/server/http.rs
View File

@@ -14,7 +14,7 @@ use yajrc::{RpcError, RpcMethod};
use crate::server::{RpcRequest, RpcResponse, SingleOrBatchRpcRequest}; use crate::server::{RpcRequest, RpcResponse, SingleOrBatchRpcRequest};
use crate::util::{internal_error, parse_error}; use crate::util::{internal_error, parse_error};
use crate::{HandleAny, Server}; use crate::{Context, HandleAny, Server};
const FALLBACK_ERROR: &str = "{\"error\":{\"code\":-32603,\"message\":\"Internal error\",\"data\":\"Failed to serialize rpc response\"}}"; const FALLBACK_ERROR: &str = "{\"error\":{\"code\":-32603,\"message\":\"Internal error\",\"data\":\"Failed to serialize rpc response\"}}";
@@ -145,7 +145,12 @@ impl<Context: Send + 'static, T: Middleware<Context> + Send + Sync> _Middleware<
} }
} }
struct DynMiddleware<Context>(Box<dyn _Middleware<Context>>); pub struct DynMiddleware<Context>(Box<dyn _Middleware<Context>>);
impl<C: Context> DynMiddleware<C> {
pub fn new<M: Middleware<C>>(middleware: M) -> Self {
Self(Box::new(middleware))
}
}
impl<Context> Clone for DynMiddleware<Context> { impl<Context> Clone for DynMiddleware<Context> {
fn clone(&self) -> Self { fn clone(&self) -> Self {
self.0.dyn_clone() self.0.dyn_clone()
@@ -177,8 +182,7 @@ impl<Context: crate::Context> Server<Context> {
} }
impl<Context: crate::Context> HttpServer<Context> { impl<Context: crate::Context> HttpServer<Context> {
pub fn middleware<T: Middleware<Context>>(mut self, middleware: T) -> Self { pub fn middleware<T: Middleware<Context>>(mut self, middleware: T) -> Self {
self.middleware self.middleware.push_back(DynMiddleware::new(middleware));
.push_back(DynMiddleware(Box::new(middleware)));
self self
} }
async fn process_http_request(&self, mut req: Request) -> Response { async fn process_http_request(&self, mut req: Request) -> Response {