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

run batch execution async

This commit is contained in:
Maximilian Schneider 2023-04-18 23:20:13 +00:00
parent 465e628a64
commit 8b27a138c9
2 changed files with 27 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/market_markers.rs
View File

@ -21,13 +21,13 @@ use solana_sdk::{
compute_budget, hash::Hash, instruction::Instruction, message::Message, signature::Keypair,
signer::Signer, transaction::Transaction,
};
use tokio::{sync::RwLock, task::JoinHandle};
use tokio::{sync::RwLock, task::{JoinHandle, self}};
use crate::{
helpers::{to_sdk_instruction, to_sp_pk},
mango::AccountKeys,
states::{PerpMarketCache, TransactionSendRecord},
tpu_manager::TpuManager,
tpu_manager::{TpuManager, self},
};
pub fn create_ask_bid_transaction(
@ -195,10 +195,13 @@ pub async fn send_mm_transactions(
batch_to_send.push((tx, record));
}
let tpu_manager = tpu_manager.clone();
task::spawn(async move{
if !tpu_manager.send_transaction_batch(&batch_to_send).await {
println!("sending failed on tpu client");
}
});
if !tpu_manager.send_transaction_batch(&batch_to_send).await {
println!("sending failed on tpu client");
}
}
}

34
src/tpu_manager.rs
View File

@ -144,29 +144,33 @@ impl TpuManager {
tpu_client.send_transaction(transaction).await
}
pub async fn send_transaction_batch(
pub async fn send_transaction_batch(
&self,
batch: &Vec<(solana_sdk::transaction::Transaction, TransactionSendRecord)>,
) -> bool {
let tpu_client = self.get_tpu_client().await;
for (tx, record) in batch {
self.stats.inc_send(&record.keeper_instruction);
self.stats
.inc_send(&record.keeper_instruction);
let tx_sent_record = self.tx_send_record.clone();
let sent = tx_sent_record.send(record.clone());
if sent.is_err() {
warn!(
"sending error on channel : {}",
sent.err().unwrap().to_string()
);
}
let tx_sent_record = self.tx_send_record.clone();
let sent = tx_sent_record.send(record.clone());
if sent.is_err() {
warn!(
"sending error on channel : {}",
sent.err().unwrap().to_string()
);
}
}
tpu_client.try_send_wire_transaction_batch(batch.iter().map(|(tx, _)| serialize(tx).expect("serialization should succeed")).collect()).await.is_ok()
tpu_client
.try_send_wire_transaction_batch(
batch
.iter()
.map(|(tx, _)| serialize(tx).expect("serialization should succeed"))
.collect(),
)
.await
.is_ok()
}
}