jsonrpc-core 0.1.0 

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

//! method and notification commands executor

use std::collections::HashMap;
use super::{Params, Value, Error, ErrorCode};

/// Should be used to handle single method call.
pub trait MethodCommand {
	fn execute(&mut self, params: Option<Params>) -> Result<Value, Error>;
}

/// Should be used to handle single notification.
pub trait NotificationCommand {
	fn execute(&mut self, params: Option<Params>);
}

/// Commands executor.
pub struct Commander {
	methods: HashMap<String, Box<MethodCommand>>,
	notifications: HashMap<String, Box<NotificationCommand>>
}

impl Commander {
	pub fn new() -> Self {
		Commander {
			methods: HashMap::new(),
			notifications: HashMap::new()
		}
	}

	pub fn add_method<C>(&mut self, name: &str, command: C) where C: MethodCommand + 'static {
		self.methods.insert(name.to_string(), Box::new(command));
	}

	pub fn add_notification<C>(&mut self, name: &str, command: C) where C: NotificationCommand + 'static {
		self.notifications.insert(name.to_string(), Box::new(command));
	}

	pub fn execute_method(&mut self, name: String, params: Option<Params>) -> Result<Value, Error> {
		match self.methods.get_mut(&name) {
			Some(command) => command.execute(params),
			None => Err(Error::new(ErrorCode::MethodNotFound))
		}
	}

	pub fn execute_notification(&mut self, name: String, params: Option<Params>) {
		if let Some(command) = self.notifications.get_mut(&name) {
			command.execute(params)
		}
	}
}