fix: PRT - fixing total stake amount taking only stake size into cons…

…ideration (#1746)
lavanet · Oct 17, 2024 · 66e7b83 · 66e7b83
1 parent 13593b4
commit 66e7b83
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/protocol/statetracker/updaters/pairing_updater.go b/protocol/statetracker/updaters/pairing_updater.go
@@ -248,12 +248,17 @@ func (pu *PairingUpdater) filterPairingListByEndpoint(ctx context.Context, curre
 			pairingEndpoints[idx] = endp
 		}
 		lavasession.SortByGeolocations(pairingEndpoints, currentGeo)
+		totalStakeAmount := provider.Stake.Amount
+		if !provider.DelegateTotal.Amount.IsNil() {
+			totalStakeAmount = totalStakeAmount.Add(provider.DelegateTotal.Amount)
+		}
+		totalStakeIncludingDelegation := sdk.Coin{Denom: provider.Stake.Denom, Amount: totalStakeAmount}
 		pairing[uint64(providerIdx)] = lavasession.NewConsumerSessionWithProvider(
 			provider.Address,
 			pairingEndpoints,
 			maxCu,
 			epoch,
-			provider.Stake,
+			totalStakeIncludingDelegation,
 		)
 	}
 	if len(pairing) == 0 {