Synnefo » snf-ganeti

{-# LANGUAGE BangPatterns #-}

{-| RPC test program.

-}

{-

Copyright (C) 2011, 2012, 2013 Google Inc.

This program is free software; you can redistribute it and/or modify

it under the terms of the GNU General Public License as published by

the Free Software Foundation; either version 2 of the License, or

(at your option) any later version.

This program is distributed in the hope that it will be useful, but

WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

General Public License for more details.

You should have received a copy of the GNU General Public License

along with this program; if not, write to the Free Software

Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

02110-1301, USA.

-}

import Control.Concurrent

import Control.Monad

import System.Console.GetOpt

import System.Environment

import System.IO

import Text.JSON (decode)

import Text.Printf

import Ganeti.BasicTypes

import Ganeti.Common

import Ganeti.Config

import Ganeti.Errors

import Ganeti.JSON

import Ganeti.Objects

import qualified Ganeti.Path as P

import Ganeti.Rpc

import Ganeti.Utils

-- | Command line options structure.

data Options = Options

  { optRpc      :: String   -- ^ RPC to execute

  , optDataFile :: FilePath -- ^ Path to the RPC serialised form

  , optVerbose  :: Bool     -- ^ Verbosity level

  , optStats    :: Bool     -- ^ Whether to show timing stats

  , optCount    :: Int      -- ^ Count of (multi) RPCs to do

  , optBatch    :: Int      -- ^ How many (multi) RPCs to run in parallel

  , optShowHelp :: Bool     -- ^ Just show the help

  , optShowComp :: Bool     -- ^ Just show the completion info

  , optShowVer  :: Bool     -- ^ Just show the program version

  } deriving Show

-- | Default values for the command line options.

defaultOptions :: Options

defaultOptions = Options

  { optRpc      = "version"

  , optDataFile = "rpc.json"

  , optVerbose  = False

  , optStats    = False

  , optCount    = 1

  , optBatch    = 1

  , optShowHelp = False

  , optShowComp = False

  , optShowVer  = False

  }

instance StandardOptions Options where

  helpRequested = optShowHelp

  verRequested  = optShowVer

  compRequested = optShowComp

  requestHelp o = o { optShowHelp = True }

  requestVer  o = o { optShowVer  = True }

  requestComp o = o { optShowComp = True }

-- | The rpcs we support. Sadly this duplicates the RPC list.

data KnownRpc = KRInstanceInfo        RpcCallInstanceInfo

              | KRAllInstancesInfo    RpcCallAllInstancesInfo

              | KRInstanceConsoleInfo RpcCallInstanceConsoleInfo

              | KRInstanceList        RpcCallInstanceList

              | KRNodeInfo            RpcCallNodeInfo

              | KRVersion             RpcCallVersion

              | KRStorageList         RpcCallStorageList

              | KRTestDelay           RpcCallTestDelay

              | KRExportList          RpcCallExportList

                deriving (Show)

-- | The command line options.

options :: [GenericOptType Options]

options =

  [ (Option "r" ["rpc"]

     (ReqArg (\ r o -> Ok o { optRpc = r }) "RPC")

     "the rpc to use [version]",

     OptComplChoices [])

  , (Option "f" ["data-file"]

     (ReqArg (\ f o -> Ok o { optDataFile = f }) "FILE")

     "the rpc serialised form [\"rpc.json\"]",

     OptComplFile)

  , (Option "v" ["verbose"]

     (NoArg (\ opts -> Ok opts { optVerbose = True}))

     "show more information when executing RPCs",

     OptComplNone)

  , (Option "t" ["stats"]

     (NoArg (\ opts -> Ok opts { optStats = True}))

     "show timing information summary",

     OptComplNone)

  , (Option "c" ["count"]

     (reqWithConversion (tryRead "reading count")

      (\count opts -> Ok opts { optCount = count }) "NUMBER")

     "Count of (multi) RPCs to execute [1]",

     OptComplInteger)

  , (Option "b" ["batch"]

     (reqWithConversion (tryRead "reading batch size")

      (\batch opts -> Ok opts { optBatch = batch }) "NUMBER")

     "Parallelisation factor for RPCs [1]",

     OptComplInteger)

  , oShowHelp

  , oShowComp

  , oShowVer

  ]

-- | Arguments we expect

arguments :: [ArgCompletion]

arguments = [ArgCompletion OptComplOneNode 1 Nothing]

-- | Log a message.

logMsg :: MVar () -> String -> IO ()

logMsg outmvar text =

  withMVar outmvar $ \_ -> do

    let p = if null text || last text /= '\n'

              then putStrLn

              else putStr

    p text

    hFlush stdout

-- | Parses a RPC.

parseRpc :: String -> String -> Result KnownRpc

parseRpc "instance_info"      f =

  fromJResult "parsing rpc" (decode f) >>= Ok . KRInstanceInfo

parseRpc "all_instances_info" f =

  fromJResult "parsing rpc" (decode f) >>= Ok . KRAllInstancesInfo

parseRpc "console_instance_info" f =

  fromJResult "parsing rpc" (decode f) >>= Ok . KRConsoleInstanceInfo

parseRpc "instance_list"      f =

  fromJResult "parsing rpc" (decode f) >>= Ok . KRInstanceList

parseRpc "node_info"          f =

  fromJResult "parsing rpc" (decode f) >>= Ok . KRNodeInfo

parseRpc "version"            f =

  fromJResult "parsing rpc" (decode f) >>= Ok . KRVersion

parseRpc "storage_list"       f =

  fromJResult "parsing rpc" (decode f) >>= Ok . KRStorageList

parseRpc "test_delay"         f =

  fromJResult "parsing rpc" (decode f) >>= Ok . KRTestDelay

parseRpc "export_list"        f =

  fromJResult "parsing rpc" (decode f) >>= Ok . KRExportList

parseRpc s _                  = Bad $ "Unknown rpc '" ++ s ++ "'"

-- | Executes a RPC. These duplicate definitions are needed due to the

-- polymorphism of 'executeRpcCall', and the binding of the result

-- based on the input rpc call.

execRpc :: [Node] -> KnownRpc -> IO [[String]]

execRpc n (KRInstanceInfo        v) = formatRpcRes `fmap` executeRpcCall n v

execRpc n (KRAllInstancesInfo    v) = formatRpcRes `fmap` executeRpcCall n v

execRpc n (KRConsoleInstanceInfo v) = formatRpcRes `fmap` executeRpcCall n v

execRpc n (KRInstanceList        v) = formatRpcRes `fmap` executeRpcCall n v

execRpc n (KRNodeInfo            v) = formatRpcRes `fmap` executeRpcCall n v

execRpc n (KRVersion             v) = formatRpcRes `fmap` executeRpcCall n v

execRpc n (KRStorageList         v) = formatRpcRes `fmap` executeRpcCall n v

execRpc n (KRTestDelay           v) = formatRpcRes `fmap` executeRpcCall n v

execRpc n (KRExportList          v) = formatRpcRes `fmap` executeRpcCall n v

-- | Helper to format the RPC result such that it can be printed by

-- 'printTable'.

formatRpcRes :: (Show b) => [(Node, ERpcError b)] -> [[String]]

formatRpcRes = map (\(n, r) -> [nodeName n, either explainRpcError show r])

-- | Main function.

main :: IO ()

main = do

  cmd_args <- getArgs

  (opts, args) <-

    parseOpts defaultOptions cmd_args "rpc-test" options arguments

  rpc <- parseRpc (optRpc opts) `liftM` readFile (optDataFile opts) >>=

         exitIfBad "parsing RPC"

  cfg_file <- P.clusterConfFile

  cfg <- loadConfig  cfg_file>>= exitIfBad "Can't load configuration"

  nodes <- exitIfBad "Can't find node" . errToResult $

            mapM (getNode cfg) args

  token <- newEmptyMVar -- semaphore for batch calls

  outmvar <- newMVar () -- token for stdout non-interleaving

  let logger = if optVerbose opts

                 then logMsg outmvar

                 else const $ return ()

  let batch = [1..optBatch opts]

      count = optCount opts

      rpcs = count * length nodes

  logger $ printf "Will execute %s multi-ops and %s RPCs"

           (show count) (show rpcs)

  tstart <- getCurrentTimeUSec

  _ <- forkIO $ mapM_ (\_ -> putMVar token ()) batch

  mapM_ (\idx -> do

           let str_idx = show idx

           logger $ "Acquiring token for run " ++ str_idx

           _ <- takeMVar token

           forkIO $ do

             start <- getCurrentTimeUSec

             logger $ "Start run " ++ str_idx

             !results <- execRpc nodes rpc

             stop <- getCurrentTimeUSec

             let delta = (fromIntegral (stop - start)::Double) / 1000

             putMVar token ()

             let stats = if optVerbose opts

                           then printf "Done run %d in %7.3fmsec\n" idx delta

                           else ""

                 table = printTable "" ["Node", "Result"]

                           results [False, False]

             logMsg outmvar $ stats ++ table

        ) [1..count]

  mapM_ (\_ -> takeMVar token) batch

  _ <- takeMVar outmvar

  when (optStats opts) $ do

    tstop <- getCurrentTimeUSec

    let delta = (fromIntegral (tstop - tstart) / 1000000)::Double

    printf "Total runtime:     %9.3fs\n" delta :: IO ()

    printf "Total mult-ops:    %9d\n" count :: IO ()

    printf "Total single RPCs: %9d\n" rpcs  :: IO ()

    printf "Multi-ops/sec:     %9.3f\n" (fromIntegral count / delta) :: IO ()

    printf "RPCs/sec:          %9.3f\n" (fromIntegral rpcs / delta) :: IO ()