108 lines
3.6 KiB
Rust
Executable File
108 lines
3.6 KiB
Rust
Executable File
use solana_hash::Hash;
|
|
use solana_sdk::{pubkey::Pubkey, signature::Keypair};
|
|
use std::sync::Arc;
|
|
|
|
use crate::swqos::FeeClient;
|
|
use crate::trading::{core::params::PumpSwapParams, factory::Protocol, BuyParams, TradeFactory};
|
|
use crate::{common::PriorityFee, SolanaRpcClient};
|
|
|
|
// Constants for compute budget
|
|
// Increased from 64KB to 256KB to handle larger transactions
|
|
const MAX_LOADED_ACCOUNTS_DATA_SIZE_LIMIT: u32 = 256 * 1024;
|
|
|
|
// Buy tokens from a Pumpswap pool
|
|
pub async fn buy(
|
|
rpc: Arc<SolanaRpcClient>,
|
|
payer: Arc<Keypair>,
|
|
mint: Pubkey,
|
|
creator: Pubkey,
|
|
amount_sol: u64,
|
|
slippage_basis_points: Option<u64>,
|
|
priority_fee: PriorityFee,
|
|
lookup_table_key: Option<Pubkey>,
|
|
recent_blockhash: Hash,
|
|
// 可选（必须全部传）
|
|
pool: Option<Pubkey>,
|
|
pool_base_token_account: Option<Pubkey>,
|
|
pool_quote_token_account: Option<Pubkey>,
|
|
user_base_token_account: Option<Pubkey>,
|
|
user_quote_token_account: Option<Pubkey>,
|
|
) -> Result<(), anyhow::Error> {
|
|
// 创建执行器
|
|
let executor = TradeFactory::create_executor(Protocol::PumpSwap);
|
|
// 创建协议特定参数
|
|
let protocol_params = Box::new(PumpSwapParams {
|
|
pool: pool,
|
|
pool_base_token_account: pool_base_token_account,
|
|
pool_quote_token_account: pool_quote_token_account,
|
|
user_base_token_account: user_base_token_account,
|
|
user_quote_token_account: user_quote_token_account,
|
|
});
|
|
// 创建买入参数
|
|
let buy_params = BuyParams {
|
|
rpc: Some(rpc.clone()),
|
|
payer: payer,
|
|
mint: mint,
|
|
creator: creator,
|
|
amount_sol: amount_sol,
|
|
slippage_basis_points: slippage_basis_points,
|
|
priority_fee: priority_fee,
|
|
lookup_table_key: lookup_table_key,
|
|
recent_blockhash: recent_blockhash,
|
|
data_size_limit: MAX_LOADED_ACCOUNTS_DATA_SIZE_LIMIT,
|
|
protocol_params,
|
|
};
|
|
// 执行买入
|
|
executor.buy(buy_params).await?;
|
|
Ok(())
|
|
}
|
|
|
|
// Buy tokens using a MEV service
|
|
pub async fn buy_with_tip(
|
|
rpc: Arc<SolanaRpcClient>,
|
|
fee_clients: Vec<Arc<FeeClient>>,
|
|
payer: Arc<Keypair>,
|
|
mint: Pubkey,
|
|
creator: Pubkey,
|
|
amount_sol: u64,
|
|
slippage_basis_points: Option<u64>,
|
|
priority_fee: PriorityFee,
|
|
lookup_table_key: Option<Pubkey>,
|
|
recent_blockhash: Hash,
|
|
// 可选（必须全部传）
|
|
pool: Option<Pubkey>,
|
|
pool_base_token_account: Option<Pubkey>,
|
|
pool_quote_token_account: Option<Pubkey>,
|
|
user_base_token_account: Option<Pubkey>,
|
|
user_quote_token_account: Option<Pubkey>,
|
|
) -> Result<(), anyhow::Error> {
|
|
// 创建执行器
|
|
let executor = TradeFactory::create_executor(Protocol::PumpSwap);
|
|
// 创建协议特定参数
|
|
let protocol_params = Box::new(PumpSwapParams {
|
|
pool: pool,
|
|
pool_base_token_account: pool_base_token_account,
|
|
pool_quote_token_account: pool_quote_token_account,
|
|
user_base_token_account: user_base_token_account,
|
|
user_quote_token_account: user_quote_token_account,
|
|
});
|
|
// 创建买入参数
|
|
let buy_params = BuyParams {
|
|
rpc: Some(rpc.clone()),
|
|
payer: payer,
|
|
mint: mint,
|
|
creator: creator,
|
|
amount_sol: amount_sol,
|
|
slippage_basis_points: slippage_basis_points,
|
|
priority_fee: priority_fee,
|
|
lookup_table_key: lookup_table_key,
|
|
recent_blockhash: recent_blockhash,
|
|
data_size_limit: MAX_LOADED_ACCOUNTS_DATA_SIZE_LIMIT,
|
|
protocol_params,
|
|
};
|
|
let buy_with_tip_params = buy_params.with_tip(fee_clients);
|
|
// 执行买入
|
|
executor.buy_with_tip(buy_with_tip_params).await?;
|
|
Ok(())
|
|
}
|