Skip to content

Commit

Permalink
Merge branch 'main' into mbreithecker/compliance
Browse files Browse the repository at this point in the history
# Conflicts:
#	testutil/integration/integration.go
#	x/global/ante_test.go
#	x/stakers/keeper/keeper_suite_protocol_bridge_test.go
  • Loading branch information
mbreithecker committed Feb 6, 2025
2 parents 0591027 + 1f27380 commit 4423a35
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 8 deletions.
18 changes: 11 additions & 7 deletions x/compliance/keeper/getters_compliance.go
Original file line number Diff line number Diff line change
@@ -1,8 +1,11 @@
package keeper

import (
"encoding/binary"

"cosmossdk.io/store/prefix"
storeTypes "cosmossdk.io/store/types"

"github.com/KYVENetwork/chain/util"
"github.com/KYVENetwork/chain/x/compliance/types"
"github.com/cosmos/cosmos-sdk/runtime"
Expand All @@ -22,12 +25,16 @@ func (k Keeper) SetMultiCoinPendingRewardsEntry(ctx sdk.Context, compliancePendi
compliancePendingRewards.Index,
), b)

// Insert the same entry with a different key prefix for query lookup
indexBytes := make([]byte, 8)
binary.BigEndian.PutUint64(indexBytes, compliancePendingRewards.Index)

// Insert the same entry with a different key prefix for query lookup
indexStore := prefix.NewStore(storeAdapter, types.MultiCoinPendingRewardsEntryKeyPrefixIndex2)
indexStore.Set(types.MultiCoinPendingRewardsKeyEntryIndex2(
compliancePendingRewards.Address,
compliancePendingRewards.Index,
), []byte{1})
), indexBytes)
}

// GetMultiCoinPendingRewardsEntry ...
Expand All @@ -53,12 +60,9 @@ func (k Keeper) GetMultiCoinPendingRewardsEntriesByIndex2(ctx sdk.Context, addre
defer iterator.Close()

for ; iterator.Valid(); iterator.Next() {
var val types.MultiCoinPendingRewardsEntry
println("key")
println(string(iterator.Key()))
println(string(iterator.Value()))
k.cdc.MustUnmarshal(iterator.Value(), &val)
list = append(list, val)
index := binary.BigEndian.Uint64(iterator.Value())
entry, _ := k.GetMultiCoinPendingRewardsEntry(ctx, index)
list = append(list, entry)
}

return
Expand Down
2 changes: 1 addition & 1 deletion x/stakers/keeper/keeper_suite_protocol_bridge_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,7 @@ var _ = Describe("keeper_suite_protocol_bridge.go", Ordered, func() {

// Assert
for k := 1; k < 10; k++ {
poolMembersCount := s.App().StakersKeeper.GetStakerCountOfPool(s.Ctx(), 0)
poolMembersCount := s.App().StakersKeeper.GetStakerCountOfPool(s.Ctx(), uint64(k))
Expect(poolMembersCount).To(Equal(uint64(0)))
}

Expand Down

0 comments on commit 4423a35

Please sign in to comment.