Statistics
| Branch: | Tag: | Revision:

root / lib / opcodes.py @ 4a96f1d1

History | View | Annotate | Download (20.3 kB)

1
#
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
"""OpCodes module
23

24
This module implements the data structures which define the cluster
25
operations - the so-called opcodes.
26

27
Every operation which modifies the cluster state is expressed via
28
opcodes.
29

30
"""
31

    
32
# this are practically structures, so disable the message about too
33
# few public methods:
34
# pylint: disable-msg=R0903
35

    
36

    
37
class BaseOpCode(object):
38
  """A simple serializable object.
39

40
  This object serves as a parent class for OpCode without any custom
41
  field handling.
42

43
  """
44
  __slots__ = []
45

    
46
  def __init__(self, **kwargs):
47
    """Constructor for BaseOpCode.
48

49
    The constructor takes only keyword arguments and will set
50
    attributes on this object based on the passed arguments. As such,
51
    it means that you should not pass arguments which are not in the
52
    __slots__ attribute for this class.
53

54
    """
55
    slots = self._all_slots()
56
    for key in kwargs:
57
      if key not in slots:
58
        raise TypeError("Object %s doesn't support the parameter '%s'" %
59
                        (self.__class__.__name__, key))
60
      setattr(self, key, kwargs[key])
61

    
62
  def __getstate__(self):
63
    """Generic serializer.
64

65
    This method just returns the contents of the instance as a
66
    dictionary.
67

68
    @rtype:  C{dict}
69
    @return: the instance attributes and their values
70

71
    """
72
    state = {}
73
    for name in self._all_slots():
74
      if hasattr(self, name):
75
        state[name] = getattr(self, name)
76
    return state
77

    
78
  def __setstate__(self, state):
79
    """Generic unserializer.
80

81
    This method just restores from the serialized state the attributes
82
    of the current instance.
83

84
    @param state: the serialized opcode data
85
    @type state:  C{dict}
86

87
    """
88
    if not isinstance(state, dict):
89
      raise ValueError("Invalid data to __setstate__: expected dict, got %s" %
90
                       type(state))
91

    
92
    for name in self._all_slots():
93
      if name not in state and hasattr(self, name):
94
        delattr(self, name)
95

    
96
    for name in state:
97
      setattr(self, name, state[name])
98

    
99
  @classmethod
100
  def _all_slots(cls):
101
    """Compute the list of all declared slots for a class.
102

103
    """
104
    slots = []
105
    for parent in cls.__mro__:
106
      slots.extend(getattr(parent, "__slots__", []))
107
    return slots
108

    
109

    
110
class OpCode(BaseOpCode):
111
  """Abstract OpCode.
112

113
  This is the root of the actual OpCode hierarchy. All clases derived
114
  from this class should override OP_ID.
115

116
  @cvar OP_ID: The ID of this opcode. This should be unique amongst all
117
               children of this class.
118
  @ivar dry_run: Whether the LU should be run in dry-run mode, i.e. just
119
                 the check steps
120

121
  """
122
  OP_ID = "OP_ABSTRACT"
123
  __slots__ = ["dry_run", "debug_level"]
124

    
125
  def __getstate__(self):
126
    """Specialized getstate for opcodes.
127

128
    This method adds to the state dictionary the OP_ID of the class,
129
    so that on unload we can identify the correct class for
130
    instantiating the opcode.
131

132
    @rtype:   C{dict}
133
    @return:  the state as a dictionary
134

135
    """
136
    data = BaseOpCode.__getstate__(self)
137
    data["OP_ID"] = self.OP_ID
138
    return data
139

    
140
  @classmethod
141
  def LoadOpCode(cls, data):
142
    """Generic load opcode method.
143

144
    The method identifies the correct opcode class from the dict-form
145
    by looking for a OP_ID key, if this is not found, or its value is
146
    not available in this module as a child of this class, we fail.
147

148
    @type data:  C{dict}
149
    @param data: the serialized opcode
150

151
    """
152
    if not isinstance(data, dict):
153
      raise ValueError("Invalid data to LoadOpCode (%s)" % type(data))
154
    if "OP_ID" not in data:
155
      raise ValueError("Invalid data to LoadOpcode, missing OP_ID")
156
    op_id = data["OP_ID"]
157
    op_class = None
158
    if op_id in OP_MAPPING:
159
      op_class = OP_MAPPING[op_id]
160
    else:
161
      raise ValueError("Invalid data to LoadOpCode: OP_ID %s unsupported" %
162
                       op_id)
