如何在Knex中扩展QueryBuilder类时访问当前上下文的事务

const knex = require('knex');

class CustomQueryBuilder extends knex.QueryBuilder {
  constructor(...args) {
    super(...args);
    this.transaction = null;
  }

  setTransaction(transaction) {
    this.transaction = transaction;
  }

  // Override other methods as needed
  // ...

  // Override execute method to use transaction if available
  execute() {
    if (this.transaction) {
      // Use transaction for query execution
      return this.transaction.query(this.toSQL());
    } else {
      // Use default query execution
      return super.execute();
    }
  }
}

// Usage example
const customKnex = knex({
  // Knex configuration
  // ...
});

const trx = customKnex.transaction();
const queryBuilder = new CustomQueryBuilder(customKnex);
queryBuilder.setTransaction(trx);

queryBuilder
  .select('*')
  .from('table')
  .then((result) => {
    // Handle query result
  })
  .catch((error) => {
    // Handle query error
  })
  .finally(() => {
    trx.commit();
  });