import qualified Ganeti.HTools.CLI as CLI
import Ganeti.HTools.IAlloc
import Ganeti.HTools.Types
+import Ganeti.HTools.Loader (RqType(..), Request(..))
-- | Command line options structure.
data Options = Options
"show help"
]
--- | Try to allocate an instance on the cluster
-tryAlloc :: (Monad m) =>
- NodeList
- -> InstanceList
- -> Instance.Instance
- -> Int
- -> m [(Maybe NodeList, [Node.Node])]
-tryAlloc nl _ inst 2 =
- let all_nodes = Container.elems nl
- all_pairs = liftM2 (,) all_nodes all_nodes
- ok_pairs = filter (\(x, y) -> Node.idx x /= Node.idx y) all_pairs
- sols = map (\(p, s) ->
- (fst $ Cluster.allocateOnPair nl inst p s, [p, s]))
- ok_pairs
- in return sols
-
-tryAlloc nl _ inst 1 =
- let all_nodes = Container.elems nl
- sols = map (\p -> (fst $ Cluster.allocateOnSingle nl inst p, [p]))
- all_nodes
- in return sols
-
-tryAlloc _ _ _ reqn = fail $ "Unsupported number of alllocation \
- \destinations required (" ++ (show reqn) ++
- "), only two supported"
-
--- | Try to allocate an instance on the cluster
-tryReloc :: (Monad m) =>
- NodeList
- -> InstanceList
- -> Int
- -> Int
- -> [Int]
- -> m [(Maybe NodeList, [Node.Node])]
-tryReloc nl il xid 1 ex_idx =
- let all_nodes = Container.elems nl
- inst = Container.find xid il
- valid_nodes = filter (not . flip elem ex_idx . idx) all_nodes
- valid_idxes = map Node.idx valid_nodes
- nl' = Container.map (\n -> if elem (Node.idx n) ex_idx then
- Node.setOffline n True
- else n) nl
- sols1 = map (\x -> let (mnl, _, _, _) =
- Cluster.applyMove nl' inst
- (Cluster.ReplaceSecondary x)
- in (mnl, [Container.find x nl'])
- ) valid_idxes
- in return sols1
-
-tryReloc _ _ _ reqn _ = fail $ "Unsupported number of relocation \
- \destinations required (" ++ (show reqn) ++
- "), only one supported"
-
-filterFails :: (Monad m) => [(Maybe NodeList, [Node.Node])]
- -> m [(NodeList, [Node.Node])]
+
+filterFails :: (Monad m) => [(Maybe Node.List, [Node.Node])]
+ -> m [(Node.List, [Node.Node])]
filterFails sols =
if null sols then fail "No nodes onto which to allocate at all"
else let sols' = filter (isJust . fst) sols
else
return $ map (\(x, y) -> (fromJust x, y)) sols'
-processResults :: (Monad m) => [(NodeList, [Node.Node])]
+processResults :: (Monad m) => [(Node.List, [Node.Node])]
-> m (String, [Node.Node])
processResults sols =
let sols' = map (\(nl', ns) -> (Cluster.compCV nl', ns)) sols
let Request rqtype nl il csf = request
new_nodes = case rqtype of
- Allocate xi reqn -> tryAlloc nl il xi reqn
+ Allocate xi reqn -> Cluster.tryAlloc nl il xi reqn
Relocate idx reqn exnodes ->
- tryReloc nl il idx reqn exnodes
+ Cluster.tryReloc nl il idx reqn exnodes
let sols = new_nodes >>= filterFails >>= processResults
let (ok, info, rn) = case sols of
Ok (info, sn) -> (True, "Request successful: " ++ info,
- map ((++ csf) . name) sn)
+ map ((++ csf) . Node.name) sn)
Bad s -> (False, "Request failed: " ++ s, [])
resp = formatResponse ok info rn
putStrLn resp