1cb2eb4c3b
Final touches on https://github.com/storj/storj/issues/5236. Change-Id: I2259ec4e7825d20db9efb36beb42d6309dee55ba
87 lines
2.4 KiB
Go
87 lines
2.4 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package metrics
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/spacemonkeygo/monkit/v3"
|
|
"github.com/zeebo/errs"
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/common/sync2"
|
|
"storj.io/storj/satellite/metabase/segmentloop"
|
|
)
|
|
|
|
var (
|
|
// Error defines the metrics chore errors class.
|
|
Error = errs.Class("metrics")
|
|
mon = monkit.Package()
|
|
)
|
|
|
|
// Config contains configurable values for metrics collection.
|
|
type Config struct {
|
|
UseRangedLoop bool `help:"whether to use ranged loop instead of segment loop" default:"false"`
|
|
}
|
|
|
|
// Chore implements the metrics chore.
|
|
//
|
|
// architecture: Chore
|
|
type Chore struct {
|
|
log *zap.Logger
|
|
config Config
|
|
Loop *sync2.Cycle
|
|
segmentLoop *segmentloop.Service
|
|
Counter *Counter
|
|
}
|
|
|
|
// NewChore creates a new instance of the metrics chore.
|
|
func NewChore(log *zap.Logger, config Config, loop *segmentloop.Service) *Chore {
|
|
return &Chore{
|
|
log: log,
|
|
config: config,
|
|
// This chore monitors segment loop, so it's fine to use very small cycle time.
|
|
Loop: sync2.NewCycle(time.Nanosecond),
|
|
segmentLoop: loop,
|
|
}
|
|
}
|
|
|
|
// Run starts the metrics chore.
|
|
func (chore *Chore) Run(ctx context.Context) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
return chore.Loop.Run(ctx, func(ctx context.Context) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
chore.Counter = NewCounter()
|
|
|
|
err = chore.segmentLoop.Monitor(ctx, chore.Counter)
|
|
if err != nil {
|
|
chore.log.Error("error joining segment loop", zap.Error(err))
|
|
return nil
|
|
}
|
|
mon.IntVal("remote_dependent_object_count").Observe(chore.Counter.RemoteObjects)
|
|
mon.IntVal("inline_object_count").Observe(chore.Counter.InlineObjects)
|
|
|
|
mon.IntVal("total_inline_bytes").Observe(chore.Counter.TotalInlineBytes) //mon:locked
|
|
mon.IntVal("total_remote_bytes").Observe(chore.Counter.TotalRemoteBytes) //mon:locked
|
|
|
|
mon.IntVal("total_inline_segments").Observe(chore.Counter.TotalInlineSegments) //mon:locked
|
|
mon.IntVal("total_remote_segments").Observe(chore.Counter.TotalRemoteSegments) //mon:locked
|
|
|
|
// TODO move this metric to a place where objects are iterated e.g. tally
|
|
// or drop it completely as we can easily get this value with redash
|
|
// mon.IntVal("total_object_count").Observe(chore.Counter.ObjectCount)
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// Close closes metrics chore.
|
|
func (chore *Chore) Close() error {
|
|
chore.Loop.Close()
|
|
return nil
|
|
}
|