ghassmo 5 yıl önce
ebeveyn
işleme
e74a4f9ebb
3 değiştirilmiş dosya ile 20 ekleme ve 11 silme
  1. 4 2
      src/bin/demowallet.rs
  2. 1 1
      src/service/mod.rs
  3. 15 8
      src/service/reqrep.rs

+ 4 - 2
src/bin/demowallet.rs

@@ -11,9 +11,11 @@ async fn start(executor: Arc<Executor<'_>>) -> Result<()> {
     client.start().await?;
     println!("connected to a server");
 
-
     let slabstore = client.get_slabstore();
-    let subscriber_task =  executor.spawn(GatewayClient::subscribe(slabstore,"127.0.0.1:4444".parse()?));
+    let subscriber_task = executor.spawn(GatewayClient::subscribe(
+        slabstore,
+        "127.0.0.1:4444".parse()?,
+    ));
 
     println!("subscriber ready");
 

+ 1 - 1
src/service/mod.rs

@@ -2,5 +2,5 @@ pub mod gateway;
 pub mod options;
 pub mod reqrep;
 
-pub use gateway::{ GatewayClient, GatewayService};
+pub use gateway::{GatewayClient, GatewayService};
 pub use options::ProgramOptions;

+ 15 - 8
src/service/reqrep.rs

@@ -56,8 +56,8 @@ impl RepProtocol {
     pub async fn start(
         &mut self,
     ) -> Result<(
-    async_channel::Sender<Reply>,
-    async_channel::Receiver<Request>,
+        async_channel::Sender<Reply>,
+        async_channel::Receiver<Request>,
     )> {
         let addr = addr_to_string(self.addr);
         self.socket.bind(addr.as_str()).await?;
@@ -150,7 +150,7 @@ impl ReqProtocol {
             Ok(reply.get_payload())
         } else {
             Err(crate::Error::ZMQError(
-                    "Couldn't parse ZmqMessage".to_string(),
+                "Couldn't parse ZmqMessage".to_string(),
             ))
         }
     }
@@ -165,16 +165,23 @@ pub struct Publisher {
 impl Publisher {
     pub fn new(addr: SocketAddr, service_name: String) -> Publisher {
         let socket = zeromq::PubSocket::new();
-        Publisher { addr, socket, service_name}
+        Publisher {
+            addr,
+            socket,
+            service_name,
+        }
     }
 
     pub async fn start(&mut self, recv_queue: async_channel::Receiver<Vec<u8>>) -> Result<()> {
         let addr = addr_to_string(self.addr);
         self.socket.bind(addr.as_str()).await?;
-        info!("{} SERVICE PUBLISHER: started - bind to {}", self.service_name, addr);
+        info!(
+            "{} SERVICE PUBLISHER: started - bind to {}",
+            self.service_name, addr
+        );
         loop {
-            let x = recv_queue.recv().await?;
-            self.publish(x).await?;
+            let msg = recv_queue.recv().await?;
+            self.publish(msg).await?;
         }
     }
 
@@ -213,7 +220,7 @@ impl Subscriber {
                 Ok(data)
             }
             None => Err(crate::Error::ZMQError(
-                    "Couldn't parse ZmqMessage".to_string(),
+                "Couldn't parse ZmqMessage".to_string(),
             )),
         }
     }