46 lines
1.2 KiB
Go
46 lines
1.2 KiB
Go
|
// Copyright (C) 2022 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package rangedlooptest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"storj.io/storj/satellite/metabase/rangedloop"
|
||
|
|
"storj.io/storj/satellite/metabase/segmentloop"
|
||
|
|
)
|
||
|
|
|
||
|
|
// SleepObserver is a subscriber to the segment loop which sleeps for every batch.
|
||
|
|
type SleepObserver struct {
|
||
|
|
Duration time.Duration
|
||
|
|
}
|
||
|
|
|
||
|
|
// Start is the callback for segment loop start.
|
||
|
|
func (c *SleepObserver) Start(ctx context.Context, time time.Time) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fork splits the observer to process a segment range.
|
||
|
|
func (c *SleepObserver) Fork(ctx context.Context) (rangedloop.Partial, error) {
|
||
|
|
return c, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Join is a noop.
|
||
|
|
func (c *SleepObserver) Join(ctx context.Context, partial rangedloop.Partial) error {
|
||
|
|
// Range done
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Finish is the callback for segment loop end.
|
||
|
|
func (c *SleepObserver) Finish(ctx context.Context) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Process sleeps for every batch of segments to simulate execution time.
|
||
|
|
func (c *SleepObserver) Process(ctx context.Context, segments []segmentloop.Segment) error {
|
||
|
|
sleepTime := time.Duration(c.Duration.Nanoseconds() * int64(len(segments)))
|
||
|
|
time.Sleep(sleepTime)
|
||
|
|
return nil
|
||
|
|
}
|