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

aquatic_udp glommio: improve AccessList update code

Browse source
This commit is contained in:
Joakim Frostegård 2021-10-19 22:58:59 +02:00
parent 38617c70f4
commit bfcdc1c842
1 changed files with 28 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

46
aquatic_udp/src/lib/glommio/handlers.rs
View file

@ -28,26 +28,10 @@ pub async fn run_request_worker(
let torrents = Rc::new(RefCell::new(TorrentMaps::default())); let torrents = Rc::new(RefCell::new(TorrentMaps::default()));
let access_list = Rc::new(RefCell::new(AccessList::default())); let access_list = Rc::new(RefCell::new(AccessList::default()));
// Periodically clean torrents and update access list
TimerActionRepeat::repeat(enclose!((config, torrents, access_list) move || { TimerActionRepeat::repeat(enclose!((config, torrents, access_list) move || {
enclose!((config, torrents, access_list) move || async move { enclose!((config, torrents, access_list) move || async move {
if config.access_list.mode.is_on(){ update_access_list(config.clone(), access_list.clone()).await;
let access_list_file = BufferedFile::open(config.access_list.path).await.unwrap();
let mut reader = StreamReaderBuilder::new(access_list_file).build();
loop {
let mut buf = String::with_capacity(42);
match reader.read_line(&mut buf).await {
Ok(_) => {
access_list.borrow_mut().insert_from_line(&buf).unwrap() // FIXME
},
Err(err) => {
}
}
}
}
torrents.borrow_mut().clean(&config, &*access_list.borrow()); torrents.borrow_mut().clean(&config, &*access_list.borrow());
@ -106,3 +90,29 @@ async fn handle_request_stream<S>(
yield_if_needed().await; yield_if_needed().await;
} }
} }
pub async fn update_access_list(
config: Config,
access_list: Rc<RefCell<AccessList>>,
){
if config.access_list.mode.is_on(){
let access_list_file = BufferedFile::open(config.access_list.path).await.unwrap();
let mut reader = StreamReaderBuilder::new(access_list_file).build();
loop {
let mut buf = String::with_capacity(42);
match reader.read_line(&mut buf).await {
Ok(_) => {
access_list.borrow_mut().insert_from_line(&buf);
},
Err(err) => {
break;
}
}
yield_if_needed().await;
}
}
}