66 lines
2.1 KiB
Go
66 lines
2.1 KiB
Go
|
package database
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"time"
|
|||
|
|
"gorm.io/gorm" // 【中文注释】: 确保 gorm 被导入，以便在函数签名中使用
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// LinkHistory GORM aodel for link_history table
|
|||
|
|
type LinkHistory struct {
|
|||
|
|
Id int `gorm:"primaryKey"`
|
|||
|
|
Type string `gorm:"type:varchar(255);not null"`
|
|||
|
|
Link string `gorm:"type:text;not null"`
|
|||
|
|
CreatedAt time.Time `gorm:"not null"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AddLinkHistory 在一个事务中添加新链接记录并修剪旧记录。
|
|||
|
|
// 它确保了操作的原子性：所有更改要么全部应用，要么全部回滚。
|
|||
|
|
func AddLinkHistory(record *LinkHistory) error {
|
|||
|
|
// 【核心修正】: 使用 GORM 的事务功能来包装所有的数据库写入和删除操作。
|
|||
|
|
// 这样可以确保数据的一致性。
|
|||
|
|
return db.Transaction(func(tx *gorm.DB) error {
|
|||
|
|
// 1. 添加新记录
|
|||
|
|
// 【重要】: 在事务内部，必须使用 tx 变量，而不是全局的 db 变量。
|
|||
|
|
if err := tx.Create(record).Error; err != nil {
|
|||
|
|
return err // 如果这里出错，事务将自动回滚
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 修剪旧记录，仅保留最近的 10 条
|
|||
|
|
var count int64
|
|||
|
|
// 【重要】: 使用 tx 进行计数
|
|||
|
|
if err := tx.Model(&LinkHistory{}).Count(&count).Error; err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if count > 10 {
|
|||
|
|
limit := int(count) - 10
|
|||
|
|
var recordsToDelete []LinkHistory
|
|||
|
|
// 【重要】: 使用 tx 查找要删除的记录
|
|||
|
|
if err := tx.Order("created_at asc").Limit(limit).Find(&recordsToDelete).Error; err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
if len(recordsToDelete) > 0 {
|
|||
|
|
// 【重要】: 使用 tx 删除记录
|
|||
|
|
if err := tx.Delete(&recordsToDelete).Error; err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 【核心修正】: 从此函数中移除了 Checkpoint() 调用。
|
|||
|
|
// 事务成功后返回 nil，GORM 会自动提交事务。
|
|||
|
|
return nil
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
// GetLinkHistory retrieves the 10 most recent link records
|
|||
|
|
func GetLinkHistory() ([]*LinkHistory, error) {
|
|||
|
|
var histories []*LinkHistory
|
|||
|
|
err := db.Order("created_at desc").Limit(10).Find(&histories).Error
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return histories, nil
|
|||
|
|
}
|