1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
extern crate hyper;
extern crate unicase;
extern crate jsonrpc_core as jsonrpc;
use std::thread;
use std::sync::Arc;
use std::io::Read;
use hyper::header::{Headers, Allow, ContentType, AccessControlAllowHeaders};
use hyper::method::Method;
use unicase::UniCase;
use self::jsonrpc::{IoHandler};
pub use hyper::header::AccessControlAllowOrigin;
struct ServerHandler {
jsonrpc_handler: Arc<IoHandler>,
cors_domain: AccessControlAllowOrigin,
}
impl ServerHandler {
fn new(jsonrpc_handler: Arc<IoHandler>, cors_domain: AccessControlAllowOrigin) -> Self {
ServerHandler {
jsonrpc_handler: jsonrpc_handler,
cors_domain: cors_domain
}
}
fn response_headers(&self) -> Headers {
let mut headers = Headers::new();
headers.set(
Allow(vec![
Method::Options, Method::Post
])
);
headers.set(ContentType::json());
headers.set(
AccessControlAllowHeaders(vec![
UniCase("origin".to_owned()),
UniCase("content-type".to_owned()),
UniCase("accept".to_owned()),
])
);
headers.set(self.cors_domain.clone());
headers
}
}
impl hyper::server::Handler for ServerHandler {
fn handle(&self, mut req: hyper::server::Request, mut res: hyper::server::Response) {
match req.method {
Method::Options => {
*res.headers_mut() = self.response_headers();
},
Method::Post => {
let mut body = String::new();
if let Err(_) = req.read_to_string(&mut body) {
*res.status_mut() = hyper::status::StatusCode::MethodNotAllowed;
return;
}
if let Some(response) = self.jsonrpc_handler.handle_request(&body) {
*res.headers_mut() = self.response_headers();
res.send(response.as_ref()).unwrap();
}
},
_ => *res.status_mut() = hyper::status::StatusCode::MethodNotAllowed
}
}
}
pub struct Server {
jsonrpc_handler: Arc<IoHandler>,
}
impl Server {
pub fn new(jsonrpc_handler: Arc<IoHandler>) -> Self {
Server {
jsonrpc_handler: jsonrpc_handler,
}
}
pub fn start(&self, addr: &str, cors_domain: AccessControlAllowOrigin, threads: usize) {
hyper::Server::http(addr).unwrap().handle_threads(ServerHandler::new(self.jsonrpc_handler.clone(), cors_domain), threads).unwrap();
}
pub fn start_async(&self, addr: &str, cors_domain: AccessControlAllowOrigin, threads: usize) {
let address = addr.to_owned();
let handler = self.jsonrpc_handler.clone();
thread::Builder::new().name("jsonrpc_http".to_string()).spawn(move || {
hyper::Server::http(address.as_ref() as &str).unwrap().handle_threads(ServerHandler::new(handler, cors_domain), threads).unwrap();
}).unwrap();
}
}