Skip to content

Commit

Permalink
Merge pull request src-d#716 from jfontan/fix/memory-leak-packing-obj…
Browse files Browse the repository at this point in the history
…ects

Clean reconstructed objects outside pack window
  • Loading branch information
mcuadros committed Jan 15, 2018
2 parents 2547d1d + 3929267 commit 861e399
Showing 1 changed file with 19 additions and 13 deletions.
32 changes: 19 additions & 13 deletions plumbing/format/packfile/delta_selector.go
Original file line number Diff line number Diff line change
Expand Up @@ -222,10 +222,16 @@ func (dw *deltaSelector) walk(
) error {
indexMap := make(map[plumbing.Hash]*deltaIndex)
for i := 0; i < len(objectsToPack); i++ {
// Clean up the index map for anything outside our pack
// window, to save memory.
// Clean up the index map and reconstructed delta objects for anything
// outside our pack window, to save memory.
if i > int(packWindow) {
delete(indexMap, objectsToPack[i-int(packWindow)].Hash())
obj := objectsToPack[i-int(packWindow)]

delete(indexMap, obj.Hash())

if obj.IsDelta() {
obj.Original = nil
}
}

target := objectsToPack[i]
Expand Down Expand Up @@ -261,6 +267,16 @@ func (dw *deltaSelector) walk(
}

func (dw *deltaSelector) tryToDeltify(indexMap map[plumbing.Hash]*deltaIndex, base, target *ObjectToPack) error {
// Original object might not be present if we're reusing a delta, so we
// ensure it is restored.
if err := dw.restoreOriginal(target); err != nil {
return err
}

if err := dw.restoreOriginal(base); err != nil {
return err
}

// If the sizes are radically different, this is a bad pairing.
if target.Size() < base.Size()>>4 {
return nil
Expand All @@ -283,16 +299,6 @@ func (dw *deltaSelector) tryToDeltify(indexMap map[plumbing.Hash]*deltaIndex, ba
return nil
}

// Original object might not be present if we're reusing a delta, so we
// ensure it is restored.
if err := dw.restoreOriginal(target); err != nil {
return err
}

if err := dw.restoreOriginal(base); err != nil {
return err
}

if _, ok := indexMap[base.Hash()]; !ok {
indexMap[base.Hash()] = new(deltaIndex)
}
Expand Down

0 comments on commit 861e399

Please sign in to comment.