163
    op = op_class()
164
    new_data = data.copy()
165
    del new_data["OP_ID"]
166
    op.__setstate__(new_data)
167
    return op
168

    
169
  def Summary(self):
170
    """Generates a summary description of this opcode.
171

172
    """
173
    # all OP_ID start with OP_, we remove that
174
    txt = self.OP_ID[3:]
175
    field_name = getattr(self, "OP_DSC_FIELD", None)
176
    if field_name:
177
      field_value = getattr(self, field_name, None)
178
      txt = "%s(%s)" % (txt, field_value)
179
    return txt
180

    
181

    
182
# cluster opcodes
183

    
184
class OpPostInitCluster(OpCode):
185
  """Post cluster initialization.
186

187
  This opcode does not touch the cluster at all. Its purpose is to run hooks
188
  after the cluster has been initialized.
189

190
  """
191
  OP_ID = "OP_CLUSTER_POST_INIT"
192
  __slots__ = []
193

    
194

    
195
class OpDestroyCluster(OpCode):
196
  """Destroy the cluster.
197

198
  This opcode has no other parameters. All the state is irreversibly
199
  lost after the execution of this opcode.
200

201
  """
202
  OP_ID = "OP_CLUSTER_DESTROY"
203
  __slots__ = []
204

    
205

    
206
class OpQueryClusterInfo(OpCode):
207
  """Query cluster information."""
208
  OP_ID = "OP_CLUSTER_QUERY"
209
  __slots__ = []
210

    
211

    
212
class OpVerifyCluster(OpCode):
213
  """Verify the cluster state.
214

215
  @type skip_checks: C{list}
216
  @ivar skip_checks: steps to be skipped from the verify process; this
217
                     needs to be a subset of
218
                     L{constants.VERIFY_OPTIONAL_CHECKS}; currently
219
                     only L{constants.VERIFY_NPLUSONE_MEM} can be passed
220

221
  """
222
  OP_ID = "OP_CLUSTER_VERIFY"
223
  __slots__ = ["skip_checks", "verbose", "error_codes",
224
               "debug_simulate_errors"]
225

    
226

    
227
class OpVerifyDisks(OpCode):
228
  """Verify the cluster disks.
229

230
  Parameters: none
231

232
  Result: a tuple of four elements:
233
    - list of node names with bad data returned (unreachable, etc.)
234
    - dict of node names with broken volume groups (values: error msg)
235
    - list of instances with degraded disks (that should be activated)
236
    - dict of instances with missing logical volumes (values: (node, vol)
237
      pairs with details about the missing volumes)
238

239
  In normal operation, all lists should be empty. A non-empty instance
240
  list (3rd element of the result) is still ok (errors were fixed) but
241
  non-empty node list means some node is down, and probably there are
242
  unfixable drbd errors.
243

244
  Note that only instances that are drbd-based are taken into
245
  consideration. This might need to be revisited in the future.
246

247
  """
248
  OP_ID = "OP_CLUSTER_VERIFY_DISKS"
249
  __slots__ = []
250

    
251

    
252
class OpRepairDiskSizes(OpCode):
253
  """Verify the disk sizes of the instances and fixes configuration
254
  mimatches.
255

256
  Parameters: optional instances list, in case we want to restrict the
257
  checks to only a subset of the instances.
258

259
  Result: a list of tuples, (instance, disk, new-size) for changed
260
  configurations.
261

262
  In normal operation, the list should be empty.
263

264
  @type instances: list
265
  @ivar instances: the list of instances to check, or empty for all instances
266

267
  """
268
  OP_ID = "OP_CLUSTER_REPAIR_DISK_SIZES"
269
  __slots__ = ["instances"]
270

    
271

    
272
class OpQueryConfigValues(OpCode):
273
  """Query cluster configuration values."""
274
  OP_ID = "OP_CLUSTER_CONFIG_QUERY"
275
  __slots__ = ["output_fields"]
276

    
277

    
278
class OpRenameCluster(OpCode):
279
  """Rename the cluster.
280

281
  @type name: C{str}
282
  @ivar name: The new name of the cluster. The name and/or the master IP
283
              address will be changed to match the new name and its IP
284
              address.
285

286
  """
287
  OP_ID = "OP_CLUSTER_RENAME"
288
  OP_DSC_FIELD = "name"
289
  __slots__ = ["name"]
