Statistics
| Branch: | Tag: | Revision:

root / lib / opcodes.py @ 8c35561f

History | View | Annotate | Download (20.9 kB)

1
#
2
#
3

    
4
# Copyright (C) 2006, 2007, 2008, 2009, 2010 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
    "drbd_helper",
303
    "enabled_hypervisors",
304
    "hvparams",
305
    "os_hvp",
306
    "beparams",
307
    "osparams",
308
    "nicparams",
309
    "candidate_pool_size",
310
    "maintain_node_health",
311
    "uid_pool",
312
    "add_uids",
313
    "remove_uids",
314
    "default_iallocator",
315
    ]
316

    
317

    
318
class OpRedistributeConfig(OpCode):
319
  """Force a full push of the cluster configuration.
320

321
  """
322
  OP_ID = "OP_CLUSTER_REDIST_CONF"
323
  __slots__ = []
324

    
325
# node opcodes
326

    
327
class OpRemoveNode(OpCode):
328
  """Remove a node.
329

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

334
  """
335
  OP_ID = "OP_NODE_REMOVE"
336
  OP_DSC_FIELD = "node_name"
337
  __slots__ = ["node_name"]
338

    
339

    
340
class OpAddNode(OpCode):
341
  """Add a node to the cluster.
342

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

361
  """
362
  OP_ID = "OP_NODE_ADD"
363
  OP_DSC_FIELD = "node_name"
364
  __slots__ = ["node_name", "primary_ip", "secondary_ip", "readd"]
365

    
366

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

    
372

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

    
378

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

    
389

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

    
400

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

    
412

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

    
426

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

    
436

    
437
class OpMigrateNode(OpCode):
438
  """Migrate all instances from a node."""
439
  OP_ID = "OP_NODE_MIGRATE"
440
  OP_DSC_FIELD = "node_name"
441
  __slots__ = [
442
    "node_name",
443
    "mode",
444
    ]
445

    
446

    
447
class OpNodeEvacuationStrategy(OpCode):
448
  """Compute the evacuation strategy for a list of nodes."""
449
  OP_ID = "OP_NODE_EVAC_STRATEGY"
450
  OP_DSC_FIELD = "nodes"
451
  __slots__ = ["nodes", "iallocator", "remote_node"]
452

    
453

    
454
# instance opcodes
455

    
456
class OpCreateInstance(OpCode):
457
  """Create an instance.
458

459
  @ivar instance_name: Instance name
460
  @ivar mode: Instance creation mode (one of L{constants.INSTANCE_CREATE_MODES})
461
  @ivar source_handshake: Signed handshake from source (remote import only)
462
  @ivar source_x509_ca: Source X509 CA in PEM format (remote import only)
463
  @ivar source_instance_name: Previous name of instance (remote import only)
464

465
  """
466
  OP_ID = "OP_INSTANCE_CREATE"
467
  OP_DSC_FIELD = "instance_name"
468
  __slots__ = [
469
    "instance_name",
470
    "os_type", "force_variant", "no_install",
471
    "pnode", "disk_template", "snode", "mode",
472
    "disks", "nics",
473
    "src_node", "src_path", "start", "identify_defaults",
474
    "wait_for_sync", "ip_check", "name_check",
475
    "file_storage_dir", "file_driver",
476
    "iallocator",
477
    "hypervisor", "hvparams", "beparams", "osparams",
478
    "source_handshake",
479
    "source_x509_ca",
480
    "source_instance_name",
481
    "dry_run",
482
    ]
483

    
484

    
485
class OpReinstallInstance(OpCode):
486
  """Reinstall an instance's OS."""
487
  OP_ID = "OP_INSTANCE_REINSTALL"
488
  OP_DSC_FIELD = "instance_name"
489
  __slots__ = ["instance_name", "os_type", "force_variant"]
490

    
491

    
492
class OpRemoveInstance(OpCode):
493
  """Remove an instance."""
