4 # Copyright (C) 2006, 2007 Google Inc.
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.
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.
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
22 # pylint: disable-msg=W0401,W0614
23 # W0401: Wildcard import ganeti.cli
24 # W0614: Unused import %s from wildcard import (since we need cli)
30 from optparse import make_option
32 from ganeti.cli import *
33 from ganeti import opcodes
34 from ganeti import constants
35 from ganeti import utils
36 from ganeti import errors
39 def Delay(opts, args):
42 @param opts: the command line options selected by the user
44 @param args: should contain only one element, the duration
47 @return: the desired exit code
50 delay = float(args[0])
51 op = opcodes.OpTestDelay(duration=delay,
52 on_master=opts.on_master,
53 on_nodes=opts.on_nodes)
59 def GenericOpCodes(opts, args):
60 """Send any opcode to the master.
62 @todo: The function is broken and needs to be converted to the
64 @param opts: the command line options selected by the user
66 @param args: should contain only one element, the path of
67 the file with the opcode definition
69 @return: the desired exit code
73 op_data = simplejson.loads(open(fname).read())
74 op_list = [opcodes.OpCode.LoadOpCode(val) for val in op_data]
75 job = opcodes.Job(op_list=op_list)
77 ToStdout("Job id: %s", jid)
84 # wait for job to complete (either by success or failure)
86 jdata = SubmitQuery(query)
88 # job not found, gone away!
94 if status in (opcodes.Job.STATUS_SUCCESS, opcodes.Job.STATUS_FAIL):
97 # sleep between checks
100 # job has finished, get and process its results
101 query["fields"].extend(["op_list", "op_status", "op_result"])
102 jdata = SubmitQuery(query)
104 # job not found, gone away!
105 ToStderr("Job lost!")
108 status, op_list, op_status, op_result = jdata[0]
109 for idx, op in enumerate(op_list):
110 ToStdout("%s %s %s %s", idx, op.OP_ID, op_status[idx], op_result[idx])
114 def TestAllocator(opts, args):
115 """Runs the test allocator opcode.
117 @param opts: the command line options selected by the user
119 @param args: should contain only one element, the iallocator name
121 @return: the desired exit code
125 disks = [{"size": utils.ParseUnit(val), "mode": 'w'}
126 for val in opts.disks.split(",")]
127 except errors.UnitParseError, err:
128 ToStderr("Invalid disks parameter '%s': %s", opts.disks, err)
131 nics = [val.split("/") for val in opts.nics.split(",")]
138 nic_dict = [{"mac": v[0], "ip": v[1], "bridge": v[2]} for v in nics]
140 if opts.tags is None:
143 opts.tags = opts.tags.split(",")
145 op = opcodes.OpTestAllocator(mode=opts.mode,
149 disk_template=opts.disk_template,
154 direction=opts.direction,
155 allocator=opts.allocator,
157 result = SubmitOpCode(op)
158 ToStdout("%s" % result)
163 'delay': (Delay, ARGS_ONE,
165 make_option("--no-master", dest="on_master", default=True,
166 action="store_false",
167 help="Do not sleep in the master code"),
168 make_option("-n", dest="on_nodes", default=[],
170 help="Select nodes to sleep on"),
172 "[opts...] <duration>", "Executes a TestDelay OpCode"),
173 'submit-job': (GenericOpCodes, ARGS_ONE,
176 "<op_list_file>", "Submits a job built from a json-file"
177 " with a list of serialized opcodes"),
178 'allocator': (TestAllocator, ARGS_ONE,
180 make_option("--dir", dest="direction",
181 default="in", choices=["in", "out"],
182 help="Show allocator input (in) or allocator"
184 make_option("--algorithm", dest="allocator",
186 help="Allocator algorithm name"),
187 make_option("-m", "--mode", default="relocate",
188 choices=["relocate", "allocate"],
189 help="Request mode, either allocate or"
191 cli_option("--mem", default=128, type="unit",
192 help="Memory size for the instance (MiB)"),
193 make_option("--disks", default="4096,4096",
194 help="Comma separated list of disk sizes (MiB)"),
195 make_option("-t", "--disk-template", default="drbd",
196 help="Select the disk template"),
197 make_option("--nics", default="00:11:22:33:44:55",
198 help="Comma separated list of nics, each nic"
199 " definition is of form mac/ip/bridge, if"
200 " missing values are replace by None"),
201 make_option("-o", "--os-type", default=None,
202 help="Select os for the instance"),
203 make_option("-p", "--vcpus", default=1, type="int",
204 help="Select number of VCPUs for the instance"),
205 make_option("--tags", default=None,
206 help="Comma separated list of tags"),
208 "{opts...} <instance>", "Executes a TestAllocator OpCode"),
212 if __name__ == '__main__':
213 sys.exit(GenericMain(commands))