fix wasm32, cleanup

eigerco · Oct 2, 2023 · 9435a78 · 9435a78
1 parent e542fdd
commit 9435a78
Showing 1 changed file with 4 additions and 11 deletions.
diff --git a/node/src/exchange/server.rs b/node/src/exchange/server.rs
@@ -14,6 +14,7 @@ use tracing::{instrument, trace};
 
 use crate::exchange::utils::{ExtendedHeaderExt, HeaderRequestExt, HeaderResponseExt};
 use crate::exchange::{ReqRespBehaviour, ResponseType};
+use crate::executor::spawn;
 use crate::store::Store;
 
 pub(super) struct ExchangeServerHandler<S, R = ReqRespBehaviour>
@@ -96,8 +97,6 @@ where
     }
 
     pub fn poll(&mut self, _: &mut Context<'_>, sender: &mut R) -> Poll<()> {
-        println!(".");
-
         while let Ok((channel, response)) = self.rx.try_recv() {
             sender.send_response(channel, response);
             //return Poll::Ready(());
@@ -109,27 +108,22 @@ where
     fn handle_request_current_head(&mut self, channel: R::Channel) {
         let store = self.store.clone();
         let tx = self.tx.clone();
-        println!("head");
 
-        let j = tokio::spawn(async move {
-            println!("inhead");
+        spawn(async move {
             let response = store
                 .get_head()
                 .await
                 .map(|head| head.to_header_response())
                 .unwrap_or_else(|_| HeaderResponse::not_found());
 
-            println!("got res");
             let _ = tx.send((channel, vec![response]));
         });
-
-        println!("{j:?}");
     }
 
     fn handle_request_by_hash(&mut self, channel: R::Channel, hash: Vec<u8>) {
         let store = self.store.clone();
         let tx = self.tx.clone();
-        tokio::spawn(async move {
+        spawn(async move {
             let response = if let Ok(hash) = Hash::from_bytes(Algorithm::Sha256, &hash) {
                 store
                     .get_by_hash(&hash)
@@ -148,7 +142,7 @@ where
         let store = self.store.clone();
         let tx = self.tx.clone();
 
-        tokio::spawn(async move {
+        spawn(async move {
             let mut responses = vec![];
             for i in origin..origin + amount {
                 let response = store
@@ -382,7 +376,6 @@ mod tests {
         type Channel = ();
 
         fn send_response(&mut self, _channel: Self::Channel, response: ResponseType) {
-            println!("sending");
             if let Some(sender) = self.0.take() {
                 let _ = sender.send(response);
             }