From 85496411f24a162a2a754e251a24d0b8eae6c230 Mon Sep 17 00:00:00 2001
From: Divam <dfordivam@gmail.com>
Date: Wed, 15 Mar 2017 11:54:45 +0900
Subject: [PATCH] Fix yesod-websockets/sample.hs example

---
 yesod-websockets/sample.hs | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/yesod-websockets/sample.hs b/yesod-websockets/sample.hs
index 84f88b2e..12e32ace 100644
--- a/yesod-websockets/sample.hs
+++ b/yesod-websockets/sample.hs
@@ -3,10 +3,10 @@ import Yesod.Core
 import Yesod.WebSockets
 import qualified Data.Text.Lazy as TL
 import Control.Monad (forever)
-import Control.Monad.Trans.Reader
 import Control.Concurrent (threadDelay)
 import Data.Time
-import Conduit
+import Data.Conduit
+import qualified Data.Conduit.List
 
 data App = App
 
@@ -25,7 +25,7 @@ timeSource = forever $ do
 getHomeR :: Handler Html
 getHomeR = do
     webSockets $ race_
-        (sourceWS $$ mapC TL.toUpper =$ sinkWSText)
+        (sourceWS $$ Data.Conduit.List.map TL.toUpper =$ sinkWSText)
         (timeSource $$ sinkWSText)
     defaultLayout $
         toWidget
@@ -43,6 +43,9 @@ getHomeR = do
                 conn.onmessage = function(e) {
                     document.write("<p>" + e.data + "</p>");
                 };
+                conn.onclose = function () {
+                    document.write("<p>Connection Closed</p>");
+                };
             |]
 
 main :: IO ()