forked from cloudfoundry/cli
/
paginated_resources.go
82 lines (68 loc) · 2.05 KB
/
paginated_resources.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
package ccv2
import (
"encoding/json"
"net/http"
"reflect"
"code.cloudfoundry.org/cli/api/cloudcontroller"
)
// PaginatedResources represents a page of resources returned by the Cloud
// Controller.
type PaginatedResources struct {
NextURL string `json:"next_url"`
ResourcesBytes json.RawMessage `json:"resources"`
resourceType reflect.Type
}
// NewPaginatedResources returns a new PaginatedResources struct with the
// given resource type.
func NewPaginatedResources(exampleResource interface{}) *PaginatedResources {
return &PaginatedResources{
resourceType: reflect.TypeOf(exampleResource),
}
}
// Resources unmarshals JSON representing a page of resources and returns a
// slice of the given resource type.
func (pr PaginatedResources) Resources() ([]interface{}, error) {
slicePtr := reflect.New(reflect.SliceOf(pr.resourceType))
err := json.Unmarshal([]byte(pr.ResourcesBytes), slicePtr.Interface())
slice := reflect.Indirect(slicePtr)
contents := make([]interface{}, 0, slice.Len())
for i := 0; i < slice.Len(); i++ {
contents = append(contents, slice.Index(i).Interface())
}
return contents, err
}
func (client Client) paginate(request *cloudcontroller.Request, obj interface{}, appendToExternalList func(interface{}) error) (Warnings, error) {
fullWarningsList := Warnings{}
for {
wrapper := NewPaginatedResources(obj)
response := cloudcontroller.Response{
DecodeJSONResponseInto: &wrapper,
}
err := client.connection.Make(request, &response)
fullWarningsList = append(fullWarningsList, response.Warnings...)
if err != nil {
return fullWarningsList, err
}
list, err := wrapper.Resources()
if err != nil {
return fullWarningsList, err
}
for _, item := range list {
err = appendToExternalList(item)
if err != nil {
return fullWarningsList, err
}
}
if wrapper.NextURL == "" {
break
}
request, err = client.newHTTPRequest(requestOptions{
URI: wrapper.NextURL,
Method: http.MethodGet,
})
if err != nil {
return fullWarningsList, err
}
}
return fullWarningsList, nil
}