-
Notifications
You must be signed in to change notification settings - Fork 7
/
roles.go
142 lines (111 loc) · 2.94 KB
/
roles.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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
package rockset
import (
"context"
"github.com/rockset/rockset-go-client/openapi"
"github.com/rockset/rockset-go-client/option"
)
// CreateRole creates a new role
//
// REST API documentation https://docs.rockset.com/rest-api/#createrole
func (rc *RockClient) CreateRole(ctx context.Context, roleName string,
options ...option.RoleOption) (openapi.Role, error) {
var err error
var resp *openapi.RoleResponse
createReq := rc.CustomRolesApi.CreateRole(ctx)
b := openapi.NewCreateRoleRequest()
b.RoleName = &roleName
b.Privileges = []openapi.Privilege{}
var opts option.RoleOptions
for _, o := range options {
o(&opts)
}
if opts.Description != nil {
b.Description = opts.Description
}
if len(opts.Privileges) > 0 {
b.Privileges = opts.Privileges
}
err = rc.Retry(ctx, func() error {
resp, _, err = createReq.Body(*b).Execute()
return err
})
if err != nil {
return openapi.Role{}, err
}
return resp.GetData(), nil
}
// UpdateRole updates a role.
//
// REST API documentation https://docs.rockset.com/rest-api/#updaterole
func (rc *RockClient) UpdateRole(ctx context.Context, roleName string,
options ...option.RoleOption) (openapi.Role, error) {
var err error
var resp *openapi.RoleResponse
createReq := rc.CustomRolesApi.UpdateRole(ctx, roleName)
b := openapi.NewUpdateRoleRequest()
var opts option.RoleOptions
for _, o := range options {
o(&opts)
}
if opts.Description != nil {
b.Description = opts.Description
}
if len(opts.Privileges) > 0 {
b.Privileges = opts.Privileges
}
err = rc.Retry(ctx, func() error {
resp, _, err = createReq.Body(*b).Execute()
return err
})
if err != nil {
return openapi.Role{}, err
}
return resp.GetData(), nil
}
// DeleteRole deletes a role.
//
// REST API documentation https://docs.rockset.com/rest-api/#deleterole
func (rc *RockClient) DeleteRole(ctx context.Context, roleName string) error {
var err error
getReq := rc.CustomRolesApi.DeleteRole(ctx, roleName)
err = rc.Retry(ctx, func() error {
_, _, err = getReq.Execute()
return err
})
if err != nil {
return err
}
return nil
}
// GetRole retrieve a role.
//
// REST API documentation https://docs.rockset.com/rest-api/#getrole
func (rc *RockClient) GetRole(ctx context.Context, roleName string) (openapi.Role, error) {
var err error
var resp *openapi.RoleResponse
getReq := rc.CustomRolesApi.GetRole(ctx, roleName)
err = rc.Retry(ctx, func() error {
resp, _, err = getReq.Execute()
return err
})
if err != nil {
return openapi.Role{}, err
}
return resp.GetData(), nil
}
// ListRoles list all roles.
//
// REST API documentation https://docs.rockset.com/rest-api/#listroles
func (rc *RockClient) ListRoles(ctx context.Context) ([]openapi.Role, error) {
var err error
var resp *openapi.ListRolesResponse
getReq := rc.CustomRolesApi.ListRoles(ctx)
err = rc.Retry(ctx, func() error {
resp, _, err = getReq.Execute()
return err
})
if err != nil {
return nil, err
}
return resp.GetData(), nil
}