From 39c56716bd030cc2d075fa3fc27d66818e6c3ec5 Mon Sep 17 00:00:00 2001 From: withtimezone Date: Thu, 25 Sep 2025 23:36:28 +0800 Subject: [PATCH] refactor: use b.Loop() to simplify the code and improve performance (#25367) Signed-off-by: withtimezone --- x/staking/bench_test.go | 3 +-- x/staking/keeper_bench_test.go | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/x/staking/bench_test.go b/x/staking/bench_test.go index 3d83f81a47..4c80b9031a 100644 --- a/x/staking/bench_test.go +++ b/x/staking/bench_test.go @@ -39,8 +39,7 @@ func benchmarkValidateGenesis(b *testing.B, n int) { validators = append(validators, validator) } - b.ResetTimer() - for i := 0; i < b.N; i++ { + for b.Loop() { genesisState := types.DefaultGenesisState() genesisState.Validators = validators if err := staking.ValidateGenesis(genesisState); err != nil { diff --git a/x/staking/keeper_bench_test.go b/x/staking/keeper_bench_test.go index f7283f566b..760b6ee6ae 100644 --- a/x/staking/keeper_bench_test.go +++ b/x/staking/keeper_bench_test.go @@ -50,8 +50,8 @@ func BenchmarkApplyAndReturnValidatorSetUpdates(b *testing.B) { require.NoError(b, keeper.SetLastValidatorPower(ctx, valAddrs[i], validator.ConsensusPower(sdk.DefaultPowerReduction))) } ctx, _ = testEnv.ctx.CacheContext() - b.ResetTimer() - for i := 0; i < b.N; i++ { + + for b.Loop() { _, _ = keeper.ApplyAndReturnValidatorSetUpdates(ctx) } }