-
Notifications
You must be signed in to change notification settings - Fork 40
/
method_list.go
125 lines (104 loc) · 2.93 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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package desktop
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 *[]Desktop
}
type ListCompleteResult struct {
Items []Desktop
}
type ListOperationOptions struct {
InitialSkip *int64
IsDescending *bool
PageSize *int64
}
func DefaultListOperationOptions() ListOperationOptions {
return ListOperationOptions{}
}
func (o ListOperationOptions) ToHeaders() *client.Headers {
out := client.Headers{}
return &out
}
func (o ListOperationOptions) ToOData() *odata.Query {
out := odata.Query{}
return &out
}
func (o ListOperationOptions) ToQuery() *client.QueryParams {
out := client.QueryParams{}
if o.InitialSkip != nil {
out.Append("initialSkip", fmt.Sprintf("%v", *o.InitialSkip))
}
if o.IsDescending != nil {
out.Append("isDescending", fmt.Sprintf("%v", *o.IsDescending))
}
if o.PageSize != nil {
out.Append("pageSize", fmt.Sprintf("%v", *o.PageSize))
}
return &out
}
// List ...
func (c DesktopClient) List(ctx context.Context, id ApplicationGroupId, options ListOperationOptions) (result ListOperationResponse, err error) {
opts := client.RequestOptions{
ContentType: "application/json; charset=utf-8",
ExpectedStatusCodes: []int{
http.StatusOK,
},
HttpMethod: http.MethodGet,
Path: fmt.Sprintf("%s/desktops", id.ID()),
OptionsObject: options,
}
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 *[]Desktop `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 DesktopClient) ListComplete(ctx context.Context, id ApplicationGroupId, options ListOperationOptions) (ListCompleteResult, error) {
return c.ListCompleteMatchingPredicate(ctx, id, options, DesktopOperationPredicate{})
}
// ListCompleteMatchingPredicate retrieves all the results and then applies the predicate
func (c DesktopClient) ListCompleteMatchingPredicate(ctx context.Context, id ApplicationGroupId, options ListOperationOptions, predicate DesktopOperationPredicate) (result ListCompleteResult, err error) {
items := make([]Desktop, 0)
resp, err := c.List(ctx, id, options)
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{
Items: items,
}
return
}