Add support for asynchronous iteration with flatMap

kriszyp · Jan 23, 2024 · f4acb0e · f4acb0e
1 parent 1afdf49
commit f4acb0e
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 4 deletions.
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "lmdb",
   "author": "Kris Zyp",
-  "version": "2.9.3-beta.5-debug",
+  "version": "2.10.0-beta.1",
   "description": "Simple, efficient, scalable, high-performance LMDB interface",
   "license": "MIT",
   "repository": {

diff --git a/util/RangeIterable.js b/util/RangeIterable.js
@@ -173,21 +173,46 @@ export class RangeIterable {
 			let isFirst = true;
 			let currentSubIterator;
 			return {
-				next() {
+				next(resolvedResult) {
 					try {
 						do {
 							if (currentSubIterator) {
-								let result = currentSubIterator.next();
+								let result;
+								if (resolvedResult) {
+									result = resolvedResult;
+									resolvedResult = undefined;
+								} else result = currentSubIterator.next();
+								if (result.then) {
+									if (!async) throw new Error('Can not synchronously iterate with asynchronous values');
+									return result.then((result) => this.next(result));
+								}
 								if (!result.done) {
 									return result;
 								}
 							}
-							let result = iterator.next();
+							let result = resolvedResult ?? iterator.next();
+							if (result.then) {
+								if (!async) throw new Error('Can not synchronously iterate with asynchronous values');
+								currentSubIterator = undefined;
+								return result.then((result) => this.next(result));
+							}
 							if (result.done) {
 								if (mappedIterable.onDone) mappedIterable.onDone();
 								return result;
 							}
 							let value = callback(result.value);
+							if (value?.then) {
+								if (!async) throw new Error('Can not synchronously iterate with asynchronous values');
+								return value.then((value) => {
+									if (Array.isArray(value) || value instanceof RangeIterable) {
+										currentSubIterator = value[Symbol.iterator]();
+										return this.next();
+									} else {
+										currentSubIterator = null;
+										return { value };
+									}
+								})
+							}
 							if (Array.isArray(value) || value instanceof RangeIterable)
 								currentSubIterator = value[Symbol.iterator]();
 							else {