494
  OP_ID = "OP_INSTANCE_REMOVE"
495
  OP_DSC_FIELD = "instance_name"
496
  __slots__ = [
497
    "instance_name",
498
    "ignore_failures",
499
    "shutdown_timeout",
500
    ]
501

    
502

    
503
class OpRenameInstance(OpCode):
504
  """Rename an instance."""
505
  OP_ID = "OP_INSTANCE_RENAME"
506
  __slots__ = [
507
    "instance_name", "ignore_ip", "new_name", "check_name",
508
    ]
509

    
510

    
511
class OpStartupInstance(OpCode):
512
  """Startup an instance."""
513
  OP_ID = "OP_INSTANCE_STARTUP"
514
  OP_DSC_FIELD = "instance_name"
515
  __slots__ = [
516
    "instance_name", "force", "hvparams", "beparams",
517
    ]
518

    
519

    
520
class OpShutdownInstance(OpCode):
521
  """Shutdown an instance."""
522
  OP_ID = "OP_INSTANCE_SHUTDOWN"
523
  OP_DSC_FIELD = "instance_name"
524
  __slots__ = ["instance_name", "timeout"]
525

    
526

    
527
class OpRebootInstance(OpCode):
528
  """Reboot an instance."""
529
  OP_ID = "OP_INSTANCE_REBOOT"
530
  OP_DSC_FIELD = "instance_name"
531
  __slots__ = [
532
    "instance_name", "reboot_type", "ignore_secondaries", "shutdown_timeout",
533
    ]
534

    
535

    
536
class OpReplaceDisks(OpCode):
537
  """Replace the disks of an instance."""
538
  OP_ID = "OP_INSTANCE_REPLACE_DISKS"
539
  OP_DSC_FIELD = "instance_name"
540
  __slots__ = [
541
    "instance_name", "remote_node", "mode", "disks", "iallocator",
542
    "early_release",
543
    ]
544

    
545

    
546
class OpFailoverInstance(OpCode):
547
  """Failover an instance."""
548
  OP_ID = "OP_INSTANCE_FAILOVER"
549
  OP_DSC_FIELD = "instance_name"
550
  __slots__ = [
551
    "instance_name", "ignore_consistency", "shutdown_timeout",
552
    ]
553

    
554

    
555
class OpMigrateInstance(OpCode):
556
  """Migrate an instance.
557

558
  This migrates (without shutting down an instance) to its secondary
559
  node.
560

561
  @ivar instance_name: the name of the instance
562
  @ivar mode: the migration mode (live, non-live or None for auto)
563

564
  """
565
  OP_ID = "OP_INSTANCE_MIGRATE"
566
  OP_DSC_FIELD = "instance_name"
567
  __slots__ = ["instance_name", "mode", "cleanup"]
568

    
569

    
570
class OpMoveInstance(OpCode):
571
  """Move an instance.
572

573
  This move (with shutting down an instance and data copying) to an
574
  arbitrary node.
575

576
  @ivar instance_name: the name of the instance
577
  @ivar target_node: the destination node
578

579
  """
580
  OP_ID = "OP_INSTANCE_MOVE"
581
  OP_DSC_FIELD = "instance_name"
582
  __slots__ = [
583
    "instance_name", "target_node", "shutdown_timeout",
584
    ]
585

    
586

    
587
class OpConnectConsole(OpCode):
588
  """Connect to an instance's console."""
589
  OP_ID = "OP_INSTANCE_CONSOLE"
590
  OP_DSC_FIELD = "instance_name"
591
  __slots__ = ["instance_name"]
592

    
593

    
594
class OpActivateInstanceDisks(OpCode):
595
  """Activate an instance's disks."""
596
  OP_ID = "OP_INSTANCE_ACTIVATE_DISKS"
597
  OP_DSC_FIELD = "instance_name"
