-
Notifications
You must be signed in to change notification settings - Fork 2
/
msg_server.go
98 lines (81 loc) · 2.54 KB
/
msg_server.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
95
96
97
98
package embedded
import (
"context"
"cosmossdk.io/store"
"github.com/sourcenetwork/sourcehub/x/acp/keeper"
"github.com/sourcenetwork/sourcehub/x/acp/types"
)
var _ (types.MsgServer) = (*msgServer)(nil)
// msgServer implements the ACP module MsgServer interface.
//
// Effectively msgServer wraps a keeper with the module's native implementation
// but additionally it calls the commit method in the commit multi store after each method call
type msgServer struct {
types.UnimplementedMsgServer
msgServer types.MsgServer
cms store.CommitMultiStore
}
// NewMsgSrever creates a message server for Embedded ACP
func NewMsgServer(k keeper.Keeper, cms store.CommitMultiStore) types.MsgServer {
srv := keeper.NewMsgServerImpl(k)
return &msgServer{
msgServer: srv,
cms: cms,
}
}
func (s *msgServer) UpdateParams(ctx context.Context, msg *types.MsgUpdateParams) (*types.MsgUpdateParamsResponse, error) {
resp, err := s.msgServer.UpdateParams(ctx, msg)
if err != nil {
return nil, err
}
s.cms.Commit()
return resp, nil
}
func (s *msgServer) CreatePolicy(ctx context.Context, msg *types.MsgCreatePolicy) (*types.MsgCreatePolicyResponse, error) {
resp, err := s.msgServer.CreatePolicy(ctx, msg)
if err != nil {
return nil, err
}
s.cms.Commit()
return resp, nil
}
func (s *msgServer) SetRelationship(ctx context.Context, msg *types.MsgSetRelationship) (*types.MsgSetRelationshipResponse, error) {
resp, err := s.msgServer.SetRelationship(ctx, msg)
if err != nil {
return nil, err
}
s.cms.Commit()
return resp, nil
}
func (s *msgServer) DeleteRelationship(ctx context.Context, msg *types.MsgDeleteRelationship) (*types.MsgDeleteRelationshipResponse, error) {
resp, err := s.msgServer.DeleteRelationship(ctx, msg)
if err != nil {
return nil, err
}
s.cms.Commit()
return resp, nil
}
func (s *msgServer) RegisterObject(ctx context.Context, msg *types.MsgRegisterObject) (*types.MsgRegisterObjectResponse, error) {
resp, err := s.msgServer.RegisterObject(ctx, msg)
if err != nil {
return nil, err
}
s.cms.Commit()
return resp, nil
}
func (s *msgServer) UnregisterObject(ctx context.Context, msg *types.MsgUnregisterObject) (*types.MsgUnregisterObjectResponse, error) {
resp, err := s.msgServer.UnregisterObject(ctx, msg)
if err != nil {
return nil, err
}
s.cms.Commit()
return resp, nil
}
func (s *msgServer) CheckAccess(ctx context.Context, msg *types.MsgCheckAccess) (*types.MsgCheckAccessResponse, error) {
resp, err := s.msgServer.CheckAccess(ctx, msg)
if err != nil {
return nil, err
}
s.cms.Commit()
return resp, nil
}