jsonrpc_macros/

delegates.rs

use std::sync::Arc;
use std::collections::HashMap;

use jsonrpc_core::{Params, Value, Error};
use jsonrpc_core::{Metadata, RemoteProcedure, RpcMethod, RpcNotification};
use jsonrpc_core::futures::{self, BoxFuture, Future};

use jsonrpc_pubsub::{self, SubscriptionId, Subscriber, PubSubMetadata};

type Data = Result<Value, Error>;
type AsyncData = BoxFuture<Value, Error>;

struct DelegateMethod<T, F> {
	delegate: Arc<T>,
	closure: F,
}

impl<T, M, F> RpcMethod<M> for DelegateMethod<T, F> where
	F: Fn(&T, Params) -> Data + 'static,
	F: Send + Sync + 'static,
	T: Send + Sync + 'static,
	M: Metadata,
{
	fn call(&self, params: Params, _meta: M) -> AsyncData {
		let closure = &self.closure;
		futures::done(closure(&self.delegate, params)).boxed()
	}
}

struct DelegateAsyncMethod<T, F> {
	delegate: Arc<T>,
	closure: F,
}

impl<T, M, F> RpcMethod<M> for DelegateAsyncMethod<T, F> where
	F: Fn(&T, Params) -> AsyncData,
	F: Send + Sync + 'static,
	T: Send + Sync + 'static,
	M: Metadata,
{
	fn call(&self, params: Params, _meta: M) -> AsyncData {
		let closure = &self.closure;
		closure(&self.delegate, params)
	}
}

struct DelegateMethodWithMeta<T, F> {
	delegate: Arc<T>,
	closure: F,
}

impl<T, M, F> RpcMethod<M> for DelegateMethodWithMeta<T, F> where
	T: Send + Sync + 'static,
	M: Metadata,
	F: Fn(&T, Params, M) -> AsyncData + Send + Sync + 'static,
{
	fn call(&self, params: Params, meta: M) -> AsyncData {
		let closure = &self.closure;
		closure(&self.delegate, params, meta)
	}
}

struct DelegateNotification<T, F> {
	delegate: Arc<T>,
	closure: F,
}

impl<T, M, F> RpcNotification<M> for DelegateNotification<T, F> where
	F: Fn(&T, Params) + 'static,
	F: Send + Sync + 'static,
	T: Send + Sync + 'static,
	M: Metadata,
{
	fn execute(&self, params: Params, _meta: M) {
		let closure = &self.closure;
		closure(&self.delegate, params)
	}
}

struct DelegateSubscribe<T, F> {
	delegate: Arc<T>,
	closure: F,
}

impl<T, M, F> jsonrpc_pubsub::SubscribeRpcMethod<M> for DelegateSubscribe<T, F> where
	T: Send + Sync + 'static,
	M: PubSubMetadata,
	F: Fn(&T, Params, M, Subscriber) + Send + Sync + 'static,
{
	fn call(&self, params: Params, meta: M, subscriber: Subscriber) {
		let closure = &self.closure;
		closure(&self.delegate, params, meta, subscriber)
	}
}

struct DelegateUnsubscribe<T, F> {
	delegate: Arc<T>,
	closure: F,
}

impl<T, F> jsonrpc_pubsub::UnsubscribeRpcMethod for DelegateUnsubscribe<T, F> where
	T: Send + Sync + 'static,
	F: Fn(&T, SubscriptionId) -> AsyncData + Send + Sync + 'static,
{
	fn call(&self, id: SubscriptionId) -> AsyncData {
		let closure = &self.closure;
		closure(&self.delegate, id)
	}
}

/// A set of RPC methods and notifications tied to single `delegate` struct.
pub struct IoDelegate<T, M = ()> where
	T: Send + Sync + 'static,
	M: Metadata,
{
	delegate: Arc<T>,
	methods: HashMap<String, RemoteProcedure<M>>,
}

impl<T, M> IoDelegate<T, M> where
	T: Send + Sync + 'static,
	M: Metadata,
{
	/// Creates new `IoDelegate`
	pub fn new(delegate: Arc<T>) -> Self {
		IoDelegate {
			delegate: delegate,
			methods: HashMap::new(),
		}
	}

	/// Adds an alias to existing method.
	/// NOTE: Aliases are not transitive, i.e. you cannot create alias to an alias.
	pub fn add_alias(&mut self, from: &str, to: &str) {
		self.methods.insert(from.into(), RemoteProcedure::Alias(to.into()));
	}

	/// Adds sync method to the delegate.
	pub fn add_method<F>(&mut self, name: &str, method: F) where
		F: Fn(&T, Params) -> Data,
		F: Send + Sync + 'static,
	{
		self.methods.insert(name.into(), RemoteProcedure::Method(Arc::new(
			DelegateMethod {
				delegate: self.delegate.clone(),
				closure: method,
			}
		)));
	}

	/// Adds async method to the delegate.
	pub fn add_async_method<F>(&mut self, name: &str, method: F) where
		F: Fn(&T, Params) -> AsyncData,
		F: Send + Sync + 'static,
	{
		self.methods.insert(name.into(), RemoteProcedure::Method(Arc::new(
			DelegateAsyncMethod {
				delegate: self.delegate.clone(),
				closure: method,
			}
		)));
	}

	/// Adds async method with metadata to the delegate.
	pub fn add_method_with_meta<F>(&mut self, name: &str, method: F) where
		F: Fn(&T, Params, M) -> AsyncData,
		F: Send + Sync + 'static,
	{
		self.methods.insert(name.into(), RemoteProcedure::Method(Arc::new(
			DelegateMethodWithMeta {
				delegate: self.delegate.clone(),
				closure: method,
			}
		)));
	}

	/// Adds notification to the delegate.
	pub fn add_notification<F>(&mut self, name: &str, notification: F) where
		F: Fn(&T, Params),
		F: Send + Sync + 'static,
	{
		self.methods.insert(name.into(), RemoteProcedure::Notification(Arc::new(
			DelegateNotification {
				delegate: self.delegate.clone(),
				closure: notification,
			}
		)));
	}
}

impl<T, M> IoDelegate<T, M> where
	T: Send + Sync + 'static,
	M: PubSubMetadata,
{
	/// Adds subscription to the delegate.
	pub fn add_subscription<Sub, Unsub>(
		&mut self,
		name: &str,
		subscribe: (&str, Sub),
		unsubscribe: (&str, Unsub),
	) where
		Sub: Fn(&T, Params, M, Subscriber),
		Sub: Send + Sync + 'static,
		Unsub: Fn(&T, SubscriptionId) -> AsyncData,
		Unsub: Send + Sync + 'static,
	{
		let (sub, unsub) = jsonrpc_pubsub::new_subscription(
			name,
			DelegateSubscribe {
				delegate: self.delegate.clone(),
				closure: subscribe.1,
			},
			DelegateUnsubscribe {
				delegate: self.delegate.clone(),
				closure: unsubscribe.1,
			}
		);
		self.add_method_with_meta(subscribe.0, move |_, params, meta| sub.call(params, meta));
		self.add_method_with_meta(unsubscribe.0, move |_, params, meta| unsub.call(params, meta));
	}
}

impl<T, M> Into<HashMap<String, RemoteProcedure<M>>> for IoDelegate<T, M> where
	T: Send + Sync + 'static,
	M: Metadata,
{
	fn into(self) -> HashMap<String, RemoteProcedure<M>> {
		self.methods
	}
}