use log::debug; use serde::{de::DeserializeOwned, Serialize}; use karyons_core::utils::random_32; use karyons_net::{dial, Conn, Endpoint}; use crate::{ codec::{Codec, CodecConfig}, message, Error, Result, JSONRPC_VERSION, }; /// Represents client config #[derive(Default)] pub struct ClientConfig { pub timeout: Option, } /// Represents an RPC client pub struct Client { codec: Codec, config: ClientConfig, } impl Client { /// Creates a new RPC client. pub fn new(conn: Conn, config: ClientConfig) -> Self { let codec_config = CodecConfig { max_allowed_buffer_size: 0, ..Default::default() }; let codec = Codec::new(conn, codec_config); Self { codec, config } } /// Creates a new RPC client using the provided endpoint. pub async fn new_with_endpoint(endpoint: &Endpoint, config: ClientConfig) -> Result { 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( &self, method: &str, params: T, ) -> Result { let id = serde_json::json!(random_32()); let request = message::Request { jsonrpc: JSONRPC_VERSION.to_string(), id, method: method.to_string(), params: serde_json::json!(params), }; let mut payload = serde_json::to_vec(&request)?; payload.push(b'\n'); self.codec.write_all(&payload).await?; debug!("--> {request}"); let mut buffer = vec![]; if let Some(t) = self.config.timeout { self.codec.read_until_timeout(&mut buffer, t).await?; } else { self.codec.read_until(&mut buffer).await?; }; let response = serde_json::from_slice::(&buffer)?; debug!("<-- {response}"); if let Some(error) = response.error { return Err(Error::CallError(error.code, error.message)); } if response.id.is_none() || response.id.unwrap() != request.id { return Err(Error::InvalidMsg("Invalid response id")); } match response.result { Some(result) => Ok(serde_json::from_value::(result)?), None => Err(Error::InvalidMsg("Invalid response result")), } } }