feat: support subgraph headers for balancer/gyroscope (#554)

KyberNetwork · Oct 24, 2024 · 99125ac · 99125ac
1 parent 12f8c70
commit 99125ac
Show file tree

Hide file tree

Showing 6 changed files with 30 additions and 24 deletions.
diff --git a/pkg/liquidity-source/balancer-v2/composable-stable/pools_list_updater.go b/pkg/liquidity-source/balancer-v2/composable-stable/pools_list_updater.go
@@ -25,10 +25,11 @@ type PoolsListUpdater struct {
 
 func NewPoolsListUpdater(config *Config, ethrpcClient *ethrpc.Client) *PoolsListUpdater {
 	sharedUpdater := shared.NewPoolsListUpdater(&shared.Config{
-		DexID:        config.DexID,
-		SubgraphAPI:  config.SubgraphAPI,
-		NewPoolLimit: config.NewPoolLimit,
-		PoolTypes:    []string{poolTypeComposableStable},
+		DexID:           config.DexID,
+		SubgraphAPI:     config.SubgraphAPI,
+		SubgraphHeaders: config.SubgraphHeaders,
+		NewPoolLimit:    config.NewPoolLimit,
+		PoolTypes:       []string{poolTypeComposableStable},
 	})
 
 	return &PoolsListUpdater{

diff --git a/pkg/liquidity-source/balancer-v2/stable/pools_list_updater.go b/pkg/liquidity-source/balancer-v2/stable/pools_list_updater.go
@@ -25,10 +25,11 @@ type PoolsListUpdater struct {
 
 func NewPoolsListUpdater(config *Config, ethrpcClient *ethrpc.Client) *PoolsListUpdater {
 	sharedUpdater := shared.NewPoolsListUpdater(&shared.Config{
-		DexID:        config.DexID,
-		SubgraphAPI:  config.SubgraphAPI,
-		NewPoolLimit: config.NewPoolLimit,
-		PoolTypes:    []string{poolTypeStable, poolTypeMetaStable},
+		DexID:           config.DexID,
+		SubgraphAPI:     config.SubgraphAPI,
+		SubgraphHeaders: config.SubgraphHeaders,
+		NewPoolLimit:    config.NewPoolLimit,
+		PoolTypes:       []string{poolTypeStable, poolTypeMetaStable},
 	})
 
 	return &PoolsListUpdater{

diff --git a/pkg/liquidity-source/balancer-v2/weighted/pools_list_updater.go b/pkg/liquidity-source/balancer-v2/weighted/pools_list_updater.go
@@ -29,10 +29,11 @@ type PoolsListUpdater struct {
 
 func NewPoolsListUpdater(config *Config, ethrpcClient *ethrpc.Client) *PoolsListUpdater {
 	sharedUpdater := shared.NewPoolsListUpdater(&shared.Config{
-		DexID:        config.DexID,
-		SubgraphAPI:  config.SubgraphAPI,
-		NewPoolLimit: config.NewPoolLimit,
-		PoolTypes:    []string{poolTypeWeighted},
+		DexID:           config.DexID,
+		SubgraphAPI:     config.SubgraphAPI,
+		SubgraphHeaders: config.SubgraphHeaders,
+		NewPoolLimit:    config.NewPoolLimit,
+		PoolTypes:       []string{poolTypeWeighted},
 	})
 
 	return &PoolsListUpdater{

diff --git a/pkg/liquidity-source/gyroscope/2clp/pools_list_updater.go b/pkg/liquidity-source/gyroscope/2clp/pools_list_updater.go
@@ -25,10 +25,11 @@ type PoolsListUpdater struct {
 
 func NewPoolsListUpdater(config *Config, ethrpcClient *ethrpc.Client) *PoolsListUpdater {
 	sharedUpdater := shared.NewPoolsListUpdater(&shared.Config{
-		DexID:        config.DexID,
-		SubgraphAPI:  config.SubgraphAPI,
-		NewPoolLimit: config.NewPoolLimit,
-		PoolTypes:    []string{poolType},
+		DexID:           config.DexID,
+		SubgraphAPI:     config.SubgraphAPI,
+		SubgraphHeaders: config.SubgraphHeaders,
+		NewPoolLimit:    config.NewPoolLimit,
+		PoolTypes:       []string{poolType},
 	})
 
 	return &PoolsListUpdater{

diff --git a/pkg/liquidity-source/gyroscope/3clp/pools_list_updater.go b/pkg/liquidity-source/gyroscope/3clp/pools_list_updater.go
@@ -25,10 +25,11 @@ type PoolsListUpdater struct {
 
 func NewPoolsListUpdater(config *Config, ethrpcClient *ethrpc.Client) *PoolsListUpdater {
 	sharedUpdater := shared.NewPoolsListUpdater(&shared.Config{
-		DexID:        config.DexID,
-		SubgraphAPI:  config.SubgraphAPI,
-		NewPoolLimit: config.NewPoolLimit,
-		PoolTypes:    []string{poolType},
+		DexID:           config.DexID,
+		SubgraphAPI:     config.SubgraphAPI,
+		SubgraphHeaders: config.SubgraphHeaders,
+		NewPoolLimit:    config.NewPoolLimit,
+		PoolTypes:       []string{poolType},
 	})
 
 	return &PoolsListUpdater{

diff --git a/pkg/liquidity-source/gyroscope/eclp/pools_list_updater.go b/pkg/liquidity-source/gyroscope/eclp/pools_list_updater.go
@@ -22,10 +22,11 @@ type PoolsListUpdater struct {
 
 func NewPoolsListUpdater(config *Config, ethrpcClient *ethrpc.Client) *PoolsListUpdater {
 	sharedUpdater := shared.NewPoolsListUpdater(&shared.Config{
-		DexID:        config.DexID,
-		SubgraphAPI:  config.SubgraphAPI,
-		NewPoolLimit: config.NewPoolLimit,
-		PoolTypes:    []string{poolType},
+		DexID:           config.DexID,
+		SubgraphAPI:     config.SubgraphAPI,
+		SubgraphHeaders: config.SubgraphHeaders,
+		NewPoolLimit:    config.NewPoolLimit,
+		PoolTypes:       []string{poolType},
 	})
 
 	return &PoolsListUpdater{