290

    
291

    
292
class OpSetClusterParams(OpCode):
293
  """Change the parameters of the cluster.
294

295
  @type vg_name: C{str} or C{None}
296
  @ivar vg_name: The new volume group name or None to disable LVM usage.
297

298
  """
299
  OP_ID = "OP_CLUSTER_SET_PARAMS"
300
  __slots__ = [
301
    "vg_name",
302
    "enabled_hypervisors",
303
    "hvparams",
304
    "os_hvp",
305
    "beparams",
306
    "nicparams",
307
    "candidate_pool_size",
308
    "maintain_node_health",
309
    "uid_pool",
310
    "add_uids",
311
    "remove_uids",
312
    ]
313

    
314

    
315
class OpRedistributeConfig(OpCode):
316
  """Force a full push of the cluster configuration.
317

318
  """
319
  OP_ID = "OP_CLUSTER_REDIST_CONF"
320
  __slots__ = []
321

    
322
# node opcodes
323

    
324
class OpRemoveNode(OpCode):
325
  """Remove a node.
326

327
  @type node_name: C{str}
328
  @ivar node_name: The name of the node to remove. If the node still has
329
                   instances on it, the operation will fail.
330

331
  """
332
  OP_ID = "OP_NODE_REMOVE"
333
  OP_DSC_FIELD = "node_name"
334
  __slots__ = ["node_name"]
335

    
336

    
337
class OpAddNode(OpCode):
338
  """Add a node to the cluster.
339

340
  @type node_name: C{str}
341
  @ivar node_name: The name of the node to add. This can be a short name,
342
                   but it will be expanded to the FQDN.
343
  @type primary_ip: IP address
344
  @ivar primary_ip: The primary IP of the node. This will be ignored when the
345
                    opcode is submitted, but will be filled during the node
346
                    add (so it will be visible in the job query).
347
  @type secondary_ip: IP address
348
  @ivar secondary_ip: The secondary IP of the node. This needs to be passed
349
                      if the cluster has been initialized in 'dual-network'
350
                      mode, otherwise it must not be given.
351
  @type readd: C{bool}
352
  @ivar readd: Whether to re-add an existing node to the cluster. If
353
               this is not passed, then the operation will abort if the node
354
               name is already in the cluster; use this parameter to 'repair'
355
               a node that had its configuration broken, or was reinstalled
356
               without removal from the cluster.
357

358
  """
359
  OP_ID = "OP_NODE_ADD"
360
  OP_DSC_FIELD = "node_name"
361
  __slots__ = ["node_name", "primary_ip", "secondary_ip", "readd"]
362

    
363

    
364
class OpQueryNodes(OpCode):
365
  """Compute the list of nodes."""
366
  OP_ID = "OP_NODE_QUERY"
367
  __slots__ = ["output_fields", "names", "use_locking"]
368

    
369

    
370
class OpQueryNodeVolumes(OpCode):
371
  """Get list of volumes on node."""
372
  OP_ID = "OP_NODE_QUERYVOLS"
373
  __slots__ = ["nodes", "output_fields"]
374

    
375

    
376
class OpQueryNodeStorage(OpCode):
377
  """Get information on storage for node(s)."""
378
  OP_ID = "OP_NODE_QUERY_STORAGE"
379
  __slots__ = [
380
    "nodes",
381
    "storage_type",
382
    "name",
383
    "output_fields",
384
    ]
385

    
386

    
387
class OpModifyNodeStorage(OpCode):
388
  """Modifies the properies of a storage unit"""
389
  OP_ID = "OP_NODE_MODIFY_STORAGE"
390
  __slots__ = [
391
    "node_name",
392
    "storage_type",
393
    "name",
394
    "changes",
395
    ]
396

    
397

    
398
class OpRepairNodeStorage(OpCode):
399
  """Repairs the volume group on a node."""
400
  OP_ID = "OP_REPAIR_NODE_STORAGE"
401
  OP_DSC_FIELD = "node_name"
402
  __slots__ = [
403
    "node_name",
404
    "storage_type",
405
    "name",
406
    "ignore_consistency",
407
    ]
408

    
409

    
410
class OpSetNodeParams(OpCode):
411
  """Change the parameters of a node."""
412
  OP_ID = "OP_NODE_SET_PARAMS"
413
  OP_DSC_FIELD = "node_name"
414
  __slots__ = [
415
    "node_name",
416
    "force",
417
    "master_candidate",
418
    "offline",
419
    "drained",
420
    "auto_promote",
421
    ]
422

    
423

    
424
class OpPowercycleNode(OpCode):
425
  """Tries to powercycle a node."""
