wood/Q58Bot
1
0
Fork 0
mirror of https://github.com/woodchen-ink/Q58Bot.git synced 2025-07-18 13:52:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复

Browse Source
This commit is contained in:
wood chen 2024-09-28 15:37:48 +08:00
parent 8661f5c774
commit 190155ff5b
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
core/database.go
View File

@ -359,24 +359,24 @@ func (d *Database) Close() error {
func (d *Database) MigrateExistingKeywords() error { func (d *Database) MigrateExistingKeywords() error {
// 检查是否已经执行过迁移 // 检查是否已经执行过迁移
var migrationDone bool var migrationDone string
err := d.db.QueryRow("SELECT value FROM config WHERE key = 'keywords_migrated'").Scan(&migrationDone) err := d.db.QueryRow("SELECT value FROM config WHERE key = 'keywords_migrated'").Scan(&migrationDone)
if err != nil && err != sql.ErrNoRows { if err != nil && err != sql.ErrNoRows {
return err return err
} }
if migrationDone { if migrationDone == "true" {
return nil // 迁移已经完成，无需再次执行 return nil // 迁移已经完成，无需再次执行
} }
// 检查旧表是否存在 // 检查旧表是否存在
var oldTableExists bool var oldTableName string
err = d.db.QueryRow("SELECT name FROM sqlite_master WHERE type='table' AND name='keywords'").Scan(&oldTableExists) err = d.db.QueryRow("SELECT name FROM sqlite_master WHERE type='table' AND name='keywords'").Scan(&oldTableName)
if err != nil && err != sql.ErrNoRows { if err != nil && err != sql.ErrNoRows {
return err return err
} }
if oldTableExists { if oldTableName == "keywords" {
// 迁移数据 // 迁移数据
_, err = d.db.Exec(`INSERT OR IGNORE INTO keywords_new (keyword, is_link, is_auto_added, added_at) _, err = d.db.Exec(`INSERT OR IGNORE INTO keywords_new (keyword, is_link, is_auto_added, added_at)
SELECT keyword, SELECT keyword,