add NewListObjectsPager method to service/dataset client (#9)

deploifai · Aug 30, 2023 · 9de9b73 · 9de9b73
1 parent ffff3ec
commit 9de9b73
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/service/dataset/list_objects.go b/service/dataset/list_objects.go
@@ -0,0 +1,25 @@
+package dataset
+
+import (
+	"context"
+	"github.com/deploifai/sdk-go/api/generated"
+	"github.com/deploifai/sdk-go/cloud_client/implementable"
+)
+
+type ListObjectsInput = implementable.ListObjectsInput
+
+func (c *Client) NewListObjectsPager(ctx context.Context, where generated.DataStorageWhereUniqueInput, input *ListObjectsInput) (pager implementable.ListObjectsPager, err error) {
+
+	dataStorage, dataStorageContainer, err := getDataStorageAndContainer(ctx, c.options.API, where)
+	if err != nil {
+		return pager, err
+	}
+
+	dataStorageClient, err := newDataStorageClient(ctx, c.options.API, dataStorage, dataStorageContainer)
+	if err != nil {
+		return pager, err
+	}
+
+	return dataStorageClient.NewListObjectsPager(input), nil
+
+}