Merge pull request #1 from corinv/master

Fix page count sometimes missing last partial page
ordermentum · Mar 7, 2019 · b549d2f · b549d2f
2 parents 90a7e7d + 692ae03
commit b549d2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -6,8 +6,8 @@ async function* batch(query = { batchSize: 1000 }) {
   if (count === 0) {
     return false;
   }
-
-  const pages = Math.max(Math.round(count / query.batchSize), 1);
+  const pagesRemainder = (count % query.batchSize) ? 1 : 0;
+  const pages = Math.floor(count / query.batchSize) + pagesRemainder;
   let page = 1;
 
   const params = Object.assign({}, query);