getDocuments with source

kakajika · May 10, 2019 · f2c0692 · f2c0692
1 parent 4f00544
commit f2c0692
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/Sources/Firestore/FIRQuery+Rx.swift b/Sources/Firestore/FIRQuery+Rx.swift
@@ -28,6 +28,23 @@ extension Reactive where Base: Query {
             return Disposables.create()
         }
     }
+
+    /**
+     * Reads the documents matching this query.
+     */
+    public func getDocuments(source: FirestoreSource) -> Observable<QuerySnapshot> {
+        return Observable<QuerySnapshot>.create { observer in
+            self.base.getDocuments(source: source) { snapshot, error in
+                if let error = error {
+                    observer.onError(error)
+                } else if let snapshot = snapshot {
+                    observer.onNext(snapshot)
+                }
+                observer.onCompleted()
+            }
+            return Disposables.create()
+        }
+    }
 
     /**
      * Reads the first document matching this query.