code.grnet.gr Git - ganeti-local/blob - src/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
  28 import Ganeti.Errors
  29 import Ganeti.Config
  30 import Ganeti.Objects
  31 import qualified Ganeti.Path as P
  32 import Ganeti.Rpc
  33 import Ganeti.Utils
  34
  35 -- | Show usage info and exit.
  36 usage :: IO ()
  37 usage = do
  38   prog <- getProgName
  39   exitErr $ "Usage: " ++ prog ++ " delay node..."
  40
  41 main :: IO ()
  42 main = do
  43   args <- getArgs
  44   (delay, nodes) <- case args of
  45                       [] -> usage >> return ("", []) -- workaround types...
  46                       _:[] -> usage >> return ("", [])
  47                       x:xs -> return (x, xs)
  48   cfg_file <- P.clusterConfFile
  49   cfg <- loadConfig  cfg_file>>= exitIfBad "Can't load configuration"
  50   let call = RpcCallTestDelay (read delay)
  51   nodes' <- exitIfBad "Can't find node" . errToResult $
  52             mapM (getNode cfg) nodes
  53   results <- executeRpcCall nodes' call
  54   putStr $ printTable "" ["Node", "Result"]
  55            (map (\(n, r) -> [nodeName n, either explainRpcError show r])
  56                 results)
  57            [False, False]