Seq6-custom

lib/dialects/abstract/query.js

@@ -331,6 +331,10 @@ class AbstractQuery {
   _logQuery(sql, debugContext, parameters) {
     const { connection, options } = this;
     const benchmark = this.sequelize.options.benchmark || options.benchmark;
+
+    const instanceType = connection.queryType === 'read' ? 'replica' : 'primary';
+    const instanceTypeMessage = `on ${instanceType} DB`;
+
     const logQueryParameters = this.sequelize.options.logQueryParameters || options.logQueryParameters;
     const startTime = Date.now();
     let logParameter = '';
@@ -355,7 +359,8 @@ class AbstractQuery {
       const afterMsg = `Executed ${fmt}`;
       debugContext(afterMsg);
       if (benchmark) {
-        this.sequelize.log(afterMsg, Date.now() - startTime, options);
+        const elapsedTime = Date.now() - startTime;
+        this.sequelize.log(`Executed ${elapsedTime > this.sequelize.options.queryThreshold ? 'SLOW QUERY ' : ''} (${instanceTypeMessage}, ${fmt}`, elapsedTime, options);
       }
     };
   }

lib/dialects/sqlite/query.js

@@ -402,6 +402,9 @@ class Query extends AbstractQuery {
         return new sequelizeErrors.TimeoutError(err, { stack: errStack });
 
       default:
+        if (err.message && err.message.includes("SQLITE_ERROR: no such table")) {
+          console.log(err.sql);
+        }
         return new sequelizeErrors.DatabaseError(err, { stack: errStack });
     }
   }

lib/hooks.js

@@ -48,7 +48,8 @@ const hookTypes = {
   beforeBulkSync: { params: 1 },
   afterBulkSync: { params: 1 },
   beforeQuery: { params: 2 },
-  afterQuery: { params: 2 }
+  afterQuery: { params: 2 },
+  beforeGetConnection: {params: 2}
 };
 exports.hooks = hookTypes;
 

lib/sequelize.js

@@ -617,7 +617,9 @@ class Sequelize {
 
       checkTransaction();
 
-      const connection = await (options.transaction ? options.transaction.connection : this.connectionManager.getConnection(options));
+      const connection = await (options.transaction ? 
+        options.transaction.connection : 
+        this.runHooks('beforeGetConnection', options, sql).then(() => this.connectionManager.getConnection(options)));
       const query = new this.dialect.Query(connection, this, options);
 
       try {