jsonrpc: Enhance the API and add support for TCP, Unix, and TLS protocols.

3 files changed, 21 insertions, 38 deletions

diff --git a/jsonrpc/src/client.rs b/jsonrpc/src/client.rs
index 939d177..0061002 100644
--- a/jsonrpc/src/client.rs
+++ b/jsonrpc/src/client.rs
@@ -2,7 +2,7 @@ use log::debug;
 use serde::{de::DeserializeOwned, Serialize};
 
 use karyons_core::util::random_32;
-use karyons_net::{dial, Conn, Endpoint};
+use karyons_net::Conn;
 
 use crate::{
     codec::{Codec, CodecConfig},
@@ -22,7 +22,7 @@ pub struct Client {
 }
 
 impl Client {
-    /// Creates a new RPC client.
+    /// Creates a new RPC client by passing a Tcp, Unix, or Tls connection.
     pub fn new(conn: Conn, config: ClientConfig) -> Self {
         let codec_config = CodecConfig {
             max_allowed_buffer_size: 0,
@@ -32,17 +32,6 @@ impl Client {
         Self { codec, config }
     }
 
-    /// Creates a new RPC client using the provided endpoint.
-    pub async fn new_with_endpoint(endpoint: &Endpoint, config: ClientConfig) -> Result<Self> {
-        let conn = dial(endpoint).await?;
-        let codec_config = CodecConfig {
-            max_allowed_buffer_size: 0,
-            ..Default::default()
-        };
-        let codec = Codec::new(conn, codec_config);
-        Ok(Self { codec, config })
-    }
-
     /// Calls the named method, waits for the response, and returns the result.
     pub async fn call<T: Serialize + DeserializeOwned, V: DeserializeOwned>(
         &self,
diff --git a/jsonrpc/src/lib.rs b/jsonrpc/src/lib.rs
index f73b5e6..65fb38f 100644
--- a/jsonrpc/src/lib.rs
+++ b/jsonrpc/src/lib.rs
@@ -1,4 +1,5 @@
-//! A fast and lightweight async [JSONRPC 2.0](https://www.jsonrpc.org/specification) implementation.
+//! A fast and lightweight async implementation of [JSON-RPC
+//! 2.0](https://www.jsonrpc.org/specification), supporting the Tcp and Unix protocols.
 //!
 //! # Example
 //!
@@ -6,6 +7,7 @@
 //! use std::sync::Arc;
 //!
 //! use serde_json::Value;
+//! use smol::net::{TcpStream, TcpListener};
 //!
 //! use karyons_jsonrpc::{JsonRPCError, Server, Client, register_service, ServerConfig, ClientConfig};
 //!
@@ -23,9 +25,9 @@
 //!     let ex = Arc::new(smol::Executor::new());
 //!
 //!     // Creates a new server
-//!     let endpoint = "tcp://127.0.0.1:60000".parse().unwrap();
+//!     let listener = TcpListener::bind("127.0.0.1:60000").await.unwrap();
 //!     let config = ServerConfig::default();
-//!     let server = Server::new_with_endpoint(&endpoint, config, ex.clone()).await.unwrap();
+//!     let server = Server::new(listener.into(), config, ex.clone());
 //!
 //!     // Register the HelloWorld service
 //!     register_service!(HelloWorld, say_hello);
@@ -39,9 +41,9 @@
 //! async {
 //!
 //!     // Creates a new client
-//!     let endpoint = "tcp://127.0.0.1:60000".parse().unwrap();
+//!     let conn = TcpStream::connect("127.0.0.1:60000").await.unwrap();
 //!     let config = ClientConfig::default();
-//!     let client = Client::new_with_endpoint(&endpoint, config).await.unwrap();
+//!     let client = Client::new(conn.into(), config);
 //!
 //!     let result: String = client.call("HelloWorld.say_hello", "world".to_string()).await.unwrap();
 //! };
@@ -55,12 +57,13 @@ pub mod message;
 mod server;
 mod service;
 
-pub const JSONRPC_VERSION: &str = "2.0";
-
-use error::{Error, Result};
-
 pub use client::{Client, ClientConfig};
 pub use codec::CodecConfig;
 pub use error::Error as JsonRPCError;
 pub use server::{Server, ServerConfig};
 pub use service::{RPCMethod, RPCService};
+
+pub use karyons_net::Endpoint;
+
+const JSONRPC_VERSION: &str = "2.0";
+use error::{Error, Result};
diff --git a/jsonrpc/src/server.rs b/jsonrpc/src/server.rs
index 05ef7da..0038e89 100644
--- a/jsonrpc/src/server.rs
+++ b/jsonrpc/src/server.rs
@@ -7,13 +7,14 @@ use karyons_core::{
     async_util::{TaskGroup, TaskResult},
     Executor,
 };
-use karyons_net::{listen, Conn, Endpoint, Listener};
+
+use karyons_net::{Conn, Listener};
 
 use crate::{
     codec::{Codec, CodecConfig},
     message,
     service::RPCService,
-    Error, Result, JSONRPC_VERSION,
+    Endpoint, Error, Result, JSONRPC_VERSION,
 };
 
 /// RPC server config
@@ -31,7 +32,7 @@ pub struct Server<'a> {
 }
 
 impl<'a> Server<'a> {
-    /// Creates a new RPC server.
+    /// Creates a new RPC server by passing a listener. It supports Tcp, Unix, and Tls.
     pub fn new(listener: Box<dyn Listener>, config: ServerConfig, ex: Executor<'a>) -> Arc<Self> {
         Arc::new(Self {
             listener,
@@ -41,19 +42,9 @@ impl<'a> Server<'a> {
         })
     }
 
-    /// Creates a new RPC server using the provided endpoint.
-    pub async fn new_with_endpoint(
-        endpoint: &Endpoint,
-        config: ServerConfig,
-        ex: Executor<'a>,
-    ) -> Result<Arc<Self>> {
-        let listener = listen(endpoint).await?;
-        Ok(Arc::new(Self {
-            listener,
-            services: RwLock::new(HashMap::new()),
-            task_group: TaskGroup::new(ex),
-            config,
-        }))
+    /// Returns the local endpoint.
+    pub fn local_endpoint(&self) -> Result<Endpoint> {
+        self.listener.local_endpoint().map_err(Error::KaryonsNet)
     }
 
     /// Starts the RPC server