feat: gear for mainnet

harsh-98 · harsh-98 · commit 949ac633383c · 2025-05-18T00:34:44.000+07:00
diff --git a/go.mod b/go.mod
@@ -5,7 +5,7 @@ go 1.23
 // toolchain go1.23.2
 
 require (
-	github.com/Gearbox-protocol/sdk-go v0.0.0-20250515050146-74e6c8e5dfad
+	github.com/Gearbox-protocol/sdk-go v0.0.0-20250517173314-c323931a15cc
 	github.com/ethereum/go-ethereum v1.13.14
 	github.com/go-playground/validator/v10 v10.4.1
 	github.com/google/go-cmp v0.6.0
diff --git a/go.sum b/go.sum
@@ -6,8 +6,8 @@ github.com/BurntSushi/toml v0.3.1 h1:WXkYYl6Yr3qBf1K79EBnL4mak0OimBfB0XUf9Vl28OQ
 github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
 github.com/DataDog/zstd v1.4.5 h1:EndNeuB0l9syBZhut0wns3gV1hL8zX8LIu6ZiVHWLIQ=
 github.com/DataDog/zstd v1.4.5/go.mod h1:1jcaCB/ufaK+sKp1NBhlGmpz41jOoPQ35bpF36t7BBo=
-github.com/Gearbox-protocol/sdk-go v0.0.0-20250515050146-74e6c8e5dfad h1:wRcGkN6RVzmxWD8lew22b4mjsddrWCCpmGGsy0xDZKQ=
-github.com/Gearbox-protocol/sdk-go v0.0.0-20250515050146-74e6c8e5dfad/go.mod h1:MZsiiSAwDdnHY6GEILYtHEqhb9wmbUUE34UBQgNlUyM=
+github.com/Gearbox-protocol/sdk-go v0.0.0-20250517173314-c323931a15cc h1:FIH9ZW6myPIcPgqTAHR6SyjdAZHqg+7/a0uyFbQ1DZ8=
+github.com/Gearbox-protocol/sdk-go v0.0.0-20250517173314-c323931a15cc/go.mod h1:MZsiiSAwDdnHY6GEILYtHEqhb9wmbUUE34UBQgNlUyM=
 github.com/Microsoft/go-winio v0.6.1 h1:9/kr64B9VUZrLm5YYwbGtUJnMgqWVOdUAXu6Migciow=
 github.com/Microsoft/go-winio v0.6.1/go.mod h1:LRdKpFKfdobln8UmuiYcKPot9D2v6svN5+sAH+4kjUM=
 github.com/OffchainLabs/go-ethereum v1.13.4-0.20240313010929-e5d8587e7227 h1:+/3TrD+q+BP36jGj2Bycdmrc/joKLNbc5ImePQzKRLM=
diff --git a/models/aggregated_block_feed/dependency.go b/models/aggregated_block_feed/dependency.go
@@ -84,8 +84,13 @@ func (q *QueryPFDependencies) getChainlinkBasedQueryUpdates(clearExtraBefore int
 }
 
 var base = []string{"WETH", "WBTC", "DAI", "USDC", "USDT", "USDC", "OHM"}
-var combo = map[string][]string{
-	"3crv": {"DAI", "USDC", "USDT"},
+var combo = map[string]struct {
+	data        []string
+	onlymainnet bool
+}{
+	"3crv":   {data: []string{"DAI", "USDC", "USDT"}},
+	"wstETH": {data: []string{"stETH"}, onlymainnet: true},
+	"rstETH": {data: []string{"stETH"}, onlymainnet: true},
 }
 
 // {"USDT": {"3crv:address"}, "USDC": {"3crv:address"}, "DAI": {"3crv:address"}, "FRAX": {"crvFRAX"}}
@@ -103,9 +108,13 @@ func (q *QueryPFDependencies) GetChainlinkTokenToUpdateToken() map[string][]stri
 			if strings.Contains(token.Symbol, sym) && token.Symbol != sym {
 				ans[token.Symbol] = append(ans[token.Symbol], token.Address)
 			}
-			if strings.Contains(strings.ToLower(token.Symbol), "3crv") {
-				for _, underlyingsym := range combo["3crv"] {
-					ans[underlyingsym] = append(ans[underlyingsym], token.Address)
+		}
+		for collection, underlyingSyms := range combo {
+			if strings.Contains(strings.ToLower(token.Symbol), strings.ToLower(collection)) { // 3crv
+				if (core.GetBaseChainId(q.client) == 1 && underlyingSyms.onlymainnet) || !underlyingSyms.onlymainnet {
+					for _, underlyingsym := range underlyingSyms.data {
+						ans[underlyingsym] = append(ans[underlyingsym], token.Address)
+					}
 				}
 			}
 		}

Original file line number	Diff line number	Diff line change
`@@ -84,8 +84,13 @@ func (q *QueryPFDependencies) getChainlinkBasedQueryUpdates(clearExtraBefore int`
`84`	`84`	`}`
`85`	`85`
`86`	`86`	`var base = []string{"WETH", "WBTC", "DAI", "USDC", "USDT", "USDC", "OHM"}`
`87`		`-var combo = map[string][]string{`
`88`		`- "3crv": {"DAI", "USDC", "USDT"},`
	`87`	`+var combo = map[string]struct {`
	`88`	`+ data []string`
	`89`	`+ onlymainnet bool`
	`90`	`+}{`
	`91`	`+ "3crv": {data: []string{"DAI", "USDC", "USDT"}},`
	`92`	`+ "wstETH": {data: []string{"stETH"}, onlymainnet: true},`
	`93`	`+ "rstETH": {data: []string{"stETH"}, onlymainnet: true},`
`89`	`94`	`}`
`90`	`95`
`91`	`96`	`// {"USDT": {"3crv:address"}, "USDC": {"3crv:address"}, "DAI": {"3crv:address"}, "FRAX": {"crvFRAX"}}`
`@@ -103,9 +108,13 @@ func (q *QueryPFDependencies) GetChainlinkTokenToUpdateToken() map[string][]stri`
`103`	`108`	`if strings.Contains(token.Symbol, sym) && token.Symbol != sym {`
`104`	`109`	`ans[token.Symbol] = append(ans[token.Symbol], token.Address)`
`105`	`110`	`}`
`106`		`- if strings.Contains(strings.ToLower(token.Symbol), "3crv") {`
`107`		`- for _, underlyingsym := range combo["3crv"] {`
`108`		`- ans[underlyingsym] = append(ans[underlyingsym], token.Address)`
	`111`	`+ }`
	`112`	`+ for collection, underlyingSyms := range combo {`
	`113`	`+ if strings.Contains(strings.ToLower(token.Symbol), strings.ToLower(collection)) { // 3crv`
	`114`	`+ if (core.GetBaseChainId(q.client) == 1 && underlyingSyms.onlymainnet) \|\| !underlyingSyms.onlymainnet {`
	`115`	`+ for _, underlyingsym := range underlyingSyms.data {`
	`116`	`+ ans[underlyingsym] = append(ans[underlyingsym], token.Address)`
	`117`	`+ }`
`109`	`118`	`}`
`110`	`119`	`}`
`111`	`120`	`}`