Synnefo » snf-ganeti

/htools/mon-collector

	htools/Ganeti/DataCollectors \

	$(APIDOC_HS_DIR)/Ganeti/DataCollectors \

HS_PROGS = htools/htools htools/mon-collector

	htools/Ganeti/DataCollectors/Drbd.hs \

	htools/Ganeti/DataCollectors/Program.hs \

	$(pkglib_python_scripts) \

	htools/mon-collector

	man/htools.1 \

	man/mon-collector.7

{-| DRBD data collector.

-}

{-

Copyright (C) 2012 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.DataCollectors.Drbd

  ( main

  , options

  , arguments

  ) where

import qualified Control.Exception as E

import Data.Attoparsec.Text.Lazy as A

import Data.Text.Lazy (pack, unpack)

import Text.JSON

import qualified Ganeti.BasicTypes as BT

import qualified Ganeti.Constants as C

import Ganeti.Block.Drbd.Parser(drbdStatusParser)

import Ganeti.Common

import Ganeti.HTools.CLI

import Ganeti.Utils

-- | The default path of the DRBD status file.

-- It is hardcoded because it is not likely to change.

defaultFile :: FilePath

defaultFile = C.drbdStatusFile

-- | The default setting for the maximum amount of not parsed character to

-- print in case of error.

-- It is set to use most of the screen estate on a standard 80x25 terminal.

-- TODO: add the possibility to set this with a command line parameter.

defaultCharNum :: Int

defaultCharNum = 80*20

options :: IO [OptType]

options = return []

-- | The list of arguments supported by the program.

arguments :: [ArgCompletion]

arguments = [ArgCompletion OptComplFile 0 (Just 1)]

-- * Command line options

-- | Main function.

main :: Options -> [String] -> IO ()

main _ args = do

  contents <-

    ((E.try . readFile $ getInputPath args) :: IO (Either IOError String)) >>=

      exitIfBad "Error reading from file" . either (BT.Bad . show) BT.Ok

  output <-

    case A.parse drbdStatusParser $ pack contents of

      A.Fail unparsedText contexts errorMessage -> exitErr $

        show (Prelude.take defaultCharNum $ unpack unparsedText) ++ "\n"

          ++ show contexts ++ "\n" ++ errorMessage

      A.Done _ drbdStatus -> return $ encode drbdStatus

  putStrLn output

  where getInputPath a =

          if null a

            then defaultFile

            else head a

{-| Small module holding program definitions for data collectors.

-}

{-

Copyright (C) 2012 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.DataCollectors.Program (personalities) where

import Ganeti.Common (ArgCompletion)

import Ganeti.HTools.CLI (OptType, Options)

import qualified Ganeti.DataCollectors.Drbd as Drbd

-- | Supported binaries.

personalities :: [(String,

                   (Options -> [String] -> IO (), IO [OptType],

                    [ArgCompletion]))]

personalities = [ ("drbd",   (Drbd.main, Drbd.options, Drbd.arguments))

                ]

{-| Main binary for all stand-alone data collectors

-}

{-

Copyright (C) 2012 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.Char (toLower)

import System.Environment

import System.IO

import Ganeti.Utils

import Ganeti.HTools.CLI (parseOpts, genericOpts)

import Ganeti.DataCollectors.Program (personalities)

-- | Display usage and exit.

usage :: String -> IO ()

usage name = do

  hPutStrLn stderr $ "Unrecognised personality '" ++ name ++ "'."

  hPutStrLn stderr "This program must be executed specifying one of the \

                    \following names as the first parameter:"

  mapM_ (hPutStrLn stderr . ("  - " ++) . fst) personalities

  exitErr "Please specify the desired role."

main :: IO ()

main = do

  cmd_args <- getArgs

  let binary =

        if null cmd_args

          then ""

          else head cmd_args

      name = map toLower binary

      boolnames = map (\(x, y) -> (x == name, Just y)) personalities

  case select Nothing boolnames of

    Nothing -> usage name

    Just (fn, options, arguments) -> do

         let actual_args = tail cmd_args

         real_options <- options

         (opts, args) <- parseOpts actual_args name (real_options ++

                           genericOpts) arguments

         fn opts args

(cluster information printer), **mon-collector**(7) (data collectors

interface).

mon-collector(7) Ganeti | Version @GANETI_VERSION@

==================================================

NAME

----

mon-collector - Command line interface for the data collectors of the

monitoring system

SYNOPSIS

--------

**mon-collector** {collector}

DESCRIPTION

-----------

``mon-collector`` is a suite of tools designed to provide a command line

interface to the data collectors implemented by the ganeti monitoring system.

``mon-collector`` is also the generic binary that must be invoked specifying,

as the first command line parameter, the name of the actual desired data

collector to be run.

When executed, ``mon-collector`` will run the specified collector and will

print its output to stdout, in JSON format.

COLLECTORS

----------

DRBD

~~~~

| drbd [*status-file*]

Collects the information about the version and status of the DRBD kernel

module, and of the disks it is managing.

If *status-file* is specified, the status will be read from that file.

Otherwise, the collector will read it from /proc/drbd.