Come si implementa un comando di spegnimento in un server WAI?

Question

Mi piacerebbe realizzare un comando 'graziosa shutdown' per la mia webapp (in contrapposizione al mio primo istinto, che è quello di chiedere solo alle persone di uccidere il processo)

I miei primi due tentativi consistevano

1. liftIO exitSuccess
2. E.yield (responseLBS statusOK [G.contentType "text/plain"] "") E.EOF

Entrambi i quali appena allegramente ha restituito un risultato al client e continuò ad ascoltare. C'è qualcosa che un'applicazione può fare per uccidere il server? Questa è anche una cosa ragionevole da voler fare?

Confesso che non ho una comprensione molto forte di iteratee, solo abbastanza per sapere che posso consumare il mio input e che Iteratee è un'istanza MonadIO.

Answer 1

1. Utilizzare un MVar. Blocca il thread principale fino a quando l'MVar è stato segnalato, quindi pulisci e esci.
2. Chiama exitImmediately. Uno dei modi più veloci per abbattere il processo, e anche terribilmente fastidioso per il debug. Non credo che i finalizzatori/parentesi/i blocchi finali verranno richiamati durante la discesa, a seconda dell'applicazione potrebbe corrompere lo stato.
3. Genera un'eccezione sul thread principale. Warp.run non rileva eccezioni, quindi funziona consentendo al gestore delle eccezioni predefinito sul thread principale (e solo il thread principale) di terminare il processo.

Come altri hanno già detto, l'utilizzo di un MVar è probabilmente l'opzione migliore. Ho incluso gli altri per completezza, ma hanno il loro posto. throwTo viene utilizzato un po 'nella libreria di base e ho lavorato su alcune applicazioni che utilizzano l'equivalente C di exitImmediately: exit(), anche se non ho incontrato alcuna app Haskell che utilizza questo metodo.

{-# LANGUAGE DeriveDataTypeable #-} 
{-# LANGUAGE OverloadedStrings #-} 

module Main (main) where 

import Control.Concurrent (MVar, ThreadId, forkIO, myThreadId, newEmptyMVar, putMVar, takeMVar) 
import Control.Exception (Exception, throwTo) 
import Control.Monad.Trans (liftIO) 
import Data.ByteString (ByteString) 
import Data.Data (Data, Typeable) 
import Data.Enumerator (Iteratee) 
import Network.HTTP.Types 
import Network.Wai as Wai 
import Network.Wai.Handler.Warp as Warp 
import System.Exit (ExitCode (ExitSuccess)) 
import System.Posix.Process (exitImmediately) 

data Shutdown = Shutdown deriving (Data, Typeable, Show) 
instance Exception Shutdown 

app :: ThreadId -> MVar() -> Request -> Iteratee ByteString IO Response 
app mainThread shutdownMVar Request{pathInfo = pathInfo} = do 
    liftIO $ case pathInfo of 
    ["shutdownByThrowing"] -> throwTo mainThread Shutdown 
    ["shutdownByMVar"]  -> putMVar shutdownMVar() 
    ["shutdownByExit"]  -> exitImmediately ExitSuccess 
    _      -> return() 
    return $ responseLBS statusOK [headerContentType "text/plain"] "ok" 

main :: IO() 
main = do 
    mainThread <- myThreadId 
    shutdownMVar <- newEmptyMVar 
    forkIO $ Warp.run 3000 (app mainThread shutdownMVar) 
    takeMVar shutdownMVar