acid-state-dist-0.1.0.1: examples/IntMasterInteractive.hs

{-# LANGUAGE TypeFamilies #-}

import Data.Acid
import Data.Acid.Centered
import Data.Acid.Centered.Master (createArchiveGlobally)
import Data.SafeCopy
import Data.Typeable

import Control.Monad (forever, replicateM_)
import System.Exit (exitSuccess)

-- state structures
import IntCommon

-- actual test
main :: IO ()
main = do
    acid <- openMasterState "127.0.0.1" 3333 (IntState 0)
    putStrLn usage
    forever $ do
        input <- getLine
        case input of
            ('x':_) -> do
                putStrLn "Bye!"
                closeAcidState acid
                exitSuccess
            ('c':_) -> do
                createCheckpoint acid
                putStrLn "Checkpoint generated."
            ('a':_) -> do
                createArchiveGlobally acid
                putStrLn "Archive generated."
            ('q':_) -> do
                val <- query acid GetState
                putStrLn $ "Current value: " ++ show val
            ('u':val) -> do
                update acid (SetState (read val :: Int))
                putStrLn "State updated."
            ('i':_) -> update acid IncrementState >> putStrLn "State incremented."
            ('k':_) -> do
                replicateM_ 1000 $ update acid IncrementState
                putStrLn "State incremented 1k times."
            _ -> putStrLn "Unknown command." >> putStrLn usage


usage :: String
usage = "Possible commands:\
        \\n  x    exit\
        \\n  c    checkpoint\
        \\n  a    archive globally\
        \\n  q    query the state\
        \\n  u v  update to value v\
        \\n  i    increment\
        \\n  k    increment 1000 times"