sgxiang
116 lines
3.1 KiB
Rust
Executable File
116 lines
3.1 KiB
Rust
Executable File
use std::sync::Arc;
|
|
|
|
use solana_client::rpc_client::RpcClient;
|
|
use solana_sdk::{commitment_config::CommitmentConfig, pubkey::Pubkey, signature::Keypair};
|
|
use serde::Deserialize;
|
|
use crate::{constants::pumpfun::trade::{DEFAULT_BUY_TIP_FEE, DEFAULT_COMPUTE_UNIT_LIMIT, DEFAULT_COMPUTE_UNIT_PRICE, DEFAULT_SELL_TIP_FEE}, swqos::FeeClient};
|
|
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub enum FeeType {
|
|
Jito,
|
|
NextBlock,
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct Cluster {
|
|
pub rpc_url: String,
|
|
pub block_engine_url: String,
|
|
pub nextblock_url: String,
|
|
pub nextblock_auth_token: String,
|
|
pub zeroslot_url: String,
|
|
pub zeroslot_auth_token: String,
|
|
pub nozomi_url: String,
|
|
pub nozomi_auth_token: String,
|
|
pub use_jito: bool,
|
|
pub use_nextblock: bool,
|
|
pub use_zeroslot: bool,
|
|
pub use_nozomi: bool,
|
|
pub priority_fee: PriorityFee,
|
|
pub commitment: CommitmentConfig,
|
|
pub lookup_table_key: Option<Pubkey>,
|
|
pub use_rpc: bool,
|
|
}
|
|
|
|
impl Cluster {
|
|
pub fn new(
|
|
rpc_url: String,
|
|
block_engine_url:
|
|
String, nextblock_url:
|
|
String, nextblock_auth_token:
|
|
String, zeroslot_url: String,
|
|
zeroslot_auth_token: String,
|
|
nozomi_url: String,
|
|
nozomi_auth_token: String,
|
|
priority_fee: PriorityFee,
|
|
commitment: CommitmentConfig,
|
|
use_jito: bool,
|
|
use_nextblock: bool,
|
|
use_zeroslot: bool,
|
|
use_nozomi: bool,
|
|
lookup_table_key: Option<Pubkey>,
|
|
use_rpc: bool,
|
|
) -> Self {
|
|
Self {
|
|
rpc_url,
|
|
block_engine_url,
|
|
nextblock_url,
|
|
nextblock_auth_token,
|
|
zeroslot_url,
|
|
zeroslot_auth_token,
|
|
nozomi_url,
|
|
nozomi_auth_token,
|
|
priority_fee,
|
|
commitment,
|
|
use_jito,
|
|
use_nextblock,
|
|
use_zeroslot,
|
|
use_nozomi,
|
|
lookup_table_key,
|
|
use_rpc,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Deserialize, Clone, PartialEq)]
|
|
pub struct PriorityFee {
|
|
pub unit_limit: u32,
|
|
pub unit_price: u64,
|
|
pub rpc_unit_limit: u32,
|
|
pub rpc_unit_price: u64,
|
|
pub buy_tip_fee: f64,
|
|
pub buy_tip_fees: Vec<f64>,
|
|
pub sell_tip_fee: f64,
|
|
}
|
|
|
|
impl Default for PriorityFee {
|
|
fn default() -> Self {
|
|
Self {
|
|
unit_limit: DEFAULT_COMPUTE_UNIT_LIMIT,
|
|
unit_price: DEFAULT_COMPUTE_UNIT_PRICE,
|
|
rpc_unit_limit: 0,
|
|
rpc_unit_price: 0,
|
|
buy_tip_fee: DEFAULT_BUY_TIP_FEE,
|
|
buy_tip_fees: vec![],
|
|
sell_tip_fee: DEFAULT_SELL_TIP_FEE
|
|
}
|
|
}
|
|
}
|
|
|
|
pub type SolanaRpcClient = solana_client::nonblocking::rpc_client::RpcClient;
|
|
|
|
pub struct MethodArgs {
|
|
pub payer: Arc<Keypair>,
|
|
pub rpc: Arc<RpcClient>,
|
|
pub nonblocking_rpc: Arc<SolanaRpcClient>,
|
|
pub jito_client: Arc<FeeClient>,
|
|
}
|
|
|
|
impl MethodArgs {
|
|
pub fn new(payer: Arc<Keypair>, rpc: Arc<RpcClient>, nonblocking_rpc: Arc<SolanaRpcClient>, jito_client: Arc<FeeClient>) -> Self {
|
|
Self { payer, rpc, nonblocking_rpc, jito_client }
|
|
}
|
|
}
|
|
|
|
pub type AnyResult<T> = anyhow::Result<T>;
|
|
|