81 lines
2.2 KiB
Go
81 lines
2.2 KiB
Go
// Copyright (C) 2023 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package queue
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"storj.io/common/storj"
|
|
"storj.io/common/uuid"
|
|
"storj.io/storj/satellite/metabase"
|
|
)
|
|
|
|
// MockRepairQueue helps testing RepairQueue.
|
|
type MockRepairQueue struct {
|
|
Segments []*InjuredSegment
|
|
}
|
|
|
|
// Insert implements RepairQueue.
|
|
func (m *MockRepairQueue) Insert(ctx context.Context, s *InjuredSegment) (alreadyInserted bool, err error) {
|
|
for _, alreadyAdded := range m.Segments {
|
|
if alreadyAdded.StreamID == s.StreamID {
|
|
return true, nil
|
|
}
|
|
}
|
|
m.Segments = append(m.Segments, s)
|
|
return false, nil
|
|
}
|
|
|
|
// InsertBatch implements RepairQueue.
|
|
func (m *MockRepairQueue) InsertBatch(ctx context.Context, segments []*InjuredSegment) (newlyInsertedSegments []*InjuredSegment, err error) {
|
|
for _, segment := range segments {
|
|
inserted, err := m.Insert(ctx, segment)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if inserted {
|
|
newlyInsertedSegments = append(newlyInsertedSegments, segment)
|
|
}
|
|
}
|
|
return newlyInsertedSegments, nil
|
|
}
|
|
|
|
// Select implements RepairQueue.
|
|
func (m *MockRepairQueue) Select(context.Context, []storj.PlacementConstraint, []storj.PlacementConstraint) (*InjuredSegment, error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
// Delete implements RepairQueue.
|
|
func (m *MockRepairQueue) Delete(ctx context.Context, s *InjuredSegment) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
// Clean implements RepairQueue.
|
|
func (m *MockRepairQueue) Clean(ctx context.Context, before time.Time) (deleted int64, err error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
// SelectN implements RepairQueue.
|
|
func (m *MockRepairQueue) SelectN(ctx context.Context, limit int) ([]InjuredSegment, error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
// Count implements RepairQueue.
|
|
func (m *MockRepairQueue) Count(ctx context.Context) (count int, err error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
// TestingSetAttemptedTime implements RepairQueue.
|
|
func (m *MockRepairQueue) TestingSetAttemptedTime(ctx context.Context, streamID uuid.UUID, position metabase.SegmentPosition, t time.Time) (rowsAffected int64, err error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
// Stat implements RepairQueue.
|
|
func (m *MockRepairQueue) Stat(ctx context.Context) ([]Stat, error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
var _ RepairQueue = &MockRepairQueue{}
|