Synnefo » snf-ganeti

    [ oPrintNodes

    , oPrintInsts

    , oPrintCommands

    , oDataFile

    , oEvacMode

    , oRapiMaster

             -> Int              -- ^ Max node name len

             -> Int              -- ^ Max instance name len

             -> [MoveJob]        -- ^ Current command list

             -> Score            -- ^ Score at which to stop

             -> Score            -- ^ Min gain limit

             -> Score            -- ^ Min score gain

             -> IO (Cluster.Table, [MoveJob]) -- ^ The resulting table

                                              -- and commands

iterateDepth ini_tbl max_rounds disk_moves inst_moves nmlen imlen

    let Cluster.Table ini_nl ini_il _ _ = ini_tbl

        allowed_next = Cluster.doNextBalance ini_tbl max_rounds min_score

        m_fin_tbl = if allowed_next

                                     nmlen imlen cur_plc fin_plc_len

                  afn = Cluster.involvedNodes ini_il cur_plc

                  upd_cmd_strs = (afn, idx, move, cmds):cmd_strs

              iterateDepth fin_tbl max_rounds disk_moves inst_moves

                           mg_limit min_gain evac_mode

        Nothing -> return (ini_tbl, cmd_strs)

-- | Displays the cluster stats.

printStats :: Node.List -> Node.List -> IO ()

printStats ini_nl fin_nl = do

            Just cdata -> return (Group.name grp, cdata)

-- | Do a few checks on the cluster data.

  -- nothing to do on an empty cluster

  when (Container.null il) $ do

         exitWith ExitSuccess

  -- hbal doesn't currently handle split clusters

    hPutStrLn stderr "Aborting."

    exitWith $ ExitFailure 1

             (Container.size nl)

  let csf = commonSuffix nl il

       printf "Note: Stripping common suffix of '%s' from names\n" csf

-- | Do a few checks on the selected group data.

             (Container.size nl)

  putStrLn $ "Selected node group: " ++ gname

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

             "Initial check done: %d bad nodes, %d bad instances.\n"

             (length bad_nodes) (length bad_instances)

checkNeedRebalance :: Options -> Score -> IO ()

checkNeedRebalance opts ini_cv = do

  let min_cv = optMinScore opts

  when (ini_cv < min_cv) $ do

         exitWith ExitSuccess

-- | Main function.

         hPutStrLn stderr "Error: this program doesn't take any arguments."

         exitWith $ ExitFailure 1

      shownodes = optShowNodes opts

      showinsts = optShowInsts opts

  ini_cdata@(ClusterData gl fixed_nl ilf ctags) <- loadExternalData opts

       putStrLn $ "Loaded cluster tags: " ++ intercalate "," ctags

  nlf <- setNodesStatus opts fixed_nl

  maybeSaveData (optSaveCluster opts) "original" "before balancing" ini_cdata

  (gname, (nl, il)) <- selectGroup opts gl nlf ilf

  maybePrintInsts showinsts "Initial" (Cluster.printInsts nl il)

  checkNeedRebalance opts ini_cv

  let imlen = maximum . map (length . Instance.alias) $ Container.elems il

      nmlen = maximum . map (length . Node.alias) $ Container.elems nl

  (fin_tbl, cmd_strs) <- iterateDepth ini_tbl (optMaxLength opts)

                         (optDiskMoves opts)

                         (optInstMoves opts)

                         (optMinGainLim opts) (optMinGain opts)

                         (optEvacMode opts)

  let (Cluster.Table fin_nl fin_il fin_cv fin_plc) = fin_tbl

                        printf "Cluster score improved from %.8f to %.8f\n"

                        ini_cv fin_cv ::String

         printf "Solution length=%d\n" (length ord_plc)

  let cmd_jobs = Cluster.splitJobs cmd_strs

  when (verbose > 3) $ printStats nl fin_nl

  eval <- maybeExecJobs opts ord_plc fin_nl il cmd_jobs

  unless eval (exitWith (ExitFailure 1))