426
  OP_ID = "OP_NODE_POWERCYCLE"
427
  OP_DSC_FIELD = "node_name"
428
  __slots__ = [
429
    "node_name",
430
    "force",
431
    ]
432

    
433

    
434
class OpEvacuateNode(OpCode):
435
  """Relocate secondary instances from a node."""
436
  OP_ID = "OP_NODE_EVACUATE"
437
  OP_DSC_FIELD = "node_name"
438
  __slots__ = [
439
    "node_name", "remote_node", "iallocator", "early_release",
440
    ]
441

    
442

    
443
class OpMigrateNode(OpCode):
444
  """Migrate all instances from a node."""
445
  OP_ID = "OP_NODE_MIGRATE"
446
  OP_DSC_FIELD = "node_name"
447
  __slots__ = [
448
    "node_name",
449
    "live",
450
    ]
451

    
452

    
453
class OpNodeEvacuationStrategy(OpCode):
454
  """Compute the evacuation strategy for a list of nodes."""
455
  OP_ID = "OP_NODE_EVAC_STRATEGY"
456
  OP_DSC_FIELD = "nodes"
457
  __slots__ = ["nodes", "iallocator", "remote_node"]
458

    
459

    
460
# instance opcodes
461

    
462
class OpCreateInstance(OpCode):
463
  """Create an instance."""
464
  OP_ID = "OP_INSTANCE_CREATE"
465
  OP_DSC_FIELD = "instance_name"
466
  __slots__ = [
467
    "instance_name",
468
    "os_type", "force_variant", "no_install",
469
    "pnode", "disk_template", "snode", "mode",
470
    "disks", "nics",
471
    "src_node", "src_path", "start", "identify_defaults",
472
    "wait_for_sync", "ip_check", "name_check",
473
    "file_storage_dir", "file_driver",
474
    "iallocator",
475
    "hypervisor", "hvparams", "beparams",
476
    "dry_run",
477
    ]
478

    
479

    
480
class OpReinstallInstance(OpCode):
481
  """Reinstall an instance's OS."""
482
  OP_ID = "OP_INSTANCE_REINSTALL"
483
  OP_DSC_FIELD = "instance_name"
484
  __slots__ = ["instance_name", "os_type", "force_variant"]
485

    
486

    
487
class OpRemoveInstance(OpCode):
488
  """Remove an instance."""
489
  OP_ID = "OP_INSTANCE_REMOVE"
490
  OP_DSC_FIELD = "instance_name"
491
  __slots__ = [
492
    "instance_name",
493
    "ignore_failures",
494
    "shutdown_timeout",
495
    ]
496

    
497

    
498
class OpRenameInstance(OpCode):
499
  """Rename an instance."""
500
  OP_ID = "OP_INSTANCE_RENAME"
501
  __slots__ = [
502
    "instance_name", "ignore_ip", "new_name",
503
    ]
504

    
505

    
506
class OpStartupInstance(OpCode):
507
  """Startup an instance."""
508
  OP_ID = "OP_INSTANCE_STARTUP"
509
  OP_DSC_FIELD = "instance_name"
510
  __slots__ = [
511
    "instance_name", "force", "hvparams", "beparams",
512
    ]
513

    
514

    
515
class OpShutdownInstance(OpCode):
516
  """Shutdown an instance."""
517
  OP_ID = "OP_INSTANCE_SHUTDOWN"
518
  OP_DSC_FIELD = "instance_name"
519
  __slots__ = ["instance_name", "timeout"]
520

    
521

    
522
class OpRebootInstance(OpCode):
523
  """Reboot an instance."""
524
  OP_ID = "OP_INSTANCE_REBOOT"
525
  OP_DSC_FIELD = "instance_name"
526
  __slots__ = [
527
    "instance_name", "reboot_type", "ignore_secondaries", "shutdown_timeout",
528
    ]
529

    
530

    
531
class OpReplaceDisks(OpCode):
532
  """Replace the disks of an instance."""
533
  OP_ID = "OP_INSTANCE_REPLACE_DISKS"
534
  OP_DSC_FIELD = "instance_name"
535
  __slots__ = [
536
    "instance_name", "remote_node", "mode", "disks", "iallocator",
537
    "early_release",
538
    ]
539

    
540

    
541
class OpFailoverInstance(OpCode):
542
  """Failover an instance."""
543
  OP_ID = "OP_INSTANCE_FAILOVER"
544
  OP_DSC_FIELD = "instance_name"
