2017-06-14 55 views
2

新主题,我在原型斯科蒂Web服务器与两个API:分叉在斯科蒂服务器

  • /add/:id开始给定ID的ASYC任务。
  • /cancel/:id杀死给定ID的任务。

基本上客户端通过提供一些Ids启动异步任务,并且也可以通过他们的Id来终止当前任务。

我用Control.Concurrent.forkIO启动一个线程,forkIO返回我存储在这是一个地图斯科蒂全球状态的ThreadIdtype AppState = Map TaskId ThreadId

/add/:id不会立即返回,而是等待任务完成并将结果返回给客户端。

我的问题是混合forkIOMonadIO m => ActionT Text m()。我需要在完成IO()行动之后拨打text :: Text -> ActionT Text m(),我通过了forkIO

这需要从MonadIO mIO这显然是一个错误,但我不能得到我的头,找到任何解决方案。

这是完整的例子:

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 

回答

3

我认为你需要将呼叫转移到text result出分叉线程的,并使用一个MVar当结果准备好沟通。因此,像

get "/add/:id" $ do 
    taskId <- fmap unpack (param "id") 
    let task = return "Hello World" 
    m <- newEmptyMVar 
    tid <- liftIO $ C.forkIO $ do 
     result <- task 
     putMVar result 
     ... 
    r <- takeMVar m 
    text r 

takeMVar将阻塞,直到MVar包含一个值。