Synnefo » snf-ganeti

{-# LANGUAGE CPP #-}

{-| Scan clusters via RAPI or LUXI and write state data files.

-}

{-

Copyright (C) 2009, 2010 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.

-}

module Main (main) where

import Data.Maybe (isJust, fromJust, fromMaybe)

import Monad

import System (exitWith, ExitCode(..))

import System.IO

import System.FilePath

import qualified System

import Text.Printf (printf)

import qualified Ganeti.HTools.Container as Container

import qualified Ganeti.HTools.Cluster as Cluster

import qualified Ganeti.HTools.Group as Group

import qualified Ganeti.HTools.Node as Node

import qualified Ganeti.HTools.Instance as Instance

#ifndef NO_CURL

import qualified Ganeti.HTools.Rapi as Rapi

#endif

import qualified Ganeti.HTools.Luxi as Luxi

import qualified Ganeti.HTools.Loader as Loader

import Ganeti.HTools.Text (serializeCluster)

import Ganeti.HTools.CLI

import Ganeti.HTools.Types

-- | Options list and functions

options :: [OptType]

options =

    [ oPrintNodes

    , oOutputDir

    , oLuxiSocket

    , oVerbose

    , oNoHeaders

    , oShowVer

    , oShowHelp

    ]

-- | Return a one-line summary of cluster state

printCluster :: Node.List -> Instance.List

             -> String

printCluster nl il =

    let (bad_nodes, bad_instances) = Cluster.computeBadItems nl il

        ccv = Cluster.compCV nl

        nodes = Container.elems nl

        insts = Container.elems il

        t_ram = sum . map Node.tMem $ nodes

        t_dsk = sum . map Node.tDsk $ nodes

        f_ram = sum . map Node.fMem $ nodes

        f_dsk = sum . map Node.fDsk $ nodes

    in

      printf "%5d %5d %5d %5d %6.0f %6d %6.0f %6d %.8f"

                 (length nodes) (length insts)

                 (length bad_nodes) (length bad_instances)

                 t_ram f_ram

                 (t_dsk / 1024) (f_dsk `div` 1024)

                 ccv

-- | Replace slashes with underscore for saving to filesystem

fixSlash :: String -> String

fixSlash = map (\x -> if x == '/' then '_' else x)

-- | Generates serialized data from loader input

processData :: Result (Group.List, Node.List, Instance.List, [String])

            -> Result (Group.List, Node.List, Instance.List, String)

processData input_data = do

  (gl, nl, il, _) <- input_data >>= Loader.mergeData [] [] []

  let (_, fix_nl) = Loader.checkData nl il

      adata = serializeCluster gl nl il

  return (gl, fix_nl, il, adata)

-- | Writes cluster data out

writeData :: Int

          -> String

          -> Options

          -> Result (Group.List, Node.List, Instance.List, String)

          -> IO Bool

writeData _ name _ (Bad err) =

  printf "\nError for %s: failed to load data. Details:\n%s\n" name err >>

  return False

writeData nlen name opts (Ok (_, nl, il, adata)) = do

  printf "%-*s " nlen name :: IO ()

  hFlush stdout

  let shownodes = optShowNodes opts

      odir = optOutPath opts

      oname = odir </> fixSlash name

  putStrLn $ printCluster nl il

  hFlush stdout

  when (isJust shownodes) $

       putStr $ Cluster.printNodes nl (fromJust shownodes)

  writeFile (oname <.> "data") adata

  return True

-- | Main function.

main :: IO ()

main = do

  cmd_args <- System.getArgs

  (opts, clusters) <- parseOpts cmd_args "hscan" options

  let local = "LOCAL"

  let nlen = if null clusters

             then length local

             else maximum . map length $ clusters

  unless (optNoHeaders opts) $

         printf "%-*s %5s %5s %5s %5s %6s %6s %6s %6s %10s\n" nlen

                "Name" "Nodes" "Inst" "BNode" "BInst" "t_mem" "f_mem"

                "t_disk" "f_disk" "Score"

  when (null clusters) $ do

         let lsock = fromMaybe defaultLuxiSocket (optLuxi opts)

         let name = local

         input_data <- Luxi.loadData lsock

         result <- writeData nlen name opts (processData input_data)

         when (not result) $ exitWith $ ExitFailure 2

#ifndef NO_CURL

  results <- mapM (\ name ->

                    do

                      input_data <- Rapi.loadData name

                      writeData nlen name opts (processData input_data)

                  ) clusters

  when (not $ all id results) $ exitWith (ExitFailure 2)

#else

  when (not $ null clusters) $ do

    putStrLn "RAPI/curl backend disabled at compile time, cannot scan clusters"

    exitWith $ ExitFailure 1

#endif