mirror of
https://github.com/grafana/grafana.git
synced 2024-12-01 21:19:28 -06:00
124 lines
3.1 KiB
Go
124 lines
3.1 KiB
Go
package cleanup
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/infra/serverlock"
|
|
"github.com/grafana/grafana/pkg/log"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/registry"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
)
|
|
|
|
type CleanUpService struct {
|
|
log log.Logger
|
|
Cfg *setting.Cfg `inject:""`
|
|
ServerLockService *serverlock.ServerLockService `inject:""`
|
|
}
|
|
|
|
func init() {
|
|
registry.RegisterService(&CleanUpService{})
|
|
}
|
|
|
|
func (srv *CleanUpService) Init() error {
|
|
srv.log = log.New("cleanup")
|
|
return nil
|
|
}
|
|
|
|
func (srv *CleanUpService) Run(ctx context.Context) error {
|
|
srv.cleanUpTmpFiles()
|
|
|
|
ticker := time.NewTicker(time.Minute * 10)
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
srv.cleanUpTmpFiles()
|
|
srv.deleteExpiredSnapshots()
|
|
srv.deleteExpiredDashboardVersions()
|
|
srv.ServerLockService.LockAndExecute(ctx, "delete old login attempts", time.Minute*10, func() {
|
|
srv.deleteOldLoginAttempts()
|
|
})
|
|
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (srv *CleanUpService) cleanUpTmpFiles() {
|
|
if _, err := os.Stat(srv.Cfg.ImagesDir); os.IsNotExist(err) {
|
|
return
|
|
}
|
|
|
|
files, err := ioutil.ReadDir(srv.Cfg.ImagesDir)
|
|
if err != nil {
|
|
srv.log.Error("Problem reading image dir", "error", err)
|
|
return
|
|
}
|
|
|
|
var toDelete []os.FileInfo
|
|
var now = time.Now()
|
|
|
|
for _, file := range files {
|
|
if srv.shouldCleanupTempFile(file.ModTime(), now) {
|
|
toDelete = append(toDelete, file)
|
|
}
|
|
}
|
|
|
|
for _, file := range toDelete {
|
|
fullPath := path.Join(srv.Cfg.ImagesDir, file.Name())
|
|
err := os.Remove(fullPath)
|
|
if err != nil {
|
|
srv.log.Error("Failed to delete temp file", "file", file.Name(), "error", err)
|
|
}
|
|
}
|
|
|
|
srv.log.Debug("Found old rendered image to delete", "deleted", len(toDelete), "kept", len(files))
|
|
}
|
|
|
|
func (srv *CleanUpService) shouldCleanupTempFile(filemtime time.Time, now time.Time) bool {
|
|
if srv.Cfg.TempDataLifetime == 0 {
|
|
return false
|
|
}
|
|
|
|
return filemtime.Add(srv.Cfg.TempDataLifetime).Before(now)
|
|
}
|
|
|
|
func (srv *CleanUpService) deleteExpiredSnapshots() {
|
|
cmd := m.DeleteExpiredSnapshotsCommand{}
|
|
if err := bus.Dispatch(&cmd); err != nil {
|
|
srv.log.Error("Failed to delete expired snapshots", "error", err.Error())
|
|
} else {
|
|
srv.log.Debug("Deleted expired snapshots", "rows affected", cmd.DeletedRows)
|
|
}
|
|
}
|
|
|
|
func (srv *CleanUpService) deleteExpiredDashboardVersions() {
|
|
cmd := m.DeleteExpiredVersionsCommand{}
|
|
if err := bus.Dispatch(&cmd); err != nil {
|
|
srv.log.Error("Failed to delete expired dashboard versions", "error", err.Error())
|
|
} else {
|
|
srv.log.Debug("Deleted old/expired dashboard versions", "rows affected", cmd.DeletedRows)
|
|
}
|
|
}
|
|
|
|
func (srv *CleanUpService) deleteOldLoginAttempts() {
|
|
if srv.Cfg.DisableBruteForceLoginProtection {
|
|
return
|
|
}
|
|
|
|
cmd := m.DeleteOldLoginAttemptsCommand{
|
|
OlderThan: time.Now().Add(time.Minute * -10),
|
|
}
|
|
if err := bus.Dispatch(&cmd); err != nil {
|
|
srv.log.Error("Problem deleting expired login attempts", "error", err.Error())
|
|
} else {
|
|
srv.log.Debug("Deleted expired login attempts", "rows affected", cmd.DeletedRows)
|
|
}
|
|
}
|