This commit is contained in:
Hardhat Chad
2025-06-06 08:57:40 -07:00
parent 2eefaeb580
commit dcc6ab6419
4 changed files with 31 additions and 31 deletions

View File

@@ -7,40 +7,40 @@ use super::Market;
impl Market {
/// Returns the constant product invariant.
pub(crate) fn k(&self) -> u128 {
(self.base.reserves() * self.quote.reserves()).saturating_sub(1)
(self.base.liquidity() * self.quote.liquidity()).saturating_sub(1)
}
/// Returns the amount of base tokens that can be bought from a given amount of quote tokens.
pub fn get_base_out(&self, quote_in: u128) -> u128 {
let base_out = self.base.reserves()
- (self.k() / (self.quote.reserves() + quote_in)).saturating_add(1);
let base_out = self.base.liquidity()
- (self.k() / (self.quote.liquidity() + quote_in)).saturating_add(1);
base_out
}
/// Returns the amount of quote tokens received from selling a given amount of base tokens.
pub fn get_quote_out(&self, base_in: u128) -> u128 {
let quote_out =
self.quote.reserves() - (self.k() / (self.base.reserves() + base_in)).saturating_add(1);
let quote_out = self.quote.liquidity()
- (self.k() / (self.base.liquidity() + base_in)).saturating_add(1);
quote_out
}
/// Returns the amount of quote tokens needed to buy a given amount of base tokens.
pub fn get_quote_in(&self, base_out: u128) -> Result<u128, OreError> {
if base_out >= self.base.reserves() {
if base_out >= self.base.liquidity() {
return Err(OreError::InsufficientVaultReserves.into());
}
let quote_in = (self.k() / (self.base.reserves() - base_out)).saturating_add(1)
- self.quote.reserves();
let quote_in = (self.k() / (self.base.liquidity() - base_out)).saturating_add(1)
- self.quote.liquidity();
Ok(quote_in)
}
/// Returns the amount of base tokens which must be sold to receive a given amount of quote tokens.
pub fn get_base_in(&self, quote_out: u128) -> Result<u128, OreError> {
if quote_out >= self.quote.reserves() {
if quote_out >= self.quote.liquidity() {
return Err(OreError::InsufficientVaultReserves.into());
}
let base_in = (self.k() / (self.quote.reserves() - quote_out)).saturating_add(1)
- self.base.reserves();
let base_in = (self.k() / (self.quote.liquidity() - quote_out)).saturating_add(1)
- self.base.liquidity();
Ok(base_in)
}
}

View File

@@ -56,7 +56,7 @@ pub struct TokenParams {
}
impl TokenParams {
pub fn reserves(&self) -> u128 {
pub fn liquidity(&self) -> u128 {
(self.balance + self.balance_virtual) as u128
}
}

View File

@@ -41,8 +41,8 @@ impl Market {
/// ```
pub fn get_virtual_limit_order(&self, direction: SwapDirection) -> VirtualLimitOrder {
// Upcast data.
let base_balance = self.base.reserves();
let quote_balance = self.quote.reserves();
let base_balance = self.base.liquidity();
let quote_balance = self.quote.liquidity();
let base_snapshot = self.snapshot.base_balance as u128;
let quote_snapshot = self.snapshot.quote_balance as u128;
@@ -126,8 +126,8 @@ impl Market {
let snapshot_slot = (slot / SLOT_WINDOW) * SLOT_WINDOW;
if snapshot_slot != self.snapshot.slot {
self.snapshot.slot = snapshot_slot;
self.snapshot.base_balance = self.base.reserves() as u64;
self.snapshot.quote_balance = self.quote.reserves() as u64;
self.snapshot.base_balance = self.base.liquidity() as u64;
self.snapshot.quote_balance = self.quote.liquidity() as u64;
}
}