chore: Fix firestore sub collection handling.

nitrictech · Jul 19, 2021 · fcc7f10 · fcc7f10
1 parent 9ee5d63
commit fcc7f10
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pkg/plugins/document/firestore/firestore.go b/pkg/plugins/document/firestore/firestore.go
@@ -164,6 +164,17 @@ func (s *FirestoreDocService) Query(collection *sdk.Collection, expressions []sd
 				Id:         docSnp.Ref.ID,
 			},
 		}
+
+		if p := docSnp.Ref.Parent.Parent; p != nil {
+			sdkDoc.Key.Collection = &sdk.Collection{
+				Name: collection.Name,
+				Parent: &sdk.Key{
+					Collection: collection.Parent.Collection,
+					Id:         p.ID,
+				},
+			}
+		}
+
 		queryResult.Documents = append(queryResult.Documents, sdkDoc)
 
 		// If query limit configured determine continue tokens