6cc2052f47
We made optimization for segment loop observers to avoid heavy monkit initialization on each call. It was applied to very often executed methods. Unfortunately we used wrong monkit method to track function times. Instead mon.Task we used mon.Func(). https://github.com/spacemonkeygo/monkit#how-it-works Change-Id: I9ca454dbd828c6b43ba09ca75c341991d2fd73a8
58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package audit
|
|
|
|
import (
|
|
"context"
|
|
"math/rand"
|
|
|
|
"storj.io/common/storj"
|
|
"storj.io/storj/satellite/metabase/segmentloop"
|
|
)
|
|
|
|
var remoteSegmentFunc = mon.Task()
|
|
|
|
var _ segmentloop.Observer = (*Collector)(nil)
|
|
|
|
// Collector uses the segment loop to add segments to node reservoirs.
|
|
type Collector struct {
|
|
Reservoirs map[storj.NodeID]*Reservoir
|
|
slotCount int
|
|
rand *rand.Rand
|
|
}
|
|
|
|
// NewCollector instantiates a segment collector.
|
|
func NewCollector(reservoirSlots int, r *rand.Rand) *Collector {
|
|
return &Collector{
|
|
Reservoirs: make(map[storj.NodeID]*Reservoir),
|
|
slotCount: reservoirSlots,
|
|
rand: r,
|
|
}
|
|
}
|
|
|
|
// LoopStarted is called at each start of a loop.
|
|
func (collector *Collector) LoopStarted(context.Context, segmentloop.LoopInfo) (err error) {
|
|
return nil
|
|
}
|
|
|
|
// RemoteSegment takes a remote segment found in metainfo and creates a reservoir for it if it doesn't exist already.
|
|
func (collector *Collector) RemoteSegment(ctx context.Context, segment *segmentloop.Segment) error {
|
|
defer remoteSegmentFunc(&ctx)(nil) // method always returns nil
|
|
|
|
for _, piece := range segment.Pieces {
|
|
res, ok := collector.Reservoirs[piece.StorageNode]
|
|
if !ok {
|
|
res = NewReservoir(collector.slotCount)
|
|
collector.Reservoirs[piece.StorageNode] = res
|
|
}
|
|
res.Sample(collector.rand, NewSegment(segment))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// InlineSegment returns nil because we're only auditing for storage nodes for now.
|
|
func (collector *Collector) InlineSegment(ctx context.Context, segment *segmentloop.Segment) (err error) {
|
|
return nil
|
|
}
|