173 lines
4.6 KiB
Go
173 lines
4.6 KiB
Go
package keeper
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/armon/go-metrics"
|
|
|
|
storetypes "github.com/cosmos/cosmos-sdk/store/types"
|
|
"github.com/cosmos/cosmos-sdk/telemetry"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/gov/types"
|
|
"github.com/cosmos/cosmos-sdk/x/gov/types/v1beta1"
|
|
)
|
|
|
|
type msgServer struct {
|
|
Keeper
|
|
}
|
|
|
|
// NewMsgServerImpl returns an implementation of the gov MsgServer interface
|
|
// for the provided Keeper.
|
|
func NewMsgServerImpl(keeper Keeper) v1beta1.MsgServer {
|
|
return &msgServer{Keeper: keeper}
|
|
}
|
|
|
|
var _ v1beta1.MsgServer = msgServer{}
|
|
|
|
func (k msgServer) SubmitProposal(goCtx context.Context, msg *v1beta1.MsgSubmitProposal) (*v1beta1.MsgSubmitProposalResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(goCtx)
|
|
proposal, err := k.Keeper.SubmitProposal(ctx, msg.GetContent())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
bytes, err := proposal.Marshal()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// ref: https://github.com/cosmos/cosmos-sdk/issues/9683
|
|
ctx.GasMeter().ConsumeGas(
|
|
3*storetypes.KVGasConfig().WriteCostPerByte*uint64(len(bytes)),
|
|
"submit proposal",
|
|
)
|
|
|
|
defer telemetry.IncrCounter(1, v1beta1.ModuleName, "proposal")
|
|
|
|
votingStarted, err := k.Keeper.AddDeposit(ctx, proposal.ProposalId, msg.GetProposer(), msg.GetInitialDeposit())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ctx.EventManager().EmitEvent(
|
|
sdk.NewEvent(
|
|
sdk.EventTypeMessage,
|
|
sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
|
|
sdk.NewAttribute(sdk.AttributeKeySender, msg.GetProposer().String()),
|
|
),
|
|
)
|
|
|
|
submitEvent := sdk.NewEvent(types.EventTypeSubmitProposal, sdk.NewAttribute(types.AttributeKeyProposalType, msg.GetContent().ProposalType()))
|
|
if votingStarted {
|
|
submitEvent = submitEvent.AppendAttributes(
|
|
sdk.NewAttribute(types.AttributeKeyVotingPeriodStart, fmt.Sprintf("%d", proposal.ProposalId)),
|
|
)
|
|
}
|
|
|
|
ctx.EventManager().EmitEvent(submitEvent)
|
|
return &v1beta1.MsgSubmitProposalResponse{
|
|
ProposalId: proposal.ProposalId,
|
|
}, nil
|
|
}
|
|
|
|
func (k msgServer) Vote(goCtx context.Context, msg *v1beta1.MsgVote) (*v1beta1.MsgVoteResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(goCtx)
|
|
accAddr, accErr := sdk.AccAddressFromBech32(msg.Voter)
|
|
if accErr != nil {
|
|
return nil, accErr
|
|
}
|
|
err := k.Keeper.AddVote(ctx, msg.ProposalId, accAddr, v1beta1.NewNonSplitVoteOption(msg.Option))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer telemetry.IncrCounterWithLabels(
|
|
[]string{v1beta1.ModuleName, "vote"},
|
|
1,
|
|
[]metrics.Label{
|
|
telemetry.NewLabel("proposal_id", strconv.Itoa(int(msg.ProposalId))),
|
|
},
|
|
)
|
|
|
|
ctx.EventManager().EmitEvent(
|
|
sdk.NewEvent(
|
|
sdk.EventTypeMessage,
|
|
sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
|
|
sdk.NewAttribute(sdk.AttributeKeySender, msg.Voter),
|
|
),
|
|
)
|
|
|
|
return &v1beta1.MsgVoteResponse{}, nil
|
|
}
|
|
|
|
func (k msgServer) VoteWeighted(goCtx context.Context, msg *v1beta1.MsgVoteWeighted) (*v1beta1.MsgVoteWeightedResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(goCtx)
|
|
accAddr, accErr := sdk.AccAddressFromBech32(msg.Voter)
|
|
if accErr != nil {
|
|
return nil, accErr
|
|
}
|
|
err := k.Keeper.AddVote(ctx, msg.ProposalId, accAddr, msg.Options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer telemetry.IncrCounterWithLabels(
|
|
[]string{v1beta1.ModuleName, "vote"},
|
|
1,
|
|
[]metrics.Label{
|
|
telemetry.NewLabel("proposal_id", strconv.Itoa(int(msg.ProposalId))),
|
|
},
|
|
)
|
|
|
|
ctx.EventManager().EmitEvent(
|
|
sdk.NewEvent(
|
|
sdk.EventTypeMessage,
|
|
sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
|
|
sdk.NewAttribute(sdk.AttributeKeySender, msg.Voter),
|
|
),
|
|
)
|
|
|
|
return &v1beta1.MsgVoteWeightedResponse{}, nil
|
|
}
|
|
|
|
func (k msgServer) Deposit(goCtx context.Context, msg *v1beta1.MsgDeposit) (*v1beta1.MsgDepositResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(goCtx)
|
|
accAddr, err := sdk.AccAddressFromBech32(msg.Depositor)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
votingStarted, err := k.Keeper.AddDeposit(ctx, msg.ProposalId, accAddr, msg.Amount)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer telemetry.IncrCounterWithLabels(
|
|
[]string{v1beta1.ModuleName, "deposit"},
|
|
1,
|
|
[]metrics.Label{
|
|
telemetry.NewLabel("proposal_id", strconv.Itoa(int(msg.ProposalId))),
|
|
},
|
|
)
|
|
|
|
ctx.EventManager().EmitEvent(
|
|
sdk.NewEvent(
|
|
sdk.EventTypeMessage,
|
|
sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
|
|
sdk.NewAttribute(sdk.AttributeKeySender, msg.Depositor),
|
|
),
|
|
)
|
|
|
|
if votingStarted {
|
|
ctx.EventManager().EmitEvent(
|
|
sdk.NewEvent(
|
|
types.EventTypeProposalDeposit,
|
|
sdk.NewAttribute(types.AttributeKeyVotingPeriodStart, fmt.Sprintf("%d", msg.ProposalId)),
|
|
),
|
|
)
|
|
}
|
|
|
|
return &v1beta1.MsgDepositResponse{}, nil
|
|
}
|