Synnefo » snf-ganeti » ganeti-local

-- | Machine-readable keys to show depending on given level.

keysData :: Level -> [String]

keysData GroupLvl = map fst groupData

keysData ClusterLvl = map fst clusterData

-- | Description to show depending on given level.

descrData :: Level -> [String]

descrData GroupLvl = map snd groupData

descrData ClusterLvl = map snd clusterData

-- | Human readable prefix for statistics.

phaseLevelDescr :: Phase -> Level -> Maybe String -> String

phaseLevelDescr phase GroupLvl (Just name) =

    printf "Statistics for group %s %s\n" name $ phaseDescr phase

phaseLevelDescr phase GroupLvl Nothing =

    printf "Statistics for group %s\n" $ phaseDescr phase

phaseLevelDescr phase ClusterLvl _ =

    printf "Cluster statistics %s\n" $ phaseDescr phase

-- | Prepare a single key given a certain level and phase of simulation.

prepareKey :: Level -> Phase -> Maybe String -> String -> String

prepareKey level phase Nothing suffix =

  printf "%s_%s_%s" (phasePrefix phase) (levelPrefix level) suffix

prepareKey level phase (Just idx) suffix =

  printf "%s_%s_%s_%s" (phasePrefix phase) (levelPrefix level) idx suffix

-- | Print all the statistics for given level and phase.

printStats :: Int            -- ^ Verbosity level

           -> Bool           -- ^ If the output should be machine readable

           -> Level          -- ^ Level on which we are printing

           -> Phase          -- ^ Current phase of simulation

           -> [String]       -- ^ Values to print

           -> Maybe String   -- ^ Additional data for groups

           -> IO ()

printStats _ True level phase values gidx = do

  let keys = map (prepareKey level phase gidx) (keysData level)

  printKeysHTC $ zip keys values

printStats verbose False level phase values name = do

  let prefix = phaseLevelDescr phase level name

      descr = descrData level

    printf "\n%s" prefix :: IO ()

    mapM_ (\(a,b) -> printf "    %s: %s\n" a b) (zip descr values)

-- | Extract name or idx from group.

extractGroupData :: Bool -> Group.Group -> String

extractGroupData True grp = printf "%d" $ Group.idx grp

extractGroupData False grp = Group.name grp

-- | Prepare values for group.

prepareGroupValues :: [Int] -> Double -> [String]

prepareGroupValues stats score =

  map (printf "%d") stats ++ [printf "%.8f" score]

-- | Prepare values for cluster.

prepareClusterValues :: Bool -> [Int] -> [Bool] -> [String]

prepareClusterValues machineread stats bstats =

  map (printf "%d")  stats ++ map (printBool machineread) bstats

-- | Print all the statistics on a group level.

printGroupStats :: Int -> Bool -> Phase -> Group.Group -> [Int] -> Double

                -> IO ()

printGroupStats verbose machineread phase grp stats score = do

  let values = prepareGroupValues stats score

      extradata = extractGroupData machineread grp

  printStats verbose machineread GroupLvl phase values (Just extradata)

printClusterStats verbose machineread phase stats needhbal canhbal = do

  let values = prepareClusterValues machineread stats [needhbal, canhbal]

  printStats verbose machineread ClusterLvl phase values Nothing