opensource/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2025-07-01 10:49:24 +08:00
Code Issues Projects Releases Wiki Activity

unexport functions

Browse Source 
License: MIT
Signed-off-by: Brian Tiger Chow <brian@perfmode.com>
This commit is contained in:
Brian Tiger Chow
2014-12-16 23:43:23 -08:00
committed by Juan Batiz-Benet
parent 9c301a2d77
commit 6e7c46a6e2
2 changed files with 7 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
exchange/bitswap/decision/engine.go
View File

@ -160,17 +160,13 @@ func (e *Engine) MessageSent(p peer.Peer, m bsmsg.BitSwapMessage) error {
return nil
}
func (e *Engine) NumBytesSentTo(p peer.Peer) uint64 {
e.lock.RLock()
defer e.lock.RUnlock()
func (e *Engine) numBytesSentTo(p peer.Peer) uint64 {
// NB not threadsafe
return e.findOrCreate(p).Accounting.BytesSent
}
func (e *Engine) NumBytesReceivedFrom(p peer.Peer) uint64 {
e.lock.RLock()
defer e.lock.RUnlock()
func (e *Engine) numBytesReceivedFrom(p peer.Peer) uint64 {
// NB not threadsafe
return e.findOrCreate(p).Accounting.BytesRecv
}

6
exchange/bitswap/decision/engine_test.go
View File

@ -44,18 +44,18 @@ func TestConsistentAccounting(t *testing.T) {
}
// Ensure sender records the change
if sender.Engine.NumBytesSentTo(receiver.Peer) == 0 {
if sender.Engine.numBytesSentTo(receiver.Peer) == 0 {
t.Fatal("Sent bytes were not recorded")
}
// Ensure sender and receiver have the same values
if sender.Engine.NumBytesSentTo(receiver.Peer) != receiver.Engine.NumBytesReceivedFrom(sender.Peer) {
if sender.Engine.numBytesSentTo(receiver.Peer) != receiver.Engine.numBytesReceivedFrom(sender.Peer) {
t.Fatal("Inconsistent book-keeping. Strategies don't agree")
}
// Ensure sender didn't record receving anything. And that the receiver
// didn't record sending anything
if receiver.Engine.NumBytesSentTo(sender.Peer) != 0 || sender.Engine.NumBytesReceivedFrom(receiver.Peer) != 0 {
if receiver.Engine.numBytesSentTo(sender.Peer) != 0 || sender.Engine.numBytesReceivedFrom(receiver.Peer) != 0 {
t.Fatal("Bert didn't send bytes to Ernie")
}
}