code.grnet.gr Git - ganeti-local/blob - htools/rpc-test.hs

   1 {-| RPC test program.
   2
   3 -}
   4
   5 {-
   6
   7 Copyright (C) 2011, 2012 Google Inc.
   8
   9 This program is free software; you can redistribute it and/or modify
  10 it under the terms of the GNU General Public License as published by
  11 the Free Software Foundation; either version 2 of the License, or
  12 (at your option) any later version.
  13
  14 This program is distributed in the hope that it will be useful, but
  15 WITHOUT ANY WARRANTY; without even the implied warranty of
  16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17 General Public License for more details.
  18
  19 You should have received a copy of the GNU General Public License
  20 along with this program; if not, write to the Free Software
  21 Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  22 02110-1301, USA.
  23
  24 -}
  25
  26 import System.Environment
  27 import System.Exit
  28 import System.IO
  29
  30 import Ganeti.Config
  31 import Ganeti.Objects
  32 import qualified Ganeti.Path as P
  33 import Ganeti.Rpc
  34 import Ganeti.Utils
  35
  36 -- | Show usage info and exit.
  37 usage :: IO ()
  38 usage = do
  39   prog <- getProgName
  40   exitErr "Usage: " ++ prog ++ " delay node..."
  41
  42 main :: IO ()
  43 main = do
  44   args <- getArgs
  45   (delay, nodes) <- case args of
  46                       [] -> usage >> return ("", []) -- workaround types...
  47                       _:[] -> usage >> return ("", [])
  48                       x:xs -> return (x, xs)
  49   cfg <- loadConfig P.clusterConfFile >>=
  50          exitIfBad "Can't load configuration"
  51   let call = RpcCallTestDelay (read delay)
  52   nodes' <- exitIfBad "Can't find node" $ mapM (getNode cfg) nodes
  53   results <- executeRpcCall nodes' call
  54   putStr $ printTable "" ["Node", "Result"]
  55            (map (\(n, r) -> [nodeName n, show r]) results) [False, False]