Include sequentially on transaction

hgourvest · Nov 27, 2020 · 3f62863 · 3f62863
1 parent 97b5c67
commit 3f62863
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 11 deletions.
diff --git a/lib/index.d.ts b/lib/index.d.ts
@@ -30,6 +30,7 @@ declare module 'node-firebird' {
     export interface Transaction {
         query(query: string, params: any[], callback: QueryCallback): void;
         execute(query: string, params: any[], callback: QueryCallback): void;
+        sequentially(query: string, params: any[], rowCallback: SequentialCallback, callback: SimpleCallback, asArray?: boolean): Database;
         commit(callback?: SimpleCallback): void;
         commitRetaining(callback?: SimpleCallback): void;
         rollback(callback?: SimpleCallback): void;

diff --git a/lib/index.js b/lib/index.js
@@ -1463,6 +1463,29 @@ Transaction.prototype.execute = function(query, params, callback, custom) {
     });
 };
 
+Transaction.prototype.sequentially = function (query, params, on, callback, asArray) {
+
+	if (params instanceof Function) {
+		asArray = callback;
+		callback = on;
+		on = params;
+		params = undefined;
+	}
+
+	if (on === undefined){
+		throw new Error('Expected "on" delegate.');
+	}
+
+	if (callback instanceof Boolean) {
+		asArray = callback;
+		callback = undefined;
+	}
+
+	var self = this;
+	self.execute(query, params, callback, { asObject: !asArray, asStream: true, on: on });
+	return self;
+};
+
 Transaction.prototype.query = function(query, params, callback) {
 
     if (params instanceof Function) {
@@ -1616,19 +1639,25 @@ Database.prototype.execute = function(query, params, callback, custom) {
 
 Database.prototype.sequentially = function(query, params, on, callback, asArray) {
 
-    if (params instanceof Function) {
-        asArray = callback;
-        callback = on;
-        on = params;
-        params = undefined;
-    }
+	if (params instanceof Function) {
+		asArray = callback;
+		callback = on;
+		on = params;
+		params = undefined;
+	}
+
+	if (on === undefined){
+		throw new Error('Expected "on" delegate.');
+	}
 
-    if (on === undefined)
-        throw new Error('Expected "on" delegate.');
+	if (callback instanceof Boolean) {
+		asArray = callback;
+		callback = undefined;
+	}
 
-    var self = this;
-    self.execute(query, params, callback, { asObject: !asArray, asStream: true, on: on });
-    return self;
+	var self = this;
+	self.execute(query, params, callback, { asObject: !asArray, asStream: true, on: on });
+	return self;
 };
 
 Database.prototype.query = function(query, params, callback) {