refactor(sqlite3): 重构SQLite数据处理逻辑

- 删除旧版SQLite.js文件
- 更新主进程文件中的数据库交互逻辑
- 优化数据库的连接方式
This commit is contained in:
ApplePine 2025-07-02 13:18:05 +08:00
parent 2582645ffa
commit 0208de66bb
2 changed files with 6 additions and 179 deletions

View File

@ -1,179 +0,0 @@
const sqlite3 = require('sqlite3').verbose()
const path = require('path')
const fs = require('fs')
const { app } = require('electron')
// 数据库路径
const dbPath = path.join(app.getPath('userData'), 'database.sqlite')
let db
// 初始化数据库
function initDatabase() {
const dbDir = path.dirname(dbPath)
if (!fs.existsSync(dbDir)) {
fs.mkdirSync(dbDir, { recursive: true })
}
db = new sqlite3.Database(dbPath)
// 创建表
db.serialize(() => {
db.run(`
CREATE TABLE IF NOT EXISTS files (
file_id INTEGER PRIMARY KEY AUTOINCREMENT,
file_name TEXT NOT NULL,
file_ext TEXT,
file_cat TEXT,
file_tags TEXT,
file_desc TEXT,
file_created_time DATETIME,
file_updated_time DATETIME,
file_path TEXT NOT NULL UNIQUE,
is_directory BOOLEAN DEFAULT 0,
added_time DATETIME DEFAULT CURRENT_TIMESTAMP
)
`)
})
console.log('Database initialized at:', dbPath)
return db
}
// 添加文件记录
function addFile(fileData) {
return new Promise((resolve, reject) => {
const stmt = db.prepare(`
INSERT INTO files (
file_name, file_ext, file_cat, file_tags, file_desc,
file_created_time, file_updated_time, file_path, is_directory
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
`)
stmt.run(
fileData.fileName,
fileData.fileExt,
fileData.fileCat,
JSON.stringify(fileData.fileTags || []),
fileData.fileDesc,
fileData.fileCreatedTime,
fileData.fileUpdatedTime,
fileData.filePath,
fileData.isDirectory ? 1 : 0,
function (err) {
if (err) {
reject(err)
} else {
resolve(this.lastID)
}
},
)
stmt.finalize()
})
}
// 搜索文件
function searchFiles(criteria) {
return new Promise((resolve, reject) => {
let query = `SELECT * FROM files WHERE 1=1`
const params = []
if (criteria.fileName) {
query += ` AND file_name LIKE ?`
params.push(`%${criteria.fileName}%`)
}
if (criteria.fileCat) {
query += ` AND file_cat = ?`
params.push(criteria.fileCat)
}
if (criteria.fileTags && criteria.fileTags.length > 0) {
const tagConditions = criteria.fileTags.map(() => `file_tags LIKE ?`).join(' OR ')
query += ` AND (${tagConditions})`
criteria.fileTags.forEach((tag) => {
params.push(`%${tag}%`)
})
}
if (criteria.startDate) {
query += ` AND added_time >= ?`
params.push(criteria.startDate)
}
if (criteria.endDate) {
query += ` AND added_time <= ?`
params.push(criteria.endDate)
}
db.all(query, params, (err, rows) => {
if (err) {
reject(err)
} else {
// 解析JSON格式的标签
const results = rows.map((row) => ({
...row,
fileTags: JSON.parse(row.file_tags || '[]'),
}))
resolve(results)
}
})
})
}
// 获取统计数据
function getStats() {
return new Promise((resolve, reject) => {
const stats = {}
// 获取文件总数
db.get('SELECT COUNT(*) as count FROM files WHERE is_directory = 0', (err, row) => {
if (err) {
reject(err)
return
}
stats.fileCount = row.count
// 获取文件夹总数
db.get('SELECT COUNT(*) as count FROM files WHERE is_directory = 1', (err, row) => {
if (err) {
reject(err)
return
}
stats.dirCount = row.count
// 获取文件类型分布
db.all(
'SELECT file_ext, COUNT(*) as count FROM files WHERE is_directory = 0 GROUP BY file_ext',
(err, rows) => {
if (err) {
reject(err)
return
}
stats.fileTypes = rows
// 获取最近添加的文件
db.all('SELECT * FROM files ORDER BY added_time DESC LIMIT 10', (err, rows) => {
if (err) {
reject(err)
} else {
stats.recentFiles = rows.map((row) => ({
...row,
fileTags: JSON.parse(row.file_tags || '[]'),
}))
resolve(stats)
}
})
},
)
})
})
})
}
module.exports = {
initDatabase,
addFile,
searchFiles,
getStats,
}

View File

@ -16,9 +16,15 @@ let mainWindow
let db let db
function createWindow() { function createWindow() {
// 设置应用图标路径
const iconPath = process.platform === 'darwin'
? path.join(__dirname, '../public/icons/mac.icns')
: path.join(__dirname, '../public/icons/win.ico')
mainWindow = new BrowserWindow({ mainWindow = new BrowserWindow({
width: 1200, width: 1200,
height: 800, height: 800,
icon: iconPath,
webPreferences: { webPreferences: {
preload: path.join(__dirname, 'preload.cjs'), preload: path.join(__dirname, 'preload.cjs'),
contextIsolation: true, contextIsolation: true,