firstDocument fix

RxSwiftCommunity · May 1, 2018 · 0e69c81 · 0e69c81
1 parent 28ae38e
commit 0e69c81
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Sources/Firestore/FIRQuery+Rx.swift b/Sources/Firestore/FIRQuery+Rx.swift
@@ -33,7 +33,9 @@ extension Reactive where Base: Query {
      * Reads the first document matching this query.
      */
     public func getFirstDocument() -> Observable<QueryDocumentSnapshot> {
-        return self.getDocuments()
+        return self.base.limit(to: 1)
+            .rx
+            .getDocuments()
             .map { snapshot in
                 guard let document = snapshot.documents.first(where: { $0.exists }) else {
                     throw NSError(domain: FirestoreErrorDomain, code: FirestoreErrorCode.notFound.rawValue, userInfo: nil)