598
  __slots__ = ["instance_name", "ignore_size"]
599

    
600

    
601
class OpDeactivateInstanceDisks(OpCode):
602
  """Deactivate an instance's disks."""
603
  OP_ID = "OP_INSTANCE_DEACTIVATE_DISKS"
604
  OP_DSC_FIELD = "instance_name"
605
  __slots__ = ["instance_name"]
606

    
607

    
608
class OpRecreateInstanceDisks(OpCode):
609
  """Deactivate an instance's disks."""
610
  OP_ID = "OP_INSTANCE_RECREATE_DISKS"
611
  OP_DSC_FIELD = "instance_name"
612
  __slots__ = ["instance_name", "disks"]
613

    
614

    
615
class OpQueryInstances(OpCode):
616
  """Compute the list of instances."""
617
  OP_ID = "OP_INSTANCE_QUERY"
618
  __slots__ = ["output_fields", "names", "use_locking"]
619

    
620

    
621
class OpQueryInstanceData(OpCode):
622
  """Compute the run-time status of instances."""
623
  OP_ID = "OP_INSTANCE_QUERY_DATA"
624
  __slots__ = ["instances", "static"]
625

    
626

    
627
class OpSetInstanceParams(OpCode):
628
  """Change the parameters of an instance."""
629
  OP_ID = "OP_INSTANCE_SET_PARAMS"
630
  OP_DSC_FIELD = "instance_name"
631
  __slots__ = [
632
    "instance_name",
633
    "hvparams", "beparams", "osparams", "force",
634
    "nics", "disks", "disk_template",
635
    "remote_node", "os_name", "force_variant",
636
    ]
637

    
638

    
639
class OpGrowDisk(OpCode):
640
  """Grow a disk of an instance."""
641
  OP_ID = "OP_INSTANCE_GROW_DISK"
642
  OP_DSC_FIELD = "instance_name"
643
  __slots__ = [
644
    "instance_name", "disk", "amount", "wait_for_sync",
645
    ]
646

    
647

    
648
# OS opcodes
649
class OpDiagnoseOS(OpCode):
650
  """Compute the list of guest operating systems."""
651
  OP_ID = "OP_OS_DIAGNOSE"
652
  __slots__ = ["output_fields", "names"]
653

    
654

    
655
# Exports opcodes
656
class OpQueryExports(OpCode):
657
  """Compute the list of exported images."""
658
  OP_ID = "OP_BACKUP_QUERY"
659
  __slots__ = ["nodes", "use_locking"]
660

    
661

    
662
class OpPrepareExport(OpCode):
663
  """Prepares an instance export.
664

665
  @ivar instance_name: Instance name
666
  @ivar mode: Export mode (one of L{constants.EXPORT_MODES})
667

668
  """
669
  OP_ID = "OP_BACKUP_PREPARE"
670
  OP_DSC_FIELD = "instance_name"
671
  __slots__ = [
672
    "instance_name", "mode",
673
    ]
674

    
675

    
676
class OpExportInstance(OpCode):
677
  """Export an instance.
678

679
  For local exports, the export destination is the node name. For remote
680
  exports, the export destination is a list of tuples, each consisting of
681
  hostname/IP address, port, HMAC and HMAC salt. The HMAC is calculated using
682
  the cluster domain secret over the value "${index}:${hostname}:${port}". The
683
  destination X509 CA must be a signed certificate.
684

685
  @ivar mode: Export mode (one of L{constants.EXPORT_MODES})
686
  @ivar target_node: Export destination
687
  @ivar x509_key_name: X509 key to use (remote export only)
688
  @ivar destination_x509_ca: Destination X509 CA in PEM format (remote export
689
                             only)
690

691
  """
692
  OP_ID = "OP_BACKUP_EXPORT"
693
  OP_DSC_FIELD = "instance_name"
