haskell - 如何在 snaplet 中使用 Network.WebSockets.Snap？

Question

如果能够使用 Network.WebSockets 那就太好了模块，但我不知道如何实际做到这一点。

使用runWebSocketsSnap :: MonadSnap m => ServerApp -> m ()来自 Network.WebSockets.Snap 的函数在我的应用程序中包含一个简单的无状态 websocket 服务器很容易:

routes :: [(ByteString, Handler App App ())]
routes = [ ("/ws", runWebSocketsSnap wsApp) ]

wsApp :: PendingConnection -> IO () -- this is the ServerApp type
wsApp pending = do
    conn <- acceptRequest pending
    forever $ do
        msg <- receiveData conn
        sendTextData conn ("Echo " `mappend` msg :: Text)

但我的目标是维护 webscket 服务器的状态(例如，已连接客户端的列表，如 http://jaspervdj.be/websockets/example.html )。或者，访问 snaplet 的酸态存储也很棒。

我的第一个想法是liftIO将 websocket 操作写入 Handler App App monad，并编写一个像这样的应用程序:

wsApp :: PendingConnection -> Handler App App ()
wsApp pending = do
    conn <- liftIO $ acceptRequest pending
    forever $ do
        msg <- liftIO $ receiveData conn
        update (SetLastMsg msg)
        liftIO $ sendTextData conn ("Stored msg in datastore.")

但是没有runWebSocketsSnap的版本这需要上述形式的应用程序，我不知道如何修改现有的应用程序( source on hackage )。在我看来，人们需要 forkIO 的替代品在 Handler App App 中执行操作相反，我对 Haskell，尤其是 Snap 中的并发性的理解到此为止......

Answer 1

runWebSocketsSnap 函数要求其参数类型为 PendingConnection -> IO ()。这意味着您无法直接访问该函数内部的应用程序数据结构。您需要做的是将信息作为参数传递给函数，如下所示。

routes = [ ("/ws", webSocketsDriver) ]

webSocketsDriver :: Handler App App ()
webSocketsDriver = do
    appState <- get
    runWebSocketsSnap (wsApp appState)

wsApp :: App -> PendingConnection -> IO ()
wsApp app pending = do
    ...