opensource/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2025-07-02 12:20:03 +08:00
Code Issues Projects Releases Wiki Activity

name findOrCreate

Browse Source 
License: MIT
Signed-off-by: Brian Tiger Chow <brian@perfmode.com>
This commit is contained in:
Brian Tiger Chow
2014-12-15 22:52:27 -08:00
committed by Juan Batiz-Benet
parent 1bced710b1
commit f028c44fd2
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
exchange/bitswap/strategy/ledgermanager.go
View File

@ -87,7 +87,7 @@ func (lm *LedgerManager) BlockIsWantedByPeer(k u.Key, p peer.Peer) bool {
lm.lock.RLock()
defer lm.lock.RUnlock()
ledger := lm.ledger(p)
ledger := lm.findOrCreate(p)
return ledger.WantListContains(k)
}
@ -97,7 +97,7 @@ func (lm *LedgerManager) MessageReceived(p peer.Peer, m bsmsg.BitSwapMessage) er
lm.lock.Lock()
defer lm.lock.Unlock()
l := lm.ledger(p)
l := lm.findOrCreate(p)
if m.Full() {
l.wantList = wl.New()
}
@ -146,7 +146,7 @@ func (lm *LedgerManager) MessageSent(p peer.Peer, m bsmsg.BitSwapMessage) error
lm.lock.Lock()
defer lm.lock.Unlock()
l := lm.ledger(p)
l := lm.findOrCreate(p)
for _, block := range m.Blocks() {
l.SentBytes(len(block.Data))
l.wantList.Remove(block.Key())
@ -160,18 +160,18 @@ func (lm *LedgerManager) NumBytesSentTo(p peer.Peer) uint64 {
lm.lock.RLock()
defer lm.lock.RUnlock()
return lm.ledger(p).Accounting.BytesSent
return lm.findOrCreate(p).Accounting.BytesSent
}
func (lm *LedgerManager) NumBytesReceivedFrom(p peer.Peer) uint64 {
lm.lock.RLock()
defer lm.lock.RUnlock()
return lm.ledger(p).Accounting.BytesRecv
return lm.findOrCreate(p).Accounting.BytesRecv
}
// ledger lazily instantiates a ledger
func (lm *LedgerManager) ledger(p peer.Peer) *ledger {
func (lm *LedgerManager) findOrCreate(p peer.Peer) *ledger {
l, ok := lm.ledgerMap[peerKey(p.Key())]
if !ok {
l = newLedger(p)