Merge pull request #91 from lytics/add-context

Add option to pass in context to getobject
lytics · Jun 3, 2021 · 1d4d62f · 1d4d62f
2 parents 02decdf + c2a4d93
commit 1d4d62f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/google/storeutils/get.go b/google/storeutils/get.go
@@ -12,8 +12,12 @@ import (
 
 // GetObject Gets a single object's bytes based on bucket and name parameters
 func GetObject(gc *storage.Client, bucket, name string) (*bytes.Buffer, error) {
+	return GetObjectWithContext(context.Background(), gc, bucket, name)
+}
 
-	rc, err := gc.Bucket(bucket).Object(name).NewReader(context.Background())
+// GetObject Gets a single object's bytes based on bucket and name parameters
+func GetObjectWithContext(ctx context.Context, gc *storage.Client, bucket, name string) (*bytes.Buffer, error) {
+	rc, err := gc.Bucket(bucket).Object(name).NewReader(ctx)
 	if err != nil {
 		if err == storage.ErrObjectNotExist {
 			return nil, cloudstorage.ErrObjectNotFound