fix roots query looping when less than limit results

ssbc · Jun 28, 2017 · 60e61c2 · 60e61c2
1 parent a6aab6f
commit 60e61c2
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/sbot/roots.js b/sbot/roots.js
@@ -150,15 +150,24 @@ module.exports = function (ssb, config) {
 
       // TRUNCATE
       if (typeof limit === 'number') {
+        var count = 0
         return pullCat([
           pull(
             stream,
-            pull.take(limit)
+            pull.take(limit),
+            pull.through(() => {
+              count += 1
+            })
           ),
 
           // send truncated marker for resuming search
-          pull.values([marker])
+          pull(
+            pull.values([marker]),
+            pull.filter(() => count === limit)
+          )
         ])
+      } else {
+        return stream
       }
     }
   }