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

return struct from NewBus

Browse Source
This commit is contained in:
Łukasz Magiera 2019-06-19 14:48:41 +02:00
parent 2f028f9607
commit 401bb25f47
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
basic.go
View File

@ -11,18 +11,18 @@ import (
///////////////////////
// BUS
type bus struct {
type BasicBus struct {
lk sync.Mutex
nodes map[reflect.Type]*node
}
func NewBus() Bus {
return &bus{
return &BasicBus{
nodes: map[reflect.Type]*node{},
}
}
func (b *bus) withNode(typ reflect.Type, cb func(*node), async func(*node)) error {
func (b *BasicBus) withNode(typ reflect.Type, cb func(*node), async func(*node)) error {
b.lk.Lock()
n, ok := b.nodes[typ]
@ -44,7 +44,7 @@ func (b *bus) withNode(typ reflect.Type, cb func(*node), async func(*node)) erro
return nil
}
func (b *bus) tryDropNode(typ reflect.Type) {
func (b *BasicBus) tryDropNode(typ reflect.Type) {
b.lk.Lock()
n, ok := b.nodes[typ]
if !ok { // already dropped
@ -64,7 +64,7 @@ func (b *bus) tryDropNode(typ reflect.Type) {
b.lk.Unlock()
}
func (b *bus) Subscribe(typedChan interface{}, opts ...SubOption) (c CancelFunc, err error) {
func (b *BasicBus) Subscribe(typedChan interface{}, opts ...SubOption) (c CancelFunc, err error) {
var settings subSettings
for _, opt := range opts {
if err := opt(&settings); err != nil {
@ -118,7 +118,7 @@ func (b *bus) Subscribe(typedChan interface{}, opts ...SubOption) (c CancelFunc,
return
}
func (b *bus) Emitter(evtType interface{}, opts ...EmitterOption) (e EmitFunc, err error) {
func (b *BasicBus) Emitter(evtType interface{}, opts ...EmitterOption) (e EmitFunc, err error) {
var settings emitterSettings
for _, opt := range opts {
opt(&settings)
@ -156,7 +156,7 @@ func (b *bus) Emitter(evtType interface{}, opts ...EmitterOption) (e EmitFunc, e
// NODE
type node struct {
// Note: make sure to NEVER lock bus.lk when this lock is held
// Note: make sure to NEVER lock BasicBus.lk when this lock is held
lk sync.RWMutex
typ reflect.Type
@ -196,4 +196,4 @@ func (n *node) emit(event interface{}) {
///////////////////////
// UTILS
var _ Bus = &bus{}
var _ Bus = &BasicBus{}

2
basic_test.go
View File

@ -156,7 +156,7 @@ func TestClosingRaces(t *testing.T) {
wg.Wait()
if len(b.(*bus).nodes) != 0 {
if len(b.(*BasicBus).nodes) != 0 {
t.Error("expected no nodes")
}
}