-
Notifications
You must be signed in to change notification settings - Fork 40
/
method_list.go
103 lines (88 loc) · 2.55 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
93
94
95
96
97
98
99
100
101
102
103
package dataconnectors
import (
"context"
"encoding/json"
"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 *[]DataConnector
}
type ListCompleteResult struct {
LatestHttpResponse *http.Response
Items []DataConnector
}
// List ...
func (c DataConnectorsClient) List(ctx context.Context, id WorkspaceId) (result ListOperationResponse, err error) {
opts := client.RequestOptions{
ContentType: "application/json; charset=utf-8",
ExpectedStatusCodes: []int{
http.StatusOK,
},
HttpMethod: http.MethodGet,
Path: fmt.Sprintf("%s/providers/Microsoft.SecurityInsights/dataConnectors", 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 *[]json.RawMessage `json:"value"`
}
if err = resp.Unmarshal(&values); err != nil {
return
}
temp := make([]DataConnector, 0)
if values.Values != nil {
for i, v := range *values.Values {
val, err := unmarshalDataConnectorImplementation(v)
if err != nil {
err = fmt.Errorf("unmarshalling item %d for DataConnector (%q): %+v", i, v, err)
return result, err
}
temp = append(temp, val)
}
}
result.Model = &temp
return
}
// ListComplete retrieves all the results into a single object
func (c DataConnectorsClient) ListComplete(ctx context.Context, id WorkspaceId) (ListCompleteResult, error) {
return c.ListCompleteMatchingPredicate(ctx, id, DataConnectorOperationPredicate{})
}
// ListCompleteMatchingPredicate retrieves all the results and then applies the predicate
func (c DataConnectorsClient) ListCompleteMatchingPredicate(ctx context.Context, id WorkspaceId, predicate DataConnectorOperationPredicate) (result ListCompleteResult, err error) {
items := make([]DataConnector, 0)
resp, err := c.List(ctx, id)
if err != nil {
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
}