Statistics
| Branch: | Tag: | Revision:

root / scripts / gnt-debug @ 30e4e741

History | View | Annotate | Download (6.6 kB)

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
"""Debugging commands"""
22

    
23
# pylint: disable-msg=W0401,W0614,C0103
24
# W0401: Wildcard import ganeti.cli
25
# W0614: Unused import %s from wildcard import (since we need cli)
26
# C0103: Invalid name gnt-backup
27

    
28
import sys
29
import simplejson
30
import time
31

    
32
from ganeti.cli import *
33
from ganeti import cli
34
from ganeti import opcodes
35
from ganeti import utils
36
from ganeti import errors
37

    
38

    
39
def Delay(opts, args):
40
  """Sleeps for a while
41

    
42
  @param opts: the command line options selected by the user
43
  @type args: list
44
  @param args: should contain only one element, the duration
45
      the sleep
46
  @rtype: int
47
  @return: the desired exit code
48

    
49
  """
50
  delay = float(args[0])
51
  op = opcodes.OpTestDelay(duration=delay,
52
                           on_master=opts.on_master,
53
                           on_nodes=opts.on_nodes)
54
  SubmitOpCode(op)
55

    
56
  return 0
57

    
58

    
59
def GenericOpCodes(opts, args):
60
  """Send any opcode to the master.
61

    
62
  @todo: The function is broken and needs to be converted to the
63
      current job queue API
64
  @param opts: the command line options selected by the user
65
  @type args: list
66
  @param args: should contain only one element, the path of
67
      the file with the opcode definition
68
  @rtype: int
69
  @return: the desired exit code
70

    
71
  """
72
  cl = cli.GetClient()
73
  jex = cli.JobExecutor(cl=cl, verbose=opts.verbose)
74

    
75
  job_cnt = 0
76
  op_cnt = 0
77
  if opts.timing_stats:
78
    ToStdout("Loading...")
79
  for job_idx in range(opts.rep_job):
80
    for fname in args:
81
      # pylint: disable-msg=W0142
82
      op_data = simplejson.loads(utils.ReadFile(fname))
83
      op_list = [opcodes.OpCode.LoadOpCode(val) for val in op_data]
84
      op_list = op_list * opts.rep_op
85
      jex.QueueJob("file %s/%d" % (fname, job_idx), *op_list)
86
      op_cnt += len(op_list)
87
      job_cnt += 1
88

    
89
  if opts.timing_stats:
90
    t1 = time.time()
91
    ToStdout("Submitting...")
92

    
93
  jex.SubmitPending()
94

    
95
  if opts.timing_stats:
96
    t2 = time.time()
97
    ToStdout("Executing...")
98

    
99
  jex.GetResults()
100
  if opts.timing_stats:
101
    t3 = time.time()
102
    ToStdout("C:op     %4d" % op_cnt)
103
    ToStdout("C:job    %4d" % job_cnt)
104
    ToStdout("T:submit %4.4f" % (t2-t1))
105
    ToStdout("T:exec   %4.4f" % (t3-t2))
106
    ToStdout("T:total  %4.4f" % (t3-t1))
107
  return 0
108

    
109

    
110
def TestAllocator(opts, args):
111
  """Runs the test allocator opcode.
112

    
113
  @param opts: the command line options selected by the user
114
  @type args: list
115
  @param args: should contain only one element, the iallocator name
116
  @rtype: int
117
  @return: the desired exit code
118

    
119
  """
120
  try:
121
    disks = [{"size": utils.ParseUnit(val), "mode": 'w'}
122
             for val in opts.disks.split(",")]
123
  except errors.UnitParseError, err:
124
    ToStderr("Invalid disks parameter '%s': %s", opts.disks, err)
125
    return 1
126

    
127
  nics = [val.split("/") for val in opts.nics.split(",")]
128
  for row in nics:
129
    while len(row) < 3:
130
      row.append(None)
131
    for i in range(3):
132
      if row[i] == '':
133
        row[i] = None
134
  nic_dict = [{"mac": v[0], "ip": v[1], "bridge": v[2]} for v in nics]
135

    
136
  if opts.tags is None:
137
    opts.tags = []
138
  else:
139
    opts.tags = opts.tags.split(",")
140

    
141
  op = opcodes.OpTestAllocator(mode=opts.mode,
142
                               name=args[0],
143
                               mem_size=opts.mem,
144
                               disks=disks,
145
                               disk_template=opts.disk_template,
146
                               nics=nic_dict,
147
                               os=opts.os,
148
                               vcpus=opts.vcpus,
149
                               tags=opts.tags,
150
                               direction=opts.direction,
151
                               allocator=opts.iallocator,
152
                               )
153
  result = SubmitOpCode(op)
154
  ToStdout("%s" % result)
155
  return 0
156

    
157

    
158
commands = {
159
  'delay': (
160
    Delay, [ArgUnknown(min=1, max=1)],
161
    [cli_option("--no-master", dest="on_master", default=True,
162
                action="store_false", help="Do not sleep in the master code"),
163
     cli_option("-n", dest="on_nodes", default=[],
164
                action="append", help="Select nodes to sleep on"),
165
     ],
166
    "[opts...] <duration>", "Executes a TestDelay OpCode"),
167
  'submit-job': (
168
    GenericOpCodes, [ArgFile(min=1)],
169
    [VERBOSE_OPT,
170
     cli_option("--op-repeat", type="int", default="1", dest="rep_op",
171
                help="Repeat the opcode sequence this number of times"),
172
     cli_option("--job-repeat", type="int", default="1", dest="rep_job",
173
                help="Repeat the job this number of times"),
174
     cli_option("--timing-stats", default=False,
175
                action="store_true", help="Show timing stats"),
176
     ],
177
    "<op_list_file...>", "Submits jobs built from json files"
178
    " containing a list of serialized opcodes"),
179
  'allocator': (
180
    TestAllocator, ARGS_ONE_INSTANCE,
181
    [cli_option("--dir", dest="direction",
182
                default="in", choices=["in", "out"],
183
                help="Show allocator input (in) or allocator"
184
                " results (out)"),
185
     IALLOCATOR_OPT,
186
     cli_option("-m", "--mode", default="relocate",
187
                choices=["relocate", "allocate"],
188
                help="Request mode, either allocate or relocate"),
189
     cli_option("--mem", default=128, type="unit",
190
                help="Memory size for the instance (MiB)"),
191
     cli_option("--disks", default="4096,4096",
192
                help="Comma separated list of disk sizes (MiB)"),
193
     DISK_TEMPLATE_OPT,
194
     cli_option("--nics", default="00:11:22:33:44:55",
195
                help="Comma separated list of nics, each nic"
196
                " definition is of form mac/ip/bridge, if"
197
                " missing values are replace by None"),
198
     OS_OPT,
199
     cli_option("-p", "--vcpus", default=1, type="int",
200
                help="Select number of VCPUs for the instance"),
201
     cli_option("--tags", default=None,
202
                help="Comma separated list of tags"),
203
     ],
204
    "{opts...} <instance>", "Executes a TestAllocator OpCode"),
205
  }
206

    
207

    
208
if __name__ == '__main__':
209
  sys.exit(GenericMain(commands))