mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
fe49182b9d
Snapshot cleanup did not work due to time.Now syntax error. Added test for it as well to catch any future errors. Added error and debug logging so that it is possible to see any errors in the future. Removed an unused configuration value and deprecated the remove expired snapshots setting.
117 lines
2.9 KiB
Go
117 lines
2.9 KiB
Go
package cleanup
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"time"
|
|
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/log"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
)
|
|
|
|
type CleanUpService struct {
|
|
log log.Logger
|
|
}
|
|
|
|
func NewCleanUpService() *CleanUpService {
|
|
return &CleanUpService{
|
|
log: log.New("cleanup"),
|
|
}
|
|
}
|
|
|
|
func (service *CleanUpService) Run(ctx context.Context) error {
|
|
service.log.Info("Initializing CleanUpService")
|
|
|
|
g, _ := errgroup.WithContext(ctx)
|
|
g.Go(func() error { return service.start(ctx) })
|
|
|
|
err := g.Wait()
|
|
service.log.Info("Stopped CleanUpService", "reason", err)
|
|
return err
|
|
}
|
|
|
|
func (service *CleanUpService) start(ctx context.Context) error {
|
|
service.cleanUpTmpFiles()
|
|
|
|
ticker := time.NewTicker(time.Minute * 10)
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
service.cleanUpTmpFiles()
|
|
service.deleteExpiredSnapshots()
|
|
service.deleteExpiredDashboardVersions()
|
|
service.deleteOldLoginAttempts()
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (service *CleanUpService) cleanUpTmpFiles() {
|
|
if _, err := os.Stat(setting.ImagesDir); os.IsNotExist(err) {
|
|
return
|
|
}
|
|
|
|
files, err := ioutil.ReadDir(setting.ImagesDir)
|
|
if err != nil {
|
|
service.log.Error("Problem reading image dir", "error", err)
|
|
return
|
|
}
|
|
|
|
var toDelete []os.FileInfo
|
|
for _, file := range files {
|
|
if file.ModTime().AddDate(0, 0, 1).Before(time.Now()) {
|
|
toDelete = append(toDelete, file)
|
|
}
|
|
}
|
|
|
|
for _, file := range toDelete {
|
|
fullPath := path.Join(setting.ImagesDir, file.Name())
|
|
err := os.Remove(fullPath)
|
|
if err != nil {
|
|
service.log.Error("Failed to delete temp file", "file", file.Name(), "error", err)
|
|
}
|
|
}
|
|
|
|
service.log.Debug("Found old rendered image to delete", "deleted", len(toDelete), "keept", len(files))
|
|
}
|
|
|
|
func (service *CleanUpService) deleteExpiredSnapshots() {
|
|
cmd := m.DeleteExpiredSnapshotsCommand{}
|
|
if err := bus.Dispatch(&cmd); err != nil {
|
|
service.log.Error("Failed to delete expired snapshots", "error", err.Error())
|
|
} else {
|
|
service.log.Debug("Deleted expired snapshots", "rows affected", cmd.DeletedRows)
|
|
}
|
|
}
|
|
|
|
func (service *CleanUpService) deleteExpiredDashboardVersions() {
|
|
cmd := m.DeleteExpiredVersionsCommand{}
|
|
if err := bus.Dispatch(&cmd); err != nil {
|
|
service.log.Error("Failed to delete expired dashboard versions", "error", err.Error())
|
|
} else {
|
|
service.log.Debug("Deleted old/expired dashboard versions", "rows affected", cmd.DeletedRows)
|
|
}
|
|
}
|
|
|
|
func (service *CleanUpService) deleteOldLoginAttempts() {
|
|
if setting.DisableBruteForceLoginProtection {
|
|
return
|
|
}
|
|
|
|
cmd := m.DeleteOldLoginAttemptsCommand{
|
|
OlderThan: time.Now().Add(time.Minute * -10),
|
|
}
|
|
if err := bus.Dispatch(&cmd); err != nil {
|
|
service.log.Error("Problem deleting expired login attempts", "error", err.Error())
|
|
} else {
|
|
service.log.Debug("Deleted expired login attempts", "rows affected", cmd.DeletedRows)
|
|
}
|
|
}
|