From 2d5a042cd4c80b1989b8235a7412110d435b1130 Mon Sep 17 00:00:00 2001 From: Valery Piashchynski Date: Thu, 17 Dec 2020 03:23:41 +0300 Subject: Fix race condition --- util/events.go | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'util') diff --git a/util/events.go b/util/events.go index 259c7ddb..676e2f40 100755 --- a/util/events.go +++ b/util/events.go @@ -1,12 +1,15 @@ package util import ( + "sync" + "github.com/spiral/roadrunner/v2/interfaces/events" ) // EventHandler helps to broadcast events to multiple listeners. type EventHandler struct { listeners []events.EventListener + sync.Mutex } func NewEventsHandler() events.Handler { @@ -15,16 +18,22 @@ func NewEventsHandler() events.Handler { // NumListeners returns number of event listeners. func (eb *EventHandler) NumListeners() int { + eb.Lock() + defer eb.Unlock() return len(eb.listeners) } // AddListener registers new event listener. func (eb *EventHandler) AddListener(listener events.EventListener) { + eb.Lock() + defer eb.Unlock() eb.listeners = append(eb.listeners, listener) } // Push broadcast events across all event listeners. func (eb *EventHandler) Push(e interface{}) { + eb.Lock() + defer eb.Unlock() for _, listener := range eb.listeners { listener(e) } -- cgit v1.2.3