jsonrpc: spread out comments and clean up

1 files changed, 5 insertions, 4 deletions

diff --git a/jsonrpc/src/server/pubsub_service.rs b/jsonrpc/src/server/pubsub_service.rs
index 5b3b50b..08d1bbb 100644
--- a/jsonrpc/src/server/pubsub_service.rs
+++ b/jsonrpc/src/server/pubsub_service.rs
@@ -1,12 +1,12 @@
-use std::{future::Future, pin::Pin};
+use std::{future::Future, pin::Pin, sync::Arc};
 
 use crate::Result;
 
-use super::channel::ArcChannel;
+use super::channel::Channel;
 
 /// Represents the RPC method
 pub type PubSubRPCMethod<'a> =
-    Box<dyn Fn(ArcChannel, String, serde_json::Value) -> PubSubRPCMethodOutput<'a> + Send + 'a>;
+    Box<dyn Fn(Arc<Channel>, String, serde_json::Value) -> PubSubRPCMethodOutput<'a> + Send + 'a>;
 type PubSubRPCMethodOutput<'a> =
     Pin<Box<dyn Future<Output = Result<serde_json::Value>> + Send + Sync + 'a>>;
 
@@ -51,7 +51,8 @@ macro_rules! impl_pubsub_rpc_service {
                 match name {
                 $(
                     stringify!($m) => {
-                        Some(Box::new(move |chan: karyon_jsonrpc::ArcChannel, method: String, params: serde_json::Value| {
+                        Some(Box::new(
+                            move |chan: std::sync::Arc<karyon_jsonrpc::Channel>, method: String, params: serde_json::Value| {
                             Box::pin(self.$m(chan, method, params))
                         }))
                     }