feat: optimize sql execution and import workflows

Author: kk1181958464

electron.vite.config.ts

@@ -18,7 +18,18 @@ export default defineConfig({
     root: resolve('src/renderer'),
     build: {
       rollupOptions: {
-        input: resolve('src/renderer/index.html')
+        input: resolve('src/renderer/index.html'),
+        output: {
+          manualChunks(id) {
+            if (!id.includes('node_modules')) return undefined
+            if (id.includes('monaco-editor') || id.includes('@monaco-editor')) return 'vendor-monaco'
+            if (id.includes('@ant-design/icons')) return 'vendor-icons'
+            if (id.includes('react') || id.includes('scheduler')) return 'vendor-react'
+            if (id.includes('zustand')) return 'vendor-state'
+            if (id.includes('lodash') || id.includes('dayjs') || id.includes('xlsx')) return 'vendor-utils'
+            return 'vendor'
+          },
+        },
       }
     },
     plugins: [react()],

src/main/index.ts

@@ -5,6 +5,7 @@ import { IPC } from '../shared/types/ipc-channels'
 import { registerAllIPC } from './ipc/index'
 import * as connectionManager from './services/connection-manager'
 import * as localStore from './services/local-store'
+import * as backupService from './services/backup'
 import * as logger from './utils/logger'
 
 let mainWindow: BrowserWindow | null = null
@@ -152,6 +153,7 @@ app.whenReady().then(() => {
   logger.info('App starting...')
   localStore.init()
   connectionManager.initializeHeartbeatInterval()
+  backupService.startScheduleRunner()
   registerAllIPC()
 
   // Window control IPC
@@ -193,6 +195,7 @@ app.on('before-quit', async () => {
   logger.info('App quitting, cleaning up...')
   tray?.destroy()
   tray = null
+  backupService.stopScheduleRunner()
   localStore.flushLocalStoreQueues()
   await connectionManager.disconnectAll()
 })

src/main/ipc/backup.ipc.ts

@@ -1,22 +1,21 @@
 import { ipcMain } from 'electron'
 import { IPC } from '../../shared/types/ipc-channels'
 import * as backup from '../services/backup'
-import * as localStore from '../services/local-store'
 
 export function registerBackupIPC() {
   ipcMain.handle(IPC.BACKUP_CREATE, async (_e, config) => {
     return backup.createBackup(config)
   })
 
-  ipcMain.handle(IPC.BACKUP_RESTORE, async (_e, connId: string, filePath: string) => {
-    return backup.restoreBackup(connId, filePath)
+  ipcMain.handle(IPC.BACKUP_RESTORE, async (_e, connId: string, filePath: string, options) => {
+    return backup.restoreBackup(connId, filePath, options)
   })
 
   ipcMain.handle(IPC.BACKUP_LIST, async (_e, connId: string) => {
     return backup.listBackups(connId)
   })
 
-  ipcMain.handle(IPC.BACKUP_SCHEDULE, async (_e, schedule) => {
-    localStore.backupSchedules.save(schedule)
+  ipcMain.handle(IPC.BACKUP_SCHEDULE, async (_e, request) => {
+    return backup.handleScheduleRequest(request)
   })
 }

src/main/ipc/import-export.ipc.ts

@@ -16,7 +16,7 @@ type ExportDataOptions = {
   insertStyle?: 'single' | 'multi' | 'ignore' | 'replace'
 }
 
-const IMPORT_EXTS = new Set(['.csv', '.tsv', '.xlsx', '.xls'])
+const IMPORT_EXTS = new Set(['.csv', '.tsv', '.xlsx', '.xls', '.sql', '.gz'])
 const EXPORT_EXTS = new Set(['.csv', '.json', '.sql', '.xlsx', '.xls'])
 
 function validateFilePath(filePath: string, allowedExts: Set<string>): void {
@@ -36,6 +36,9 @@ export function registerImportExportIPC() {
   ipcMain.handle(IPC.IMPORT_FILE, async (_e, connId: string, db: string, table: string, filePath: string, options?: ImportOptions) => {
     validateFilePath(filePath, IMPORT_EXTS)
     const ext = path.extname(filePath).toLowerCase()
+    if (ext === '.sql' || filePath.toLowerCase().endsWith('.sql.gz')) {
+      return importExport.importSQL(connId, db, filePath, options)
+    }
     if (ext === '.csv' || ext === '.tsv') {
       return importExport.importCSV(connId, db, table, filePath, options)
     }
@@ -44,6 +47,10 @@ export function registerImportExportIPC() {
 
   ipcMain.handle(IPC.IMPORT_PREVIEW, async (_e, filePath: string) => {
     validateFilePath(filePath, IMPORT_EXTS)
+    const lower = filePath.toLowerCase()
+    if (lower.endsWith('.sql') || lower.endsWith('.sql.gz')) {
+      return { columns: [], rows: [], totalRows: 0 }
+    }
     return importExport.previewImport(filePath)
   })