mirror of
https://github.com/ilyakooo0/urbit.git
synced 2024-12-15 10:02:47 +03:00
705 lines
20 KiB
Haskell
705 lines
20 KiB
Haskell
{-|
|
|
Low-Level IPC flows for interacting with the serf process.
|
|
|
|
- Serf process can be started and shutdown with `start` and `stop`.
|
|
- You can ask the serf what it's last event was with
|
|
`serfLastEventBlocking`.
|
|
- A running serf can be asked to compact it's heap or take a snapshot.
|
|
- You can scry into a running serf.
|
|
- A running serf can be asked to execute a boot sequence, replay from
|
|
existing events, and run a ship with `boot`, `replay`, and `run`.
|
|
|
|
The `run` and `replay` flows will do batching of events to keep the
|
|
IPC pipe full.
|
|
|
|
```
|
|
|%
|
|
:: +writ: from king to serf
|
|
::
|
|
+$ gang (unit (set ship))
|
|
+$ writ
|
|
$% $: %live
|
|
$% [%cram eve=@]
|
|
[%exit cod=@]
|
|
[%save eve=@]
|
|
[%pack ~]
|
|
== ==
|
|
[%peek mil=@ now=@da lyc=gang pat=path]
|
|
[%play eve=@ lit=(list ?((pair @da ovum) *))]
|
|
[%work mil=@ job=(pair @da ovum)]
|
|
==
|
|
:: +plea: from serf to king
|
|
::
|
|
+$ plea
|
|
$% [%live ~]
|
|
[%ripe [pro=%1 hon=@ nok=@] eve=@ mug=@]
|
|
[%slog pri=@ ?(cord tank)]
|
|
$: %peek
|
|
$% [%done dat=(unit (cask))]
|
|
[%bail dud=goof]
|
|
== ==
|
|
$: %play
|
|
$% [%done mug=@]
|
|
[%bail eve=@ mug=@ dud=goof]
|
|
== ==
|
|
$: %work
|
|
$% [%done eve=@ mug=@ fec=(list ovum)]
|
|
[%swap eve=@ mug=@ job=(pair @da ovum) fec=(list ovum)]
|
|
[%bail lud=(list goof)]
|
|
== ==
|
|
==
|
|
```
|
|
-}
|
|
|
|
module Urbit.Vere.Serf.IPC
|
|
( Serf
|
|
, start
|
|
, stop
|
|
, serfLastEventBlocking
|
|
, snapshot
|
|
, compact
|
|
, scry
|
|
, boot
|
|
, replay
|
|
, run
|
|
, swim
|
|
, sendSIGINT
|
|
, module Urbit.Vere.Serf.Types
|
|
)
|
|
where
|
|
|
|
import Urbit.Prelude hiding ((<|))
|
|
|
|
import Data.Bits
|
|
import Data.Conduit
|
|
import System.Process
|
|
import Urbit.Vere.Serf.Types
|
|
|
|
import Control.Monad.STM (retry)
|
|
import Control.Monad.Trans.Resource (MonadResource, allocate, runResourceT)
|
|
import Data.Sequence (Seq((:<|), (:|>)))
|
|
import Foreign.Marshal.Alloc (alloca)
|
|
import Foreign.Ptr (castPtr)
|
|
import Foreign.Storable (peek, poke)
|
|
import RIO.Prelude (decodeUtf8Lenient)
|
|
import System.Posix.Signals (sigINT, sigKILL, signalProcess)
|
|
import Urbit.Arvo (Ev, FX)
|
|
import Urbit.Noun.Time (Wen)
|
|
|
|
import qualified Data.ByteString as BS
|
|
import qualified Data.ByteString.Unsafe as BS
|
|
import qualified System.IO.Error as IO
|
|
import qualified Urbit.Noun.Time as Time
|
|
|
|
|
|
-- Serf API --------------------------------------------------------------------
|
|
|
|
data Serf = Serf
|
|
{ serfSend :: Handle
|
|
, serfRecv :: Handle
|
|
, serfProc :: ProcessHandle
|
|
, serfSlog :: Slog -> IO ()
|
|
, serfLock :: MVar (Maybe SerfState)
|
|
}
|
|
|
|
|
|
-- Internal Protocol Types -----------------------------------------------------
|
|
|
|
data Live
|
|
= LExit Atom -- exit status code
|
|
| LSave EventId
|
|
| LCram EventId
|
|
| LPack ()
|
|
deriving (Show)
|
|
|
|
data Play
|
|
= PDone Mug
|
|
| PBail PlayBail
|
|
deriving (Show)
|
|
|
|
data Scry
|
|
= SDone (Maybe (Term, Noun))
|
|
| SBail Goof
|
|
deriving (Show)
|
|
|
|
data Work
|
|
= WDone EventId Mug FX
|
|
| WSwap EventId Mug (Wen, Noun) FX
|
|
| WBail [Goof]
|
|
deriving (Show)
|
|
|
|
data Writ
|
|
= WLive Live
|
|
| WPeek Atom Wen Gang Path
|
|
| WPlay EventId [Noun]
|
|
| WWork Atom Wen Ev
|
|
deriving (Show)
|
|
|
|
data Plea
|
|
= PLive ()
|
|
| PRipe SerfInfo
|
|
| PSlog Slog
|
|
| PPeek Scry
|
|
| PPlay Play
|
|
| PWork Work
|
|
deriving (Show)
|
|
|
|
deriveNoun ''Live
|
|
deriveNoun ''Play
|
|
deriveNoun ''Scry
|
|
deriveNoun ''Work
|
|
deriveNoun ''Writ
|
|
deriveNoun ''Plea
|
|
|
|
|
|
-- Access Current Serf State ---------------------------------------------------
|
|
|
|
serfLastEventBlocking :: Serf -> IO EventId
|
|
serfLastEventBlocking Serf{serfLock} = readMVar serfLock >>= \case
|
|
Nothing -> throwIO SerfNotRunning
|
|
Just ss -> pure (ssLast ss)
|
|
|
|
|
|
-- Low Level IPC Functions -----------------------------------------------------
|
|
|
|
fromRightExn :: (Exception e, MonadIO m) => Either a b -> (a -> e) -> m b
|
|
fromRightExn (Left m) exn = throwIO (exn m)
|
|
fromRightExn (Right x) _ = pure x
|
|
|
|
-- TODO Support Big Endian
|
|
sendLen :: Serf -> Int -> IO ()
|
|
sendLen s i = do
|
|
w <- evaluate (fromIntegral i :: Word64)
|
|
withWord64AsByteString w (hPut (serfSend s))
|
|
where
|
|
withWord64AsByteString :: Word64 -> (ByteString -> IO a) -> IO a
|
|
withWord64AsByteString w k = alloca $ \wp -> do
|
|
poke wp w
|
|
bs <- BS.unsafePackCStringLen (castPtr wp, 8)
|
|
k bs
|
|
|
|
sendBytes :: Serf -> ByteString -> IO ()
|
|
sendBytes s bs = handle onIOError $ do
|
|
sendLen s (length bs)
|
|
hPut (serfSend s) bs
|
|
hFlush (serfSend s)
|
|
where
|
|
onIOError :: IOError -> IO ()
|
|
onIOError = const (throwIO SerfConnectionClosed)
|
|
|
|
recvBytes :: Serf -> Word64 -> IO ByteString
|
|
recvBytes serf = BS.hGet (serfRecv serf) . fromIntegral
|
|
|
|
recvLen :: Serf -> IO Word64
|
|
recvLen w = do
|
|
bs <- BS.hGet (serfRecv w) 8
|
|
case length bs of
|
|
8 -> BS.unsafeUseAsCString bs (peek @Word64 . castPtr)
|
|
_ -> throwIO SerfConnectionClosed
|
|
|
|
recvResp :: Serf -> IO ByteString
|
|
recvResp serf = do
|
|
len <- recvLen serf
|
|
recvBytes serf len
|
|
|
|
|
|
-- Send Writ / Recv Plea -------------------------------------------------------
|
|
|
|
sendWrit :: Serf -> Writ -> IO ()
|
|
sendWrit s = sendBytes s . jamBS . toNoun
|
|
|
|
recvPlea :: Serf -> IO Plea
|
|
recvPlea w = do
|
|
b <- recvResp w
|
|
n <- fromRightExn (cueBS b) (const $ BadPleaAtom $ bytesAtom b)
|
|
p <- fromRightExn (fromNounErr @Plea n) (\(p, m) -> BadPleaNoun n p m)
|
|
pure p
|
|
|
|
recvPleaHandlingSlog :: Serf -> IO Plea
|
|
recvPleaHandlingSlog serf = loop
|
|
where
|
|
loop = recvPlea serf >>= \case
|
|
PSlog info -> serfSlog serf info >> loop
|
|
other -> pure other
|
|
|
|
|
|
-- Higher-Level IPC Functions --------------------------------------------------
|
|
|
|
recvRipe :: Serf -> IO SerfInfo
|
|
recvRipe serf = recvPleaHandlingSlog serf >>= \case
|
|
PRipe ripe -> pure ripe
|
|
plea -> throwIO (UnexpectedPlea (toNoun plea) "expecting %play")
|
|
|
|
recvPlay :: Serf -> IO Play
|
|
recvPlay serf = recvPleaHandlingSlog serf >>= \case
|
|
PPlay play -> pure play
|
|
plea -> throwIO (UnexpectedPlea (toNoun plea) "expecting %play")
|
|
|
|
recvLive :: Serf -> IO ()
|
|
recvLive serf = recvPleaHandlingSlog serf >>= \case
|
|
PLive () -> pure ()
|
|
plea -> throwIO (UnexpectedPlea (toNoun plea) "expecting %live")
|
|
|
|
recvWork :: Serf -> IO Work
|
|
recvWork serf = do
|
|
recvPleaHandlingSlog serf >>= \case
|
|
PWork work -> pure work
|
|
plea -> throwIO (UnexpectedPlea (toNoun plea) "expecting %work")
|
|
|
|
recvPeek :: Serf -> IO (Maybe (Term, Noun))
|
|
recvPeek serf = do
|
|
recvPleaHandlingSlog serf >>= \case
|
|
PPeek (SDone peek) -> pure peek
|
|
-- XX produce error
|
|
PPeek (SBail dud) -> throwIO (PeekBail dud)
|
|
plea -> throwIO (UnexpectedPlea (toNoun plea) "expecting %peek")
|
|
|
|
|
|
-- Request-Response Points -- These don't touch the lock -----------------------
|
|
|
|
sendSnapshotRequest :: Serf -> EventId -> IO ()
|
|
sendSnapshotRequest serf eve = do
|
|
sendWrit serf (WLive $ LSave eve)
|
|
recvLive serf
|
|
|
|
sendCompactionRequest :: Serf -> IO ()
|
|
sendCompactionRequest serf = do
|
|
sendWrit serf (WLive $ LPack ())
|
|
recvLive serf
|
|
|
|
sendScryRequest :: Serf -> Wen -> Gang -> Path -> IO (Maybe (Term, Noun))
|
|
sendScryRequest serf w g p = do
|
|
sendWrit serf (WPeek 0 w g p)
|
|
recvPeek serf
|
|
|
|
sendShutdownRequest :: Serf -> Atom -> IO ()
|
|
sendShutdownRequest serf exitCode = do
|
|
sendWrit serf (WLive $ LExit exitCode)
|
|
pure ()
|
|
|
|
|
|
-- Starting the Serf -----------------------------------------------------------
|
|
|
|
compileFlags :: [Flag] -> Word
|
|
compileFlags = foldl' (\acc flag -> setBit acc (fromEnum flag)) 0
|
|
|
|
readStdErr :: Handle -> (Text -> IO ()) -> IO () -> IO ()
|
|
readStdErr h onLine onClose = loop
|
|
where
|
|
loop = do
|
|
IO.tryIOError (BS.hGetLine h >>= onLine . decodeUtf8Lenient) >>= \case
|
|
Left exn -> onClose
|
|
Right () -> loop
|
|
|
|
start :: Config -> IO (Serf, SerfInfo)
|
|
start (Config exePax pierPath flags onSlog onStdr onDead) = do
|
|
(Just i, Just o, Just e, p) <- createProcess pSpec
|
|
void $ async (readStdErr e onStdr onDead)
|
|
vLock <- newEmptyMVar
|
|
let serf = Serf i o p onSlog vLock
|
|
info <- recvRipe serf
|
|
putMVar vLock (Just $ siStat info)
|
|
pure (serf, info)
|
|
where
|
|
diskKey = ""
|
|
config = show (compileFlags flags)
|
|
rock = "0" -- XX support loading from rock
|
|
cache = "50000" -- XX support memo-cache size
|
|
args = ["serf", pierPath, diskKey, config, cache, rock]
|
|
pSpec = (proc exePax args) { std_in = CreatePipe
|
|
, std_out = CreatePipe
|
|
, std_err = CreatePipe
|
|
}
|
|
|
|
|
|
-- Taking the SerfState Lock ---------------------------------------------------
|
|
|
|
takeLock :: MonadIO m => Serf -> m SerfState
|
|
takeLock serf = io $ do
|
|
takeMVar (serfLock serf) >>= \case
|
|
Nothing -> putMVar (serfLock serf) Nothing >> throwIO SerfNotRunning
|
|
Just ss -> pure ss
|
|
|
|
serfLockTaken
|
|
:: MonadResource m => Serf -> m (IORef (Maybe SerfState), SerfState)
|
|
serfLockTaken serf = snd <$> allocate take release
|
|
where
|
|
take = (,) <$> newIORef Nothing <*> takeLock serf
|
|
release (rv, _) = do
|
|
mRes <- readIORef rv
|
|
when (mRes == Nothing) (forcefullyKillSerf serf)
|
|
putMVar (serfLock serf) mRes
|
|
|
|
withSerfLock
|
|
:: MonadResource m => Serf -> (SerfState -> m (SerfState, a)) -> m a
|
|
withSerfLock serf act = do
|
|
(vState , initialState) <- serfLockTaken serf
|
|
(newState, result ) <- act initialState
|
|
writeIORef vState (Just newState)
|
|
pure result
|
|
|
|
withSerfLockIO :: Serf -> (SerfState -> IO (SerfState, a)) -> IO a
|
|
withSerfLockIO s a = runResourceT (withSerfLock s (io . a))
|
|
|
|
|
|
-- SIGINT ----------------------------------------------------------------------
|
|
|
|
sendSIGINT :: Serf -> IO ()
|
|
sendSIGINT serf = do
|
|
getPid (serfProc serf) >>= \case
|
|
Nothing -> pure ()
|
|
Just pid -> do
|
|
io $ signalProcess sigINT pid
|
|
|
|
|
|
-- Killing the Serf ------------------------------------------------------------
|
|
|
|
|
|
{-|
|
|
Ask the serf to shutdown. If it takes more than 2s, kill it with
|
|
SIGKILL.
|
|
-}
|
|
stop :: HasLogFunc e => Serf -> RIO e ()
|
|
stop serf = do
|
|
race_ niceKill (wait2sec >> forceKill)
|
|
where
|
|
wait2sec = threadDelay 2_000_000
|
|
|
|
niceKill = do
|
|
logTrace "Asking serf to shut down"
|
|
io (gracefullyKillSerf serf)
|
|
logTrace "Serf went down when asked."
|
|
|
|
forceKill = do
|
|
logTrace "Serf taking too long to go down, kill with fire (SIGTERM)."
|
|
io (forcefullyKillSerf serf)
|
|
logTrace "Serf process killed with SIGTERM."
|
|
|
|
{-|
|
|
Kill the serf by taking the lock, then asking for it to exit.
|
|
-}
|
|
gracefullyKillSerf :: Serf -> IO ()
|
|
gracefullyKillSerf serf@Serf{..} = do
|
|
finalState <- takeMVar serfLock
|
|
sendShutdownRequest serf 0
|
|
waitForProcess serfProc
|
|
pure ()
|
|
|
|
{-|
|
|
Kill the serf by sending it a SIGKILL.
|
|
-}
|
|
forcefullyKillSerf :: Serf -> IO ()
|
|
forcefullyKillSerf serf = do
|
|
getPid (serfProc serf) >>= \case
|
|
Nothing -> pure ()
|
|
Just pid -> do
|
|
io $ signalProcess sigKILL pid
|
|
io $ void $ waitForProcess (serfProc serf)
|
|
|
|
|
|
-- Flows for Interacting with the Serf -----------------------------------------
|
|
|
|
{-|
|
|
Ask the serf to write a snapshot to disk.
|
|
-}
|
|
snapshot :: Serf -> IO ()
|
|
snapshot serf = withSerfLockIO serf $ \ss -> do
|
|
sendSnapshotRequest serf (ssLast ss)
|
|
pure (ss, ())
|
|
|
|
{-|
|
|
Ask the serf to de-duplicate and de-fragment it's heap.
|
|
-}
|
|
compact :: Serf -> IO ()
|
|
compact serf = withSerfLockIO serf $ \ss -> do
|
|
sendCompactionRequest serf
|
|
pure (ss, ())
|
|
|
|
{-|
|
|
Peek into the serf state.
|
|
-}
|
|
scry :: Serf -> Wen -> Gang -> Path -> IO (Maybe (Term, Noun))
|
|
scry serf w g p = withSerfLockIO serf $ \ss -> do
|
|
(ss,) <$> sendScryRequest serf w g p
|
|
|
|
|
|
{-|
|
|
Given a list of boot events, send them to to the serf in a single
|
|
%play message. They must all be sent in a single %play event so that
|
|
the serf can determine the length of the boot sequence.
|
|
-}
|
|
boot :: Serf -> [Noun] -> IO (Maybe PlayBail)
|
|
boot serf@Serf {..} seq = do
|
|
withSerfLockIO serf $ \ss -> do
|
|
sendWrit serf (WPlay 1 seq)
|
|
recvPlay serf >>= \case
|
|
PBail bail -> pure (ss, Just bail)
|
|
PDone mug -> pure (SerfState (fromIntegral $ length seq) mug, Nothing)
|
|
|
|
{-|
|
|
Given a stream of nouns (from the event log), feed them into the serf
|
|
in batches of size `batchSize`.
|
|
|
|
- On `%bail` response, return early.
|
|
- On IPC errors, kill the serf and rethrow.
|
|
- On success, return `Nothing`.
|
|
-}
|
|
replay
|
|
:: forall m
|
|
. (MonadResource m, MonadUnliftIO m, MonadIO m)
|
|
=> Int
|
|
-> (Int -> IO ())
|
|
-> Serf
|
|
-> ConduitT Noun Void m (Maybe PlayBail)
|
|
replay batchSize cb serf = do
|
|
withSerfLock serf $ \ss -> do
|
|
(r, ss') <- loop ss
|
|
pure (ss', r)
|
|
where
|
|
loop :: SerfState -> ConduitT Noun Void m (Maybe PlayBail, SerfState)
|
|
loop ss@(SerfState lastEve lastMug) = do
|
|
awaitBatch batchSize >>= \case
|
|
[] -> pure (Nothing, SerfState lastEve lastMug)
|
|
evs -> do
|
|
let nexEve = lastEve + 1
|
|
let newEve = lastEve + fromIntegral (length evs)
|
|
io $ sendWrit serf (WPlay nexEve evs)
|
|
io (recvPlay serf) >>= \case
|
|
PBail bail -> pure (Just bail, SerfState lastEve lastMug)
|
|
PDone newMug -> do
|
|
io (cb $ length evs)
|
|
loop (SerfState newEve newMug)
|
|
|
|
{-|
|
|
TODO If this is slow, use a mutable vector instead of reversing a list.
|
|
-}
|
|
awaitBatch :: Monad m => Int -> ConduitT i o m [i]
|
|
awaitBatch = go []
|
|
where
|
|
go acc 0 = pure (reverse acc)
|
|
go acc n = await >>= \case
|
|
Nothing -> pure (reverse acc)
|
|
Just x -> go (x:acc) (n-1)
|
|
|
|
|
|
-- Special Replay for Collecting FX --------------------------------------------
|
|
|
|
{-|
|
|
This does event-log replay using the running IPC flow so that we
|
|
can collect effects.
|
|
|
|
We don't tolerate replacement events or bails since we are actually
|
|
replaying the log, so we just throw exceptions in those cases.
|
|
-}
|
|
swim
|
|
:: forall m
|
|
. (MonadIO m, MonadUnliftIO m, MonadResource m)
|
|
=> Serf
|
|
-> ConduitT (Wen, Ev) (EventId, FX) m ()
|
|
swim serf = do
|
|
withSerfLock serf $ \SerfState {..} -> do
|
|
(, ()) <$> loop ssHash ssLast
|
|
where
|
|
loop
|
|
:: Mug
|
|
-> EventId
|
|
-> ConduitT (Wen, Ev) (EventId, FX) m SerfState
|
|
loop mug eve = await >>= \case
|
|
Nothing -> do
|
|
pure (SerfState eve mug)
|
|
Just (wen, evn) -> do
|
|
io (sendWrit serf (WWork 0 wen evn))
|
|
io (recvWork serf) >>= \case
|
|
WBail goofs -> do
|
|
throwIO (BailDuringReplay eve goofs)
|
|
WSwap eid hash (wen, noun) fx -> do
|
|
throwIO (SwapDuringReplay eid hash (wen, noun) fx)
|
|
WDone eid hash fx -> do
|
|
yield (eid, fx)
|
|
loop hash eid
|
|
|
|
|
|
|
|
-- Running Ship Flow -----------------------------------------------------------
|
|
|
|
{-|
|
|
TODO Don't take snapshot until event log has processed current event.
|
|
-}
|
|
run
|
|
:: Serf
|
|
-> Int
|
|
-> STM EventId
|
|
-> STM RunReq
|
|
-> ((Fact, FX) -> STM ())
|
|
-> (Maybe Ev -> STM ())
|
|
-> IO ()
|
|
run serf maxBatchSize getLastEvInLog onInput sendOn spin = topLoop
|
|
where
|
|
topLoop :: IO ()
|
|
topLoop = atomically onInput >>= \case
|
|
RRWork workErr -> doWork workErr
|
|
RRSave () -> doSave
|
|
RRKill () -> doKill
|
|
RRPack () -> doPack
|
|
RRScry w g p k -> doScry w g p k
|
|
|
|
doPack :: IO ()
|
|
doPack = compact serf >> topLoop
|
|
|
|
waitForLog :: IO ()
|
|
waitForLog = do
|
|
serfLast <- serfLastEventBlocking serf
|
|
atomically $ do
|
|
logLast <- getLastEvInLog
|
|
when (logLast < serfLast) retry
|
|
|
|
doSave :: IO ()
|
|
doSave = waitForLog >> snapshot serf >> topLoop
|
|
|
|
doKill :: IO ()
|
|
doKill = waitForLog >> snapshot serf >> pure ()
|
|
|
|
doScry :: Wen -> Gang -> Path -> (Maybe (Term, Noun) -> IO ()) -> IO ()
|
|
doScry w g p k = (scry serf w g p >>= k) >> topLoop
|
|
|
|
doWork :: EvErr -> IO ()
|
|
doWork firstWorkErr = do
|
|
que <- newTBMQueueIO 1
|
|
() <- atomically (writeTBMQueue que firstWorkErr)
|
|
tWork <- async (processWork serf maxBatchSize que onWorkResp spin)
|
|
flip onException (cancel tWork) $ do
|
|
nexSt <- workLoop que
|
|
wait tWork
|
|
nexSt
|
|
|
|
workLoop :: TBMQueue EvErr -> IO (IO ())
|
|
workLoop que = atomically onInput >>= \case
|
|
RRKill () -> atomically (closeTBMQueue que) >> pure doKill
|
|
RRSave () -> atomically (closeTBMQueue que) >> pure doSave
|
|
RRPack () -> atomically (closeTBMQueue que) >> pure doPack
|
|
RRScry w g p k -> atomically (closeTBMQueue que) >> pure (doScry w g p k)
|
|
RRWork workErr -> atomically (writeTBMQueue que workErr) >> workLoop que
|
|
|
|
onWorkResp :: Wen -> EvErr -> Work -> IO ()
|
|
onWorkResp wen (EvErr evn err) = \case
|
|
WDone eid hash fx -> do
|
|
io $ err (RunOkay eid)
|
|
atomically $ sendOn ((Fact eid hash wen (toNoun evn)), fx)
|
|
WSwap eid hash (wen, noun) fx -> do
|
|
io $ err (RunSwap eid hash wen noun fx)
|
|
atomically $ sendOn (Fact eid hash wen noun, fx)
|
|
WBail goofs -> do
|
|
io $ err (RunBail goofs)
|
|
|
|
{-|
|
|
Given:
|
|
|
|
- A stream of incoming requests
|
|
- A sequence of in-flight requests that haven't been responded to
|
|
- A maximum number of in-flight requests.
|
|
|
|
Wait until the number of in-fligh requests is smaller than the maximum,
|
|
and then take the next item from the stream of requests.
|
|
-}
|
|
pullFromQueueBounded :: Int -> TVar (Seq a) -> TBMQueue b -> STM (Maybe b)
|
|
pullFromQueueBounded maxSize vInFlight queue = do
|
|
inFlight <- length <$> readTVar vInFlight
|
|
if inFlight >= maxSize
|
|
then retry
|
|
else readTBMQueue queue
|
|
|
|
{-|
|
|
Given
|
|
|
|
- `maxSize`: The maximum number of jobs to send to the serf before
|
|
getting a response.
|
|
- `q`: A bounded queue (which can be closed)
|
|
- `onResp`: a callback to call for each response from the serf.
|
|
- `spin`: a callback to tell the terminal driver which event is
|
|
currently being processed.
|
|
|
|
Pull jobs from the queue and send them to the serf (eagerly, up to
|
|
`maxSize`) and call the callback with each response from the serf.
|
|
|
|
When the queue is closed, wait for the serf to respond to all pending
|
|
work, and then return.
|
|
|
|
Whenever the serf is idle, call `spin Nothing` and whenever the serf
|
|
is working on an event, call `spin (Just ev)`.
|
|
-}
|
|
processWork
|
|
:: Serf
|
|
-> Int
|
|
-> TBMQueue EvErr
|
|
-> (Wen -> EvErr -> Work -> IO ())
|
|
-> (Maybe Ev -> STM ())
|
|
-> IO ()
|
|
processWork serf maxSize q onResp spin = do
|
|
vDoneFlag <- newTVarIO False
|
|
vInFlightQueue <- newTVarIO empty
|
|
recvThread <- async (recvLoop serf vDoneFlag vInFlightQueue spin)
|
|
flip onException (print "KILLING: processWork" >> cancel recvThread) $ do
|
|
loop vInFlightQueue vDoneFlag
|
|
wait recvThread
|
|
where
|
|
loop :: TVar (Seq (Ev, Work -> IO ())) -> TVar Bool -> IO ()
|
|
loop vInFlight vDone = do
|
|
atomically (pullFromQueueBounded maxSize vInFlight q) >>= \case
|
|
Nothing -> do
|
|
atomically (writeTVar vDone True)
|
|
Just evErr@(EvErr ev _) -> do
|
|
now <- Time.now
|
|
let cb = onResp now evErr
|
|
atomically $ modifyTVar' vInFlight (:|> (ev, cb))
|
|
sendWrit serf (WWork 0 now ev)
|
|
loop vInFlight vDone
|
|
|
|
{-|
|
|
Given:
|
|
|
|
- `vDone`: A flag that no more work will be sent to the serf.
|
|
|
|
- `vWork`: A list of work requests that have been sent to the serf,
|
|
haven't been responded to yet.
|
|
|
|
If the serf has responded to all work requests, and no more work is
|
|
going to be sent to the serf, then return.
|
|
|
|
If we are going to send more work to the serf, but the queue is empty,
|
|
then wait.
|
|
|
|
If work requests have been sent to the serf, take the first one,
|
|
wait for a response from the serf, call the associated callback,
|
|
and repeat the whole process.
|
|
-}
|
|
recvLoop
|
|
:: Serf
|
|
-> TVar Bool
|
|
-> TVar (Seq (Ev, Work -> IO ()))
|
|
-> (Maybe Ev -> STM ())
|
|
-> IO ()
|
|
recvLoop serf vDone vWork spin = do
|
|
withSerfLockIO serf \SerfState {..} -> do
|
|
loop ssLast ssHash
|
|
where
|
|
loop eve mug = do
|
|
atomically $ do
|
|
whenM (null <$> readTVar vWork) $ do
|
|
spin Nothing
|
|
atomically takeCallback >>= \case
|
|
Nothing -> pure (SerfState eve mug, ())
|
|
Just (curEve, cb) -> do
|
|
atomically (spin (Just curEve))
|
|
recvWork serf >>= \case
|
|
work@(WDone eid hash _) -> cb work >> loop eid hash
|
|
work@(WSwap eid hash _ _) -> cb work >> loop eid hash
|
|
work@(WBail _) -> cb work >> loop eve mug
|
|
|
|
takeCallback :: STM (Maybe (Ev, Work -> IO ()))
|
|
takeCallback = do
|
|
((,) <$> readTVar vDone <*> readTVar vWork) >>= \case
|
|
(False, Empty ) -> retry
|
|
(True , Empty ) -> pure Nothing
|
|
(_ , (e, x) :<| xs) -> writeTVar vWork xs $> Just (e, x)
|
|
(_ , _ ) -> error "impossible"
|