feat: Mellow LRTs on balancerf

ojo-network · Jul 5, 2024 · 7272d91 · 7272d91
1 parent f338617
commit 7272d91
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 18 deletions.
diff --git a/client/spot_prices.go b/client/spot_prices.go
@@ -91,9 +91,9 @@ func (c *Client) QueryBalancerSpotPrice(p pool.Pool, blockNum uint64) indexer.Sp
 		c.reportError(fmt.Errorf("error initializing %s pool caller: %w", p.ExchangePair(), err))
 		return indexer.SpotPrice{}
 	}
-	poolRate, err := poolCaller.GetRate(nil)
+	poolRate, err := poolCaller.GetTokenRate(nil, common.HexToAddress(p.TokenAddress))
 	if err != nil {
-		c.reportError(fmt.Errorf("error getting %s pool balance: %w", p.ExchangePair(), err))
+		c.reportError(fmt.Errorf("error getting %s token rate from pool: %w", p.ExchangePair(), err))
 		return indexer.SpotPrice{}
 	}
 	return indexer.SpotPrice{

diff --git a/client/swaps.go b/client/swaps.go
@@ -125,23 +125,13 @@ func (c *Client) WatchBalancerSwapEvent(p pool.Pool) error {
 		return err
 	}
 
-	// retreive token addresses of pool
-	vaultCaller, err := vault.NewPoolCaller(vaultAddress, c.ethClient)
-	if err != nil {
-		return err
-	}
-	poolTokens, err := vaultCaller.GetPoolTokens(nil, poolId)
-	if err != nil {
-		return err
-	}
-
 	// build parameters for swap event subscription
 	poolIdParam := make([][32]byte, 1)
 	poolIdParam[0] = poolId
 	tokenInParam := make([]common.Address, 1)
-	tokenInParam[0] = poolTokens.Tokens[0]
+	tokenInParam[0] = common.HexToAddress(p.TokenAddress)
 	tokenOutParam := make([]common.Address, 1)
-	tokenOutParam[0] = poolTokens.Tokens[1]
+	tokenOutParam[0] = common.HexToAddress(p.TokenAddress)
 
 	vaultFilterer, err := vault.NewPoolFilterer(vaultAddress, c.ethClient)
 	if err != nil {
@@ -173,7 +163,7 @@ func (c *Client) WatchBalancerSwapEvent(p pool.Pool) error {
 
 			swap := p.ConvertBalancerEventToSwap(event, poolRate)
 			spotPrice := p.ConvertBalancerEventToSpotPrice(event, poolRate)
-			c.logger.Info().Interface("balancer swap", swap).Msg("algebra swap event received")
+			c.logger.Info().Interface("balancer swap", swap).Msg("balancer swap event received")
 			c.indexer.AddSwap(swap)
 			c.indexer.AddPrice(spotPrice)
 		}

diff --git a/pool/pool.go b/pool/pool.go
@@ -14,6 +14,7 @@ import (
 
 type Pool struct {
 	Address      string `yaml:"address"`
+	TokenAddress string `yaml:"token_address"`
 	Base         string `yaml:"base"`
 	Quote        string `yaml:"quote"`
 	BaseDecimal  uint64 `yaml:"base_decimal"`

diff --git a/sample-config.yaml b/sample-config.yaml
@@ -67,9 +67,10 @@ exchanges:
     node_urls:
       - "wss://mainnet.infura.io/ws/v3/6849a09aeeb044b592d46bcdce07ccef"
     pools:
-      - address: "0x596192bB6e41802428Ac943D2f1476C1Af25CC0E"
-        base: "EZETH"
-        quote: "WETH"
+      - address: "0x4216d5900a6109bba48418b5e2AB6cc4e61Cf477"
+        token_address: "0x84631c0d0081FDe56DeB72F6DE77abBbF6A9f93a"
+        base: "RE7LRT"
+        quote: "ETH"
         base_decimal: 18
         quote_decimal: 18
         invert_price: false