fix

babylonlabs-io · Oct 25, 2024 · 441cc9c · 441cc9c
1 parent 88ef031
commit 441cc9c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/x/finality/keeper/grpc_query.go b/x/finality/keeper/grpc_query.go
@@ -44,7 +44,7 @@ func (k Keeper) FinalityProviderPowerAtHeight(ctx context.Context, req *types.Qu
 	}
 
 	sdkCtx := sdk.UnwrapSDKContext(ctx)
-	power := k.getVotingPower(sdkCtx, fpBTCPK.MustMarshal(), req.Height)
+	power := k.GetVotingPower(sdkCtx, fpBTCPK.MustMarshal(), req.Height)
 
 	return &types.QueryFinalityProviderPowerAtHeightResponse{VotingPower: power}, nil
 }

diff --git a/x/finality/keeper/power_table.go b/x/finality/keeper/power_table.go
@@ -17,7 +17,8 @@ func (k Keeper) SetVotingPower(ctx context.Context, fpBTCPK []byte, height uint6
 	store.Set(fpBTCPK, sdk.Uint64ToBigEndian(power))
 }
 
-func (k Keeper) getVotingPower(ctx context.Context, fpBTCPK []byte, height uint64) uint64 {
+// GetVotingPower gets the voting power of a given finality provider at a given Babylon height
+func (k Keeper) GetVotingPower(ctx context.Context, fpBTCPK []byte, height uint64) uint64 {
 	store := k.votingPowerBbnBlockHeightStore(ctx, height)
 	powerBytes := store.Get(fpBTCPK)
 	if len(powerBytes) == 0 {
@@ -26,14 +27,6 @@ func (k Keeper) getVotingPower(ctx context.Context, fpBTCPK []byte, height uint6
 	return sdk.BigEndianToUint64(powerBytes)
 }
 
-// GetVotingPower gets the voting power of a given finality provider at a given Babylon height
-func (k Keeper) GetVotingPower(ctx context.Context, fpBTCPK []byte, height uint64) uint64 {
-	if !k.BTCStakingKeeper.HasFinalityProvider(ctx, fpBTCPK) {
-		return 0
-	}
-	return k.getVotingPower(ctx, fpBTCPK, height)
-}
-
 // GetCurrentVotingPower gets the voting power of a given finality provider at the current height
 // NOTE: it's possible that the voting power table is 1 block behind CometBFT, e.g., when `BeginBlock`
 // hasn't executed yet