mirror of https://github.com/statping/statping
51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/statping/statping/utils"
|
|
"time"
|
|
|
|
_ "github.com/jinzhu/gorm/dialects/mysql"
|
|
_ "github.com/jinzhu/gorm/dialects/postgres"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
var (
|
|
log = utils.Log.WithField("type", "database")
|
|
)
|
|
|
|
// Maintenance will automatically delete old records from 'failures' and 'hits'
|
|
// this function is currently set to delete records 7+ days old every 60 minutes
|
|
func Maintenance() {
|
|
dur := utils.Params.GetDuration("REMOVE_AFTER")
|
|
interval := utils.Params.GetDuration("CLEANUP_INTERVAL")
|
|
|
|
log.Infof("Database Cleanup runs every %s and will remove records older than %s", interval.String(), dur.String())
|
|
ticker := interval
|
|
|
|
for {
|
|
select {
|
|
case <-time.After(ticker):
|
|
deleteAfter := utils.Now().Add(-dur)
|
|
|
|
log.Infof("Deleting failures older than %s", deleteAfter.String())
|
|
deleteAllSince("failures", deleteAfter)
|
|
|
|
log.Infof("Deleting hits older than %s", deleteAfter.String())
|
|
deleteAllSince("hits", deleteAfter)
|
|
|
|
ticker = interval
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// deleteAllSince will delete a specific table's records based on a time.
|
|
func deleteAllSince(table string, date time.Time) {
|
|
sql := fmt.Sprintf("DELETE FROM %s WHERE created_at < '%s'", table, database.FormatTime(date))
|
|
log.Info(sql)
|
|
if err := database.Exec(sql).Error(); err != nil {
|
|
log.WithField("query", sql).Errorln(err)
|
|
}
|
|
}
|