80 lines
2.0 KiB
Go
80 lines
2.0 KiB
Go
|
// Copyright (C) 2020 Storj Labs, Inc.
|
||
|
// See LICENSE for copying information.
|
||
|
|
||
|
package projectbwcleanup
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"time"
|
||
|
|
||
|
"github.com/spacemonkeygo/monkit/v3"
|
||
|
"github.com/zeebo/errs"
|
||
|
"go.uber.org/zap"
|
||
|
|
||
|
"storj.io/common/sync2"
|
||
|
"storj.io/storj/satellite/accounting"
|
||
|
)
|
||
|
|
||
|
var mon = monkit.Package()
|
||
|
|
||
|
// Config is a configuration struct for the Chore.
|
||
|
type Config struct {
|
||
|
Interval time.Duration `help:"how often to remove unused project bandwidth rollups" default:"168h"`
|
||
|
RetainMonths int `help:"number of months of project bandwidth rollups to retain, not including the current month" default:"2"`
|
||
|
}
|
||
|
|
||
|
// Chore to remove unused project bandwidth rollups.
|
||
|
//
|
||
|
// architecture: Chore
|
||
|
type Chore struct {
|
||
|
log *zap.Logger
|
||
|
db accounting.ProjectAccounting
|
||
|
config Config
|
||
|
|
||
|
Loop *sync2.Cycle
|
||
|
}
|
||
|
|
||
|
// NewChore creates new chore for removing unused project bandwidth rollups.
|
||
|
func NewChore(log *zap.Logger, db accounting.ProjectAccounting, config Config) *Chore {
|
||
|
|
||
|
return &Chore{
|
||
|
log: log,
|
||
|
db: db,
|
||
|
config: config,
|
||
|
|
||
|
Loop: sync2.NewCycle(config.Interval),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Run starts the chore.
|
||
|
func (chore *Chore) Run(ctx context.Context) (err error) {
|
||
|
defer mon.Task()(&ctx)(&err)
|
||
|
return chore.Loop.Run(ctx, func(ctx context.Context) error {
|
||
|
err := chore.RunOnce(ctx)
|
||
|
if err != nil {
|
||
|
chore.log.Error("error removing project bandwidth rollups", zap.Error(err))
|
||
|
}
|
||
|
return nil
|
||
|
})
|
||
|
}
|
||
|
|
||
|
// RunOnce removes unused project bandwidth rollups.
|
||
|
func (chore *Chore) RunOnce(ctx context.Context) (err error) {
|
||
|
defer mon.Task()(&ctx)(&err)
|
||
|
|
||
|
if chore.config.RetainMonths < 0 {
|
||
|
return errs.New("retain months cannot be less than 0")
|
||
|
}
|
||
|
|
||
|
beforeTime := time.Now().UTC().AddDate(0, -chore.config.RetainMonths, 0)
|
||
|
beforeMonth := time.Date(beforeTime.Year(), beforeTime.Month(), 1, 0, 0, 0, 0, time.UTC)
|
||
|
|
||
|
return chore.db.DeleteProjectAllocatedBandwidthBefore(ctx, beforeMonth)
|
||
|
}
|
||
|
|
||
|
// Close stops the chore.
|
||
|
func (chore *Chore) Close() error {
|
||
|
chore.Loop.Close()
|
||
|
return nil
|
||
|
}
|