Statistics
| Branch: | Tag: | Revision:

root / lib / opcodes.py @ 9e5442ce

History | View | Annotate | Download (17.2 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
    for key in kwargs:
56
      if key not in self.__slots__:
57
        raise TypeError("Object %s doesn't support the parameter '%s'" %
58
                        (self.__class__.__name__, key))
59
      setattr(self, key, kwargs[key])
60

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

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

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

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

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

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

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

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

    
91
    for name in self.__slots__:
92
      if name not in state:
93
        delattr(self, name)
94

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

    
98

    
99
class OpCode(BaseOpCode):
100
  """Abstract OpCode.
101

102
  This is the root of the actual OpCode hierarchy. All clases derived
103
  from this class should override OP_ID.
104

105
  @cvar OP_ID: The ID of this opcode. This should be unique amongst all
106
               children of this class.
107
  @ivar dry_run: Whether the LU should be run in dry-run mode, i.e. just
108
                 the check steps
109

110
  """
111
  OP_ID = "OP_ABSTRACT"
112
  __slots__ = BaseOpCode.__slots__ + ["dry_run"]
113

    
114
  def __getstate__(self):
115
    """Specialized getstate for opcodes.
116

117
    This method adds to the state dictionary the OP_ID of the class,
118
    so that on unload we can identify the correct class for
119
    instantiating the opcode.
120

121
    @rtype:   C{dict}
122
    @return:  the state as a dictionary
123

124
    """
125
    data = BaseOpCode.__getstate__(self)
126
    data["OP_ID"] = self.OP_ID
127
    return data
128

    
129
  @classmethod
130
  def LoadOpCode(cls, data):
131
    """Generic load opcode method.
132

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

137
    @type data:  C{dict}
138
    @param data: the serialized opcode
139

140
    """
141
    if not isinstance(data, dict):
142
      raise ValueError("Invalid data to LoadOpCode (%s)" % type(data))
143
    if "OP_ID" not in data:
144
      raise ValueError("Invalid data to LoadOpcode, missing OP_ID")
145
    op_id = data["OP_ID"]
146
    op_class = None
147
    if op_id in OP_MAPPING:
148
      op_class = OP_MAPPING[op_id]
149
    else:
150
      raise ValueError("Invalid data to LoadOpCode: OP_ID %s unsupported" %
151
                       op_id)
152
    op = op_class()
153
    new_data = data.copy()
154
    del new_data["OP_ID"]
155
    op.__setstate__(new_data)
156
    return op
157

    
158
  def Summary(self):
159
    """Generates a summary description of this opcode.
160

161
    """
162
    # all OP_ID start with OP_, we remove that
163
    txt = self.OP_ID[3:]
164
    field_name = getattr(self, "OP_DSC_FIELD", None)
165
    if field_name:
166
      field_value = getattr(self, field_name, None)
167
      txt = "%s(%s)" % (txt, field_value)
168
    return txt
169

    
170

    
171
# cluster opcodes
172

    
173
class OpDestroyCluster(OpCode):
174
  """Destroy the cluster.
175

176
  This opcode has no other parameters. All the state is irreversibly
177
  lost after the execution of this opcode.
178

179
  """
180
  OP_ID = "OP_CLUSTER_DESTROY"
181
  __slots__ = OpCode.__slots__ + []
182

    
183

    
184
class OpQueryClusterInfo(OpCode):
185
  """Query cluster information."""
186
  OP_ID = "OP_CLUSTER_QUERY"
187
  __slots__ = OpCode.__slots__ + []
188

    
189

    
190
class OpVerifyCluster(OpCode):
191
  """Verify the cluster state.
192

193
  @type skip_checks: C{list}
194
  @ivar skip_checks: steps to be skipped from the verify process; this
195
                     needs to be a subset of
196
                     L{constants.VERIFY_OPTIONAL_CHECKS}; currently
197
                     only L{constants.VERIFY_NPLUSONE_MEM} can be passed
198

199
  """
200
  OP_ID = "OP_CLUSTER_VERIFY"
201
  __slots__ = OpCode.__slots__ + ["skip_checks"]
202

    
203

    
204
class OpVerifyDisks(OpCode):
205
  """Verify the cluster disks.
206

207
  Parameters: none
208

209
  Result: a tuple of four elements:
210
    - list of node names with bad data returned (unreachable, etc.)
211
    - dict of node names with broken volume groups (values: error msg)
212
    - list of instances with degraded disks (that should be activated)
213
    - dict of instances with missing logical volumes (values: (node, vol)
214
      pairs with details about the missing volumes)
215

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

221
  Note that only instances that are drbd-based are taken into
222
  consideration. This might need to be revisited in the future.
223

224
  """
225
  OP_ID = "OP_CLUSTER_VERIFY_DISKS"
226
  __slots__ = OpCode.__slots__ + []
227

    
228

    
229
class OpQueryConfigValues(OpCode):
230
  """Query cluster configuration values."""
231
  OP_ID = "OP_CLUSTER_CONFIG_QUERY"
232
  __slots__ = OpCode.__slots__ + ["output_fields"]
233

    
234

    
235
class OpRenameCluster(OpCode):
236
  """Rename the cluster.
237

238
  @type name: C{str}
239
  @ivar name: The new name of the cluster. The name and/or the master IP
240
              address will be changed to match the new name and its IP
241
              address.
242

243
  """
244
  OP_ID = "OP_CLUSTER_RENAME"
245
  OP_DSC_FIELD = "name"
246
  __slots__ = OpCode.__slots__ + ["name"]
247

    
248

    
249
class OpSetClusterParams(OpCode):
250
  """Change the parameters of the cluster.
251

252
  @type vg_name: C{str} or C{None}
253
  @ivar vg_name: The new volume group name or None to disable LVM usage.
254

255
  """
256
  OP_ID = "OP_CLUSTER_SET_PARAMS"
257
  __slots__ = OpCode.__slots__ + [
258
    "vg_name",
259
    "enabled_hypervisors",
260
    "hvparams",
261
    "beparams",
262
    "nicparams",
263
    "candidate_pool_size",
264
    ]
265

    
266

    
267
class OpRedistributeConfig(OpCode):
268
  """Force a full push of the cluster configuration.
269

270
  """
271
  OP_ID = "OP_CLUSTER_REDIST_CONF"
272
  __slots__ = OpCode.__slots__ + [
273
    ]
274

    
275
# node opcodes
276

    
277
class OpRemoveNode(OpCode):
278
  """Remove a node.
279

280
  @type node_name: C{str}
281
  @ivar node_name: The name of the node to remove. If the node still has
282
                   instances on it, the operation will fail.
283

284
  """
285
  OP_ID = "OP_NODE_REMOVE"
286
  OP_DSC_FIELD = "node_name"
287
  __slots__ = OpCode.__slots__ + ["node_name"]
288

    
289

    
290
class OpAddNode(OpCode):
291
  """Add a node to the cluster.
292

293
  @type node_name: C{str}
294
  @ivar node_name: The name of the node to add. This can be a short name,
295
                   but it will be expanded to the FQDN.
296
  @type primary_ip: IP address
297
  @ivar primary_ip: The primary IP of the node. This will be ignored when the
298
                    opcode is submitted, but will be filled during the node
299
                    add (so it will be visible in the job query).
300
  @type secondary_ip: IP address
301
  @ivar secondary_ip: The secondary IP of the node. This needs to be passed
302
                      if the cluster has been initialized in 'dual-network'
303
                      mode, otherwise it must not be given.
304
  @type readd: C{bool}
305
  @ivar readd: Whether to re-add an existing node to the cluster. If
306
               this is not passed, then the operation will abort if the node
307
               name is already in the cluster; use this parameter to 'repair'
308
               a node that had its configuration broken, or was reinstalled
309
               without removal from the cluster.
310

311
  """
312
  OP_ID = "OP_NODE_ADD"
313
  OP_DSC_FIELD = "node_name"
314
  __slots__ = OpCode.__slots__ + [
315
    "node_name", "primary_ip", "secondary_ip", "readd",
316
    ]
317

    
318

    
319
class OpQueryNodes(OpCode):
320
  """Compute the list of nodes."""
321
  OP_ID = "OP_NODE_QUERY"
322
  __slots__ = OpCode.__slots__ + ["output_fields", "names", "use_locking"]
323

    
324

    
325
class OpQueryNodeVolumes(OpCode):
326
  """Get list of volumes on node."""
327
  OP_ID = "OP_NODE_QUERYVOLS"
328
  __slots__ = OpCode.__slots__ + ["nodes", "output_fields"]
329

    
330

    
331
class OpQueryNodeStorage(OpCode):
332
  """Get information on storage for node(s)."""
333
  OP_ID = "OP_NODE_QUERY_STORAGE"
334
  __slots__ = OpCode.__slots__ + [
335
    "nodes",
336
    "storage_type",
337
    "name",
338
    "output_fields",
339
    ]
340

    
341

    
342
class OpSetNodeParams(OpCode):
343
  """Change the parameters of a node."""
344
  OP_ID = "OP_NODE_SET_PARAMS"
345
  OP_DSC_FIELD = "node_name"
346
  __slots__ = OpCode.__slots__ + [
347
    "node_name",
348
    "force",
349
    "master_candidate",
350
    "offline",
351
    "drained",
352
    ]
353

    
354

    
355
class OpPowercycleNode(OpCode):
356
  """Tries to powercycle a node."""
357
  OP_ID = "OP_NODE_POWERCYCLE"
358
  OP_DSC_FIELD = "node_name"
359
  __slots__ = OpCode.__slots__ + [
360
    "node_name",
361
    "force",
362
    ]
363

    
364

    
365
class OpEvacuateNode(OpCode):
366
  """Relocate secondary instances from a node."""
367
  OP_ID = "OP_NODE_EVACUATE"
368
  OP_DSC_FIELD = "node_name"
369
  __slots__ = OpCode.__slots__ + [
370
    "node_name", "remote_node", "iallocator",
371
    ]
372

    
373

    
374
class OpMigrateNode(OpCode):
375
  """Migrate all instances from a node."""
376
  OP_ID = "OP_NODE_MIGRATE"
377
  OP_DSC_FIELD = "node_name"
378
  __slots__ = OpCode.__slots__ + [
379
    "node_name",
380
    "live",
381
    ]
382

    
383

    
384
# instance opcodes
385

    
386
class OpCreateInstance(OpCode):
387
  """Create an instance."""
388
  OP_ID = "OP_INSTANCE_CREATE"
389
  OP_DSC_FIELD = "instance_name"
390
  __slots__ = OpCode.__slots__ + [
391
    "instance_name", "os_type", "pnode",
392
    "disk_template", "snode", "mode",
393
    "disks", "nics",
394
    "src_node", "src_path", "start",
395
    "wait_for_sync", "ip_check",
396
    "file_storage_dir", "file_driver",
397
    "iallocator",
398
    "hypervisor", "hvparams", "beparams",
399
    "dry_run",
400
    ]
401

    
402

    
403
class OpReinstallInstance(OpCode):
404
  """Reinstall an instance's OS."""
405
  OP_ID = "OP_INSTANCE_REINSTALL"
406
  OP_DSC_FIELD = "instance_name"
407
  __slots__ = OpCode.__slots__ + ["instance_name", "os_type"]
408

    
409

    
410
class OpRemoveInstance(OpCode):
411
  """Remove an instance."""
412
  OP_ID = "OP_INSTANCE_REMOVE"
413
  OP_DSC_FIELD = "instance_name"
414
  __slots__ = OpCode.__slots__ + ["instance_name", "ignore_failures"]
415

    
416

    
417
class OpRenameInstance(OpCode):
418
  """Rename an instance."""
419
  OP_ID = "OP_INSTANCE_RENAME"
420
  __slots__ = OpCode.__slots__ + [
421
    "instance_name", "ignore_ip", "new_name",
422
    ]
423

    
424

    
425
class OpStartupInstance(OpCode):
426
  """Startup an instance."""
427
  OP_ID = "OP_INSTANCE_STARTUP"
428
  OP_DSC_FIELD = "instance_name"
429
  __slots__ = OpCode.__slots__ + [
430
    "instance_name", "force", "hvparams", "beparams",
431
    ]
432

    
433

    
434
class OpShutdownInstance(OpCode):
435
  """Shutdown an instance."""
436
  OP_ID = "OP_INSTANCE_SHUTDOWN"
437
  OP_DSC_FIELD = "instance_name"
438
  __slots__ = OpCode.__slots__ + ["instance_name"]
439

    
440

    
441
class OpRebootInstance(OpCode):
442
  """Reboot an instance."""
443
  OP_ID = "OP_INSTANCE_REBOOT"
444
  OP_DSC_FIELD = "instance_name"
445
  __slots__ = OpCode.__slots__ + [
446
    "instance_name", "reboot_type", "ignore_secondaries",
447
    ]
448

    
449

    
450
class OpReplaceDisks(OpCode):
451
  """Replace the disks of an instance."""
452
  OP_ID = "OP_INSTANCE_REPLACE_DISKS"
453
  OP_DSC_FIELD = "instance_name"
454
  __slots__ = OpCode.__slots__ + [
455
    "instance_name", "remote_node", "mode", "disks", "iallocator",
456
    ]
457

    
458

    
459
class OpFailoverInstance(OpCode):
460
  """Failover an instance."""
461
  OP_ID = "OP_INSTANCE_FAILOVER"
462
  OP_DSC_FIELD = "instance_name"
463
  __slots__ = OpCode.__slots__ + ["instance_name", "ignore_consistency"]
464

    
465

    
466
class OpMigrateInstance(OpCode):
467
  """Migrate an instance.
468

469
  This migrates (without shutting down an instance) to its secondary
470
  node.
471

472
  @ivar instance_name: the name of the instance
473

474
  """
475
  OP_ID = "OP_INSTANCE_MIGRATE"
476
  OP_DSC_FIELD = "instance_name"
477
  __slots__ = OpCode.__slots__ + ["instance_name", "live", "cleanup"]
478

    
479

    
480
class OpConnectConsole(OpCode):
481
  """Connect to an instance's console."""
482
  OP_ID = "OP_INSTANCE_CONSOLE"
483
  OP_DSC_FIELD = "instance_name"
484
  __slots__ = OpCode.__slots__ + ["instance_name"]
485

    
486

    
487
class OpActivateInstanceDisks(OpCode):
488
  """Activate an instance's disks."""
489
  OP_ID = "OP_INSTANCE_ACTIVATE_DISKS"
490
  OP_DSC_FIELD = "instance_name"
491
  __slots__ = OpCode.__slots__ + ["instance_name"]
492

    
493

    
494
class OpDeactivateInstanceDisks(OpCode):
495
  """Deactivate an instance's disks."""
496
  OP_ID = "OP_INSTANCE_DEACTIVATE_DISKS"
497
  OP_DSC_FIELD = "instance_name"
498
  __slots__ = OpCode.__slots__ + ["instance_name"]
499

    
500

    
501
class OpQueryInstances(OpCode):
502
  """Compute the list of instances."""
503
  OP_ID = "OP_INSTANCE_QUERY"
504
  __slots__ = OpCode.__slots__ + ["output_fields", "names", "use_locking"]
505

    
506

    
507
class OpQueryInstanceData(OpCode):
508
  """Compute the run-time status of instances."""
509
  OP_ID = "OP_INSTANCE_QUERY_DATA"
510
  __slots__ = OpCode.__slots__ + ["instances", "static"]
511

    
512

    
513
class OpSetInstanceParams(OpCode):
514
  """Change the parameters of an instance."""
515
  OP_ID = "OP_INSTANCE_SET_PARAMS"
516
  OP_DSC_FIELD = "instance_name"
517
  __slots__ = OpCode.__slots__ + [
518
    "instance_name",
519
    "hvparams", "beparams", "force",
520
    "nics", "disks",
521
    ]
522

    
523

    
524
class OpGrowDisk(OpCode):
525
  """Grow a disk of an instance."""
526
  OP_ID = "OP_INSTANCE_GROW_DISK"
527
  OP_DSC_FIELD = "instance_name"
528
  __slots__ = OpCode.__slots__ + [
529
    "instance_name", "disk", "amount", "wait_for_sync",
530
    ]
531

    
532

    
533
# OS opcodes
534
class OpDiagnoseOS(OpCode):
535
  """Compute the list of guest operating systems."""
536
  OP_ID = "OP_OS_DIAGNOSE"
537
  __slots__ = OpCode.__slots__ + ["output_fields", "names"]
538

    
539

    
540
# Exports opcodes
541
class OpQueryExports(OpCode):
542
  """Compute the list of exported images."""
543
  OP_ID = "OP_BACKUP_QUERY"
544
  __slots__ = OpCode.__slots__ + ["nodes", "use_locking"]
545

    
546

    
547
class OpExportInstance(OpCode):
548
  """Export an instance."""
549
  OP_ID = "OP_BACKUP_EXPORT"
550
  OP_DSC_FIELD = "instance_name"
551
  __slots__ = OpCode.__slots__ + ["instance_name", "target_node", "shutdown"]
552

    
553

    
554
class OpRemoveExport(OpCode):
555
  """Remove an instance's export."""
556
  OP_ID = "OP_BACKUP_REMOVE"
557
  OP_DSC_FIELD = "instance_name"
558
  __slots__ = OpCode.__slots__ + ["instance_name"]
559

    
560

    
561
# Tags opcodes
562
class OpGetTags(OpCode):
563
  """Returns the tags of the given object."""
564
  OP_ID = "OP_TAGS_GET"
565
  OP_DSC_FIELD = "name"
566
  __slots__ = OpCode.__slots__ + ["kind", "name"]
567

    
568

    
569
class OpSearchTags(OpCode):
570
  """Searches the tags in the cluster for a given pattern."""
571
  OP_ID = "OP_TAGS_SEARCH"
572
  OP_DSC_FIELD = "pattern"
573
  __slots__ = OpCode.__slots__ + ["pattern"]
574

    
575

    
576
class OpAddTags(OpCode):
577
  """Add a list of tags on a given object."""
578
  OP_ID = "OP_TAGS_SET"
579
  __slots__ = OpCode.__slots__ + ["kind", "name", "tags"]
580

    
581

    
582
class OpDelTags(OpCode):
583
  """Remove a list of tags from a given object."""
584
  OP_ID = "OP_TAGS_DEL"
585
  __slots__ = OpCode.__slots__ + ["kind", "name", "tags"]
586

    
587

    
588
# Test opcodes
589
class OpTestDelay(OpCode):
590
  """Sleeps for a configured amount of time.
591

592
  This is used just for debugging and testing.
593

594
  Parameters:
595
    - duration: the time to sleep
596
    - on_master: if true, sleep on the master
597
    - on_nodes: list of nodes in which to sleep
598

599
  If the on_master parameter is true, it will execute a sleep on the
600
  master (before any node sleep).
601

602
  If the on_nodes list is not empty, it will sleep on those nodes
603
  (after the sleep on the master, if that is enabled).
604

605
  As an additional feature, the case of duration < 0 will be reported
606
  as an execution error, so this opcode can be used as a failure
607
  generator. The case of duration == 0 will not be treated specially.
608

609
  """
610
  OP_ID = "OP_TEST_DELAY"
611
  OP_DSC_FIELD = "duration"
612
  __slots__ = OpCode.__slots__ + ["duration", "on_master", "on_nodes"]
613

    
614

    
615
class OpTestAllocator(OpCode):
616
  """Allocator framework testing.
617

618
  This opcode has two modes:
619
    - gather and return allocator input for a given mode (allocate new
620
      or replace secondary) and a given instance definition (direction
621
      'in')
622
    - run a selected allocator for a given operation (as above) and
623
      return the allocator output (direction 'out')
624

625
  """
626
  OP_ID = "OP_TEST_ALLOCATOR"
627
  OP_DSC_FIELD = "allocator"
628
  __slots__ = OpCode.__slots__ + [
629
    "direction", "mode", "allocator", "name",
630
    "mem_size", "disks", "disk_template",
631
    "os", "tags", "nics", "vcpus", "hypervisor",
632
    ]
633

    
634
OP_MAPPING = dict([(v.OP_ID, v) for v in globals().values()
635
                   if (isinstance(v, type) and issubclass(v, OpCode) and
636
                       hasattr(v, "OP_ID"))])