coosld/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2026-02-24 20:07:45 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

pin: Remove double bookkeeping of refcount keys

Browse Source 
License: MIT
Signed-off-by: Jeromy <jeromyj@gmail.com>
This commit is contained in:
Tommi Virtanen 2015-05-08 17:10:46 -07:00 committed by Jeromy
parent d6a61529ca
commit fecfb76cdf
1 changed files with 6 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pin/indirect.go
View File

@ -3,17 +3,14 @@ package pin
import (
ds "github.com/ipfs/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-datastore"
key "github.com/ipfs/go-ipfs/blocks/key"
"github.com/ipfs/go-ipfs/blocks/set"
)
type indirectPin struct {
blockset set.BlockSet
refCounts map[key.Key]int
}
func newIndirectPin() *indirectPin {
return &indirectPin{
blockset: set.NewSimpleBlockSet(),
refCounts: make(map[key.Key]int),
}
}
@ -36,7 +33,7 @@ func loadIndirPin(d ds.Datastore, k ds.Key) (*indirectPin, error) {
}
// log.Debugf("indirPin keys: %#v", keys)
return &indirectPin{blockset: set.SimpleSetFromKeys(keys), refCounts: refcnt}, nil
return &indirectPin{refCounts: refcnt}, nil
}
func storeIndirPin(d ds.Datastore, k ds.Key, p *indirectPin) error {
@ -49,11 +46,7 @@ func storeIndirPin(d ds.Datastore, k ds.Key, p *indirectPin) error {
}
func (i *indirectPin) Increment(k key.Key) {
c := i.refCounts[k]
i.refCounts[k] = c + 1
if c <= 0 {
i.blockset.AddBlock(k)
}
i.refCounts[k]++
}
func (i *indirectPin) Decrement(k key.Key) {
@ -61,16 +54,15 @@ func (i *indirectPin) Decrement(k key.Key) {
log.Warningf("pinning: bad call: asked to unpin nonexistent indirect key: %v", k)
return
}
c := i.refCounts[k] - 1
i.refCounts[k] = c
if c <= 0 {
i.blockset.RemoveBlock(k)
i.refCounts[k]--
if i.refCounts[k] == 0 {
delete(i.refCounts, k)
}
}
func (i *indirectPin) HasKey(k key.Key) bool {
return i.blockset.HasKey(k)
_, found := i.refCounts[k]
return found
}
func (i *indirectPin) GetRefs() map[key.Key]int {