-
Notifications
You must be signed in to change notification settings - Fork 42
/
method_list.go
92 lines (77 loc) · 2.19 KB
/
method_list.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
package datastores
import (
"context"
"fmt"
"net/http"
"github.com/hashicorp/go-azure-sdk/sdk/client"
"github.com/hashicorp/go-azure-sdk/sdk/odata"
)
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See NOTICE.txt in the project root for license information.
type ListOperationResponse struct {
HttpResponse *http.Response
OData *odata.OData
Model *[]Datastore
}
type ListCompleteResult struct {
LatestHttpResponse *http.Response
Items []Datastore
}
// List ...
func (c DataStoresClient) List(ctx context.Context, id ClusterId) (result ListOperationResponse, err error) {
opts := client.RequestOptions{
ContentType: "application/json; charset=utf-8",
ExpectedStatusCodes: []int{
http.StatusOK,
},
HttpMethod: http.MethodGet,
Path: fmt.Sprintf("%s/dataStores", id.ID()),
}
req, err := c.Client.NewRequest(ctx, opts)
if err != nil {
return
}
var resp *client.Response
resp, err = req.ExecutePaged(ctx)
if resp != nil {
result.OData = resp.OData
result.HttpResponse = resp.Response
}
if err != nil {
return
}
var values struct {
Values *[]Datastore `json:"value"`
}
if err = resp.Unmarshal(&values); err != nil {
return
}
result.Model = values.Values
return
}
// ListComplete retrieves all the results into a single object
func (c DataStoresClient) ListComplete(ctx context.Context, id ClusterId) (ListCompleteResult, error) {
return c.ListCompleteMatchingPredicate(ctx, id, DatastoreOperationPredicate{})
}
// ListCompleteMatchingPredicate retrieves all the results and then applies the predicate
func (c DataStoresClient) ListCompleteMatchingPredicate(ctx context.Context, id ClusterId, predicate DatastoreOperationPredicate) (result ListCompleteResult, err error) {
items := make([]Datastore, 0)
resp, err := c.List(ctx, id)
if err != nil {
result.LatestHttpResponse = resp.HttpResponse
err = fmt.Errorf("loading results: %+v", err)
return
}
if resp.Model != nil {
for _, v := range *resp.Model {
if predicate.Matches(v) {
items = append(items, v)
}
}
}
result = ListCompleteResult{
LatestHttpResponse: resp.HttpResponse,
Items: items,
}
return
}