systemd integration

package.yaml

@@ -121,6 +121,7 @@ dependencies:
   - jose-jwt
   - mono-traversable
   - lens-aeson
+  - systemd
 
 other-extensions:
   - GeneralizedNewtypeDeriving

src/Application.hs

@@ -24,7 +24,8 @@ import Language.Haskell.TH.Syntax           (qLocation)
 import Network.Wai (Middleware)
 import Network.Wai.Handler.Warp             (Settings, defaultSettings,
                                              defaultShouldDisplayException,
-                                             runSettings, setHost,
+                                             runSettings, runSettingsSocket, setHost,
+                                             setBeforeMainLoop,
                                              setOnException, setPort, getPort)
 import Network.Wai.Middleware.RequestLogger (Destination (Logger),
                                              IPAddrSource (..),
@@ -71,6 +72,8 @@ import qualified Data.Aeson as Aeson
 import System.Exit (exitFailure)
 
 import qualified Database.Memcached.Binary.IO as Memcached
+
+import qualified System.Systemd.Daemon as Systemd
   
 -- Import all relevant handler modules here.
 -- (HPack takes care to add new modules to our cabal file nowadays.)
@@ -291,6 +294,7 @@ warpSettings :: UniWorX -> Settings
 warpSettings foundation = defaultSettings
     & setPort (foundation ^. _appPort)
     & setHost (foundation ^. _appHost)
+    & setBeforeMainLoop (void Systemd.notifyReady)
     & setOnException (\_req e ->
         when (defaultShouldDisplayException e) $ do
           logger <- readTVarIO . snd $ appLogger foundation
@@ -338,7 +342,12 @@ appMain = runResourceT $ do
     app <- makeApplication foundation
 
     -- Run the application with Warp
-    liftIO $ runSettings (warpSettings foundation) app
+    activatedSockets <- liftIO Systemd.getActivatedSockets
+    liftIO $ case activatedSockets of
+      Just [sock]
+        -> runSettingsSocket (warpSettings foundation) sock app
+      _other
+        -> runSettings (warpSettings foundation) app
 
 
 --------------------------------------------------------------

stack.yaml

@@ -49,4 +49,6 @@ extra-deps:
   - quickcheck-classes-0.4.14
   - semirings-0.2.1.1
 
+  - systemd-1.1.2
+
 resolver: lts-10.5