code.grnet.gr Git - ganeti-local/blob - scripts/gnt-debug

   1 #!/usr/bin/python
   2 #
   3
   4 # Copyright (C) 2006, 2007 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
  22 # pylint: disable-msg=W0401,W0614
  23 # W0401: Wildcard import ganeti.cli
  24 # W0614: Unused import %s from wildcard import (since we need cli)
  25
  26 import sys
  27 import simplejson
  28 import time
  29
  30 from optparse import make_option
  31
  32 from ganeti.cli import *
  33 from ganeti import cli
  34 from ganeti import opcodes
  35 from ganeti import constants
  36 from ganeti import utils
  37 from ganeti import errors
  38
  39
  40 def Delay(opts, args):
  41   """Sleeps for a while
  42
  43   @param opts: the command line options selected by the user
  44   @type args: list
  45   @param args: should contain only one element, the duration
  46       the sleep
  47   @rtype: int
  48   @return: the desired exit code
  49
  50   """
  51   delay = float(args[0])
  52   op = opcodes.OpTestDelay(duration=delay,
  53                            on_master=opts.on_master,
  54                            on_nodes=opts.on_nodes)
  55   SubmitOpCode(op)
  56
  57   return 0
  58
  59
  60 def GenericOpCodes(opts, args):
  61   """Send any opcode to the master.
  62
  63   @todo: The function is broken and needs to be converted to the
  64       current job queue API
  65   @param opts: the command line options selected by the user
  66   @type args: list
  67   @param args: should contain only one element, the path of
  68       the file with the opcode definition
  69   @rtype: int
  70   @return: the desired exit code
  71
  72   """
  73   cl = cli.GetClient()
  74   job_data = []
  75   job_ids = []
  76   for fname in args:
  77     op_data = simplejson.loads(open(fname).read())
  78     op_list = [opcodes.OpCode.LoadOpCode(val) for val in op_data]
  79     job_data.append((fname, op_list))
  80   for fname, op_list in job_data:
  81     jid = cli.SendJob(op_list, cl=cl)
  82     ToStdout("File '%s', job id: %s", fname, jid)
  83     job_ids.append(jid)
  84   for jid in job_ids:
  85     ToStdout("Waiting for job id %s", jid)
  86     cli.PollJob(jid, cl=cl)
  87   return 0
  88
  89
  90 def TestAllocator(opts, args):
  91   """Runs the test allocator opcode.
  92
  93   @param opts: the command line options selected by the user
  94   @type args: list
  95   @param args: should contain only one element, the iallocator name
  96   @rtype: int
  97   @return: the desired exit code
  98
  99   """
 100   try:
 101     disks = [{"size": utils.ParseUnit(val), "mode": 'w'}
 102              for val in opts.disks.split(",")]
 103   except errors.UnitParseError, err:
 104     ToStderr("Invalid disks parameter '%s': %s", opts.disks, err)
 105     return 1
 106
 107   nics = [val.split("/") for val in opts.nics.split(",")]
 108   for row in nics:
 109     while len(row) < 3:
 110       row.append(None)
 111     for i in range(3):
 112       if row[i] == '':
 113         row[i] = None
 114   nic_dict = [{"mac": v[0], "ip": v[1], "bridge": v[2]} for v in nics]
 115
 116   if opts.tags is None:
 117     opts.tags = []
 118   else:
 119     opts.tags = opts.tags.split(",")
 120
 121   op = opcodes.OpTestAllocator(mode=opts.mode,
 122                                name=args[0],
 123                                mem_size=opts.mem,
 124                                disks=disks,
 125                                disk_template=opts.disk_template,
 126                                nics=nic_dict,
 127                                os=opts.os_type,
 128                                vcpus=opts.vcpus,
 129                                tags=opts.tags,
 130                                direction=opts.direction,
 131                                allocator=opts.allocator,
 132                                )
 133   result = SubmitOpCode(op)
 134   ToStdout("%s" % result)
 135   return 0
 136
 137
 138 commands = {
 139   'delay': (Delay, ARGS_ONE,
 140             [DEBUG_OPT,
 141              make_option("--no-master", dest="on_master", default=True,
 142                          action="store_false",
 143                          help="Do not sleep in the master code"),
 144              make_option("-n", dest="on_nodes", default=[],
 145                          action="append",
 146                          help="Select nodes to sleep on"),
 147              ],
 148             "[opts...] <duration>", "Executes a TestDelay OpCode"),
 149   'submit-job': (GenericOpCodes, ARGS_ATLEAST(1),
 150                  [DEBUG_OPT,
 151                   ],
 152                  "<op_list_file...>", "Submits jobs built from json files"
 153                  " containing a list of serialized opcodes"),
 154   'allocator': (TestAllocator, ARGS_ONE,
 155                 [DEBUG_OPT,
 156                  make_option("--dir", dest="direction",
 157                              default="in", choices=["in", "out"],
 158                              help="Show allocator input (in) or allocator"
 159                              " results (out)"),
 160                  make_option("--algorithm", dest="allocator",
 161                              default=None,
 162                              help="Allocator algorithm name"),
 163                  make_option("-m", "--mode", default="relocate",
 164                              choices=["relocate", "allocate"],
 165                              help="Request mode, either allocate or"
 166                              " relocate"),
 167                  cli_option("--mem", default=128, type="unit",
 168                             help="Memory size for the instance (MiB)"),
 169                  make_option("--disks", default="4096,4096",
 170                              help="Comma separated list of disk sizes (MiB)"),
 171                  make_option("-t", "--disk-template", default="drbd",
 172                              help="Select the disk template"),
 173                  make_option("--nics", default="00:11:22:33:44:55",
 174                              help="Comma separated list of nics, each nic"
 175                              " definition is of form mac/ip/bridge, if"
 176                              " missing values are replace by None"),
 177                  make_option("-o", "--os-type", default=None,
 178                              help="Select os for the instance"),
 179                  make_option("-p", "--vcpus", default=1, type="int",
 180                              help="Select number of VCPUs for the instance"),
 181                  make_option("--tags", default=None,
 182                              help="Comma separated list of tags"),
 183                  ],
 184                 "{opts...} <instance>", "Executes a TestAllocator OpCode"),
 185   }
 186
 187
 188 if __name__ == '__main__':
 189   sys.exit(GenericMain(commands))