63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
|
// Copyright (C) 2019 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
// Package bandwidth implements bandwidth usage rollup loop.
|
||
|
|
package bandwidth
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"go.uber.org/zap"
|
||
|
|
monkit "gopkg.in/spacemonkeygo/monkit.v2"
|
||
|
|
|
||
|
|
"storj.io/storj/internal/sync2"
|
||
|
|
)
|
||
|
|
|
||
|
|
var mon = monkit.Package()
|
||
|
|
|
||
|
|
// Config defines parameters for storage node Collector.
|
||
|
|
type Config struct {
|
||
|
|
Interval time.Duration `help:"how frequently bandwidth usage rollups are calculated" default:"1h0m0s"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// Service implements
|
||
|
|
type Service struct {
|
||
|
|
log *zap.Logger
|
||
|
|
db DB
|
||
|
|
Loop sync2.Cycle
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewService creates a new bandwidth service.
|
||
|
|
func NewService(log *zap.Logger, db DB, config Config) *Service {
|
||
|
|
return &Service{
|
||
|
|
log: log,
|
||
|
|
db: db,
|
||
|
|
Loop: *sync2.NewCycle(config.Interval),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Run starts the background process for rollups of bandwidth usage
|
||
|
|
func (service *Service) Run(ctx context.Context) (err error) {
|
||
|
|
defer mon.Task()(&ctx)(&err)
|
||
|
|
return service.Loop.Run(ctx, service.Rollup)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Rollup calls bandwidth DB Rollup method and logs any errors
|
||
|
|
func (service *Service) Rollup(ctx context.Context) (err error) {
|
||
|
|
defer mon.Task()(&ctx)(&err)
|
||
|
|
|
||
|
|
service.log.Info("Performing bandwidth usage rollups")
|
||
|
|
err = service.db.Rollup(ctx)
|
||
|
|
if err != nil {
|
||
|
|
service.log.Error("Could not rollup bandwidth usage", zap.Error(err))
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Close stops the background process for rollups of bandwidth usage
|
||
|
|
func (service *Service) Close() (err error) {
|
||
|
|
service.Loop.Close()
|
||
|
|
return nil
|
||
|
|
}
|