61 lines
1.7 KiB
Go
61 lines
1.7 KiB
Go
package keeper
|
|
|
|
import (
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
"github.com/cosmos/cosmos-sdk/x/group"
|
|
"github.com/cosmos/cosmos-sdk/x/group/errors"
|
|
"github.com/cosmos/cosmos-sdk/x/group/internal/orm"
|
|
)
|
|
|
|
// Tally is a function that tallies a proposal by iterating through its votes,
|
|
// and returns the tally result without modifying the proposal or any state.
|
|
func (q Keeper) Tally(ctx sdk.Context, p group.Proposal, groupId uint64) (group.TallyResult, error) {
|
|
// If proposal has already been tallied and updated, then its status is
|
|
// closed, in which case we just return the previously stored result.
|
|
if p.Status == group.PROPOSAL_STATUS_CLOSED {
|
|
return p.FinalTallyResult, nil
|
|
}
|
|
|
|
it, err := q.voteByProposalIndex.Get(ctx.KVStore(q.key), p.Id)
|
|
if err != nil {
|
|
return group.TallyResult{}, err
|
|
}
|
|
defer it.Close()
|
|
|
|
tallyResult := group.DefaultTallyResult()
|
|
|
|
var vote group.Vote
|
|
for {
|
|
_, err = it.LoadNext(&vote)
|
|
if errors.ErrORMIteratorDone.Is(err) {
|
|
break
|
|
}
|
|
if err != nil {
|
|
return group.TallyResult{}, err
|
|
}
|
|
|
|
var member group.GroupMember
|
|
err := q.groupMemberTable.GetOne(ctx.KVStore(q.key), orm.PrimaryKey(&group.GroupMember{
|
|
GroupId: groupId,
|
|
Member: &group.Member{Address: vote.Voter},
|
|
}), &member)
|
|
|
|
switch {
|
|
case sdkerrors.ErrNotFound.Is(err):
|
|
// If the member left the group after voting, then we simply skip the
|
|
// vote.
|
|
continue
|
|
case err != nil:
|
|
// For any other errors, we stop and return the error.
|
|
return group.TallyResult{}, err
|
|
}
|
|
|
|
if err := tallyResult.Add(vote, member.Member.Weight); err != nil {
|
|
return group.TallyResult{}, sdkerrors.Wrap(err, "add new vote")
|
|
}
|
|
}
|
|
|
|
return tallyResult, nil
|
|
}
|