idun/pkg/buckets/buckets.go

138 lines
2.4 KiB
Go
Raw Normal View History

2023-10-12 08:20:53 +00:00
package buckets
import (
"errors"
"fmt"
"idun/pkg/storage"
"sort"
"time"
)
type Bucket interface {
AddFile(storage.File) error
Execute() error
ToString() string
GetFilesToDelete() ([]storage.File, error)
}
type UnlimitBucket struct {
start time.Time
end time.Time
}
func (u UnlimitBucket) AddFile(f storage.File) error {
t, err := f.GetTime()
if err != nil {
return err
}
if t.Before(u.start) || t.After(u.end) {
return storage.ErrFileNotInBucketTime
}
return nil
}
func (u UnlimitBucket) Execute() error {
return nil
}
func (b UnlimitBucket) GetFilesToDelete() ([]storage.File, error) {
return []storage.File{}, nil
}
func (u UnlimitBucket) ToString() string {
return fmt.Sprintf("%v to %v (Unlimit)", u.start, u.end)
}
type DeleteBucket struct {
start time.Time
end time.Time
files []storage.File
}
func (d *DeleteBucket) AddFile(f storage.File) error {
t, err := f.GetTime()
if err != nil {
return err
}
if t.Before(d.start) || t.After(d.end) {
return storage.ErrFileNotInBucketTime
}
d.files = append(d.files, f)
return nil
}
func (d DeleteBucket) Execute() error {
return nil
}
func (d DeleteBucket) ToString() string {
return fmt.Sprintf("%v to %v (Delete)", d.start, d.end)
}
func (b DeleteBucket) GetFilesToDelete() ([]storage.File, error) {
return b.files, nil
}
type TimeBucket struct {
files []storage.File
start time.Time
end time.Time
}
func (b TimeBucket) ToString() string {
return fmt.Sprintf("%v to %v", b.start, b.end)
}
func (b *TimeBucket) AddFile(f storage.File) error {
t, err := f.GetTime()
if err != nil {
return err
}
if t.Before(b.start) || t.After(b.end) {
return storage.ErrFileNotInBucketTime
}
b.files = append(b.files, f)
return nil
}
func (b TimeBucket) Execute() error {
if len(b.files) <= 1 {
return nil
}
return nil
}
func (b TimeBucket) GetFilesToDelete() ([]storage.File, error) {
toDelete := []storage.File{}
sort.SliceStable(b.files, func(i, j int) bool {
t1, _ := b.files[i].GetTime()
t2, _ := b.files[j].GetTime()
return t1.After(t2)
})
for i, f := range b.files {
if i == 0 {
continue
}
toDelete = append(toDelete, f)
}
return toDelete, nil
}
func NewTimeBucket(start time.Time, end time.Time) (TimeBucket, error) {
if start.After(end) {
return TimeBucket{}, errors.New("start time must be before end time")
}
t := TimeBucket{
start: start,
end: end,
}
return t, nil
}