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

refactor(database): modify CleanupExpiredLinks to return rows affected

Browse Source
This commit is contained in:
wood chen 2024-10-20 22:07:25 +08:00
parent e3cfd8e4bc
commit af013336ce
2 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
core/database.go
View File

@ -122,14 +122,20 @@ func (d *Database) RemoveKeyword(keyword string) (bool, error) {
return rowsAffected > 0, nil
}
func (d *Database) CleanupExpiredLinks() error {
func (d *Database) CleanupExpiredLinks() (int64, error) {
twoMonthsAgo := time.Now().AddDate(0, -2, 0)
_, err := d.db.Exec("DELETE FROM keywords WHERE is_link = TRUE AND is_auto_added = TRUE AND added_at < ?", twoMonthsAgo)
result, err := d.db.Exec("DELETE FROM keywords WHERE is_link = TRUE AND is_auto_added = TRUE AND added_at < ?", twoMonthsAgo)
if err != nil {
return err
return 0, err
}
rowsAffected, err := result.RowsAffected()
if err != nil {
return 0, err
}
d.invalidateCache("keywords")
return nil
return rowsAffected, nil
}
func (d *Database) GetAllKeywords() ([]string, error) {

4
service/scheduled_tasks.go
View File

@ -28,10 +28,10 @@ func periodicCleanup() {
}
func cleanupExpiredLinks() {
err := core.DB.CleanupExpiredLinks()
rowsAffected, err := core.DB.CleanupExpiredLinks()
if err != nil {
log.Printf("清理过期链接时发生错误: %v", err)
} else {
log.Println("已成功清理过期链接")
log.Printf("已成功清理 %d 条过期链接", rowsAffected)
}
}