feat : add DefaultGas for Iziswap (#580)

KyberNetwork · Nov 11, 2024 · 383da40 · 383da40
1 parent 3549700
commit 383da40
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/pkg/source/iziswap/constant.go b/pkg/source/iziswap/constant.go
@@ -23,8 +23,10 @@ const (
 	SIMULATOR_PT_RANGE = 2000
 )
 
-var zeroBI = big.NewInt(0)
-
+var (
+	DefaultGas = Gas{Swap: 210000}
+	zeroBI     = big.NewInt(0)
+)
 var pointDeltas = map[int]int{
 	100:   1,
 	400:   8,

diff --git a/pkg/source/iziswap/pool_simulator.go b/pkg/source/iziswap/pool_simulator.go
@@ -17,6 +17,7 @@ import (
 type PoolSimulator struct {
 	pool.Pool
 	PoolInfo swap.PoolInfo
+	gas      Gas
 }
 
 func NewPoolSimulator(entityPool entity.Pool) (*PoolSimulator, error) {
@@ -107,6 +108,7 @@ func (p *PoolSimulator) CalcAmountOut(param pool.CalcAmountOutParams) (*pool.Cal
 				Token:  tokenAmountIn.Token,
 				Amount: nil,
 			},
+			Gas: p.gas.Swap,
 			SwapInfo: iZiSwapInfo{
 				nextPoint:      ret.CurrentPoint,
 				nextLiquidity:  new(big.Int).Set(ret.Liquidity),
@@ -134,6 +136,7 @@ func (p *PoolSimulator) CalcAmountOut(param pool.CalcAmountOutParams) (*pool.Cal
 				Token:  tokenAmountIn.Token,
 				Amount: nil,
 			},
+			Gas: p.gas.Swap,
 			SwapInfo: iZiSwapInfo{
 				nextPoint:      ret.CurrentPoint,
 				nextLiquidity:  new(big.Int).Set(ret.Liquidity),
@@ -200,6 +203,7 @@ func (p *PoolSimulator) CalcAmountIn(param pool.CalcAmountInParams) (*pool.CalcA
 				Token:  tokenAmountOut.Token,
 				Amount: nil,
 			},
+			Gas:      p.gas.Swap,
 			SwapInfo: swapInfo,
 		}, nil
 	} else {
@@ -230,6 +234,7 @@ func (p *PoolSimulator) CalcAmountIn(param pool.CalcAmountInParams) (*pool.CalcA
 				Token:  tokenAmountOut.Token,
 				Amount: nil,
 			},
+			Gas: p.gas.Swap,
 			SwapInfo: iZiSwapInfo{
 				nextPoint:      ret.CurrentPoint,
 				nextLiquidity:  new(big.Int).Set(ret.Liquidity),

diff --git a/pkg/source/iziswap/type.go b/pkg/source/iziswap/type.go
@@ -77,3 +77,7 @@ type Metadata struct {
 type Meta struct {
 	LimitPoint int `json:"limitPoint"`
 }
+
+type Gas struct {
+	Swap int64
+}