optimized blocking issue during bulk log data deletion

Signed-off-by: wozulong <>
This commit is contained in:
wozulong
2024-09-10 17:24:44 +08:00
parent 208bc5e794
commit a60f209c85
2 changed files with 22 additions and 4 deletions

View File

@@ -247,7 +247,25 @@ func SumUsedToken(logType int, startTimestamp int64, endTimestamp int64, modelNa
return token
}
func DeleteOldLog(targetTimestamp int64) (int64, error) {
result := LOG_DB.Where("created_at < ?", targetTimestamp).Delete(&Log{})
return result.RowsAffected, result.Error
func DeleteOldLog(ctx context.Context, targetTimestamp int64, limit int) (int64, error) {
var total int64 = 0
for {
if nil != ctx.Err() {
return total, ctx.Err()
}
result := LOG_DB.Where("created_at < ?", targetTimestamp).Limit(limit).Delete(&Log{})
if nil != result.Error {
return total, result.Error
}
if 0 == result.RowsAffected {
break
}
total += result.RowsAffected
}
return total, nil
}