Synnefo » snf-ganeti

{-| Implementation of the LUXI loader.

-}

{-

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

    (

      loadData

    ) where

import qualified Control.Exception as E

import Text.JSON.Types

import qualified Ganeti.Luxi as L

import Ganeti.HTools.Loader

import Ganeti.HTools.Types

import qualified Ganeti.HTools.Node as Node

import qualified Ganeti.HTools.Instance as Instance

import Ganeti.HTools.Utils (fromJVal, annotateResult, tryFromObj, asJSObject)

-- * Utility functions

-- | Ensure a given JSValue is actually a JSArray.

toArray :: (Monad m) => JSValue -> m [JSValue]

toArray v =

    case v of

      JSArray arr -> return arr

      o -> fail ("Invalid input, expected array but got " ++ show o)

-- * Data querying functionality

-- | The input data for node query.

queryNodesMsg :: JSValue

queryNodesMsg =

    let nnames = JSArray []

        fnames = ["name",

                  "mtotal", "mnode", "mfree",

                  "dtotal", "dfree",

                  "ctotal",

                  "offline", "drained"]

        fields = JSArray $ map (JSString . toJSString) fnames

        use_locking = JSBool False

    in JSArray [nnames, fields, use_locking]

-- | The input data for instance query.

queryInstancesMsg :: JSValue

queryInstancesMsg =

    let nnames = JSArray []

        fnames = ["name",

                  "disk_usage", "be/memory", "be/vcpus",

                  "status", "pnode", "snodes", "tags", "oper_ram"]

        fields = JSArray $ map (JSString . toJSString) fnames

        use_locking = JSBool False

    in JSArray [nnames, fields, use_locking]

-- | The input data for cluster query

queryClusterInfoMsg :: JSValue

queryClusterInfoMsg = JSArray []

-- | Wraper over callMethod doing node query.

queryNodes :: L.Client -> IO (Result JSValue)

queryNodes = L.callMethod L.QueryNodes queryNodesMsg

-- | Wraper over callMethod doing instance query.

queryInstances :: L.Client -> IO (Result JSValue)

queryInstances = L.callMethod L.QueryInstances queryInstancesMsg

queryClusterInfo :: L.Client -> IO (Result JSValue)

queryClusterInfo = L.callMethod L.QueryClusterInfo queryClusterInfoMsg

-- | Parse a instance list in JSON format.

getInstances :: NameAssoc

             -> JSValue

             -> Result [(String, Instance.Instance)]

getInstances ktn arr = toArray arr >>= mapM (parseInstance ktn)

-- | Construct an instance from a JSON object.

parseInstance :: [(String, Ndx)]

              -> JSValue

              -> Result (String, Instance.Instance)

parseInstance ktn (JSArray [ name, disk, mem, vcpus

                           , status, pnode, snodes, tags, oram ]) = do

  xname <- annotateResult "Parsing new instance" (fromJVal name)

  let convert v = annotateResult ("Instance '" ++ xname ++ "'") (fromJVal v)

  xdisk <- convert disk

  xmem <- (case oram of

             JSRational _ _ -> convert oram

             _ -> convert mem)

  xvcpus <- convert vcpus

  xpnode <- convert pnode >>= lookupNode ktn xname

  xsnodes <- convert snodes::Result [JSString]

  snode <- (if null xsnodes then return Node.noSecondary

            else lookupNode ktn xname (fromJSString $ head xsnodes))

  xrunning <- convert status

  xtags <- convert tags

  let inst = Instance.create xname xmem xdisk xvcpus

             xrunning xtags xpnode snode

  return (xname, inst)

parseInstance _ v = fail ("Invalid instance query result: " ++ show v)

-- | Parse a node list in JSON format.

getNodes :: JSValue -> Result [(String, Node.Node)]

getNodes arr = toArray arr >>= mapM parseNode

-- | Construct a node from a JSON object.

parseNode :: JSValue -> Result (String, Node.Node)

parseNode (JSArray [ name, mtotal, mnode, mfree, dtotal, dfree

                   , ctotal, offline, drained ])

    = do

  xname <- annotateResult "Parsing new node" (fromJVal name)

  let convert v = annotateResult ("Node '" ++ xname ++ "'") (fromJVal v)

  xoffline <- convert offline

  xdrained <- convert drained

  node <- (if xoffline || xdrained

           then return $ Node.create xname 0 0 0 0 0 0 True

           else do

             xmtotal  <- convert mtotal

             xmnode   <- convert mnode

             xmfree   <- convert mfree

             xdtotal  <- convert dtotal

             xdfree   <- convert dfree

             xctotal  <- convert ctotal

             return $ Node.create xname xmtotal xmnode xmfree

                    xdtotal xdfree xctotal False)

  return (xname, node)

parseNode v = fail ("Invalid node query result: " ++ show v)

getClusterTags :: JSValue -> Result [String]

getClusterTags v = do

  let errmsg = "Parsing cluster info"

  obj <- annotateResult errmsg $ asJSObject v

  tryFromObj errmsg (fromJSObject obj) "tags"

-- * Main loader functionality

-- | Builds the cluster data from an URL.

loadData :: String -- ^ Unix socket to use as source

         -> IO (Result (Node.AssocList, Instance.AssocList, [String]))

loadData master =

  E.bracket

       (L.getClient master)

       L.closeClient

       (\s -> do

          nodes <- queryNodes s

          instances <- queryInstances s

          cinfo <- queryClusterInfo s

          return $ do -- Result monad

            node_data <- nodes >>= getNodes

            let (node_names, node_idx) = assignIndices node_data

            inst_data <- instances >>= getInstances node_names

            let (_, inst_idx) = assignIndices inst_data

            ctags <- cinfo >>= getClusterTags

            return (node_idx, inst_idx, ctags)

       )