use super::{RpcxFn, Server};
#[allow(unused_imports)]
use rpcx_protocol::*;
use std::net::TcpStream;
impl Server {
pub fn add_register_plugin(&mut self, p: Box<dyn RegisterPlugin + Send + Sync>) {
let mut plugins = self.register_plugins.write().unwrap();
plugins.push(p);
}
pub fn add_connect_plugin(&mut self, p: Box<dyn ConnectPlugin + Send + Sync>) {
let mut plugins = self.connect_plugins.write().unwrap();
plugins.push(p);
}
}
pub trait RegisterPlugin {
fn register_fn(
&mut self,
service_path: &str,
service_method: &str,
meta: String,
f: RpcxFn,
) -> Result<()>;
}
pub trait ConnectPlugin {
fn connected(&mut self, conn: &TcpStream) -> Result<()>;
}