/
grafana.go
286 lines (233 loc) · 9.26 KB
/
grafana.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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
package aws
import (
"context"
"fmt"
"time"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/service/grafana"
"github.com/aws/aws-sdk-go-v2/service/grafana/types"
"github.com/aws/smithy-go/middleware"
smithytime "github.com/aws/smithy-go/time"
smithywaiter "github.com/aws/smithy-go/waiter"
"github.com/jmespath/go-jmespath"
)
type GrafanaClient struct {
*grafana.Client
}
func NewGrafanaClient() *GrafanaClient {
return &GrafanaClient{grafana.NewFromConfig(GetConfig())}
}
func (c *GrafanaClient) CreateWorkspace(name string, auth []string, roleArn string) (*types.WorkspaceDescription, error) {
result, err := c.Client.CreateWorkspace(context.Background(), &grafana.CreateWorkspaceInput{
AccountAccessType: types.AccountAccessTypeCurrentAccount,
AuthenticationProviders: toAuthenticationProviderTypes(auth),
PermissionType: types.PermissionTypeServiceManaged,
WorkspaceDataSources: []types.DataSourceType{types.DataSourceTypePrometheus},
WorkspaceName: aws.String(name),
WorkspaceRoleArn: aws.String(roleArn),
})
if err != nil {
return nil, err
}
err = NewWorkspaceActiveWaiter(c.Client).Wait(context.Background(),
&grafana.DescribeWorkspaceInput{WorkspaceId: result.Workspace.Id},
5*time.Minute,
)
return result.Workspace, err
}
func (c *GrafanaClient) DeleteWorkspace(id string) error {
_, err := c.Client.DeleteWorkspace(context.Background(), &grafana.DeleteWorkspaceInput{
WorkspaceId: aws.String(id),
})
return err
}
func (c *GrafanaClient) DescribeWorkspace(id string) (*types.WorkspaceDescription, error) {
result, err := c.Client.DescribeWorkspace(context.Background(), &grafana.DescribeWorkspaceInput{
WorkspaceId: aws.String(id),
})
if err != nil {
return nil, err
}
return result.Workspace, nil
}
func (c *GrafanaClient) ListWorkspaces() ([]types.WorkspaceSummary, error) {
workspaces := []types.WorkspaceSummary{}
pageNum := 0
paginator := grafana.NewListWorkspacesPaginator(c.Client, &grafana.ListWorkspacesInput{})
for paginator.HasMorePages() && pageNum < maxPages {
out, err := paginator.NextPage(context.Background())
if err != nil {
return nil, err
}
workspaces = append(workspaces, out.Workspaces...)
pageNum++
}
return workspaces, nil
}
func (c *GrafanaClient) UpdateWorkspaceAuthentication(id, samlMetadataUrl string) error {
result, err := c.Client.DescribeWorkspace(context.Background(), &grafana.DescribeWorkspaceInput{
WorkspaceId: aws.String(id),
})
if err != nil {
return err
}
_, err = c.Client.UpdateWorkspaceAuthentication(context.Background(), &grafana.UpdateWorkspaceAuthenticationInput{
AuthenticationProviders: result.Workspace.Authentication.Providers,
SamlConfiguration: &types.SamlConfiguration{
IdpMetadata: &types.IdpMetadataMemberUrl{
Value: samlMetadataUrl,
},
AssertionAttributes: &types.AssertionAttributes{
Role: aws.String("role"),
},
RoleValues: &types.RoleValues{
Admin: []string{"admin"},
},
},
WorkspaceId: aws.String(id),
})
return err
}
func toAuthenticationProviderTypes(s []string) []types.AuthenticationProviderTypes {
apt := make([]types.AuthenticationProviderTypes, len(s))
for i, v := range s {
apt[i] = types.AuthenticationProviderTypes(v)
}
return apt
}
// DescribeWorkspaceAPIClient is a client that implements the DescribeWorkspace
// operation.
type DescribeWorkspaceAPIClient interface {
DescribeWorkspace(context.Context, *grafana.DescribeWorkspaceInput, ...func(*grafana.Options)) (*grafana.DescribeWorkspaceOutput, error)
}
var _ DescribeWorkspaceAPIClient = (*grafana.Client)(nil)
type WorkspaceActiveWaiterOptions struct {
// Set of options to modify how an operation is invoked. These apply to all
// operations invoked for this client. Use functional options on operation call to
// modify this list for per operation behavior.
APIOptions []func(*middleware.Stack) error
// MinDelay is the minimum amount of time to delay between retries. If unset,
// WorkspaceActiveWaiter will use default minimum delay of 60 seconds. Note
// that MinDelay must resolve to a value lesser than or equal to the MaxDelay.
MinDelay time.Duration
// MaxDelay is the maximum amount of time to delay between retries. If unset or set
// to zero, WorkspaceActiveWaiter will use default max delay of 120 seconds.
// Note that MaxDelay must resolve to value greater than or equal to the MinDelay.
MaxDelay time.Duration
// LogWaitAttempts is used to enable logging for waiter retry attempts
LogWaitAttempts bool
// Retryable is function that can be used to override the service defined
// waiter-behavior based on operation output, or returned error. This function is
// used by the waiter to decide if a state is retryable or a terminal state. By
// default service-modeled logic will populate this option. This option can thus be
// used to define a custom waiter state with fall-back to service-modeled waiter
// state mutators.The function returns an error in case of a failure state. In case
// of retry state, this function returns a bool value of true and nil error, while
// in case of success it returns a bool value of false and nil error.
Retryable func(context.Context, *grafana.DescribeWorkspaceInput, *grafana.DescribeWorkspaceOutput, error) (bool, error)
}
// WorkspaceActiveWaiter defines the waiters for WorkspaceActive
type WorkspaceActiveWaiter struct {
client DescribeWorkspaceAPIClient
options WorkspaceActiveWaiterOptions
}
// NewWorkspaceActiveWaiter constructs a WorkspaceActiveWaiter.
func NewWorkspaceActiveWaiter(client DescribeWorkspaceAPIClient, optFns ...func(*WorkspaceActiveWaiterOptions)) *WorkspaceActiveWaiter {
options := WorkspaceActiveWaiterOptions{}
options.APIOptions = append(options.APIOptions, WaiterLogger{}.AddLogger)
options.MinDelay = 2 * time.Second
options.MaxDelay = 5 * time.Second
options.Retryable = workspaceActiveStateRetryable
for _, fn := range optFns {
fn(&options)
}
return &WorkspaceActiveWaiter{
client: client,
options: options,
}
}
// Wait calls the waiter function for WorkspaceActive waiter. The maxWaitDur
// is the maximum wait duration the waiter will wait. The maxWaitDur is required
// and must be greater than zero.
func (w *WorkspaceActiveWaiter) Wait(ctx context.Context, params *grafana.DescribeWorkspaceInput, maxWaitDur time.Duration, optFns ...func(*WorkspaceActiveWaiterOptions)) error {
_, err := w.WaitForOutput(ctx, params, maxWaitDur, optFns...)
return err
}
// WaitForOutput calls the waiter function for CertificateValidated waiter and
// returns the output of the successful operation. The maxWaitDur is the maximum
// wait duration the waiter will wait. The maxWaitDur is required and must be
// greater than zero.
func (w *WorkspaceActiveWaiter) WaitForOutput(ctx context.Context, params *grafana.DescribeWorkspaceInput, maxWaitDur time.Duration, optFns ...func(*WorkspaceActiveWaiterOptions)) (*grafana.DescribeWorkspaceOutput, error) {
if maxWaitDur <= 0 {
return nil, fmt.Errorf("maximum wait time for waiter must be greater than zero")
}
options := w.options
for _, fn := range optFns {
fn(&options)
}
if options.MaxDelay <= 0 {
options.MaxDelay = 120 * time.Second
}
if options.MinDelay > options.MaxDelay {
return nil, fmt.Errorf("minimum waiter delay %v must be lesser than or equal to maximum waiter delay of %v", options.MinDelay, options.MaxDelay)
}
ctx, cancelFn := context.WithTimeout(ctx, maxWaitDur)
defer cancelFn()
logger := smithywaiter.Logger{}
remainingTime := maxWaitDur
var attempt int64
for {
attempt++
apiOptions := options.APIOptions
start := time.Now()
if options.LogWaitAttempts {
logger.Attempt = attempt
apiOptions = append([]func(*middleware.Stack) error{}, options.APIOptions...)
apiOptions = append(apiOptions, logger.AddLogger)
}
out, err := w.client.DescribeWorkspace(ctx, params, func(o *grafana.Options) {
o.APIOptions = append(o.APIOptions, apiOptions...)
})
retryable, err := options.Retryable(ctx, params, out, err)
if err != nil {
return nil, err
}
if !retryable {
return out, nil
}
remainingTime -= time.Since(start)
if remainingTime < options.MinDelay || remainingTime <= 0 {
break
}
// compute exponential backoff between waiter retries
delay, err := smithywaiter.ComputeDelay(
attempt, options.MinDelay, options.MaxDelay, remainingTime,
)
if err != nil {
return nil, fmt.Errorf("error computing waiter delay, %w", err)
}
remainingTime -= delay
// sleep for the delay amount before invoking a request
if err := smithytime.SleepWithContext(ctx, delay); err != nil {
return nil, fmt.Errorf("request cancelled while waiting, %w", err)
}
}
return nil, fmt.Errorf("exceeded max wait time for WorkspaceActive waiter")
}
func workspaceActiveStateRetryable(ctx context.Context, input *grafana.DescribeWorkspaceInput, output *grafana.DescribeWorkspaceOutput, err error) (bool, error) {
if err == nil {
pathValue, err := jmespath.Search("workspace.status", output)
if err != nil {
return false, fmt.Errorf("error evaluating waiter state: %w", err)
}
expectedValue := "ACTIVE"
value, ok := pathValue.(types.WorkspaceStatus)
if !ok {
return false, fmt.Errorf("waiter comparator expected types.WorkspaceStatus value, got %T", pathValue)
}
if string(value) == expectedValue {
return false, nil
}
}
return true, nil
}