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

fix

Browse Source
This commit is contained in:
wood chen 2024-09-28 15:41:21 +08:00
parent bb6447e975
commit fd0df99204
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
core/database.go
View File

@ -44,15 +44,15 @@ func NewDatabase() (*Database, error) {
func (d *Database) createTables() error { func (d *Database) createTables() error {
queries := []string{ queries := []string{
`CREATE TABLE IF NOT EXISTS keywords_new ( `CREATE TABLE IF NOT EXISTS keywords (
id INTEGER PRIMARY KEY, id INTEGER PRIMARY KEY,
keyword TEXT UNIQUE, keyword TEXT UNIQUE,
is_link BOOLEAN DEFAULT FALSE, is_link BOOLEAN DEFAULT FALSE,
is_auto_added BOOLEAN DEFAULT FALSE, is_auto_added BOOLEAN DEFAULT FALSE,
added_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP added_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
)`, )`,
`CREATE INDEX IF NOT EXISTS idx_keyword ON keywords_new(keyword)`, `CREATE INDEX IF NOT EXISTS idx_keyword ON keywords(keyword)`,
`CREATE INDEX IF NOT EXISTS idx_added_at ON keywords_new(added_at)`, `CREATE INDEX IF NOT EXISTS idx_added_at ON keywords(added_at)`,
`CREATE TABLE IF NOT EXISTS whitelist ( `CREATE TABLE IF NOT EXISTS whitelist (
id INTEGER PRIMARY KEY, id INTEGER PRIMARY KEY,
domain TEXT UNIQUE domain TEXT UNIQUE
@ -405,8 +405,6 @@ func (d *Database) MigrateExistingKeywords() error {
if contains(columns, col) { if contains(columns, col) {
insertColumns = append(insertColumns, col) insertColumns = append(insertColumns, col)
selectColumns = append(selectColumns, col) selectColumns = append(selectColumns, col)
} else {
selectColumns = append(selectColumns, "FALSE")
} }
} }
@ -427,12 +425,18 @@ func (d *Database) MigrateExistingKeywords() error {
if err != nil { if err != nil {
return err return err
} }
} } else {
// 如果旧表不存在，创建一个空的新表
// 重命名新表 _, err = d.db.Exec(`CREATE TABLE IF NOT EXISTS keywords (
_, err = d.db.Exec("ALTER TABLE keywords_new RENAME TO keywords") id INTEGER PRIMARY KEY,
if err != nil { keyword TEXT UNIQUE,
return err is_link BOOLEAN DEFAULT FALSE,
is_auto_added BOOLEAN DEFAULT FALSE,
added_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
)`)
if err != nil {
return err
}
} }
// 更新配置，标记迁移已完成 // 更新配置，标记迁移已完成