Implement message type filter for listener

client/client.go

@@ -96,9 +96,11 @@ func (client *Client) receiver() {
 
 		needGc := false
 		for _, listener := range client.listenerStore.Listeners() {
-			if listener.IsActive() {
+			if listener.IsActive() && listener.Updates != nil && typ.GetType() == listener.Filter.GetType() { // All updates go to Updates channel if type == filter
 				listener.Updates <- typ
-			} else {
+			} else if listener.IsActive() && listener.RawUpdates != nil { // All updates go to RawUpdates channel if filter is empty
+				listener.RawUpdates <- typ
+			} else if !listener.IsActive() { // GC inactive listener
 				needGc = true
 			}
 		}
@@ -137,7 +139,18 @@ func (client *Client) Send(req Request) (*Response, error) {
 func (client *Client) GetListener() *Listener {
 	listener := &Listener{
 		isActive:   true,
-		Updates:  make(chan Type, 1000),
+		RawUpdates: make(chan Type, 1000),
+	}
+	client.listenerStore.Add(listener)
+
+	return listener
+}
+
+func (client *Client) AddEventReceiver(msgType Type, channelCapacity int) *Listener {
+	listener := &Listener{
+		isActive:   true,
+		Updates:    make(chan Type, channelCapacity),
+		Filter:     msgType,
 	}
 	client.listenerStore.Add(listener)
 

client/listener.go

@@ -48,6 +48,8 @@ type Listener struct {
 	mu         sync.Mutex
 	isActive   bool
 	Updates    chan Type
+	RawUpdates chan Type
+	Filter     Type
 }
 
 func (listener *Listener) Close() {
@@ -55,8 +57,13 @@ func (listener *Listener) Close() {
 	defer listener.mu.Unlock()
 
 	listener.isActive = false
+	if listener.Updates != nil {
 		close(listener.Updates)
 	}
+	if listener.RawUpdates != nil {
+		close(listener.RawUpdates)
+	}
+}
 
 func (listener *Listener) IsActive() bool {
 	listener.mu.Lock()