Add event listener using query object

src/query-builder.js

@@ -12,6 +12,7 @@ const {
 /** @typedef { import('./model') } Model */
 /** @typedef { import('./events/pipeline') } EventsPipeline */
 /** @typedef { import('./events/event') } Event */
+/** @typedef { import('./events/pipeline').EventListener } EventListener */
 
 /**
  * @callback Scope
@@ -206,6 +207,25 @@ class QueryBuilder extends BaseQueryBuilder {
         return new FetchingEvent()
     }
   }
+
+  /**
+   * @param {String} eventName
+   * @param {EventListener|Function} listener
+   * @param {Object} [options]
+   * @param {Boolean} [options.native=false]
+   * @return {QueryBuilder}
+   */
+  on (eventName, listener, options = {}) {
+    const { native = false } = options
+
+    if (native) {
+      return BaseQueryBuilder.prototype.on.call(this, eventName, listener)
+    }
+
+    this.events.on(eventName, listener)
+
+    return this
+  }
 }
 
 /**

tests/queries.test.js

@@ -1,4 +1,5 @@
 const test = require('ava')
+const sinon = require('sinon')
 const setupDb = require('./setup-db')
 const { equalQueries } = require('./assertions')
 const { createKex } = require('./utils')
@@ -36,3 +37,25 @@ test('forbid using table()', t => {
       .table('foo')
   })
 })
+
+test('events | add listener', async t => {
+  const User = createKex(t).createModel('User')
+
+  const listener = sinon.stub()
+
+  await User.query()
+    .on('fetching', listener)
+
+  t.true(listener.calledOnce)
+})
+
+test('events | add knex native listener', async t => {
+  const User = createKex(t).createModel('User')
+
+  const listener = sinon.stub()
+
+  await User.query()
+    .on('query', listener, { native: true })
+
+  t.true(listener.calledOnce)
+})