We consider a cluster dedicated, if all nodes in all allocatable
groups, or the desired group if specified, have the exclusive
storage flag set.
Signed-off-by: Klaus Aehlig <aeh...@google.com>
---
Makefile.am | 1 +
src/Ganeti/HTools/Dedicated.hs | 61 ++++++++++++++++++++++++++++++++++++++++++
2 files changed, 62 insertions(+)
create mode 100644 src/Ganeti/HTools/Dedicated.hs
diff --git a/Makefile.am b/Makefile.am
index 4744e15..4a929d8 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -890,6 +890,7 @@ HS_LIB_SRCS = \
src/Ganeti/HTools/CLI.hs \
src/Ganeti/HTools/Cluster.hs \
src/Ganeti/HTools/Container.hs \
+ src/Ganeti/HTools/Dedicated.hs \
src/Ganeti/HTools/ExtLoader.hs \
src/Ganeti/HTools/Graph.hs \
src/Ganeti/HTools/Group.hs \
diff --git a/src/Ganeti/HTools/Dedicated.hs b/src/Ganeti/HTools/Dedicated.hs
new file mode 100644
index 0000000..a718db7
--- /dev/null
+++ b/src/Ganeti/HTools/Dedicated.hs
@@ -0,0 +1,61 @@
+{-| Implementation of special handling of dedicated clusters.
+
+-}
+
+{-
+
+Copyright (C) 2014 Google Inc.
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are
+met:
+
+1. Redistributions of source code must retain the above copyright notice,
+this list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright
+notice, this list of conditions and the following disclaimer in the
+documentation and/or other materials provided with the distribution.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
+IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
+TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
+CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+-}
+
+module Ganeti.HTools.Dedicated
+ ( isDedicated
+ ) where
+
+import Control.Applicative (liftA2)
+import qualified Data.Foldable as F
+import qualified Data.IntMap as IntMap
+import qualified Data.IntSet as IntSet
+
+import qualified Ganeti.HTools.Group as Group
+import qualified Ganeti.HTools.Loader as Loader
+import qualified Ganeti.HTools.Node as Node
+import qualified Ganeti.HTools.Types as T
+
+-- | Given a cluster description and maybe a group name, decide
+-- if that group, or all allocatable groups if no group is given,
+-- is dedicated.
+isDedicated :: Loader.ClusterData -> Maybe String -> Bool
+isDedicated cdata maybeGroup =
+ let groups =
+ IntMap.keysSet
+ . IntMap.filter (maybe ((/=) T.AllocUnallocable . Group.allocPolicy)
+ (\name -> (==) name . Group.name) maybeGroup)
+ $ Loader.cdGroups cdata
+ in F.all (liftA2 (||) Node.exclStorage
+ $ not . (`IntSet.member` groups) . Node.group)
+ $ Loader.cdNodes cdata
--
2.2.0.rc0.207.ga3a616c
