67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
// Copyright (C) 2021 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package audit
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/common/errs2"
|
|
"storj.io/common/rpc"
|
|
"storj.io/common/rpc/rpcstatus"
|
|
"storj.io/storj/satellite/metabase"
|
|
)
|
|
|
|
// PieceAudit is piece audit status.
|
|
type PieceAudit int
|
|
|
|
const (
|
|
// PieceAuditUnknown is unknown piece audit.
|
|
PieceAuditUnknown PieceAudit = iota
|
|
// PieceAuditFailure is failed piece audit.
|
|
PieceAuditFailure
|
|
// PieceAuditOffline is offline node piece audit.
|
|
PieceAuditOffline
|
|
// PieceAuditContained is online but unresponsive node piece audit.
|
|
PieceAuditContained
|
|
// PieceAuditSuccess is successful piece audit.
|
|
PieceAuditSuccess
|
|
)
|
|
|
|
// Pieces contains pieces structured by piece audit.
|
|
type Pieces struct {
|
|
Successful metabase.Pieces
|
|
Failed metabase.Pieces
|
|
Offline metabase.Pieces
|
|
Contained metabase.Pieces
|
|
Unknown metabase.Pieces
|
|
}
|
|
|
|
// PieceAuditFromErr returns piece audit based on error.
|
|
func PieceAuditFromErr(err error) PieceAudit {
|
|
if err == nil {
|
|
return PieceAuditSuccess
|
|
}
|
|
|
|
if rpc.Error.Has(err) {
|
|
switch {
|
|
case errs.Is(err, context.DeadlineExceeded), errs2.IsRPC(err, rpcstatus.Unknown):
|
|
return PieceAuditOffline
|
|
default:
|
|
// TODO: is this path not reachable?
|
|
return PieceAuditUnknown
|
|
}
|
|
}
|
|
|
|
switch {
|
|
case errs2.IsRPC(err, rpcstatus.NotFound):
|
|
return PieceAuditFailure
|
|
case errs2.IsRPC(err, rpcstatus.DeadlineExceeded):
|
|
return PieceAuditContained
|
|
default:
|
|
return PieceAuditUnknown
|
|
}
|
|
}
|