1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
##
## Func cloud availability application.
## Given machine requirements, returns the best
## location in the cloud to run koan.
## Depends on the storage and virt modules.
##
## Copyright 2008, Red Hat, Inc
## Matt Hicks <mhicks@redhat.com>
## +AUTHORS
##
## This software may be freely redistributed under the terms of the GNU
## general public license.
##
## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
##
import os.path
import optparse
import sys
import random
import func.overlord.client as func_client
import func.utils as utils
class FindResources(object):
def __init__(self):
pass
def run(self,args):
p = optparse.OptionParser()
p.add_option("-v", "--verbose",
dest="verbose",
action="store_true",
help="provide extra output")
p.add_option("-s", "--server-spec",
dest="server_spec",
default="*",
help="run against specific servers, default: '*'")
p.add_option("-m", "--memory",
dest="memory",
default="512",
help="the memory requirements in megabytes, default: '512'")
p.add_option("-a", "--arch",
dest="arch",
default="i386",
help="the architecture requirements, default: 'i386'")
(options, args) = p.parse_args(args)
self.options = options
# convert the memory and storage to integers for later comparisons
memory = int(options.memory)
arch = options.arch
# see what hosts have enough RAM
avail_hosts = {}
host_freemem = func_client.Client(options.server_spec).virt.freemem()
for (host, freemem) in host_freemem.iteritems():
if utils.is_error(freemem):
print "-- connection refused: %s" % host
continue
# Take an additional 256M off the freemem to keep
# Domain-0 stable (shrinking it to 256M can cause
# it to crash under load)
if (freemem-256) >= memory:
avail_hosts[host] = {'memory': freemem}
# Default the dest_host to nothing
dest_host = None
# see what hosts have the right architecture
arch_hosts = {}
host_arch = func_client.Client(options.server_spec).command.run('uname -i')
for (host, output) in host_arch.iteritems():
if utils.is_error(output):
print "-- connection refused: %s" % host
continue
host_arch = output[1].rstrip()
# If the host_arch is 64 bit, allow 32 bit machines on it
if host_arch == arch or (host_arch == "x86_64" and arch == "i386"):
arch_hosts[host] = host
if len(avail_hosts) > 0:
# Find the host that is the closest memory match
# and matching architecture
for (host, attrs) in avail_hosts.iteritems():
if arch_hosts.has_key(host):
if not dest_host:
dest_host = [host, attrs['memory']]
else:
if attrs['memory'] < dest_host[1]:
# Use the better match
dest_host = [host, attrs['memory']]
return dest_host
if __name__ == "__main__":
inv = FindResources()
inv.run(sys.argv)
|
