middlewares/
middlewares.rs

1use jsonrpc_core::futures_util::{future::Either, FutureExt};
2use jsonrpc_core::*;
3use std::future::Future;
4use std::sync::atomic::{self, AtomicUsize};
5use std::time::Instant;
6
7#[derive(Clone, Debug)]
8struct Meta(usize);
9impl Metadata for Meta {}
10
11#[derive(Default)]
12struct MyMiddleware(AtomicUsize);
13impl Middleware<Meta> for MyMiddleware {
14	type Future = FutureResponse;
15	type CallFuture = middleware::NoopCallFuture;
16
17	fn on_request<F, X>(&self, request: Request, meta: Meta, next: F) -> Either<Self::Future, X>
18	where
19		F: FnOnce(Request, Meta) -> X + Send,
20		X: Future<Output = Option<Response>> + Send + 'static,
21	{
22		let start = Instant::now();
23		let request_number = self.0.fetch_add(1, atomic::Ordering::SeqCst);
24		println!("Processing request {}: {:?}, {:?}", request_number, request, meta);
25
26		Either::Left(Box::pin(next(request, meta).map(move |res| {
27			println!("Processing took: {:?}", start.elapsed());
28			res
29		})))
30	}
31}
32
33pub fn main() {
34	let mut io = MetaIoHandler::with_middleware(MyMiddleware::default());
35
36	io.add_method_with_meta("say_hello", |_params: Params, meta: Meta| async move {
37		Ok(Value::String(format!("Hello World: {}", meta.0)))
38	});
39
40	let request = r#"{"jsonrpc": "2.0", "method": "say_hello", "params": [42, 23], "id": 1}"#;
41	let response = r#"{"jsonrpc":"2.0","result":"Hello World: 5","id":1}"#;
42
43	let headers = 5;
44	assert_eq!(
45		io.handle_request_sync(request, Meta(headers)),
46		Some(response.to_owned())
47	);
48}
middlewares/middlewares.rs

middlewares/
middlewares.rs
