libp2p/go-eventbus
1
0
Fork 0
mirror of https://github.com/libp2p/go-eventbus.git synced 2025-03-25 12:10:06 +08:00
Code Issues Projects Releases Wiki Activity

Use reflect.Type in node map

Browse Source
This commit is contained in:
Łukasz Magiera 2019-06-19 13:51:25 +02:00
parent 8b50ba1149
commit 71ffb0ebf1
1 changed files with 6 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
basic.go
View File

@ -13,24 +13,22 @@ import (
type bus struct {
lk sync.Mutex
nodes map[string]*node
nodes map[reflect.Type]*node
}
func NewBus() Bus {
return &bus{
nodes: map[string]*node{},
nodes: map[reflect.Type]*node{},
}
}
func (b *bus) withNode(typ reflect.Type, cb func(*node), async func(*node)) error {
path := typePath(typ)
b.lk.Lock()
n, ok := b.nodes[path]
n, ok := b.nodes[typ]
if !ok {
n = newNode(typ)
b.nodes[path] = n
b.nodes[typ] = n
}
n.lk.Lock()
@ -47,10 +45,8 @@ func (b *bus) withNode(typ reflect.Type, cb func(*node), async func(*node)) erro
}
func (b *bus) tryDropNode(typ reflect.Type) {
path := typePath(typ)
b.lk.Lock()
n, ok := b.nodes[path]
n, ok := b.nodes[typ]
if !ok { // already dropped
b.lk.Unlock()
return
@ -64,7 +60,7 @@ func (b *bus) tryDropNode(typ reflect.Type) {
}
n.lk.Unlock()
delete(b.nodes, path)
delete(b.nodes, typ)
b.lk.Unlock()
}
@ -201,8 +197,4 @@ func (n *node) emit(event interface{}) {
///////////////////////
// UTILS
func typePath(t reflect.Type) string {
return t.PkgPath() + "/" + t.String()
}
var _ Bus = &bus{}