blockworks-foundation
/
lite-rpc
mirror of https://github.com/blockworks-foundation/lite-rpc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

drain queues

This commit is contained in:
GroovieGermanikus 2023-07-28 14:47:38 +02:00
parent 8ea7428afd
commit 7cae76bb3d
1 changed files with 22 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
quic-forward-proxy/src/proxy.rs
View File

@ -233,11 +233,22 @@ async fn tx_forwarder(tpu_quic_client: TpuQuicClient, mut transaction_channel: R
let packet = receiver.recv().await.unwrap();
assert_eq!(packet.tpu_address, tpu_address, "routing error");
debug!("forwarding transaction batch of size {} to address {}", packet.transactions.len(), packet.tpu_address);
let mut transactions_batch = packet.transactions;
let mut batch_size = 1;
while let Ok(more) = receiver.try_recv() {
transactions_batch.extend(more.transactions);
batch_size += 1;
}
if batch_size > 1 {
debug!("encountered batch of size {}", batch_size);
}
debug!("forwarding transaction batch of size {} to address {}", transactions_batch.len(), packet.tpu_address);
// TODo move send_txs_to_tpu_static to tpu_quic_client
let result = timeout(Duration::from_millis(500),
send_txs_to_tpu_static(&auto_connection, &packet.transactions)).await;
send_txs_to_tpu_static(&auto_connection, &transactions_batch)).await;
// .expect("timeout sending data to TPU node");
debug!("send_txs_to_tpu_static result {:?} - loop over errors", result);
@ -251,8 +262,17 @@ async fn tx_forwarder(tpu_quic_client: TpuQuicClient, mut transaction_channel: R
} // -- new agent
let agent_channel = agents.get(&tpu_address).unwrap();
agent_channel.send(forward_packet).await.unwrap();
let mut batch_size = 1;
while let Ok(more) = transaction_channel.try_recv() {
agent_channel.send(more).await.unwrap();
batch_size += 1;
}
if batch_size > 1 {
debug!("encountered batch of size {}", batch_size);
}
// check if the tpu has already a task+queue running, if not start one, sort+queue packets by tpu address