middlewares/
middlewares.rs

1extern crate bitconch_jsonrpc_core as jsonrpc_core;
2
3use std::time::Instant;
4use std::sync::atomic::{self, AtomicUsize};
5use jsonrpc_core::*;
6use jsonrpc_core::futures::Future;
7use jsonrpc_core::futures::future::Either;
8
9#[derive(Clone, Debug)]
10struct Meta(usize);
11impl Metadata for Meta {}
12
13#[derive(Default)]
14struct MyMiddleware(AtomicUsize);
15impl Middleware<Meta> for MyMiddleware {
16	type Future = FutureResponse;
17
18	fn on_request<F, X>(&self, request: Request, meta: Meta, next: F) -> Either<Self::Future, X> where
19		F: FnOnce(Request, Meta) -> X + Send,
20		X: Future<Item=Option<Response>, Error=()> + 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::A(Box::new(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| {
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(request, Meta(headers)).wait().unwrap(),
46		Some(response.to_owned())
47	);
48}