Connect hook

Author: AndrewBurian

stream.go

@@ -131,6 +131,9 @@ func (s *Stream) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 
 	// wait for the client to exit or be shutdown
 	s.Register(c)
+	if s.clientConnectHook != nil {
+		s.clientConnectHook(r, c)
+	}
 	c.Wait()
 	s.Remove(c)
 }
@@ -161,6 +164,10 @@ func (s *Stream) TopicHandler(topics []string) http.HandlerFunc {
 			s.Subscribe(topic, c)
 		}
 
+		if s.clientConnectHook != nil {
+			s.clientConnectHook(r, c)
+		}
+
 		// wait for the client to exit or be shutdown
 		c.Wait()
 		s.Remove(c)