aboutsummaryrefslogtreecommitdiffstats
path: root/event
diff options
context:
space:
mode:
authorFelix Lange <fjl@twurst.com>2014-10-17 00:59:28 +0800
committerFelix Lange <fjl@twurst.com>2014-10-17 00:59:28 +0800
commitfa84e50ddb8e64d4cb92d58e235cfed13761f21e (patch)
treed0dc6c7e0d31f49b0d3ac3cc2b7bb8550ea2351e /event
parentade980912da4afb7d92f845b8d41955851791dc9 (diff)
downloaddexon-fa84e50ddb8e64d4cb92d58e235cfed13761f21e.tar
dexon-fa84e50ddb8e64d4cb92d58e235cfed13761f21e.tar.gz
dexon-fa84e50ddb8e64d4cb92d58e235cfed13761f21e.tar.bz2
dexon-fa84e50ddb8e64d4cb92d58e235cfed13761f21e.tar.lz
dexon-fa84e50ddb8e64d4cb92d58e235cfed13761f21e.tar.xz
dexon-fa84e50ddb8e64d4cb92d58e235cfed13761f21e.tar.zst
dexon-fa84e50ddb8e64d4cb92d58e235cfed13761f21e.zip
event: panic for duplicate type
Diffstat (limited to 'event')
-rw-r--r--event/event.go6
-rw-r--r--event/event_test.go15
2 files changed, 20 insertions, 1 deletions
diff --git a/event/event.go b/event/event.go
index d11a0e9bd..540fbba65 100644
--- a/event/event.go
+++ b/event/event.go
@@ -3,6 +3,7 @@ package event
import (
"errors"
+ "fmt"
"reflect"
"sync"
)
@@ -40,6 +41,7 @@ var ErrMuxClosed = errors.New("event: mux closed")
func (mux *TypeMux) Subscribe(types ...interface{}) Subscription {
sub := newsub(mux)
mux.mutex.Lock()
+ defer mux.mutex.Unlock()
if mux.stopped {
close(sub.postC)
} else {
@@ -49,13 +51,15 @@ func (mux *TypeMux) Subscribe(types ...interface{}) Subscription {
for _, t := range types {
rtyp := reflect.TypeOf(t)
oldsubs := mux.subm[rtyp]
+ if find(oldsubs, sub) != -1 {
+ panic(fmt.Sprintf("event: duplicate type %s in Subscribe", rtyp))
+ }
subs := make([]*muxsub, len(oldsubs)+1)
copy(subs, oldsubs)
subs[len(oldsubs)] = sub
mux.subm[rtyp] = subs
}
}
- mux.mutex.Unlock()
return sub
}
diff --git a/event/event_test.go b/event/event_test.go
index f65aaa0a2..c7c0266c1 100644
--- a/event/event_test.go
+++ b/event/event_test.go
@@ -60,6 +60,21 @@ func TestUnsubscribeUnblockPost(t *testing.T) {
}
}
+func TestSubscribeDuplicateType(t *testing.T) {
+ mux := new(TypeMux)
+ expected := "event: duplicate type event.testEvent in Subscribe"
+
+ defer func() {
+ err := recover()
+ if err == nil {
+ t.Errorf("Subscribe didn't panic for duplicate type")
+ } else if err != expected {
+ t.Errorf("panic mismatch: got %#v, expected %#v", err, expected)
+ }
+ }()
+ mux.Subscribe(testEvent(1), testEvent(2))
+}
+
func TestMuxConcurrent(t *testing.T) {
rand.Seed(time.Now().Unix())
mux := new(TypeMux)