YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-01 02:05:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

aquatic_http: support multiple request workers

Browse source
This commit is contained in:
Joakim Frostegård 2020-08-13 01:50:18 +02:00
parent 875eb122c7
commit 2b4cd2dea1
3 changed files with 24 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

28
aquatic_http/src/lib/lib.rs
View file

@ -22,10 +22,21 @@ pub const APP_NAME: &str = "aquatic_http: HTTP/TLS BitTorrent tracker";
pub fn run(config: Config) -> anyhow::Result<()> {
let opt_tls_acceptor = create_tls_acceptor(&config.network.tls)?;
let state = State::default();
start_workers(config.clone(), state.clone())?;
loop {
::std::thread::sleep(Duration::from_secs(config.cleaning.interval));
tasks::clean_torrents(&state);
}
}
pub fn start_workers(config: Config, state: State) -> anyhow::Result<()> {
let opt_tls_acceptor = create_tls_acceptor(&config.network.tls)?;
let (request_channel_sender, request_channel_receiver) = ::crossbeam_channel::unbounded();
let mut out_message_senders = Vec::new();
@ -96,11 +107,14 @@ pub fn run(config: Config) -> anyhow::Result<()> {
let response_channel_sender = ResponseChannelSender::new(out_message_senders);
{
for i in 0..config.request_workers {
let config = config.clone();
let state = state.clone();
let request_channel_receiver = request_channel_receiver.clone();
let response_channel_sender = response_channel_sender.clone();
let wakers = wakers.clone();
Builder::new().name("request".to_string()).spawn(move || {
Builder::new().name(format!("request-{:02}", i + 1)).spawn(move || {
handler::run_request_worker(
config,
state,
@ -111,10 +125,6 @@ pub fn run(config: Config) -> anyhow::Result<()> {
})?;
}
loop {
::std::thread::sleep(Duration::from_secs(config.cleaning.interval));
tasks::clean_torrents(&state);
}
Ok(())
}