-
Notifications
You must be signed in to change notification settings - Fork 0
/
sdk_handler.go
94 lines (72 loc) · 2.42 KB
/
sdk_handler.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
package keeper
import (
context "context"
sdkdistr "github.com/cosmos/cosmos-sdk/x/distribution/types"
"github.com/nephirim/fanfury-sdk/v2/x/lsnative/distribution/types"
)
type SdkMsgHandler interface {
sdkdistr.MsgServer
}
// NewSdkMsgHandlerWrapper returns SdkMsgHandler that implements MsgServer for
// vanilla sdk distribution keeper handler, allowing us to route legacy messages to a forked keeper.
func NewSdkMsgHandlerWrapper(handler types.MsgServer) SdkMsgHandler {
return sdkMsgHandlerWrapper{
handler: handler,
}
}
var _ SdkMsgHandler = sdkMsgHandlerWrapper{}
type sdkMsgHandlerWrapper struct {
handler types.MsgServer
}
// FundCommunityPool implements SdkMsgHandler
func (h sdkMsgHandlerWrapper) FundCommunityPool(
ctx context.Context,
sdkMsg *sdkdistr.MsgFundCommunityPool,
) (*sdkdistr.MsgFundCommunityPoolResponse, error) {
msg := types.MsgFundCommunityPool(*sdkMsg)
res, err := h.handler.FundCommunityPool(ctx, &msg)
if err != nil {
return nil, err
}
sdkResp := (sdkdistr.MsgFundCommunityPoolResponse)(*res)
return &sdkResp, nil
}
// SetWithdrawAddress implements SdkMsgHandler
func (h sdkMsgHandlerWrapper) SetWithdrawAddress(
ctx context.Context,
sdkMsg *sdkdistr.MsgSetWithdrawAddress,
) (*sdkdistr.MsgSetWithdrawAddressResponse, error) {
msg := types.MsgSetWithdrawAddress(*sdkMsg)
res, err := h.handler.SetWithdrawAddress(ctx, &msg)
if err != nil {
return nil, err
}
sdkResp := (sdkdistr.MsgSetWithdrawAddressResponse)(*res)
return &sdkResp, nil
}
// WithdrawDelegatorReward implements SdkMsgHandler
func (h sdkMsgHandlerWrapper) WithdrawDelegatorReward(
ctx context.Context,
sdkMsg *sdkdistr.MsgWithdrawDelegatorReward,
) (*sdkdistr.MsgWithdrawDelegatorRewardResponse, error) {
msg := types.MsgWithdrawDelegatorReward(*sdkMsg)
res, err := h.handler.WithdrawDelegatorReward(ctx, &msg)
if err != nil {
return nil, err
}
sdkResp := (sdkdistr.MsgWithdrawDelegatorRewardResponse)(*res)
return &sdkResp, nil
}
// WithdrawValidatorCommission implements SdkMsgHandler
func (h sdkMsgHandlerWrapper) WithdrawValidatorCommission(
ctx context.Context,
sdkMsg *sdkdistr.MsgWithdrawValidatorCommission,
) (*sdkdistr.MsgWithdrawValidatorCommissionResponse, error) {
msg := types.MsgWithdrawValidatorCommission(*sdkMsg)
res, err := h.handler.WithdrawValidatorCommission(ctx, &msg)
if err != nil {
return nil, err
}
sdkResp := (sdkdistr.MsgWithdrawValidatorCommissionResponse)(*res)
return &sdkResp, nil
}