gmsol-model 0.9.0

GMX-Solana is an extension of GMX on the Solana blockchain.
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

use num_traits::{CheckedAdd, Zero};

/// Claimable collateral amounts.
#[derive(Debug, Clone, Copy)]
#[cfg_attr(
    feature = "anchor-lang",
    derive(anchor_lang::AnchorDeserialize, anchor_lang::AnchorSerialize)
)]
pub struct ClaimableCollateral<T> {
    output_token_amount: T,
    secondary_output_token_amount: T,
}

#[cfg(feature = "gmsol-utils")]
impl<T: gmsol_utils::InitSpace> gmsol_utils::InitSpace for ClaimableCollateral<T> {
    const INIT_SPACE: usize = 2 * T::INIT_SPACE;
}

impl<T> ClaimableCollateral<T> {
    /// Get output token amount.
    pub fn output_token_amount(&self) -> &T {
        &self.output_token_amount
    }

    /// Get secondary output token amount.
    pub fn secondary_output_token_amount(&self) -> &T {
        &self.secondary_output_token_amount
    }

    /// Try to add amount.
    pub fn try_add_amount(&mut self, amount: &T, is_output_token: bool) -> crate::Result<&mut Self>
    where
        T: CheckedAdd,
    {
        let current = if is_output_token {
            &mut self.output_token_amount
        } else {
            &mut self.secondary_output_token_amount
        };
        *current = current.checked_add(amount).ok_or(crate::Error::Overflow)?;
        Ok(self)
    }
}

impl<T: Zero> Default for ClaimableCollateral<T> {
    fn default() -> Self {
        Self {
            output_token_amount: Zero::zero(),
            secondary_output_token_amount: Zero::zero(),
        }
    }
}