hunterlong/statup

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
// env: REMOVE_AFTER - golang duration parsed time for deleting records older than REMOVE_AFTER duration from now
// env: CLEANUP_INTERVAL - golang duration parsed time for checking old records routine
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)
    }
}