gcpfirestore - Add missing resource header when running query

google · Dec 12, 2023 · 85aa2eb · 85aa2eb
1 parent e86400e
commit 85aa2eb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docstore/gcpfirestore/query.go b/docstore/gcpfirestore/query.go
@@ -51,7 +51,7 @@ func (c *collection) newDocIterator(ctx context.Context, q *driver.Query) (*docI
 		}
 	}
 	ctx, cancel := context.WithCancel(ctx)
-	sc, err := c.client.RunQuery(ctx, req)
+	sc, err := c.client.RunQuery(withResourceHeader(ctx, c.dbPath), req)
 	if err != nil {
 		cancel()
 		return nil, err
@@ -65,7 +65,7 @@ func (c *collection) newDocIterator(ctx context.Context, q *driver.Query) (*docI
 	}, nil
 }
 
-////////////////////////////////////////////////////////////////
+// //////////////////////////////////////////////////////////////
 // The code below is adapted from cloud.google.com/go/firestore.
 
 type docIterator struct {