add RpcModule::remove (#1416)

* rpc: remove method * remove internal remove method --------- Co-authored-by: Niklas Adolfsson <[email protected]>
paritytech · Jun 26, 2024 · b7d601f · b7d601f
1 parent 2e71ce8
commit b7d601f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/core/src/server/rpc_module.rs b/core/src/server/rpc_module.rs
@@ -552,6 +552,14 @@ impl<Context: Send + Sync + 'static> RpcModule<Context> {
 		)
 	}
 
+	/// Removes the method if it exists.
+	///
+	/// Be aware that a subscription consist of two methods, `subscribe` and `unsubscribe` and
+	/// it's the caller responsibility to remove both `subscribe` and `unsubscribe` methods for subscriptions.
+	pub fn remove_method(&mut self, method_name: &'static str) -> Option<MethodCallback> {
+		self.methods.mut_callbacks().remove(method_name)
+	}
+
 	/// Register a new asynchronous RPC method, which computes the response with the given callback.
 	///
 	/// ## Examples