storj/pkg/audit/service.go

95 lines
2.4 KiB
Go
Raw Normal View History

2019-01-24 20:15:10 +00:00
// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package audit
import (
"context"
2018-10-16 21:02:18 +01:00
"time"
"go.uber.org/zap"
"storj.io/storj/pkg/overlay"
"storj.io/storj/pkg/pointerdb"
"storj.io/storj/pkg/provider"
2019-01-23 19:58:44 +00:00
"storj.io/storj/pkg/statdb"
"storj.io/storj/pkg/transport"
)
2019-01-23 19:58:44 +00:00
// Config contains configurable values for audit service
type Config struct {
MaxRetriesStatDB int `help:"max number of times to attempt updating a statdb batch" default:"3"`
Interval time.Duration `help:"how frequently segments are audited" default:"30s"`
}
// Service helps coordinate Cursor and Verifier to run the audit process continuously
type Service struct {
log *zap.Logger
Cursor *Cursor
Verifier *Verifier
Reporter reporter
ticker *time.Ticker
}
// NewService instantiates a Service with access to a Cursor and Verifier
2019-01-23 19:58:44 +00:00
func NewService(log *zap.Logger, sdb statdb.DB, interval time.Duration, maxRetries int, pointers *pointerdb.Service, allocation *pointerdb.AllocationSigner, transport transport.Client, overlay *overlay.Cache, identity *provider.FullIdentity) (service *Service, err error) {
// TODO: instead of overlay.Client use overlay.Service
cursor := NewCursor(pointers, allocation, identity)
2018-11-07 01:16:43 +00:00
verifier := NewVerifier(transport, overlay, identity)
2019-01-23 19:58:44 +00:00
reporter, err := NewReporter(sdb, maxRetries)
if err != nil {
return nil, err
}
2018-10-16 21:02:18 +01:00
return &Service{
log: log,
Cursor: cursor,
2018-10-16 21:02:18 +01:00
Verifier: verifier,
Reporter: reporter,
ticker: time.NewTicker(interval),
2018-10-16 21:02:18 +01:00
}, nil
}
// Run runs auditing service
func (service *Service) Run(ctx context.Context) (err error) {
2018-10-16 21:02:18 +01:00
defer mon.Task()(&ctx)(&err)
service.log.Info("Audit cron is starting up")
for {
err := service.process(ctx)
if err != nil {
service.log.Error("process", zap.Error(err))
}
select {
case <-service.ticker.C:
case <-ctx.Done():
return ctx.Err()
2018-10-16 21:02:18 +01:00
}
}
}
// process picks a random stripe and verifies correctness
func (service *Service) process(ctx context.Context) error {
stripe, err := service.Cursor.NextStripe(ctx)
if err != nil {
return err
}
2018-11-07 01:16:43 +00:00
if stripe == nil {
return nil
2018-11-07 01:16:43 +00:00
}
verifiedNodes, err := service.Verifier.verify(ctx, stripe)
if err != nil {
return err
}
// TODO(moby) we need to decide if we want to do something with nodes that the reporter failed to update
_, err = service.Reporter.RecordAudits(ctx, verifiedNodes)
if err != nil {
return err
}
2018-10-16 21:02:18 +01:00
return nil
}