feat: add log level option

subframe7536 · Oct 11, 2024 · 90b3f45 · 90b3f45
1 parent d3bc46c
commit 90b3f45
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/logger.ts b/src/logger.ts
@@ -12,6 +12,11 @@ export type LoggerParams = {
 }
 
 export type LoggerOptions = {
+  /**
+   * Toggle log or setup log level
+   * @default 'error'
+   */
+  enable?: boolean | LogEvent['level']
   /**
    * Log functions
    * @param data logger params, see {@link LoggerParams}
@@ -44,11 +49,14 @@ export type LoggerOptions = {
  * })
  */
 export function createKyselyLogger(
-  options: LoggerOptions,
+  options: LoggerOptions = {},
 ): (event: LogEvent) => void {
-  const { logger = console.log, merge, logQueryNode } = options
+  const { enable = 'error', logger = console.log, merge, logQueryNode } = options
 
   return (event: LogEvent) => {
+    if (!enable || (typeof enable === 'string' && event.level !== enable)) {
+      return
+    }
     const { level, queryDurationMillis, query: { parameters, sql, query } } = event
     const questionMarker = '_Q_'
     const err = level === 'error' ? event.error : undefined