mirror of
https://github.com/restic/restic.git
synced 2024-12-25 09:18:55 +00:00
137 lines
2.6 KiB
Go
137 lines
2.6 KiB
Go
package restic
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/restic/restic/internal/crypto"
|
|
"github.com/restic/restic/internal/errors"
|
|
)
|
|
|
|
// Blob is one part of a file or a tree.
|
|
type Blob struct {
|
|
BlobHandle
|
|
Length uint
|
|
Offset uint
|
|
UncompressedLength uint
|
|
}
|
|
|
|
func (b Blob) String() string {
|
|
return fmt.Sprintf("<Blob (%v) %v, offset %v, length %v, uncompressed length %v>",
|
|
b.Type, b.ID.Str(), b.Offset, b.Length, b.UncompressedLength)
|
|
}
|
|
|
|
func (b Blob) DataLength() uint {
|
|
if b.UncompressedLength != 0 {
|
|
return b.UncompressedLength
|
|
}
|
|
return uint(crypto.PlaintextLength(int(b.Length)))
|
|
}
|
|
|
|
func (b Blob) IsCompressed() bool {
|
|
return b.UncompressedLength != 0
|
|
}
|
|
|
|
// PackedBlob is a blob stored within a file.
|
|
type PackedBlob struct {
|
|
Blob
|
|
PackID ID
|
|
}
|
|
|
|
// BlobHandle identifies a blob of a given type.
|
|
type BlobHandle struct {
|
|
ID ID
|
|
Type BlobType
|
|
}
|
|
|
|
func (h BlobHandle) String() string {
|
|
return fmt.Sprintf("<%s/%s>", h.Type, h.ID.Str())
|
|
}
|
|
|
|
func NewRandomBlobHandle() BlobHandle {
|
|
return BlobHandle{ID: NewRandomID(), Type: DataBlob}
|
|
}
|
|
|
|
// BlobType specifies what a blob stored in a pack is.
|
|
type BlobType uint8
|
|
|
|
// These are the blob types that can be stored in a pack.
|
|
const (
|
|
InvalidBlob BlobType = iota
|
|
DataBlob
|
|
TreeBlob
|
|
NumBlobTypes // Number of types. Must be last in this enumeration.
|
|
)
|
|
|
|
func (t BlobType) String() string {
|
|
switch t {
|
|
case DataBlob:
|
|
return "data"
|
|
case TreeBlob:
|
|
return "tree"
|
|
case InvalidBlob:
|
|
return "invalid"
|
|
}
|
|
|
|
return fmt.Sprintf("<BlobType %d>", t)
|
|
}
|
|
|
|
// MarshalJSON encodes the BlobType into JSON.
|
|
func (t BlobType) MarshalJSON() ([]byte, error) {
|
|
switch t {
|
|
case DataBlob:
|
|
return []byte(`"data"`), nil
|
|
case TreeBlob:
|
|
return []byte(`"tree"`), nil
|
|
}
|
|
|
|
return nil, errors.New("unknown blob type")
|
|
}
|
|
|
|
// UnmarshalJSON decodes the BlobType from JSON.
|
|
func (t *BlobType) UnmarshalJSON(buf []byte) error {
|
|
switch string(buf) {
|
|
case `"data"`:
|
|
*t = DataBlob
|
|
case `"tree"`:
|
|
*t = TreeBlob
|
|
default:
|
|
return errors.New("unknown blob type")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// BlobHandles is an ordered list of BlobHandles that implements sort.Interface.
|
|
type BlobHandles []BlobHandle
|
|
|
|
func (h BlobHandles) Len() int {
|
|
return len(h)
|
|
}
|
|
|
|
func (h BlobHandles) Less(i, j int) bool {
|
|
for k, b := range h[i].ID {
|
|
if b == h[j].ID[k] {
|
|
continue
|
|
}
|
|
|
|
if b < h[j].ID[k] {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
return h[i].Type < h[j].Type
|
|
}
|
|
|
|
func (h BlobHandles) Swap(i, j int) {
|
|
h[i], h[j] = h[j], h[i]
|
|
}
|
|
|
|
func (h BlobHandles) String() string {
|
|
elements := make([]string, 0, len(h))
|
|
for _, e := range h {
|
|
elements = append(elements, e.String())
|
|
}
|
|
return fmt.Sprintf("%v", elements)
|
|
}
|