feat: gear for mainnet

Author: harsh-98

debts/engine.go

@@ -153,6 +153,12 @@ func (eng *DebtEngine) CalculateDebt() {
 			}
 			// #C3
 			sessionSnapshot := eng.lastCSS[session.ID]
+			if sessionSnapshot == nil {
+				if session.ID != "0x430789365519A006B110727C1c95032475e26Dbf_21786407_69" {
+					log.Warnf("Session(%s) not found in last css", session.ID, blockNum)
+				}
+				continue
+			}
 			caValue := utils.GetFloat64Decimal(
 				eng.GetAmountInUSD(
 					session.CreditManager,
@@ -191,6 +197,9 @@ func (eng *DebtEngine) CalculateDebt() {
 			if (session.ClosedAt != 0 && session.ClosedAt <= blockNum) || session.Since > blockNum {
 				continue
 			}
+			if eng.lastCSS[sessionId] == nil {
+				continue
+			}
 			cmAddr := session.CreditManager
 			// pool cum index is when the pool is not registered
 			if cmToPoolDetails[cmAddr] != nil {

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-20250517181031-ee1eaae730e0
 	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

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-20250517181031-ee1eaae730e0 h1:6jcCjdLuwxgv4MQDu1Dg2TvuFuEiophpR3g/h8zCnQ4=
+github.com/Gearbox-protocol/sdk-go v0.0.0-20250517181031-ee1eaae730e0/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=

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)
+					}
 				}
 			}
 		}