Pass EventSource connection to child components

1 files changed, 7 insertions, 4 deletions

diff --git a/vnext/src/components/Thread.js b/vnext/src/components/Thread.js
index 4946cf5f..d61fd87f 100644
--- a/vnext/src/components/Thread.js
+++ b/vnext/src/components/Thread.js
@@ -22,8 +22,12 @@ export default function Thread(props) {
   const [loading, setLoading] = useState(false);
   const [active, setActive] = useState(0);
   useEffect(() => {
+    props.connection.addEventListener('msg', onReply);
     setActive(0);
     loadReplies();
+    return () => {
+      props.connection.removeEventListener('msg', onReply);
+    }
   }, []);
   let loadReplies = () => {
     document.body.scrollTop = 0;
@@ -49,11 +53,10 @@ export default function Thread(props) {
         console.log(ex);
       });
   }
-  let onReply = (msg) => {
+  let onReply = (json) => {
+    const msg = JSON.parse(json);
     if (msg.mid == message.mid) {
-      this.setState({
-        replies: [...this.state.replies, msg]
-      });
+      setReplies([...this.state.replies, msg]);
     }
   }
   let postComment = (template) => {