Statistics
| Branch: | Tag: | Revision:

root / doc / examples / dumb-allocator @ 2a139bb0

History | View | Annotate | Download (2.9 kB)

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())