80 lines
2 KiB
Go
80 lines
2 KiB
Go
package logger
|
|
|
|
import (
|
|
"github.com/FAU-CDI/wisski-distillery/internal/component"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/component/sql"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/models"
|
|
"github.com/FAU-CDI/wisski-distillery/pkg/environment"
|
|
"github.com/tkw1536/goprogram/lib/collection"
|
|
)
|
|
|
|
// Logger is responsible for logging backups and snapshots
|
|
type Logger struct {
|
|
component.ComponentBase
|
|
|
|
SQL *sql.SQL
|
|
}
|
|
|
|
func (*Logger) Name() string { return "snapshots-log" }
|
|
|
|
// For retrieves (and prunes) the ExportLog.
|
|
// Slug determines if entries for Backups (empty slug)
|
|
// or a specific Instance (non-empty slug) are returned.
|
|
func (log *Logger) For(slug string) (exports []models.Export, err error) {
|
|
exports, err = log.Log()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return collection.Filter(exports, func(s models.Export) bool {
|
|
return s.Slug == slug
|
|
}), nil
|
|
}
|
|
|
|
// Log retrieves (and prunes) all entries in the snapshot log.
|
|
func (log *Logger) Log() ([]models.Export, error) {
|
|
// query the table!
|
|
table, err := log.SQL.QueryTable(false, models.ExportTable)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// find all the exports
|
|
var exports []models.Export
|
|
res := table.Find(&exports)
|
|
if res.Error != nil {
|
|
return nil, res.Error
|
|
}
|
|
|
|
// partition out the exports that have been deleted!
|
|
parts := collection.Partition(exports, func(s models.Export) bool {
|
|
_, err := log.Core.Environment.Stat(s.Path)
|
|
return !environment.IsNotExist(err)
|
|
})
|
|
|
|
// go and delete them!
|
|
if len(parts[false]) > 0 {
|
|
if err := table.Delete(parts[false]).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
// return the ones that still exist
|
|
return parts[true], nil
|
|
}
|
|
|
|
// AddToExportLog adds the provided export to the log.
|
|
func (log *Logger) Add(export models.Export) error {
|
|
// find the table
|
|
table, err := log.SQL.QueryTable(false, models.ExportTable)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// and save it!
|
|
res := table.Create(&export)
|
|
if res.Error != nil {
|
|
return res.Error
|
|
}
|
|
return nil
|
|
}
|