2
新主题,我在原型斯科蒂Web服务器与两个API:分叉在斯科蒂服务器
/add/:id
开始给定ID的ASYC任务。/cancel/:id
杀死给定ID的任务。
基本上客户端通过提供一些Ids启动异步任务,并且也可以通过他们的Id来终止当前任务。
我用Control.Concurrent.forkIO
启动一个线程,forkIO
返回我存储在这是一个地图斯科蒂全球状态的ThreadId
:type AppState = Map TaskId ThreadId
。
/add/:id
不会立即返回,而是等待任务完成并将结果返回给客户端。
我的问题是混合forkIO
与MonadIO m => ActionT Text m()
。我需要在完成IO()
行动之后拨打text :: Text -> ActionT Text m()
,我通过了forkIO
。
这需要从MonadIO m
到IO
这显然是一个错误,但我不能得到我的头,找到任何解决方案。
这是完整的例子:
import qualified Control.Concurrent as C
import qualified Control.Concurrent.STM as STM
import Control.Monad.Trans.Reader (ReaderT(..), runReaderT)
import Control.Monad.Trans (MonadIO)
import Control.Monad.Reader (MonadReader, lift, liftIO, ask)
import qualified Data.Map as M
import Data.Text.Lazy (Text, pack, unpack)
import Web.Scotty.Trans
type TaskId = String
type AppState = M.Map TaskId C.ThreadId
newtype WebM a = WebM { runWebM :: ReaderT (STM.TVar AppState) IO a }
deriving (Applicative, Functor, Monad, MonadIO, MonadReader (STM.TVar AppState))
app :: ScottyT Text WebM()
app = do
get "/add/:id" $ do
taskId <- fmap unpack (param "id")
let task = return "Hello World" -- just a dummy IO
tid <- liftIO $ C.forkIO $ do
result <- task
-- Couldn't match type ‘ActionT Text m’ with ‘IO’
lift $ modify' $ M.delete taskId -- remove the completed task from the state
text result -- return the result to the client
return() -- forkIO :: IO() -> IO ThreadId
lift $ modify' $ M.insert taskId tid -- immedialtey add the new task to the state
get "/cancel/:id" $ do
taskId <- fmap unpack (param "id")
dic <- lift $ gets id
maybe
(text $ pack (taskId ++ " Not Found"))
(
\ tid -> do
liftIO $ C.killThread tid
lift $ modify' $ M.delete taskId -- remove the cancelled task from the state
text $ pack (taskId ++ " Cancelled")
)
(M.lookup taskId dic)
gets :: (AppState -> b) -> WebM b
gets f = fmap f (ask >>= liftIO . STM.readTVarIO)
modify' :: (AppState -> AppState) -> WebM()
modify' f = ask >>= liftIO . STM.atomically . flip STM.modifyTVar' f
main :: IO()
main = do
dic <- STM.newTVarIO M.empty
let runActionToIO m = runReaderT (runWebM m) dic
scottyT 3000 runActionToIO app