From 74e3a51fbe69b469998a64dea35ed91138d98a78 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20Frosteg=C3=A5rd?= <joakim.frostegard@gmail.com>
Date: Tue, 7 Apr 2020 13:50:10 +0200
Subject: [PATCH] aquatic: add and use Config.poll_event_capacity

---
 aquatic/src/lib/common.rs  | 1 -
 aquatic/src/lib/config.rs  | 2 ++
 aquatic/src/lib/network.rs | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/aquatic/src/lib/common.rs b/aquatic/src/lib/common.rs
index 0c953fb..b4fba40 100644
--- a/aquatic/src/lib/common.rs
+++ b/aquatic/src/lib/common.rs
@@ -9,7 +9,6 @@ use indexmap::IndexMap;
 pub use bittorrent_udp::types::*;
 
 
-pub const EVENT_CAPACITY: usize = 4096;
 pub const MAX_PACKET_SIZE: usize = 4096;
 
 
diff --git a/aquatic/src/lib/config.rs b/aquatic/src/lib/config.rs
index b486247..18cb9a5 100644
--- a/aquatic/src/lib/config.rs
+++ b/aquatic/src/lib/config.rs
@@ -6,6 +6,7 @@ pub struct Config {
     pub address: SocketAddr,
     pub num_threads: usize,
     pub recv_buffer_size: usize,
+    pub poll_event_capacity: usize,
     pub max_scrape_torrents: u8,
     pub max_response_peers: usize,
     pub statistics_interval: u64,
@@ -17,6 +18,7 @@ impl Default for Config {
         Self {
             address: SocketAddr::from(([127, 0, 0, 1], 3000)),
             num_threads: 4,
+            poll_event_capacity: 4096,
             recv_buffer_size: 4096 * 16,
             max_scrape_torrents: 255,
             max_response_peers: 255,
diff --git a/aquatic/src/lib/network.rs b/aquatic/src/lib/network.rs
index 0860840..5ecc495 100644
--- a/aquatic/src/lib/network.rs
+++ b/aquatic/src/lib/network.rs
@@ -62,7 +62,7 @@ pub fn run_event_loop(
         .register(&mut socket, Token(token_num), interests)
         .unwrap();
 
-    let mut events = Events::with_capacity(EVENT_CAPACITY);
+    let mut events = Events::with_capacity(config.poll_event_capacity);
 
     let mut connect_requests: Vec<(ConnectRequest, SocketAddr)> = Vec::with_capacity(1024);
     let mut announce_requests: Vec<(AnnounceRequest, SocketAddr)> = Vec::with_capacity(1024);