66 lines
2.2 KiB
Rust
Executable File
66 lines
2.2 KiB
Rust
Executable File
use anyhow::anyhow;
|
|
use solana_hash::Hash;
|
|
use solana_sdk::{instruction::Instruction, signature::Keypair, signer::Signer};
|
|
use solana_system_interface::instruction::advance_nonce_account;
|
|
|
|
use crate::common::nonce_cache::NonceCache;
|
|
|
|
/// 添加nonce消费指令到指令集合中
|
|
///
|
|
/// 只有提供了nonce_pubkey时才使用nonce功能
|
|
/// 如果nonce被锁定、已使用或未准备好，将返回错误
|
|
/// 成功时会锁定并标记nonce为已使用
|
|
pub fn add_nonce_instruction(
|
|
instructions: &mut Vec<Instruction>,
|
|
payer: &Keypair,
|
|
) -> Result<(), anyhow::Error> {
|
|
let nonce_cache = NonceCache::get_instance();
|
|
let nonce_info = nonce_cache.get_nonce_info();
|
|
|
|
// 只检查nonce_account是否存在
|
|
if let Some(nonce_pubkey) = nonce_info.nonce_account {
|
|
// 暂不加锁
|
|
// if nonce_info.lock {
|
|
// return Err(anyhow!("Nonce is locked"));
|
|
// }
|
|
if nonce_info.used {
|
|
return Err(anyhow!("Nonce is used"));
|
|
}
|
|
if nonce_info.current_nonce == Hash::default() {
|
|
return Err(anyhow!("Nonce is not ready"));
|
|
}
|
|
// if nonce_info.next_buy_time == 0 || chrono::Utc::now().timestamp() < nonce_info.next_buy_time {
|
|
// return Err(anyhow!("Nonce is not ready"));
|
|
// }
|
|
// 加锁 - 暂不加锁
|
|
// nonce_cache.lock();
|
|
|
|
// 创建Solana系统nonce推进指令 - 使用系统程序ID
|
|
let nonce_advance_ix = advance_nonce_account(&nonce_pubkey, &payer.pubkey());
|
|
|
|
instructions.push(nonce_advance_ix);
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// 获取用于交易的blockhash
|
|
/// 如果使用了nonce账户，返回nonce中的blockhash，否则返回传入的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
|
|
} else {
|
|
recent_blockhash
|
|
}
|
|
}
|
|
|
|
/// 检查是否使用nonce账户
|
|
pub fn is_using_nonce() -> bool {
|
|
let nonce_cache = NonceCache::get_instance();
|
|
let nonce_info = nonce_cache.get_nonce_info();
|
|
nonce_info.nonce_account.is_some()
|
|
}
|