Skip generator if non-unique transactions (#25039)

This commit is contained in:
sakridge 2022-05-10 09:22:24 -05:00 committed by GitHub
parent e27e9fb538
commit 6cfa19e2e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions

View File

@ -240,16 +240,16 @@ fn run_dos(
data.resize(params.data_size, 0);
}
DataType::Transaction => {
let tp = params.transaction_params;
let tp = params.transaction_params.clone();
info!("{:?}", tp);
transaction_generator = Some(TransactionGenerator::new(tp));
let tx = transaction_generator
.as_mut()
.unwrap()
.generate(payer, &rpc_client);
let mut tg = TransactionGenerator::new(tp);
let tx = tg.generate(payer, &rpc_client);
info!("{:?}", tx);
data = bincode::serialize(&tx).unwrap();
if params.transaction_params.unique_transactions {
transaction_generator = Some(tg);
}
}
DataType::GetAccountInfo => {}
DataType::GetProgramAccounts => {}
@ -287,7 +287,7 @@ fn run_dos(
}
if let Some(tg) = transaction_generator.as_mut() {
let tx = tg.generate(payer, &rpc_client);
info!("{:?}", tx);
debug!("{:?}", tx);
data = bincode::serialize(&tx).unwrap();
}
let res = socket.send_to(&data, target);
@ -350,7 +350,7 @@ struct DosClientParameters {
transaction_params: TransactionParams,
}
#[derive(Args, Serialize, Deserialize, Debug, Default)]
#[derive(Clone, Args, Serialize, Deserialize, Debug, Default)]
#[clap(rename_all = "kebab-case")]
struct TransactionParams {
#[clap(