-
Notifications
You must be signed in to change notification settings - Fork 38
/
method_listbydatabase.go
130 lines (109 loc) · 3.37 KB
/
method_listbydatabase.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
package databasecolumns
import (
"context"
"fmt"
"net/http"
"github.com/hashicorp/go-azure-helpers/resourcemanager/commonids"
"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 ListByDatabaseOperationResponse struct {
HttpResponse *http.Response
OData *odata.OData
Model *[]DatabaseColumn
}
type ListByDatabaseCompleteResult struct {
Items []DatabaseColumn
}
type ListByDatabaseOperationOptions struct {
Column *[]string
OrderBy *[]string
Schema *[]string
Table *[]string
}
func DefaultListByDatabaseOperationOptions() ListByDatabaseOperationOptions {
return ListByDatabaseOperationOptions{}
}
func (o ListByDatabaseOperationOptions) ToHeaders() *client.Headers {
out := client.Headers{}
return &out
}
func (o ListByDatabaseOperationOptions) ToOData() *odata.Query {
out := odata.Query{}
return &out
}
func (o ListByDatabaseOperationOptions) ToQuery() *client.QueryParams {
out := client.QueryParams{}
if o.Column != nil {
out.Append("column", fmt.Sprintf("%v", *o.Column))
}
if o.OrderBy != nil {
out.Append("orderBy", fmt.Sprintf("%v", *o.OrderBy))
}
if o.Schema != nil {
out.Append("schema", fmt.Sprintf("%v", *o.Schema))
}
if o.Table != nil {
out.Append("table", fmt.Sprintf("%v", *o.Table))
}
return &out
}
// ListByDatabase ...
func (c DatabaseColumnsClient) ListByDatabase(ctx context.Context, id commonids.SqlDatabaseId, options ListByDatabaseOperationOptions) (result ListByDatabaseOperationResponse, err error) {
opts := client.RequestOptions{
ContentType: "application/json; charset=utf-8",
ExpectedStatusCodes: []int{
http.StatusOK,
},
HttpMethod: http.MethodGet,
Path: fmt.Sprintf("%s/columns", 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 *[]DatabaseColumn `json:"value"`
}
if err = resp.Unmarshal(&values); err != nil {
return
}
result.Model = values.Values
return
}
// ListByDatabaseComplete retrieves all the results into a single object
func (c DatabaseColumnsClient) ListByDatabaseComplete(ctx context.Context, id commonids.SqlDatabaseId, options ListByDatabaseOperationOptions) (ListByDatabaseCompleteResult, error) {
return c.ListByDatabaseCompleteMatchingPredicate(ctx, id, options, DatabaseColumnOperationPredicate{})
}
// ListByDatabaseCompleteMatchingPredicate retrieves all the results and then applies the predicate
func (c DatabaseColumnsClient) ListByDatabaseCompleteMatchingPredicate(ctx context.Context, id commonids.SqlDatabaseId, options ListByDatabaseOperationOptions, predicate DatabaseColumnOperationPredicate) (result ListByDatabaseCompleteResult, err error) {
items := make([]DatabaseColumn, 0)
resp, err := c.ListByDatabase(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 = ListByDatabaseCompleteResult{
Items: items,
}
return
}