code.grnet.gr Git - ganeti-local/blob - doc/examples/dumb-allocator

   1 #!/usr/bin/python
   2 #
   3
   4 # Copyright (C) 2008 Google Inc.
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful, but
  12 # WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 # General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program; if not, write to the Free Software
  18 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  19 # 02110-1301, USA.
  20
  21 """Simple first-fit allocator for ganeti instance allocation framework.
  22
  23 This allocator just iterates over the nodes and selects the first one
  24 that fits in both memory and disk space, without any consideration for
  25 equal spread or VCPU oversubscription.
  26
  27 """
  28 import simplejson
  29 import sys
  30
  31
  32 def SelectNode(nodes, request, to_skip):
  33   """Select a node for the given instance
  34
  35   """
  36   disk_size = request["disk_space_total"]
  37   selected = None
  38   for nname, ninfo in nodes.iteritems():
  39     if nname in to_skip:
  40       continue
  41     if request["memory"] > ninfo["free_memory"]:
  42       continue
  43     if disk_size > ninfo["free_disk"]:
  44       continue
  45     selected = nname
  46     break
  47   return selected
  48
  49
  50 def OutputError(text, exit_code=1):
  51   """Builds an error response with a given info message.
  52
  53   """
  54   error = {
  55     "success": False,
  56     "info": text,
  57     "nodes": [],
  58     }
  59   print simplejson.dumps(error, indent=2)
  60   return exit_code
  61
  62
  63 def main():
  64   """Main function.
  65
  66   """
  67   if len(sys.argv) < 2:
  68     print >> sys.stderr, "Usage: %s cluster.json" % (sys.argv[0])
  69     return 1
  70
  71   data = simplejson.load(open(sys.argv[1]))
  72
  73   nodes =  data["nodes"]
  74   request = data["request"]
  75   req_type = request["type"]
  76   if req_type == "allocate":
  77     forbidden_nodes = []
  78     inst_data = request
  79   elif req_type == "relocate":
  80     idict = data["instances"][request["name"]]
  81     forbidden_nodes = idict["nodes"]
  82     inst_data = idict
  83     inst_data["disk_space_total"] = request["disk_space_total"]
  84   else:
  85     return OutputError("Unsupported allocator mode '%s'" % req_type)
  86
  87   result_nodes = []
  88   while len(result_nodes) < request["required_nodes"]:
  89     new_selection = SelectNode(nodes, inst_data, result_nodes+forbidden_nodes)
  90     if new_selection is None:
  91       return OutputError("Can't find a suitable node for position %s"
  92                          " (already selected: %s)" %
  93                          (len(result_nodes) + 1, ", ".join(result_nodes)),
  94                          exit_code=0)
  95     result_nodes.append(new_selection)
  96
  97   result = {
  98           "success": True,
  99           "info": "Allocation successful",
 100           "nodes": result_nodes,
 101           }
 102   print simplejson.dumps(result, indent=2)
 103   return 0
 104
 105 if __name__ == "__main__":
 106     sys.exit(main())