opensource/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2025-06-30 09:59:13 +08:00
Code Issues Projects Releases Wiki Activity

fix(bitswap:testnet) use peer.Map

Browse Source
This commit is contained in:
Brian Tiger Chow
2014-09-19 14:46:15 -07:00
parent a8a7caa088
commit 57e088bbff
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
exchange/bitswap/testnet/routing.go
View File

@ -24,13 +24,13 @@ type RoutingServer interface {
func VirtualRoutingServer() RoutingServer {
return &hashTable{
m: make(map[u.Key]map[*peer.Peer]bool),
providers: make(map[u.Key]peer.Map),
}
}
type hashTable struct {
lock sync.RWMutex
m map[u.Key]map[*peer.Peer]bool
lock sync.RWMutex
providers map[u.Key]peer.Map
}
var TODO = errors.New("TODO")
@ -39,11 +39,11 @@ func (rs *hashTable) Announce(p *peer.Peer, k u.Key) error {
rs.lock.Lock()
defer rs.lock.Unlock()
_, ok := rs.m[k]
_, ok := rs.providers[k]
if !ok {
rs.m[k] = make(map[*peer.Peer]bool)
rs.providers[k] = make(peer.Map)
}
rs.m[k][p] = true
rs.providers[k][p.Key()] = p
return nil
}
@ -51,11 +51,11 @@ func (rs *hashTable) Providers(k u.Key) []*peer.Peer {
rs.lock.RLock()
defer rs.lock.RUnlock()
ret := make([]*peer.Peer, 0)
peerset, ok := rs.m[k]
peerset, ok := rs.providers[k]
if !ok {
return ret
}
for peer, _ := range peerset {
for _, peer := range peerset {
ret = append(ret, peer)
}
return ret