2
2
mirror of https://github.com/octoleo/restic.git synced 2024-12-23 03:18:55 +00:00

Rework function for listing packs

This commit is contained in:
Alexander Neumann 2016-08-25 21:08:16 +02:00
parent e07ae7631c
commit 9f752b8306
5 changed files with 13 additions and 30 deletions

View File

@ -676,13 +676,13 @@ func checkPack(r *repository.Repository, id backend.ID) error {
return fmt.Errorf("Pack ID does not match, want %v, got %v", id.Str(), hash.Str()) return fmt.Errorf("Pack ID does not match, want %v, got %v", id.Str(), hash.Str())
} }
unpacker, err := pack.NewUnpacker(r.Key(), pack.BufferLoader(buf)) blobs, err := pack.List(r.Key(), pack.BufferLoader(buf))
if err != nil { if err != nil {
return err return err
} }
var errs []error var errs []error
for i, blob := range unpacker.Entries { for i, blob := range blobs {
debug.Log("Checker.checkPack", " check blob %d: %v", i, blob.ID.Str()) debug.Log("Checker.checkPack", " check blob %d: %v", i, blob.ID.Str())
plainBuf := make([]byte, blob.Length) plainBuf := make([]byte, blob.Length)

View File

@ -228,22 +228,13 @@ func (p *Packer) String() string {
return fmt.Sprintf("<Packer %d blobs, %d bytes>", len(p.blobs), p.bytes) return fmt.Sprintf("<Packer %d blobs, %d bytes>", len(p.blobs), p.bytes)
} }
// Unpacker is used to read individual blobs from a pack.
type Unpacker struct {
rd io.ReadSeeker
Entries []Blob
k *crypto.Key
}
const ( const (
preloadHeaderSize = 2048 preloadHeaderSize = 2048
maxHeaderSize = 16 * 1024 * 1024 maxHeaderSize = 16 * 1024 * 1024
) )
// NewUnpacker returns a pointer to Unpacker which can be used to read // List returns the list of entries found in a pack file.
// individual Blobs from a pack. func List(k *crypto.Key, ldr Loader) (entries []Blob, err error) {
func NewUnpacker(k *crypto.Key, ldr Loader) (*Unpacker, error) {
var err error
// read the last 2048 byte, this will mostly be enough for the header, so // read the last 2048 byte, this will mostly be enough for the header, so
// we do not need another round trip. // we do not need another round trip.
@ -294,8 +285,6 @@ func NewUnpacker(k *crypto.Key, ldr Loader) (*Unpacker, error) {
rd := bytes.NewReader(hdr) rd := bytes.NewReader(hdr)
var entries []Blob
pos := uint(0) pos := uint(0)
for { for {
e := headerEntry{} e := headerEntry{}
@ -328,11 +317,5 @@ func NewUnpacker(k *crypto.Key, ldr Loader) (*Unpacker, error) {
pos += uint(e.Length) pos += uint(e.Length)
} }
up := &Unpacker{ return entries, nil
rd: rd,
k: k,
Entries: entries,
}
return up, nil
} }

View File

@ -63,13 +63,13 @@ func verifyBlobs(t testing.TB, bufs []Buf, k *crypto.Key, ldr pack.Loader, packS
Equals(t, uint(written), packSize) Equals(t, uint(written), packSize)
// read and parse it again // read and parse it again
np, err := pack.NewUnpacker(k, ldr) entries, err := pack.List(k, ldr)
OK(t, err) OK(t, err)
Equals(t, len(np.Entries), len(bufs)) Equals(t, len(entries), len(bufs))
var buf []byte var buf []byte
for i, b := range bufs { for i, b := range bufs {
e := np.Entries[i] e := entries[i]
Equals(t, b.id, e.ID) Equals(t, b.id, e.ID)
if len(buf) < int(e.Length) { if len(buf) < int(e.Length) {

View File

@ -32,14 +32,14 @@ func Repack(repo *Repository, packs backend.IDSet, keepBlobs pack.BlobSet) (err
debug.Log("Repack", "pack %v loaded (%d bytes)", packID.Str(), len(buf)) debug.Log("Repack", "pack %v loaded (%d bytes)", packID.Str(), len(buf))
unpck, err := pack.NewUnpacker(repo.Key(), pack.BufferLoader(buf)) blobs, err := pack.List(repo.Key(), pack.BufferLoader(buf))
if err != nil { if err != nil {
return err return err
} }
debug.Log("Repack", "processing pack %v, blobs: %v", packID.Str(), len(unpck.Entries)) debug.Log("Repack", "processing pack %v, blobs: %v", packID.Str(), len(blobs))
var plaintext []byte var plaintext []byte
for _, entry := range unpck.Entries { for _, entry := range blobs {
h := pack.Handle{ID: entry.ID, Type: entry.Type} h := pack.Handle{ID: entry.ID, Type: entry.Type}
if !keepBlobs.Has(h) { if !keepBlobs.Has(h) {
continue continue

View File

@ -556,12 +556,12 @@ func (r *Repository) ListPack(id backend.ID) ([]pack.Blob, int64, error) {
ldr := pack.BackendLoader{Backend: r.Backend(), Handle: h} ldr := pack.BackendLoader{Backend: r.Backend(), Handle: h}
unpacker, err := pack.NewUnpacker(r.Key(), ldr) blobs, err := pack.List(r.Key(), ldr)
if err != nil { if err != nil {
return nil, 0, err return nil, 0, err
} }
return unpacker.Entries, blobInfo.Size, nil return blobs, blobInfo.Size, nil
} }
// Delete calls backend.Delete() if implemented, and returns an error // Delete calls backend.Delete() if implemented, and returns an error