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

aquatic_udp: glommio: read access list on start

Browse source
This commit is contained in:
Joakim Frostegård 2021-10-23 13:04:14 +02:00
parent 119cd0fe46
commit d0be89388c
4 changed files with 29 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aquatic_udp/src/lib/glommio/mod.rs
View file

@ -5,6 +5,7 @@
use std::sync::{atomic::AtomicUsize, Arc};
use aquatic_common::access_list::AccessList;
use glommio::channels::channel_mesh::MeshBuilder;
use glommio::prelude::*;
@ -23,6 +24,12 @@ pub fn run(config: Config) -> anyhow::Result<()> {
});
}
let access_list = if config.access_list.mode.is_on() {
AccessList::create_from_path(&config.access_list.path).expect("Load access list")
} else {
AccessList::default()
};
let num_peers = config.socket_workers + config.request_workers;
let request_mesh_builder = MeshBuilder::partial(num_peers, SHARED_CHANNEL_SIZE);
@ -37,6 +44,7 @@ pub fn run(config: Config) -> anyhow::Result<()> {
let request_mesh_builder = request_mesh_builder.clone();
let response_mesh_builder = response_mesh_builder.clone();
let num_bound_sockets = num_bound_sockets.clone();
let access_list = access_list.clone();
let mut builder = LocalExecutorBuilder::default();
@ -50,6 +58,7 @@ pub fn run(config: Config) -> anyhow::Result<()> {
request_mesh_builder,
response_mesh_builder,
num_bound_sockets,
access_list,
)
.await
});
@ -61,6 +70,7 @@ pub fn run(config: Config) -> anyhow::Result<()> {
let config = config.clone();
let request_mesh_builder = request_mesh_builder.clone();
let response_mesh_builder = response_mesh_builder.clone();
let access_list = access_list.clone();
let mut builder = LocalExecutorBuilder::default();
@ -70,7 +80,13 @@ pub fn run(config: Config) -> anyhow::Result<()> {
}
let executor = builder.spawn(|| async move {
handlers::run_request_worker(config, request_mesh_builder, response_mesh_builder).await
handlers::run_request_worker(
config,
request_mesh_builder,
response_mesh_builder,
access_list,
)
.await
});
executors.push(executor);