Synnefo » snf-ganeti

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

-}

{-

Copyright (C) 2009, 2010, 2011 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 Ganeti.HTools.Program.Hscan (main) where

import Control.Monad

import Data.Maybe (isJust, fromJust, fromMaybe)

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.Node as Node

import qualified Ganeti.HTools.Instance as Instance

import qualified Ganeti.HTools.Rapi as Rapi

import qualified Ganeti.HTools.Luxi as Luxi

import Ganeti.HTools.Loader (checkData, mergeData, ClusterData(..))

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 :: ClusterData -> Result ClusterData

processData input_data = do

  cdata@(ClusterData _ nl il _) <- mergeData [] [] [] [] input_data

  let (_, fix_nl) = checkData nl il

  return cdata { cdNodes = fix_nl }

-- | Writes cluster data out.

writeData :: Int

          -> String

          -> Options

          -> Result ClusterData

          -> 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 cdata) = do

  let fixdata = processData cdata

  case fixdata of

    Bad err -> printf "\nError for %s: failed to process data. Details:\n%s\n"

               name err >> return False

    Ok processed -> writeDataInner nlen name opts cdata processed

-- | Inner function for writing cluster data to disk.

writeDataInner :: Int

               -> String

               -> Options

               -> ClusterData

               -> ClusterData

               -> IO Bool

writeDataInner nlen name opts cdata fixdata = do

  let (ClusterData _ nl il _) = fixdata

  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") (serializeCluster cdata)

  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 input_data

         unless result $ exitWith $ ExitFailure 2

  results <- mapM (\name -> Rapi.loadData name >>= writeData nlen name opts)

             clusters

  unless (all id results) $ exitWith (ExitFailure 2)