fix(dialect-generic-sqlite): custom message for rest type

Author: subframe7536

packages/dialect-generic-sqlite/README.md

@@ -212,7 +212,7 @@ import { createNodeOnMessageCallback } from 'kysely-generic-sqlite/worker-helper
 
 createNodeOnMessageCallback<{}, Database>(
   data => createSqliteExecutor(':memory:'),
-  ([type, exec, data1, data2, data3]) => {
+  (exec, type, data1, data2, data3) => {
     if (type === 'test') {
       exec.db.pragma('optimize')
       console.log(data1) // 'your-data'

packages/dialect-generic-sqlite/src/worker/types.ts

@@ -94,8 +94,8 @@ export type InitFn<
  * Function that handle all message
  */
 export type MessageHandleFn<DB = unknown> = (
-  type: string,
   exec: IGenericSqlite<DB>,
+  type: string,
   data1: unknown,
   data2: unknown,
   data3: unknown

packages/dialect-generic-sqlite/src/worker/utils.ts

@@ -15,7 +15,7 @@ import {
  * Create generic message handler
  * @param init Function that init sqlite executor
  * @param post Function that post message to main thread
- * @param message Handle all messages. If returning data is not `undefined` and `null`, it will be set as second element of first param
+ * @param message Handle custom messages. If returning data is not `undefined` and `null`, it will be set as second element of first param
  */
 export function createGenericOnMessageCallback<T extends Record<string, unknown>, DB = unknown>(
   init: InitFn<T, DB>,
@@ -50,11 +50,13 @@ export function createGenericOnMessageCallback<T extends Record<string, unknown>
           ret[0] = endEvent
           break
         }
-      }
-      if (message) {
-        const data = await message(type, db, data1, data2, data3)
-        if (data !== undefined && data !== null) {
-          ret[1] = data
+        default: {
+          if (message) {
+            const data = await message(db, type, data1, data2, data3)
+            if (data !== undefined && data !== null) {
+              ret[1] = data
+            }
+          }
         }
       }
     } catch (error) {