fixed panic to send closed channel

opensaasstudio · Jul 7, 2017 · 32aeb37 · 32aeb37
1 parent a0625ee
commit 32aeb37
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/manager/client.go b/manager/client.go
@@ -110,7 +110,7 @@ func (cm *ClientManager) SendPayload(payload event.Payload) {
 				wg2.Add(1)
 				go func(client chan event.Payload) {
 					defer wg2.Done()
-					client <- payload
+					sendPayloadSafety(client, payload)
 				}(client)
 			}
 			wg2.Wait()
@@ -119,6 +119,16 @@ func (cm *ClientManager) SendPayload(payload event.Payload) {
 	wg.Wait()
 }
 
+func sendPayloadSafety(client chan event.Payload, payload event.Payload) {
+	defer func() {
+		if err := recover(); err != nil {
+			return
+		}
+	}()
+	client <- payload
+	return
+}
+
 const heartBeatEvent = "heartbeat"
 
 func (cm *ClientManager) SendHeartBeat() {