mb-solana 0.0.10

my solana lib helpers and cli tools
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

use rand::{prelude::{SliceRandom, StdRng}, SeedableRng};
use rust_decimal::Decimal;
use tokio::runtime::Builder;
use typed_builder::TypedBuilder;

use crate::rpc;

#[derive(TypedBuilder)]
pub struct SolBalanceParams {
    pub account: String,

    #[builder(default = vec!["https://api.mainnet-beta.solana.com".to_string()])]
    pub nodes: Vec<String>,

    #[builder(default = vec![])]
    pub proxies: Vec<String>,

    #[builder(default = 5)]
    pub attempts: u8,

    #[builder(default = 10)]
    pub timeout: u8,
}

#[derive(TypedBuilder)]
pub struct SplTokenBalanceParams {
    pub owner: String,
    pub mint: String,

    #[builder(default = vec!["https://api.mainnet-beta.solana.com".to_string()])]
    pub nodes: Vec<String>,

    #[builder(default = vec![])]
    pub proxies: Vec<String>,

    #[builder(default = 5)]
    pub attempts: u8,

    #[builder(default = 10)]
    pub timeout: u8,
}

pub async fn sol_balance(params: SolBalanceParams) -> Option<u64> {
    let mut rng = StdRng::from_entropy();
    for _ in 0..params.attempts {
        let node = params.nodes.choose(&mut rng).unwrap();
        let proxy = params.proxies.choose(&mut rng).map(|p| p.as_str());
        if let Ok(balance) = rpc::sol_balance(node, &params.account, params.timeout, proxy).await {
            return Some(balance);
        }
    }
    None
}

pub fn sol_balance_sync(params: SolBalanceParams) -> Option<u64> {
    Builder::new_multi_thread().enable_all().build().unwrap().block_on(sol_balance(params))
}

pub async fn spl_token_balance(params: SplTokenBalanceParams) -> Option<Decimal> {
    let mut rng = StdRng::from_entropy();
    for _ in 0..params.attempts {
        let node = params.nodes.choose(&mut rng).unwrap();
        let proxy = params.proxies.choose(&mut rng).map(|p| p.as_str());
        if let Ok(balance) = rpc::spl_token_balance(node, &params.owner, &params.mint, params.timeout, proxy).await {
            return Some(balance);
        }
    }
    None
}

pub fn spl_token_balance_sync(params: SplTokenBalanceParams) -> Option<Decimal> {
    Builder::new_multi_thread().enable_all().build().unwrap().block_on(spl_token_balance(params))
}