chore: improved staker sorting

KYVENetwork · Dec 20, 2024 · 87df706 · 87df706
1 parent 1e3c457
commit 87df706
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/x/query/keeper/grpc_query_stakers_by_pool.go b/x/query/keeper/grpc_query_stakers_by_pool.go
@@ -29,8 +29,10 @@ func (k Keeper) StakersByPool(c context.Context, req *types.QueryStakersByPoolRe
 		stakers = append(stakers, *k.GetFullStaker(ctx, valaccount.Staker))
 	}
 
-	sort.Slice(stakers, func(i, j int) bool {
-		return k.stakerKeeper.GetValidatorPoolStake(ctx, stakers[i].Address, req.PoolId) > k.stakerKeeper.GetValidatorPoolStake(ctx, stakers[j].Address, req.PoolId)
+	stakes := k.stakerKeeper.GetValidatorPoolStakes(ctx, req.PoolId)
+
+	sort.SliceStable(stakers, func(i, j int) bool {
+		return stakes[stakers[i].Address] > stakes[stakers[j].Address]
 	})
 
 	return &types.QueryStakersByPoolResponse{Stakers: stakers}, nil