storj/satellite/repair/queue/queue.go
Erik van Velzen 10d71a8a3c satellite/satellitedb: outline for batch insert
Part of https://github.com/storj/storj/issues/4727

Change-Id: I1a9ad3b009f363e37f5e68e810074eecb7448db3
2022-05-09 11:39:52 +00:00

49 lines
1.7 KiB
Go

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package queue
import (
"context"
"time"
"storj.io/common/uuid"
"storj.io/storj/satellite/metabase"
)
// InjuredSegment contains information about segment which
// should be repaired.
type InjuredSegment struct {
StreamID uuid.UUID
Position metabase.SegmentPosition
SegmentHealth float64
AttemptedAt *time.Time
UpdatedAt time.Time
InsertedAt time.Time
}
// RepairQueue implements queueing for segments that need repairing.
// Implementation can be found at satellite/satellitedb/repairqueue.go.
//
// architecture: Database
type RepairQueue interface {
// Insert adds an injured segment.
Insert(ctx context.Context, s *InjuredSegment) (alreadyInserted bool, err error)
// InsertBatch adds multiple injured segments
InsertBatch(ctx context.Context, segments []*InjuredSegment) (newlyInsertedSegments []*InjuredSegment, err error)
// Select gets an injured segment.
Select(ctx context.Context) (*InjuredSegment, error)
// Delete removes an injured segment.
Delete(ctx context.Context, s *InjuredSegment) error
// Clean removes all segments last updated before a certain time
Clean(ctx context.Context, before time.Time) (deleted int64, err error)
// SelectN lists limit amount of injured segments.
SelectN(ctx context.Context, limit int) ([]InjuredSegment, error)
// Count counts the number of segments in the repair queue.
Count(ctx context.Context) (count int, err error)
// TestingSetAttemptedTime sets attempted time for a segment.
TestingSetAttemptedTime(ctx context.Context, streamID uuid.UUID, position metabase.SegmentPosition, t time.Time) (rowsAffected int64, err error)
}