¿Cómo implemento un comando de apagado en un servidor WAI?

Question

me gustaría implementar un comando de 'un apagado' para mi aplicación web (en comparación con mi primer instinto, que es simplemente pedir a la gente para matar el proceso)

Mis primeros dos intentos consistieron en

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

Ambos de los cuales solo alegremente devuelto un resultado al cliente y continuó escuchando. ¿Hay algo que una aplicación pueda hacer para matar el servidor? ¿Es esto algo razonable que quieras hacer?

Confieso que no tengo una comprensión muy sólida de iteratee, solo lo suficiente como para saber que puedo consumir mi entrada y que Iteratee es una instancia de MonadIO.

Answer 1

1. Utilice un MVar. Bloquee el hilo principal hasta que se haya señalizado el MVar, luego limpie y salga.
2. Llamada exitImmediately. Una de las maneras más rápidas de derribar el proceso, y también terriblemente molesto para depurar. No creo que los finalizadores/corchetes/bloques finalmente se llamarán en el camino hacia abajo, dependiendo de su aplicación puede corromper el estado.
3. Lanza una excepción al hilo principal. Warp.run no capta excepciones, por lo que esto funciona al permitir que el manejador de excepciones predeterminado en el hilo principal (y solo el hilo principal) finalice el proceso.

Como han mencionado otros, usar un MVar es probablemente la mejor opción. Incluí los otros para completar, pero tienen su lugar. throwTo se usa algo en la biblioteca base y he trabajado en algunas aplicaciones que usan el equivalente en C de exitImmediately: exit(), aunque no he encontrado ninguna aplicación Haskell que use este método.

{-# 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