2
2
mirror of https://github.com/octoleo/restic.git synced 2024-12-27 20:45:19 +00:00

index: Add TreePacks()

This commit is contained in:
Alexander Neumann 2017-09-23 14:47:40 +02:00
parent 56fccecd06
commit 1eaad6cebb

View File

@ -17,6 +17,7 @@ import (
type Index struct { type Index struct {
m sync.Mutex m sync.Mutex
pack map[restic.BlobHandle][]indexEntry pack map[restic.BlobHandle][]indexEntry
treePacks restic.IDs
final bool // set to true for all indexes read from the backend ("finalized") final bool // set to true for all indexes read from the backend ("finalized")
id restic.ID // set to the ID of the index when it's finalized id restic.ID // set to the ID of the index when it's finalized
@ -437,6 +438,11 @@ func (idx *Index) Dump(w io.Writer) error {
return nil return nil
} }
// TreePacks returns a list of packs that contain only tree blobs.
func (idx *Index) TreePacks() restic.IDs {
return idx.treePacks
}
// isErrOldIndex returns true if the error may be caused by an old index // isErrOldIndex returns true if the error may be caused by an old index
// format. // format.
func isErrOldIndex(err error) bool { func isErrOldIndex(err error) bool {
@ -469,6 +475,8 @@ func DecodeIndex(buf []byte) (idx *Index, err error) {
idx = NewIndex() idx = NewIndex()
for _, pack := range idxJSON.Packs { for _, pack := range idxJSON.Packs {
var data, tree bool
for _, blob := range pack.Blobs { for _, blob := range pack.Blobs {
idx.store(restic.PackedBlob{ idx.store(restic.PackedBlob{
Blob: restic.Blob{ Blob: restic.Blob{
@ -479,6 +487,17 @@ func DecodeIndex(buf []byte) (idx *Index, err error) {
}, },
PackID: pack.ID, PackID: pack.ID,
}) })
switch blob.Type {
case restic.DataBlob:
data = true
case restic.TreeBlob:
tree = true
}
}
if !data && tree {
idx.treePacks = append(idx.treePacks, pack.ID)
} }
} }
idx.supersedes = idxJSON.Supersedes idx.supersedes = idxJSON.Supersedes
@ -501,6 +520,8 @@ func DecodeOldIndex(buf []byte) (idx *Index, err error) {
idx = NewIndex() idx = NewIndex()
for _, pack := range list { for _, pack := range list {
var data, tree bool
for _, blob := range pack.Blobs { for _, blob := range pack.Blobs {
idx.store(restic.PackedBlob{ idx.store(restic.PackedBlob{
Blob: restic.Blob{ Blob: restic.Blob{
@ -511,6 +532,17 @@ func DecodeOldIndex(buf []byte) (idx *Index, err error) {
}, },
PackID: pack.ID, PackID: pack.ID,
}) })
switch blob.Type {
case restic.DataBlob:
data = true
case restic.TreeBlob:
tree = true
}
}
if !data && tree {
idx.treePacks = append(idx.treePacks, pack.ID)
} }
} }
idx.final = true idx.final = true