feat: add flexible nonce parameter support
- Add nonce_account and current_nonce to trade parameters - Remove hardcoded NonceCache dependency from nonce_manager - Update examples and documentation for new nonce usage - Fix nonce documentation errors and improve clarity
This commit is contained in:
@@ -1,16 +1,37 @@
|
||||
use std::sync::Arc;
|
||||
|
||||
use solana_sdk::{message::AddressLookupTableAccount, pubkey::Pubkey};
|
||||
|
||||
use crate::common::address_lookup_cache::get_address_lookup_table_account;
|
||||
use crate::common::{
|
||||
address_lookup_cache::{get_address_lookup_table_account, AddressLookupTableCache},
|
||||
SolanaRpcClient,
|
||||
};
|
||||
|
||||
/// Get address lookup table account list
|
||||
/// If lookup_table_key is provided, get the corresponding account, otherwise return empty list
|
||||
pub async fn get_address_lookup_table_accounts(
|
||||
rpc: Option<Arc<SolanaRpcClient>>,
|
||||
lookup_table_key: Option<Pubkey>,
|
||||
) -> Vec<AddressLookupTableAccount> {
|
||||
match lookup_table_key {
|
||||
Some(key) => {
|
||||
let account = get_address_lookup_table_account(&key).await;
|
||||
vec![account]
|
||||
if account.addresses.len() == 0 {
|
||||
if rpc.is_some() {
|
||||
let _ = AddressLookupTableCache::get_instance()
|
||||
.set_address_lookup_table(rpc.unwrap(), &key)
|
||||
.await;
|
||||
let new_account = get_address_lookup_table_account(&key).await;
|
||||
if new_account.addresses.len() == 0 {
|
||||
return Vec::new();
|
||||
} else {
|
||||
return vec![new_account];
|
||||
}
|
||||
} else {
|
||||
return Vec::new();
|
||||
}
|
||||
}
|
||||
return vec![account];
|
||||
}
|
||||
None => Vec::new(),
|
||||
}
|
||||
|
||||
@@ -1,10 +1,7 @@
|
||||
use anyhow::anyhow;
|
||||
use solana_hash::Hash;
|
||||
use solana_sdk::{instruction::Instruction, signature::Keypair, signer::Signer};
|
||||
use solana_sdk::{instruction::Instruction, pubkey::Pubkey, signature::Keypair, signer::Signer};
|
||||
use solana_system_interface::instruction::advance_nonce_account;
|
||||
|
||||
use crate::common::nonce_cache::NonceCache;
|
||||
|
||||
/// Add nonce advance instruction to the instruction set
|
||||
///
|
||||
/// Nonce functionality is only used when nonce_pubkey is provided
|
||||
@@ -13,36 +10,25 @@ use crate::common::nonce_cache::NonceCache;
|
||||
pub fn add_nonce_instruction(
|
||||
instructions: &mut Vec<Instruction>,
|
||||
payer: &Keypair,
|
||||
nonce_account: Option<Pubkey>,
|
||||
current_nonce: Option<Hash>,
|
||||
) -> Result<(), anyhow::Error> {
|
||||
let nonce_cache = NonceCache::get_instance();
|
||||
let nonce_info = nonce_cache.get_nonce_info();
|
||||
|
||||
// Only check if nonce_account exists
|
||||
if let Some(nonce_pubkey) = nonce_info.nonce_account {
|
||||
if nonce_info.used {
|
||||
return Err(anyhow!("Nonce is used"));
|
||||
}
|
||||
if nonce_info.current_nonce == Hash::default() {
|
||||
return Err(anyhow!("Nonce is not ready"));
|
||||
}
|
||||
|
||||
// Create Solana system nonce advance instruction - using system program ID
|
||||
let nonce_advance_ix = advance_nonce_account(&nonce_pubkey, &payer.pubkey());
|
||||
|
||||
if nonce_account.is_some() && current_nonce.is_some() {
|
||||
let nonce_advance_ix = advance_nonce_account(&nonce_account.unwrap(), &payer.pubkey());
|
||||
instructions.push(nonce_advance_ix);
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Get blockhash for transaction
|
||||
/// If nonce account is used, return blockhash from nonce, otherwise return the provided recent_blockhash
|
||||
pub fn get_transaction_blockhash(recent_blockhash: Hash) -> Hash {
|
||||
let nonce_cache = NonceCache::get_instance();
|
||||
let nonce_info = nonce_cache.get_nonce_info();
|
||||
|
||||
if nonce_info.nonce_account.is_some() {
|
||||
nonce_info.current_nonce
|
||||
pub fn get_transaction_blockhash(
|
||||
recent_blockhash: Hash,
|
||||
nonce_account: Option<Pubkey>,
|
||||
current_nonce: Option<Hash>,
|
||||
) -> Hash {
|
||||
if nonce_account.is_some() && current_nonce.is_some() {
|
||||
current_nonce.unwrap()
|
||||
} else {
|
||||
recent_blockhash
|
||||
}
|
||||
|
||||
@@ -16,11 +16,12 @@ use super::{
|
||||
compute_budget_manager::compute_budget_instructions,
|
||||
nonce_manager::{add_nonce_instruction, get_transaction_blockhash},
|
||||
};
|
||||
use crate::trading::MiddlewareManager;
|
||||
use crate::{common::SolanaRpcClient, trading::MiddlewareManager};
|
||||
|
||||
/// Build standard RPC transaction
|
||||
pub async fn build_transaction(
|
||||
payer: Arc<Keypair>,
|
||||
rpc: Option<Arc<SolanaRpcClient>>,
|
||||
unit_limit: u32,
|
||||
unit_price: u64,
|
||||
business_instructions: Vec<Instruction>,
|
||||
@@ -33,11 +34,15 @@ pub async fn build_transaction(
|
||||
with_tip: bool,
|
||||
tip_account: &Pubkey,
|
||||
tip_amount: f64,
|
||||
nonce_account: Option<Pubkey>,
|
||||
current_nonce: Option<Hash>,
|
||||
) -> Result<VersionedTransaction, anyhow::Error> {
|
||||
let mut instructions = Vec::with_capacity(business_instructions.len() + 5);
|
||||
|
||||
// Add nonce instruction
|
||||
if let Err(e) = add_nonce_instruction(&mut instructions, payer.as_ref()) {
|
||||
if let Err(e) =
|
||||
add_nonce_instruction(&mut instructions, payer.as_ref(), nonce_account, current_nonce)
|
||||
{
|
||||
return Err(e);
|
||||
}
|
||||
|
||||
@@ -62,11 +67,11 @@ pub async fn build_transaction(
|
||||
instructions.extend(business_instructions);
|
||||
|
||||
// Get blockhash for transaction
|
||||
let blockhash =
|
||||
if is_buy { get_transaction_blockhash(recent_blockhash) } else { recent_blockhash };
|
||||
let blockhash = get_transaction_blockhash(recent_blockhash, nonce_account, current_nonce);
|
||||
|
||||
// Get address lookup table accounts
|
||||
let address_lookup_table_accounts = get_address_lookup_table_accounts(lookup_table_key).await;
|
||||
let address_lookup_table_accounts =
|
||||
get_address_lookup_table_accounts(rpc, lookup_table_key).await;
|
||||
|
||||
// Build transaction
|
||||
build_versioned_transaction(
|
||||
|
||||
Reference in New Issue
Block a user