jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 19:09:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(rpc): add more helper functions to modle regsitry (#1380)

Browse Source 
Co-authored-by: Georgios Konstantopoulos <me@gakonst.com>
This commit is contained in:
Matthias Seitz
2023-02-15 21:07:09 +01:00
committed by GitHub
parent a08a7b0110
commit ba20759062
1 changed files with 65 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
crates/rpc/rpc-builder/src/lib.rs
View File

@ -265,7 +265,7 @@ impl RpcModuleSelection {
Network: NetworkInfo + Peers + Clone + 'static,
{
let mut registry = RethModuleRegistry::new(client, pool, network);
registry.module(self)
registry.module_for(self)
}
/// Returns an iterator over all configured [RethRpcModule]
@ -357,27 +357,86 @@ pub struct RethModuleRegistry<Client, Pool, Network> {
// === impl RethModuleRegistry ===
impl<Client, Pool, Network> RethModuleRegistry<Client, Pool, Network> {
/// Creates a new, empty instance.
pub fn new(client: Client, pool: Pool, network: Network) -> Self {
Self { client, pool, network, eth_api: None, modules: Default::default() }
}
/// Returns all installed methods
pub fn methods(&self) -> Vec<Methods> {
self.modules.values().cloned().collect()
}
/// Returns a merged RpcModule
pub fn module(&self) -> RpcModule<()> {
let mut module = RpcModule::new(());
for methods in self.modules.values().cloned() {
module.merge(methods).expect("No conflicts");
}
module
}
}
impl<Client, Pool, Network> RethModuleRegistry<Client, Pool, Network>
where
Network: NetworkInfo + Peers + Clone + 'static,
{
/// Register Admin Namespace
pub fn register_admin(&mut self) -> &mut Self {
self.modules
.insert(RethRpcModule::Admin, AdminApi::new(self.network.clone()).into_rpc().into());
self
}
/// Register Web3 Namespace
pub fn register_web3(&mut self) -> &mut Self {
self.modules
.insert(RethRpcModule::Web3, Web3Api::new(self.network.clone()).into_rpc().into());
self
}
}
impl<Client, Pool, Network> RethModuleRegistry<Client, Pool, Network>
where
Client: BlockProvider + StateProviderFactory + Clone + 'static,
Pool: TransactionPool + Clone + 'static,
Network: NetworkInfo + Peers + Clone + 'static,
{
/// Creates a new, empty instance
pub fn new(client: Client, pool: Pool, network: Network) -> Self {
Self { client, pool, network, eth_api: None, modules: Default::default() }
/// Register Eth Namespace
pub fn register_eth(&mut self) -> &mut Self {
let eth_api = self.eth_api();
self.modules.insert(RethRpcModule::Eth, eth_api.into_rpc().into());
self
}
/// Register Debug Namespace
pub fn register_debug(&mut self) -> &mut Self {
let eth_api = self.eth_api();
self.modules.insert(RethRpcModule::Debug, DebugApi::new(eth_api).into_rpc().into());
self
}
/// Register Net Namespace
pub fn register_net(&mut self) -> &mut Self {
let eth_api = self.eth_api();
self.modules.insert(
RethRpcModule::Net,
NetApi::new(self.network.clone(), eth_api).into_rpc().into(),
);
self
}
/// Helper function to create a [RpcModule] if it's not `None`
fn maybe_module(&mut self, config: Option<&RpcModuleSelection>) -> Option<RpcModule<()>> {
let config = config?;
let module = self.module(config);
let module = self.module_for(config);
Some(module)
}
/// Populates a new [RpcModule] based on the selected [RethRpcModule]s in the given
/// [RpcModuleSelection]
pub fn module(&mut self, config: &RpcModuleSelection) -> RpcModule<()> {
pub fn module_for(&mut self, config: &RpcModuleSelection) -> RpcModule<()> {
let mut module = RpcModule::new(());
let all_methods = self.reth_methods(config.iter_selection());
for methods in all_methods {