Context cancellation handling (#131)

* Return connect() when context is cancelled * Check ctx.Err() is err.canceled * Allow to run PostDisconnectCallback
signalfx · Aug 24, 2021 · 5defcae · 5defcae
1 parent df8ce38
commit 5defcae
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/signalflow/conn.go b/signalflow/conn.go
@@ -82,6 +82,11 @@ func (c *wsConn) Run() {
 				c.PostDisconnectCallback()
 			}
 
+			if c.ctx.Err() == context.Canceled {
+				log.Printf("Context cancelled, stop reconnecting.")
+				return
+			}
+
 			var err error
 			conn, err = c.connect()
 			if err != nil {