545
  __slots__ = [
546
    "instance_name", "ignore_consistency", "shutdown_timeout",
547
    ]
548

    
549

    
550
class OpMigrateInstance(OpCode):
551
  """Migrate an instance.
552

553
  This migrates (without shutting down an instance) to its secondary
554
  node.
555

556
  @ivar instance_name: the name of the instance
557

558
  """
559
  OP_ID = "OP_INSTANCE_MIGRATE"
560
  OP_DSC_FIELD = "instance_name"
561
  __slots__ = ["instance_name", "live", "cleanup"]
562

    
563

    
564
class OpMoveInstance(OpCode):
565
  """Move an instance.
566

567
  This move (with shutting down an instance and data copying) to an
568
  arbitrary node.
569

570
  @ivar instance_name: the name of the instance
571
  @ivar target_node: the destination node
572

573
  """
574
  OP_ID = "OP_INSTANCE_MOVE"
575
  OP_DSC_FIELD = "instance_name"
576
  __slots__ = [
577
    "instance_name", "target_node", "shutdown_timeout",
578
    ]
579

    
580

    
581
class OpConnectConsole(OpCode):
582
  """Connect to an instance's console."""
583
  OP_ID = "OP_INSTANCE_CONSOLE"
584
  OP_DSC_FIELD = "instance_name"
585
  __slots__ = ["instance_name"]
586

    
587

    
588
class OpActivateInstanceDisks(OpCode):
589
  """Activate an instance's disks."""
590
  OP_ID = "OP_INSTANCE_ACTIVATE_DISKS"
591
  OP_DSC_FIELD = "instance_name"
592
  __slots__ = ["instance_name", "ignore_size"]
593

    
594

    
595
class OpDeactivateInstanceDisks(OpCode):
596
  """Deactivate an instance's disks."""
597
  OP_ID = "OP_INSTANCE_DEACTIVATE_DISKS"
598
  OP_DSC_FIELD = "instance_name"
599
  __slots__ = ["instance_name"]
600

    
601

    
602
class OpRecreateInstanceDisks(OpCode):
603
  """Deactivate an instance's disks."""
604
  OP_ID = "OP_INSTANCE_RECREATE_DISKS"
605
  OP_DSC_FIELD = "instance_name"
606
  __slots__ = ["instance_name", "disks"]
607

    
608

    
609
class OpQueryInstances(OpCode):
610
  """Compute the list of instances."""
611
  OP_ID = "OP_INSTANCE_QUERY"
612
  __slots__ = ["output_fields", "names", "use_locking"]
613

    
614

    
615
class OpQueryInstanceData(OpCode):
616
  """Compute the run-time status of instances."""
617
  OP_ID = "OP_INSTANCE_QUERY_DATA"
618
  __slots__ = ["instances", "static"]
619

    
620

    
621
class OpSetInstanceParams(OpCode):
622
  """Change the parameters of an instance."""
623
  OP_ID = "OP_INSTANCE_SET_PARAMS"
624
  OP_DSC_FIELD = "instance_name"
625
  __slots__ = [
626
    "instance_name",
627
    "hvparams", "beparams", "force",
628
    "nics", "disks", "disk_template",
629
    "remote_node", "os_name", "force_variant",
630
    ]
631

    
632

    
633
class OpGrowDisk(OpCode):
634
  """Grow a disk of an instance."""
635
  OP_ID = "OP_INSTANCE_GROW_DISK"
636
  OP_DSC_FIELD = "instance_name"
637
  __slots__ = [
638
    "instance_name", "disk", "amount", "wait_for_sync",
639
    ]
640

    
641

    
642
# OS opcodes
643
class OpDiagnoseOS(OpCode):
644
  """Compute the list of guest operating systems."""
645
  OP_ID = "OP_OS_DIAGNOSE"
646
  __slots__ = ["output_fields", "names"]
647

    
648

    
649
# Exports opcodes
650
class OpQueryExports(OpCode):
651
  """Compute the list of exported images."""
652
  OP_ID = "OP_BACKUP_QUERY"
653
  __slots__ = ["nodes", "use_locking"]
654

    
655

    
656
class OpPrepareExport(OpCode):
657
  """Prepares an instance export.
658

659
  @ivar instance_name: Instance name
660
  @ivar mode: Export mode (one of L{constants.EXPORT_MODES})
661

662
  """
663
  OP_ID = "OP_BACKUP_PREPARE"
664
  OP_DSC_FIELD = "instance_name"
