From cb17fd9382310bb2fcd1b229599ccc7b09f650ec Mon Sep 17 00:00:00 2001 From: voidlizard Date: Thu, 14 Nov 2024 10:51:32 +0300 Subject: [PATCH] code cleanup --- hbs2-peer/app/BlockDownloadNew.hs | 18 ++++++++---------- hbs2-peer/app/RPC2.hs | 2 +- 2 files changed, 9 insertions(+), 11 deletions(-) diff --git a/hbs2-peer/app/BlockDownloadNew.hs b/hbs2-peer/app/BlockDownloadNew.hs index fb17ae15..d8fd7335 100644 --- a/hbs2-peer/app/BlockDownloadNew.hs +++ b/hbs2-peer/app/BlockDownloadNew.hs @@ -366,7 +366,7 @@ downloadFromPeerRec t bu0 cache env h0 peer = do bu <- lift $ getCurrentBurst bm t0 <- getTimeCoarse - w <- lift $ downloadFromPeer t bu cache env (coerce h) peer + w <- lift $ downloadFromPeer bu cache env (coerce h) peer t1 <- getTimeCoarse let dt = toMicroSeconds $ TimeoutTS (t1 - t0) atomically $ modifyTVar toq ( dt : ) @@ -402,20 +402,18 @@ downloadFromPeerRec t bu0 cache env h0 peer = do pure $ Right () -downloadFromPeer :: forall e t cache m . ( e ~ L4Proto - , MonadUnliftIO m - , IsTimeout t - , BlockSizeCache e cache - ) - => Timeout t - -> Int +downloadFromPeer :: forall e cache m . ( e ~ L4Proto + , MonadUnliftIO m + , BlockSizeCache e cache + ) + => Int -> cache -> PeerEnv e -> Hash HbSync -> Peer e -> m (Either (DownloadError e) ByteString) -downloadFromPeer t bu cache env h peer = liftIO $ withPeerM env do +downloadFromPeer bu cache env h peer = liftIO $ withPeerM env do pd@PeerData{..} <- find (KnownPeerKey peer) id >>= orThrow (UnknownPeerError peer) @@ -832,7 +830,7 @@ downloadDispatcher brains env = flip runContT pure do bu <- lift $ getCurrentBurst bm t0 <- getTimeCoarse - r <- lift $ downloadFromPeer (TimeoutSec 10) bu (KnownSize size) env (coerce hx) p + r <- lift $ downloadFromPeer bu (KnownSize size) env (coerce hx) p t1 <- getTimeCoarse case r of diff --git a/hbs2-peer/app/RPC2.hs b/hbs2-peer/app/RPC2.hs index eb560493..f40620f3 100644 --- a/hbs2-peer/app/RPC2.hs +++ b/hbs2-peer/app/RPC2.hs @@ -150,7 +150,7 @@ instance (e ~ L4Proto, MonadUnliftIO m, HasRpcContext PeerAPI RPC2Context m) => peer <- either (const $ exit (mkSym "error:invalid-address")) pure peer' - what <- lift $ downloadFromPeer defChunkWaitMax 4 rpcBrains rpcPeerEnv (coerce blk) peer + what <- lift $ downloadFromPeer 50 rpcBrains rpcPeerEnv (coerce blk) peer case what of Left e -> pure $ mkList @C [ mkSym "error" , mkStr (show e) ]