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

Question

能够从snaplet内部使用Network.WebSockets模块会很好，但我无法弄清楚如何实际执行它。

使用来自Network.WebSockets.Snap的runWebSocketsSnap :: MonadSnap m => ServerApp -> m ()函数，很容易在我的应用程序中包含一个简单的无状态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的酸状态存储器将是很好的。

我的第一个想法是将websocket操作liftIO到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采用上述形式的应用程序，我无法弄清楚如何修改现有的（ hackage上的源代码 ）。 在我看来，需要替代forkIO ，而不是在Handler App App monad中执行操作，但我对Haskell的理解，特别是Snap中的并发性在这里结束......

Answer 1

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

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

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

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