rado 0.1.39 → 0.1.40

package/dist/driver/better-sqlite3.js

@@ -6,6 +6,9 @@ var PreparedStatement = class {
   constructor(stmt) {
     this.stmt = stmt;
   }
+  iterate(params = []) {
+    return this.stmt.iterate(...params);
+  }
   all(params = []) {
     return this.stmt.all(...params);
   }

package/dist/driver/sql.js.js

@@ -7,12 +7,13 @@ var PreparedStatement = class {
     this.db = db;
     this.stmt = stmt;
   }
-  all(params) {
+  *iterate(params) {
     this.stmt.bind(params);
-    const res = [];
     while (this.stmt.step())
-      res.push(this.stmt.getAsObject());
-    return res;
+      yield this.stmt.getAsObject();
+  }
+  all(params) {
+    return Array.from(this.iterate(params));
   }
   run(params) {
     this.stmt.run(params);

package/dist/driver/sqlite3.js

@@ -6,6 +6,37 @@ var PreparedStatement = class {
   constructor(stmt) {
     this.stmt = stmt;
   }
+  async *iterate(params) {
+    let rows = [], done = false, error;
+    let resolve;
+    let promise = new Promise((r) => resolve = r);
+    this.stmt.each(
+      params,
+      (err, row) => {
+        if (err)
+          error = err;
+        else
+          rows.push(row);
+        resolve();
+      },
+      () => {
+        done = true;
+        resolve();
+      }
+    );
+    while (true) {
+      const mustWait = !(rows.length || error || done);
+      if (mustWait)
+        await promise;
+      promise = new Promise((r) => resolve = r);
+      if (error)
+        throw error;
+      if (rows.length)
+        yield rows.shift();
+      if (done)
+        return;
+    }
+  }
   all(params) {
     return new Promise((resolve, reject) => {
       this.stmt.all(params, (err, rows) => {

package/dist/lib/Driver.d.ts

@@ -41,6 +41,7 @@ declare abstract class SyncDriver extends DriverBase {
     get<T>(cursor: Cursor.SelectMultiple<T>): T | null;
     get<T>(cursor: Cursor<T>): T;
     get<T>(strings: TemplateStringsArray, ...params: Array<any>): T;
+    iterate<T>(cursor: Cursor.SelectMultiple<T>): Iterable<T>;
     transaction<T>(run: (query: SyncDriver) => T): T;
     toAsync(): SyncWrapper;
 }
@@ -48,6 +49,7 @@ interface SyncPreparedStatement {
     run(params?: Array<any>): {
         rowsAffected: number;
     };
+    iterate<T>(params?: Array<any>): Iterable<T>;
     all<T>(params?: Array<any>): Array<T>;
     get<T>(params?: Array<any>): T;
     execute(params?: Array<any>): void;
@@ -71,6 +73,7 @@ declare abstract class AsyncDriver extends DriverBase {
     get<T>(cursor: Cursor.SelectMultiple<T>): Promise<T | null>;
     get<T>(cursor: Cursor<T>): Promise<T>;
     get<T>(strings: TemplateStringsArray, ...params: Array<any>): Promise<T>;
+    iterate<T>(cursor: Cursor.SelectMultiple<T>): AsyncIterable<T>;
     transaction<T>(run: (query: AsyncDriver) => T): Promise<T>;
 }
 declare class SyncWrapper extends AsyncDriver {
@@ -86,6 +89,7 @@ interface AsyncPreparedStatement {
     run(params?: Array<any>): Promise<{
         rowsAffected: number;
     }>;
+    iterate<T>(params?: Array<any>): AsyncIterable<T>;
     all<T>(params?: Array<any>): Promise<Array<T>>;
     get<T>(params?: Array<any>): Promise<T>;
     execute(params?: Array<any>): Promise<void>;

package/dist/lib/Driver.js

@@ -141,6 +141,12 @@ var SyncDriver = class extends DriverBase {
   get(...args) {
     return super.get(...args);
   }
+  *iterate(cursor) {
+    const stmt = this.prepareStatement(this.formatter.compile(cursor.query()));
+    for (const row of stmt.iterate()) {
+      yield JSON.parse(row.result).result;
+    }
+  }
   transaction(run) {
     const id = `t${this.transactionId++}`;
     this.executeQuery(
@@ -240,6 +246,12 @@ var AsyncDriver = class extends DriverBase {
   get(...args) {
     return super.get(...args);
   }
+  async *iterate(cursor) {
+    const stmt = this.prepareStatement(this.formatter.compile(cursor.query()));
+    for await (const row of stmt.iterate()) {
+      yield JSON.parse(row.result).result;
+    }
+  }
   async transaction(run) {
     const id = `t${this.transactionId++}`;
     const [connection, release] = this.isolate();
@@ -266,6 +278,10 @@ var SyncPreparedStatementWrapper = class {
   constructor(stmt) {
     this.stmt = stmt;
   }
+  async *iterate(params) {
+    for (const row of this.stmt.iterate(params))
+      yield row;
+  }
   async run(params) {
     return this.stmt.run(params);
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "rado",
-  "version": "0.1.39",
+  "version": "0.1.40",
   "license": "MIT",
   "type": "module",
   "exports": {