694
  __slots__ = [
695
    # TODO: Rename target_node as it changes meaning for different export modes
696
    # (e.g. "destination")
697
    "instance_name", "target_node", "shutdown", "shutdown_timeout",
698
    "remove_instance",
699
    "ignore_remove_failures",
700
    "mode",
701
    "x509_key_name",
702
    "destination_x509_ca",
703
    ]
704

    
705

    
706
class OpRemoveExport(OpCode):
707
  """Remove an instance's export."""
708
  OP_ID = "OP_BACKUP_REMOVE"
709
  OP_DSC_FIELD = "instance_name"
710
  __slots__ = ["instance_name"]
711

    
712

    
713
# Tags opcodes
714
class OpGetTags(OpCode):
715
  """Returns the tags of the given object."""
716
  OP_ID = "OP_TAGS_GET"
717
  OP_DSC_FIELD = "name"
718
  __slots__ = ["kind", "name"]
719

    
720

    
721
class OpSearchTags(OpCode):
722
  """Searches the tags in the cluster for a given pattern."""
723
  OP_ID = "OP_TAGS_SEARCH"
724
  OP_DSC_FIELD = "pattern"
725
  __slots__ = ["pattern"]
726

    
727

    
728
class OpAddTags(OpCode):
729
  """Add a list of tags on a given object."""
730
  OP_ID = "OP_TAGS_SET"
731
  __slots__ = ["kind", "name", "tags"]
732

    
733

    
734
class OpDelTags(OpCode):
735
  """Remove a list of tags from a given object."""
736
  OP_ID = "OP_TAGS_DEL"
737
  __slots__ = ["kind", "name", "tags"]
738

    
739

    
740
# Test opcodes
741
class OpTestDelay(OpCode):
742
  """Sleeps for a configured amount of time.
743

744
  This is used just for debugging and testing.
745

746
  Parameters:
747
    - duration: the time to sleep
748
    - on_master: if true, sleep on the master
749
    - on_nodes: list of nodes in which to sleep
750

751
  If the on_master parameter is true, it will execute a sleep on the
752
  master (before any node sleep).
753

754
  If the on_nodes list is not empty, it will sleep on those nodes
755
  (after the sleep on the master, if that is enabled).
756

757
  As an additional feature, the case of duration < 0 will be reported
758
  as an execution error, so this opcode can be used as a failure
759
  generator. The case of duration == 0 will not be treated specially.
760

761
  """
762
  OP_ID = "OP_TEST_DELAY"
763
  OP_DSC_FIELD = "duration"
764
  __slots__ = ["duration", "on_master", "on_nodes", "repeat"]
765

    
766

    
767
class OpTestAllocator(OpCode):
768
  """Allocator framework testing.
769

770
  This opcode has two modes:
771
    - gather and return allocator input for a given mode (allocate new
772
      or replace secondary) and a given instance definition (direction
773
      'in')
774
    - run a selected allocator for a given operation (as above) and
775
      return the allocator output (direction 'out')
776

777
  """
778
  OP_ID = "OP_TEST_ALLOCATOR"
779
  OP_DSC_FIELD = "allocator"
780
  __slots__ = [
781
    "direction", "mode", "allocator", "name",
782
    "mem_size", "disks", "disk_template",
783
    "os", "tags", "nics", "vcpus", "hypervisor",
784
    "evac_nodes",
785
    ]
786

    
787

    
788
class OpTestJobqueue(OpCode):
789
  """Utility opcode to test some aspects of the job queue.
790

791
  """
792
  OP_ID = "OP_TEST_JQUEUE"
793
  __slots__ = [
794
    "notify_waitlock",
795
    "notify_exec",
796
    "log_messages",
797
    "fail",
798
    ]
799

    
800

    
801
OP_MAPPING = dict([(v.OP_ID, v) for v in globals().values()
802
                   if (isinstance(v, type) and issubclass(v, OpCode) and
803
                       hasattr(v, "OP_ID"))])