vecna
/
lox
mirror of https://gitlab.torproject.org/vecna/lox.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Incorporate impl Stream for ResourceStream changes from rdsys-api-backend

This commit is contained in:
onyinyang 2023-03-22 19:14:56 -04:00
parent 47f645a4f9
commit a2d50e29d8
No known key found for this signature in database
GPG Key ID: 156A6435430C2036
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
crates/lox-distributor/src/main.rs
View File

@ -1,4 +1,5 @@
use futures::future;
use futures::StreamExt;
use hyper::{
body,
body::Bytes,
@ -30,6 +31,7 @@ use std::{
sync::{Arc, Mutex},
time::Duration,
};
use tokio::{
signal, spawn,
sync::{broadcast, mpsc, oneshot},
@ -398,11 +400,11 @@ async fn rdsys_stream(
// in the config.json file.
// TODO: ensure this stream gracefully shutdowns on the ctrl_c command.
async fn rdsys_sender(rtype: ResourceInfo, tx: mpsc::Sender<ResourceDiff>) {
let rstream = start_stream(rtype.endpoint, rtype.name, rtype.token, rtype.types)
let mut rstream = start_stream(rtype.endpoint, rtype.name, rtype.token, rtype.types)
.await
.expect("rdsys stream initialization failed. Start rdsys or check config.json");
sleep(Duration::from_millis(1)).await;
for diff in rstream {
while let Some(diff) = rstream.next().await {
println!("Received diff: {:?}", diff); //send this through a channel
tx.send(diff).await.unwrap();
sleep(Duration::from_secs(1)).await;