From 0c6781ba4b0248b53079ce3e66d04822a028a5f6 Mon Sep 17 00:00:00 2001
From: Igor Katson <igor.katson@gmail.com>
Date: Sun, 19 Nov 2023 20:59:01 +0000
Subject: [PATCH] Saving

---
 crates/librqbit/src/torrent_state.rs | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/crates/librqbit/src/torrent_state.rs b/crates/librqbit/src/torrent_state.rs
index 8c50566..b70f49b 100644
--- a/crates/librqbit/src/torrent_state.rs
+++ b/crates/librqbit/src/torrent_state.rs
@@ -866,12 +866,9 @@ impl PeerConnectionHandler for PeerHandler {
             Message::Choke => self.on_i_am_choked(self.addr),
             Message::Unchoke => self.on_i_am_unchoked(self.addr),
             Message::Interested => self.on_peer_interested(self.addr),
-            Message::Piece(piece) => {
-                timeit("on_received_piece", || {
-                    self.on_received_piece(self.addr, piece)
-                        .context("on_received_piece")
-                })?;
-            }
+            Message::Piece(piece) => self
+                .on_received_piece(self.addr, piece)
+                .context("on_received_piece")?,
             Message::KeepAlive => {
                 debug!("keepalive received");
             }