enhance db operation code style

packages/ui/src/main/flow/OPEN_SETTING_WINDOW/ipc/index.ts

@@ -259,9 +259,7 @@ export default function initIpc () {
     return checkCookieListFormat(cookies)
   })
 
-  ipcMain.handle('connect-db', async () => {
+  ipcMain.handle('get-auto-start-chat-record', async () => {
-    const worker = await initDbWorker()
-    console.log(worker)
     const a = await getAutoStartChatRecord()
     console.log(a)
   })

packages/ui/src/main/flow/OPEN_SETTING_WINDOW/utils/db/index.ts

@@ -9,11 +9,11 @@ export const initDbWorker = () => {
     worker = createDbWorker()
     workerExitCode = null
     return new Promise((resolve, reject) => {
-      worker.once('exit', (exitCode) => {
+      worker!.once('exit', (exitCode) => {
         workerExitCode = exitCode
         worker = null
       })
-      worker.on('message', function handler(data) {
+      worker!.on('message', function handler(data) {
         if (data.type === 'DB_INIT_SUCCESS') {
           resolve(worker)
           // attach more event
@@ -34,16 +34,16 @@ export const initDbWorker = () => {
 const createWorkerPromise = async (data) => {
   await initDbWorker()
   const uuid = randomUUID()
-  worker.postMessage({
+  worker!.postMessage({
     _uuid: uuid,
     ...data
   })
   return new Promise((resolve) => {
-    worker.on('message', function handler(data) {
+    worker!.on('message', function handler(data) {
       const { _uuid, ...payload } = data ?? {}
       if (_uuid === uuid) {
         resolve(payload)
-        worker!.off('message', handler)
+        worker?.off('message', handler)
       }
     })
   })

packages/ui/src/main/flow/OPEN_SETTING_WINDOW/utils/db/worker/index.ts

@@ -25,11 +25,11 @@ dbInitPromise.then(
 )
 
 const payloadHandler = {
-  getAutoStartChatRecord(payload, callback) {
+  async getAutoStartChatRecord(payload) {
-    callback({
+    return {
       x: 'zzzz',
       ...payload
-    })
+    }
   }
 }
 
@@ -37,17 +37,14 @@ async function attachMessageHandler() {
   if (!dataSource) {
     await dbInitPromise
   }
-  parentPort?.on('message', (event) => {
+  parentPort?.on('message', async (event) => {
     const { _uuid, ...restObj } = event
     const { type } = event
 
-    const callback = (result) => {
+    const result = await payloadHandler[type](restObj)
-      parentPort?.postMessage({
+    parentPort?.postMessage({
-        _uuid,
+      _uuid,
-        data: result
+      data: result
-      })
+    })
-    }
-
-    payloadHandler[type](restObj, callback)
   })
 }

packages/ui/src/renderer/src/page/BootstrapSplash/index.vue

@@ -53,7 +53,7 @@ onMounted(async () => {
   })
 })
 
-electron.ipcRenderer.invoke('connect-db').then(() => {
+electron.ipcRenderer.invoke('get-auto-start-chat-record').then(() => {
   debugger
 }, () => {
   debugger