665
  __slots__ = [
666
    "instance_name", "mode",
667
    ]
668

    
669

    
670
class OpExportInstance(OpCode):
671
  """Export an instance.
672

673
  For local exports, the export destination is the node name. For remote
674
  exports, the export destination is a list of tuples, each consisting of
675
  hostname/IP address, port, HMAC and HMAC salt. The HMAC is calculated using
676
  the cluster domain secret over the value "${index}:${hostname}:${port}". The
677
  destination X509 CA must be a signed certificate.
678

679
  @ivar mode: Export mode (one of L{constants.EXPORT_MODES})
680
  @ivar target_node: Export destination
681
  @ivar x509_key_name: X509 key to use (remote export only)
682
  @ivar destination_x509_ca: Destination X509 CA in PEM format (remote export
683
                             only)
684

685
  """
686
  OP_ID = "OP_BACKUP_EXPORT"
687
  OP_DSC_FIELD = "instance_name"
688
  __slots__ = [
689
    # TODO: Rename target_node as it changes meaning for different export modes
690
    # (e.g. "destination")
691
    "instance_name", "target_node", "shutdown", "shutdown_timeout",
692
    "remove_instance",
693
    "ignore_remove_failures",
694
    "mode",
695
    "x509_key_name",
696
    "destination_x509_ca",
697
    ]
698

    
699

    
700
class OpRemoveExport(OpCode):
701
  """Remove an instance's export."""
702
  OP_ID = "OP_BACKUP_REMOVE"
703
  OP_DSC_FIELD = "instance_name"
704
  __slots__ = ["instance_name"]
705

    
706

    
707
# Tags opcodes
708
class OpGetTags(OpCode):
709
  """Returns the tags of the given object."""
710
  OP_ID = "OP_TAGS_GET"
711
  OP_DSC_FIELD = "name"
712
  __slots__ = ["kind", "name"]
713

    
714

    
715
class OpSearchTags(OpCode):
716
  """Searches the tags in the cluster for a given pattern."""
717
  OP_ID = "OP_TAGS_SEARCH"
718
  OP_DSC_FIELD = "pattern"
719
  __slots__ = ["pattern"]
720

    
721

    
722
class OpAddTags(OpCode):
723
  """Add a list of tags on a given object."""
724
  OP_ID = "OP_TAGS_SET"
725
  __slots__ = ["kind", "name", "tags"]
726

    
727

    
728
class OpDelTags(OpCode):
729
  """Remove a list of tags from a given object."""
730
  OP_ID = "OP_TAGS_DEL"
731
  __slots__ = ["kind", "name", "tags"]
732

    
733

    
734
# Test opcodes
735
class OpTestDelay(OpCode):
736
  """Sleeps for a configured amount of time.
737

738
  This is used just for debugging and testing.
739

740
  Parameters:
741
    - duration: the time to sleep
742
    - on_master: if true, sleep on the master
743
    - on_nodes: list of nodes in which to sleep
744

745
  If the on_master parameter is true, it will execute a sleep on the
746
  master (before any node sleep).
747

748
  If the on_nodes list is not empty, it will sleep on those nodes
749
  (after the sleep on the master, if that is enabled).
750

751
  As an additional feature, the case of duration < 0 will be reported
752
  as an execution error, so this opcode can be used as a failure
753
  generator. The case of duration == 0 will not be treated specially.
754

755
  """
756
  OP_ID = "OP_TEST_DELAY"
757
  OP_DSC_FIELD = "duration"
758
  __slots__ = ["duration", "on_master", "on_nodes"]
759

    
760

    
761
class OpTestAllocator(OpCode):
762
  """Allocator framework testing.
763

764
  This opcode has two modes:
765
    - gather and return allocator input for a given mode (allocate new
766
      or replace secondary) and a given instance definition (direction
767
      'in')
768
    - run a selected allocator for a given operation (as above) and
769
      return the allocator output (direction 'out')
770

771
  """
772
  OP_ID = "OP_TEST_ALLOCATOR"
773
  OP_DSC_FIELD = "allocator"
774
  __slots__ = [
775
    "direction", "mode", "allocator", "name",
776
    "mem_size", "disks", "disk_template",
777
    "os", "tags", "nics", "vcpus", "hypervisor",
778
    "evac_nodes",
779
    ]
780

    
781

    
782
OP_MAPPING = dict([(v.OP_ID, v) for v in globals().values()
783
                   if (isinstance(v, type) and issubclass(v, OpCode) and
784
                       hasattr(v, "OP_ID"))])