Merge pull request #61 from libp2p/event-bus-option

add a WithEventBus constructor option

Author: marten-seemann

blank.go

@@ -37,7 +37,8 @@ type BlankHost struct {
 }
 
 type config struct {
-	cmgr connmgr.ConnManager
+	cmgr     connmgr.ConnManager
+	eventBus event.Bus
 }
 
 type Option = func(cfg *config)
@@ -48,6 +49,12 @@ func WithConnectionManager(cmgr connmgr.ConnManager) Option {
 	}
 }
 
+func WithEventBus(eventBus event.Bus) Option {
+	return func(cfg *config) {
+		cfg.eventBus = eventBus
+	}
+}
+
 func NewBlankHost(n network.Network, options ...Option) *BlankHost {
 	cfg := config{
 		cmgr: &connmgr.NullConnMgr{},
@@ -57,10 +64,12 @@ func NewBlankHost(n network.Network, options ...Option) *BlankHost {
 	}
 
 	bh := &BlankHost{
-		n:        n,
-		cmgr:     cfg.cmgr,
-		mux:      mstream.NewMultistreamMuxer(),
-		eventbus: eventbus.NewBus(),
+		n:    n,
+		cmgr: cfg.cmgr,
+		mux:  mstream.NewMultistreamMuxer(),
+	}
+	if bh.eventbus == nil {
+		bh.eventbus = eventbus.NewBus()
 	}
 
 	// subscribe the connection manager to network notifications (has no effect with NullConnMgr)