ws: rename request workers to swarm workers

This commit is contained in:
Joakim Frostegård 2022-07-04 11:55:53 +02:00
parent 224d50e98b
commit 12fc8bcf1e
5 changed files with 18 additions and 18 deletions

View file

@ -13,12 +13,12 @@ use aquatic_toml_config::TomlConfig;
#[serde(default)] #[serde(default)]
pub struct Config { pub struct Config {
/// Socket workers receive requests from the socket, parse them and send /// Socket workers receive requests from the socket, parse them and send
/// them on to the request workers. They then receive responses from the /// them on to the swarm workers. They then receive responses from the
/// request workers, encode them and send them back over the socket. /// swarm workers, encode them and send them back over the socket.
pub socket_workers: usize, pub socket_workers: usize,
/// Request workers receive a number of requests from socket workers, /// Swarm workers receive a number of requests from socket workers,
/// generate responses and send them back to the socket workers. /// generate responses and send them back to the socket workers.
pub request_workers: usize, pub swarm_workers: usize,
pub log_level: LogLevel, pub log_level: LogLevel,
pub network: NetworkConfig, pub network: NetworkConfig,
pub protocol: ProtocolConfig, pub protocol: ProtocolConfig,
@ -32,7 +32,7 @@ impl Default for Config {
fn default() -> Self { fn default() -> Self {
Self { Self {
socket_workers: 1, socket_workers: 1,
request_workers: 1, swarm_workers: 1,
log_level: LogLevel::default(), log_level: LogLevel::default(),
network: NetworkConfig::default(), network: NetworkConfig::default(),
protocol: ProtocolConfig::default(), protocol: ProtocolConfig::default(),

View file

@ -32,7 +32,7 @@ pub fn run(config: Config) -> ::anyhow::Result<()> {
update_access_list(&config.access_list, &state.access_list)?; update_access_list(&config.access_list, &state.access_list)?;
let num_peers = config.socket_workers + config.request_workers; let num_peers = config.socket_workers + config.swarm_workers;
let request_mesh_builder = MeshBuilder::partial(num_peers, SHARED_IN_CHANNEL_SIZE); let request_mesh_builder = MeshBuilder::partial(num_peers, SHARED_IN_CHANNEL_SIZE);
let response_mesh_builder = MeshBuilder::partial(num_peers, SHARED_IN_CHANNEL_SIZE * 16); let response_mesh_builder = MeshBuilder::partial(num_peers, SHARED_IN_CHANNEL_SIZE * 16);
@ -59,7 +59,7 @@ pub fn run(config: Config) -> ::anyhow::Result<()> {
let placement = get_worker_placement( let placement = get_worker_placement(
&config.cpu_pinning, &config.cpu_pinning,
config.socket_workers, config.socket_workers,
config.request_workers, config.swarm_workers,
WorkerIndex::SocketWorker(i), WorkerIndex::SocketWorker(i),
)?; )?;
let builder = LocalExecutorBuilder::new(placement).name("socket"); let builder = LocalExecutorBuilder::new(placement).name("socket");
@ -82,7 +82,7 @@ pub fn run(config: Config) -> ::anyhow::Result<()> {
executors.push(executor); executors.push(executor);
} }
for i in 0..(config.request_workers) { for i in 0..(config.swarm_workers) {
let sentinel = sentinel.clone(); let sentinel = sentinel.clone();
let config = config.clone(); let config = config.clone();
let state = state.clone(); let state = state.clone();
@ -92,14 +92,14 @@ pub fn run(config: Config) -> ::anyhow::Result<()> {
let placement = get_worker_placement( let placement = get_worker_placement(
&config.cpu_pinning, &config.cpu_pinning,
config.socket_workers, config.socket_workers,
config.request_workers, config.swarm_workers,
WorkerIndex::SwarmWorker(i), WorkerIndex::SwarmWorker(i),
)?; )?;
let builder = LocalExecutorBuilder::new(placement).name("request"); let builder = LocalExecutorBuilder::new(placement).name("request");
let executor = builder let executor = builder
.spawn(move || async move { .spawn(move || async move {
workers::request::run_request_worker( workers::swarm::run_swarm_worker(
sentinel, sentinel,
config, config,
state, state,
@ -117,7 +117,7 @@ pub fn run(config: Config) -> ::anyhow::Result<()> {
set_affinity_for_util_worker( set_affinity_for_util_worker(
&config.cpu_pinning, &config.cpu_pinning,
config.socket_workers, config.socket_workers,
config.request_workers, config.swarm_workers,
)?; )?;
} }

View file

@ -1,2 +1,2 @@
pub mod request;
pub mod socket; pub mod socket;
pub mod swarm;

View file

@ -352,7 +352,7 @@ impl ConnectionReader {
) )
.await .await
.unwrap(); .unwrap();
::log::info!("sent message to request worker"); ::log::info!("sent message to swarm worker");
} else { } else {
self.send_error_response("Info hash not allowed".into(), Some(info_hash)) self.send_error_response("Info hash not allowed".into(), Some(info_hash))
.await?; .await?;
@ -404,7 +404,7 @@ impl ConnectionReader {
.send_to(consumer_index, (meta, in_message)) .send_to(consumer_index, (meta, in_message))
.await .await
.unwrap(); .unwrap();
::log::info!("sent message to request worker"); ::log::info!("sent message to swarm worker");
} }
} }
} }
@ -541,7 +541,7 @@ impl ConnectionWriter {
} }
fn calculate_in_message_consumer_index(config: &Config, info_hash: InfoHash) -> usize { fn calculate_in_message_consumer_index(config: &Config, info_hash: InfoHash) -> usize {
(info_hash.0[0] as usize) % config.request_workers (info_hash.0[0] as usize) % config.swarm_workers
} }
fn create_tcp_listener( fn create_tcp_listener(

View file

@ -127,7 +127,7 @@ impl TorrentMaps {
} }
} }
pub async fn run_request_worker( pub async fn run_swarm_worker(
_sentinel: PanicSentinel, _sentinel: PanicSentinel,
config: Config, config: Config,
state: State, state: State,
@ -223,14 +223,14 @@ async fn handle_request_stream<S>(
}; };
for (meta, out_message) in out_messages.drain(..) { for (meta, out_message) in out_messages.drain(..) {
::log::info!("request worker trying to send OutMessage to socket worker"); ::log::info!("swarm worker trying to send OutMessage to socket worker");
out_message_senders out_message_senders
.send_to(meta.out_message_consumer_id.0, (meta, out_message)) .send_to(meta.out_message_consumer_id.0, (meta, out_message))
.await .await
.expect("failed sending out_message to socket worker"); .expect("failed sending out_message to socket worker");
::log::info!("request worker sent OutMessage to socket worker"); ::log::info!("swarm worker sent OutMessage to socket worker");
} }
}, },
) )