-
Notifications
You must be signed in to change notification settings - Fork 209
/
query.go
128 lines (102 loc) · 2.62 KB
/
query.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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
package cli
import (
"context"
sdkclient "github.com/cosmos/cosmos-sdk/client"
"github.com/cosmos/cosmos-sdk/client/flags"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/spf13/cobra"
types "github.com/akash-network/akash-api/go/node/audit/v1beta3"
aclient "github.com/akash-network/node/client"
)
func GetQueryCmd() *cobra.Command {
cmd := &cobra.Command{
Use: types.ModuleName,
Short: "Audit query commands",
SuggestionsMinimumDistance: 2,
RunE: sdkclient.ValidateCmd,
}
cmd.AddCommand(
cmdGetProviders(),
cmdGetProvider(),
)
return cmd
}
func cmdGetProviders() *cobra.Command {
cmd := &cobra.Command{
Use: "list",
Short: "Query for all providers",
RunE: func(cmd *cobra.Command, args []string) error {
ctx := cmd.Context()
cctx, err := sdkclient.GetClientQueryContext(cmd)
if err != nil {
return err
}
qq, err := aclient.DiscoverQueryClient(ctx, cctx)
if err != nil {
return err
}
pageReq, err := sdkclient.ReadPageRequest(cmd.Flags())
if err != nil {
return err
}
params := &types.QueryAllProvidersAttributesRequest{
Pagination: pageReq,
}
res, err := qq.AllProvidersAttributes(context.Background(), params)
if err != nil {
return err
}
return qq.ClientContext().PrintProto(res)
},
}
flags.AddQueryFlagsToCmd(cmd)
flags.AddPaginationFlagsToCmd(cmd, "providers")
return cmd
}
func cmdGetProvider() *cobra.Command {
cmd := &cobra.Command{
Use: "get [owner address] [auditor address]",
Short: "Query provider",
Args: cobra.RangeArgs(1, 2),
RunE: func(cmd *cobra.Command, args []string) error {
ctx := cmd.Context()
cctx, err := sdkclient.GetClientQueryContext(cmd)
if err != nil {
return err
}
qq, err := aclient.DiscoverQueryClient(ctx, cctx)
if err != nil {
return err
}
owner, err := sdk.AccAddressFromBech32(args[0])
if err != nil {
return err
}
var res *types.QueryProvidersResponse
if len(args) == 1 {
res, err = qq.ProviderAttributes(context.Background(),
&types.QueryProviderAttributesRequest{
Owner: owner.String(),
},
)
} else {
var auditor sdk.AccAddress
if auditor, err = sdk.AccAddressFromBech32(args[1]); err != nil {
return err
}
res, err = qq.ProviderAuditorAttributes(context.Background(),
&types.QueryProviderAuditorRequest{
Auditor: auditor.String(),
Owner: owner.String(),
},
)
}
if err != nil {
return err
}
return qq.ClientContext().PrintProto(res)
},
}
flags.AddQueryFlagsToCmd(cmd)
return cmd
}