Merge pull request #343 from rawmind0/pagination

Added ListAll function on types_template.go to depaginate results on collection
rancher · Feb 10, 2020 · 40f8d46 · 40f8d46
2 parents 99f434a + 9035b1e
commit 40f8d46
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/generator/type_template.go b/generator/type_template.go
@@ -38,6 +38,7 @@ type {{.schema.CodeName}}Client struct {
 
 type {{.schema.CodeName}}Operations interface {
     List(opts *types.ListOpts) (*{{.schema.CodeName}}Collection, error)
+    ListAll(opts *types.ListOpts) (*{{.schema.CodeName}}Collection, error)
     Create(opts *{{.schema.CodeName}}) (*{{.schema.CodeName}}, error)
     Update(existing *{{.schema.CodeName}}, updates interface{}) (*{{.schema.CodeName}}, error)
     Replace(existing *{{.schema.CodeName}}) (*{{.schema.CodeName}}, error)
@@ -98,6 +99,23 @@ func (c *{{.schema.CodeName}}Client) List(opts *types.ListOpts) (*{{.schema.Code
     return resp, err
 }
 
+func (c *{{.schema.CodeName}}Client) ListAll(opts *types.ListOpts) (*{{.schema.CodeName}}Collection, error) {
+    resp := &{{.schema.CodeName}}Collection{}
+    resp, err := c.List(opts)
+    if err != nil {
+        return resp, err
+    }
+    data := resp.Data
+    for resp, err = resp.Next(); resp != nil && err == nil; resp, err = resp.Next() {
+        data = append(data, resp.Data...)
+    }
+    if err != nil {
+        return resp, err
+    }
+    resp.Data = data
+    return resp, err
+}
+
 func (cc *{{.schema.CodeName}}Collection) Next() (*{{.schema.CodeName}}Collection, error) {
     if cc != nil && cc.Pagination != nil && cc.Pagination.Next != "" {
         resp := &{{.schema.CodeName}}Collection{}