add priceLimit to pool extra (#602)

* add priceLimit to pool extra * fix
KyberNetwork · Nov 20, 2024 · b2f8e3f · b2f8e3f
1 parent 24c7ba1
commit b2f8e3f
Show file tree

Hide file tree

Showing 16 changed files with 56 additions and 12 deletions.
diff --git a/pkg/source/algebrav1/pool_simulator.go b/pkg/source/algebrav1/pool_simulator.go
@@ -171,8 +171,10 @@ func (p *PoolSimulator) UpdateBalance(params pool.UpdateBalanceParams) {
 	p.globalState = si.GlobalState
 }
 
-func (p *PoolSimulator) GetMetaInfo(tokenIn string, tokenOut string) interface{} {
+func (p *PoolSimulator) GetMetaInfo(tokenIn string, _ string) interface{} {
+	zeroForOne := strings.EqualFold(tokenIn, p.Info.Tokens[0])
 	return PoolMeta{
 		BlockNumber: p.Pool.Info.BlockNumber,
+		PriceLimit:  p.getSqrtPriceLimit(zeroForOne),
 	}
 }
diff --git a/pkg/source/algebrav1/type.go b/pkg/source/algebrav1/type.go
@@ -132,7 +132,8 @@ type StateUpdate struct {
 }
 
 type PoolMeta struct {
-	BlockNumber uint64 `json:"blockNumber"`
+	BlockNumber uint64   `json:"blockNumber"`
+	PriceLimit  *big.Int `json:"priceLimit"`
 }
 
 func transformTickRespToTick(tickResp TickResp) (v3Entities.Tick, error) {

diff --git a/pkg/source/nuriv2/pool_simulator.go b/pkg/source/nuriv2/pool_simulator.go
@@ -203,5 +203,8 @@ func (p *PoolSimulator) UpdateBalance(params pool.UpdateBalanceParams) {
 }
 
 func (p *PoolSimulator) GetMetaInfo(tokenIn string, tokenOut string) interface{} {
-	return nil
+	zeroForOne := strings.EqualFold(tokenIn, p.V3Pool.Token0.Address.String())
+	return PoolMeta{
+		PriceLimit: p.getSqrtPriceLimit(zeroForOne),
+	}
 }
diff --git a/pkg/source/nuriv2/type.go b/pkg/source/nuriv2/type.go
@@ -68,6 +68,10 @@ type Extra struct {
 	Ticks        []Tick   `json:"ticks"`
 }
 
+type PoolMeta struct {
+	PriceLimit *big.Int `json:"priceLimit"`
+}
+
 type Slot0 struct {
 	SqrtPriceX96               *big.Int `json:"sqrtPriceX96"`
 	Tick                       *big.Int `json:"tick"`

diff --git a/pkg/source/pancakev3/pool_simulator.go b/pkg/source/pancakev3/pool_simulator.go
@@ -300,8 +300,13 @@ func (p *PoolSimulator) UpdateBalance(params pool.UpdateBalanceParams) {
 	p.V3Pool.TickCurrent = si.nextStateTickCurrent
 }
 
-func (p *PoolSimulator) GetMetaInfo(_ string, _ string) interface{} {
+func (p *PoolSimulator) GetMetaInfo(tokenIn string, _ string) interface{} {
+	zeroForOne := strings.EqualFold(tokenIn, p.V3Pool.Token0.Address.String())
+	var priceLimit v3Utils.Uint160
+	_ = p.getSqrtPriceLimit(zeroForOne, &priceLimit)
+
 	return PoolMeta{
 		BlockNumber: p.Pool.Info.BlockNumber,
+		PriceLimit:  priceLimit.ToBig(),
 	}
 }
diff --git a/pkg/source/pancakev3/pool_simulator_bigint.go b/pkg/source/pancakev3/pool_simulator_bigint.go
@@ -285,8 +285,10 @@ func (p *PoolSimulatorBigInt) UpdateBalance(params pool.UpdateBalanceParams) {
 	p.V3Pool.TickCurrent = si.nextStateTickCurrent
 }
 
-func (p *PoolSimulatorBigInt) GetMetaInfo(_ string, _ string) interface{} {
+func (p *PoolSimulatorBigInt) GetMetaInfo(tokenIn string, _ string) interface{} {
+	zeroForOne := strings.EqualFold(tokenIn, p.V3Pool.Token0.Address.String())
 	return PoolMeta{
 		BlockNumber: p.Pool.Info.BlockNumber,
+		PriceLimit:  p.getSqrtPriceLimit(zeroForOne),
 	}
 }
diff --git a/pkg/source/pancakev3/type.go b/pkg/source/pancakev3/type.go
@@ -108,7 +108,8 @@ type FetchRPCResult struct {
 }
 
 type PoolMeta struct {
-	BlockNumber uint64 `json:"blockNumber"`
+	BlockNumber uint64   `json:"blockNumber"`
+	PriceLimit  *big.Int `json:"priceLimit"`
 }
 
 func transformTickRespToTick(tickResp TickResp) (Tick, error) {

diff --git a/pkg/source/ramsesv2/pool_simulator.go b/pkg/source/ramsesv2/pool_simulator.go
@@ -203,5 +203,8 @@ func (p *PoolSimulator) UpdateBalance(params pool.UpdateBalanceParams) {
 }
 
 func (p *PoolSimulator) GetMetaInfo(tokenIn string, tokenOut string) interface{} {
-	return nil
+	zeroForOne := strings.EqualFold(tokenIn, p.V3Pool.Token0.Address.String())
+	return PoolMeta{
+		PriceLimit: p.getSqrtPriceLimit(zeroForOne),
+	}
 }
diff --git a/pkg/source/ramsesv2/type.go b/pkg/source/ramsesv2/type.go
@@ -66,6 +66,10 @@ type Extra struct {
 	Ticks        []Tick   `json:"ticks"`
 }
 
+type PoolMeta struct {
+	PriceLimit *big.Int `json:"priceLimit"`
+}
+
 type Slot0 struct {
 	SqrtPriceX96               *big.Int `json:"sqrtPriceX96"`
 	Tick                       *big.Int `json:"tick"`

diff --git a/pkg/source/slipstream/pool_simulator.go b/pkg/source/slipstream/pool_simulator.go
@@ -284,7 +284,11 @@ func (p *PoolSimulator) UpdateBalance(params pool.UpdateBalanceParams) {
 }
 
 func (p *PoolSimulator) GetMetaInfo(tokenIn string, tokenOut string) interface{} {
+	zeroForOne := strings.EqualFold(tokenIn, p.V3Pool.Token0.Address.String())
+	var priceLimit v3Utils.Uint160
+	_ = p.getSqrtPriceLimit(zeroForOne, &priceLimit)
 	return PoolMeta{
 		BlockNumber: p.Pool.Info.BlockNumber,
+		PriceLimit:  priceLimit.ToBig(),
 	}
 }
diff --git a/pkg/source/slipstream/type.go b/pkg/source/slipstream/type.go
@@ -104,7 +104,8 @@ type FetchRPCResult struct {
 }
 
 type PoolMeta struct {
-	BlockNumber uint64 `json:"blockNumber"`
+	BlockNumber uint64   `json:"blockNumber"`
+	PriceLimit  *big.Int `json:"priceLimit"`
 }
 
 func transformTickRespToTick(tickResp TickResp) (Tick, error) {

diff --git a/pkg/source/solidly-v3/pool_simulator.go b/pkg/source/solidly-v3/pool_simulator.go
@@ -276,5 +276,8 @@ func (p *PoolSimulator) UpdateBalance(params pool.UpdateBalanceParams) {
 }
 
 func (p *PoolSimulator) GetMetaInfo(tokenIn string, tokenOut string) interface{} {
-	return nil
+	zeroForOne := strings.EqualFold(tokenIn, p.V3Pool.Token0.Address.String())
+	return PoolMeta{
+		PriceLimit: p.getSqrtPriceLimit(zeroForOne),
+	}
 }
diff --git a/pkg/source/solidly-v3/type.go b/pkg/source/solidly-v3/type.go
@@ -63,6 +63,10 @@ type Extra struct {
 	Ticks        []Tick   `json:"ticks"`
 }
 
+type PoolMeta struct {
+	PriceLimit *big.Int `json:"priceLimit"`
+}
+
 type Slot0 struct {
 	SqrtPriceX96 *big.Int `json:"sqrtPriceX96"`
 	Tick         *big.Int `json:"tick"`

diff --git a/pkg/source/uniswapv3/pool_simulator.go b/pkg/source/uniswapv3/pool_simulator.go
@@ -301,8 +301,12 @@ func (p *PoolSimulator) UpdateBalance(params pool.UpdateBalanceParams) {
 	p.V3Pool.TickCurrent = si.nextStateTickCurrent
 }
 
-func (p *PoolSimulator) GetMetaInfo(_ string, _ string) interface{} {
+func (p *PoolSimulator) GetMetaInfo(tokenIn string, _ string) interface{} {
+	zeroForOne := strings.EqualFold(tokenIn, p.V3Pool.Token0.Address.String())
+	var priceLimit v3Utils.Uint160
+	_ = p.getSqrtPriceLimit(zeroForOne, &priceLimit)
 	return PoolMeta{
 		BlockNumber: p.Pool.Info.BlockNumber,
+		PriceLimit:  priceLimit.ToBig(),
 	}
 }
diff --git a/pkg/source/uniswapv3/pool_simulator_bigint.go b/pkg/source/uniswapv3/pool_simulator_bigint.go
@@ -285,8 +285,10 @@ func (p *PoolSimulatorBigInt) UpdateBalance(params pool.UpdateBalanceParams) {
 	p.V3Pool.TickCurrent = si.nextStateTickCurrent
 }
 
-func (p *PoolSimulatorBigInt) GetMetaInfo(_ string, _ string) interface{} {
+func (p *PoolSimulatorBigInt) GetMetaInfo(tokenIn string, _ string) interface{} {
+	zeroForOne := strings.EqualFold(tokenIn, p.V3Pool.Token0.Address.String())
 	return PoolMeta{
 		BlockNumber: p.Pool.Info.BlockNumber,
+		PriceLimit:  p.getSqrtPriceLimit(zeroForOne),
 	}
 }
diff --git a/pkg/source/uniswapv3/type.go b/pkg/source/uniswapv3/type.go
@@ -112,7 +112,8 @@ type FetchRPCResult struct {
 }
 
 type PoolMeta struct {
-	BlockNumber uint64 `json:"blockNumber"`
+	BlockNumber uint64   `json:"blockNumber"`
+	PriceLimit  *big.Int `json:"priceLimit"`
 }
 
 func transformTickRespToTick(tickResp TickResp) (Tick, error) {