-#!/usr/bin/python
+#
#
# Copyright (C) 2006, 2007 Google Inc.
# few public methods:
# pylint: disable-msg=R0903
-class OpCode(object):
- """Abstract OpCode"""
- OP_ID = "OP_ABSTRACT"
+
+class BaseJO(object):
+ """A simple serializable object.
+
+ This object serves as a parent class for both OpCode and Job since
+ they are serialized in the same way.
+
+ """
__slots__ = []
def __init__(self, **kwargs):
for key in kwargs:
if key not in self.__slots__:
- raise TypeError, ("OpCode %s doesn't support the parameter '%s'" %
- (self.__class__.__name__, key))
+ raise TypeError("Object %s doesn't support the parameter '%s'" %
+ (self.__class__.__name__, key))
setattr(self, key, kwargs[key])
+ def __getstate__(self):
+ state = {}
+ for name in self.__slots__:
+ if hasattr(self, name):
+ state[name] = getattr(self, name)
+ return state
+
+ def __setstate__(self, state):
+ if not isinstance(state, dict):
+ raise ValueError("Invalid data to __setstate__: expected dict, got %s" %
+ type(state))
+
+ for name in self.__slots__:
+ if name not in state:
+ delattr(self, name)
+
+ for name in state:
+ setattr(self, name, state[name])
+
+
+class Job(BaseJO):
+ """Job definition structure
+
+ The Job definitions has two sets of parameters:
+ - the parameters of the job itself (all filled by server):
+ - job_id,
+ - status: pending, running, successfull, failed, aborted
+ - opcode parameters:
+ - op_list, list of opcodes, clients creates this
+ - op_status, status for each opcode, server fills in
+ - op_result, result for each opcode, server fills in
+
+ """
+ STATUS_PENDING = 1
+ STATUS_RUNNING = 2
+ STATUS_SUCCESS = 3
+ STATUS_FAIL = 4
+ STATUS_ABORT = 5
+
+ __slots__ = [
+ "job_id",
+ "status",
+ "op_list",
+ "op_status",
+ "op_result",
+ ]
+
+ def __getstate__(self):
+ """Specialized getstate for jobs
+
+ """
+ data = BaseJO.__getstate__(self)
+ if "op_list" in data:
+ data["op_list"] = [op.__getstate__() for op in data["op_list"]]
+ return data
+
+ def __setstate__(self, state):
+ """Specialized setstate for jobs
+
+ """
+ BaseJO.__setstate__(self, state)
+ if "op_list" in state:
+ self.op_list = [OpCode.LoadOpCode(op) for op in state["op_list"]]
+
+
+class OpCode(BaseJO):
+ """Abstract OpCode"""
+ OP_ID = "OP_ABSTRACT"
+ __slots__ = []
+
+ def __getstate__(self):
+ """Specialized getstate for opcodes.
+
+ """
+ data = BaseJO.__getstate__(self)
+ data["OP_ID"] = self.OP_ID
+ return data
+
+ @classmethod
+ def LoadOpCode(cls, data):
+ """Generic load opcode method.
+
+ """
+ if not isinstance(data, dict):
+ raise ValueError("Invalid data to LoadOpCode (%s)" % type(data))
+ if "OP_ID" not in data:
+ raise ValueError("Invalid data to LoadOpcode, missing OP_ID")
+ op_id = data["OP_ID"]
+ op_class = None
+ for item in globals().values():
+ if (isinstance(item, type) and
+ issubclass(item, cls) and
+ hasattr(item, "OP_ID") and
+ getattr(item, "OP_ID") == op_id):
+ op_class = item
+ break
+ if op_class is None:
+ raise ValueError("Invalid data to LoadOpCode: OP_ID %s unsupported" %
+ op_id)
+ op = op_class()
+ new_data = data.copy()
+ del new_data["OP_ID"]
+ op.__setstate__(new_data)
+ return op
+
class OpInitCluster(OpCode):
"""Initialise the cluster."""
OP_ID = "OP_CLUSTER_INIT"
__slots__ = ["cluster_name", "secondary_ip", "hypervisor_type",
- "vg_name", "mac_prefix", "def_bridge", "master_netdev"]
+ "vg_name", "mac_prefix", "def_bridge", "master_netdev",
+ "file_storage_dir"]
class OpDestroyCluster(OpCode):
class OpQueryClusterInfo(OpCode):
- """Initialise the cluster."""
+ """Query cluster information."""
OP_ID = "OP_CLUSTER_QUERY"
__slots__ = []
class OpClusterCopyFile(OpCode):
- """Initialise the cluster."""
+ """Copy a file to multiple nodes."""
OP_ID = "OP_CLUSTER_COPYFILE"
__slots__ = ["nodes", "filename"]
class OpRunClusterCommand(OpCode):
- """Initialise the cluster."""
+ """Run a command on multiple nodes."""
OP_ID = "OP_CLUSTER_RUNCOMMAND"
__slots__ = ["nodes", "command"]
class OpVerifyCluster(OpCode):
- """Initialise the cluster."""
+ """Verify the cluster state."""
OP_ID = "OP_CLUSTER_VERIFY"
+ __slots__ = ["skip_checks"]
+
+
+class OpVerifyDisks(OpCode):
+ """Verify the cluster disks.
+
+ Parameters: none
+
+ Result: two lists:
+ - list of node names with bad data returned (unreachable, etc.)
+ - dist of node names with broken volume groups (values: error msg)
+ - list of instances with degraded disks (that should be activated)
+ - dict of instances with missing logical volumes (values: (node, vol)
+ pairs with details about the missing volumes)
+
+ In normal operation, all lists should be empty. A non-empty instance
+ list (3rd element of the result) is still ok (errors were fixed) but
+ non-empty node list means some node is down, and probably there are
+ unfixable drbd errors.
+
+ Note that only instances that are drbd-based are taken into
+ consideration. This might need to be revisited in the future.
+
+ """
+ OP_ID = "OP_CLUSTER_VERIFY_DISKS"
__slots__ = []
class OpMasterFailover(OpCode):
- """Initialise the cluster."""
+ """Do a master failover."""
OP_ID = "OP_CLUSTER_MASTERFAILOVER"
__slots__ = []
class OpDumpClusterConfig(OpCode):
- """Initialise the cluster."""
+ """Dump the cluster configuration."""
OP_ID = "OP_CLUSTER_DUMPCONFIG"
__slots__ = []
+class OpRenameCluster(OpCode):
+ """Rename the cluster."""
+ OP_ID = "OP_CLUSTER_RENAME"
+ __slots__ = ["name"]
+
+
+class OpSetClusterParams(OpCode):
+ """Change the parameters of the cluster."""
+ OP_ID = "OP_CLUSTER_SET_PARAMS"
+ __slots__ = ["vg_name"]
+
+
+# node opcodes
+
class OpRemoveNode(OpCode):
"""Remove a node."""
OP_ID = "OP_NODE_REMOVE"
class OpAddNode(OpCode):
"""Add a node."""
OP_ID = "OP_NODE_ADD"
- __slots__ = ["node_name", "primary_ip", "secondary_ip"]
+ __slots__ = ["node_name", "primary_ip", "secondary_ip", "readd"]
class OpQueryNodes(OpCode):
"""Compute the list of nodes."""
OP_ID = "OP_NODE_QUERY"
- __slots__ = ["output_fields"]
-
-
-class OpQueryNodeData(OpCode):
- """Compute the node info."""
- OP_ID = "OP_NODE_INFO"
- __slots__ = ["nodes"]
+ __slots__ = ["output_fields", "names"]
class OpQueryNodeVolumes(OpCode):
# instance opcodes
class OpCreateInstance(OpCode):
- """Compute the list of instances."""
+ """Create an instance."""
OP_ID = "OP_INSTANCE_CREATE"
- __slots__ = ["instance_name", "mem_size", "disk_size", "os_type", "pnode",
- "disk_template", "snode", "swap_size", "mode",
- "vcpus", "ip", "bridge", "src_node", "src_path", "start",
- "wait_for_sync"]
+ __slots__ = [
+ "instance_name", "mem_size", "disk_size", "os_type", "pnode",
+ "disk_template", "snode", "swap_size", "mode",
+ "vcpus", "ip", "bridge", "src_node", "src_path", "start",
+ "wait_for_sync", "ip_check", "mac",
+ "kernel_path", "initrd_path", "hvm_boot_order",
+ "file_storage_dir", "file_driver",
+ "iallocator",
+ ]
class OpReinstallInstance(OpCode):
- """Reinstall an instance."""
+ """Reinstall an instance's OS."""
OP_ID = "OP_INSTANCE_REINSTALL"
- __slots__ = ["instance_name"]
+ __slots__ = ["instance_name", "os_type"]
class OpRemoveInstance(OpCode):
"""Remove an instance."""
OP_ID = "OP_INSTANCE_REMOVE"
- __slots__ = ["instance_name"]
+ __slots__ = ["instance_name", "ignore_failures"]
+
+
+class OpRenameInstance(OpCode):
+ """Rename an instance."""
+ OP_ID = "OP_INSTANCE_RENAME"
+ __slots__ = ["instance_name", "ignore_ip", "new_name"]
class OpStartupInstance(OpCode):
- """Remove an instance."""
+ """Startup an instance."""
OP_ID = "OP_INSTANCE_STARTUP"
__slots__ = ["instance_name", "force", "extra_args"]
class OpShutdownInstance(OpCode):
- """Remove an instance."""
+ """Shutdown an instance."""
OP_ID = "OP_INSTANCE_SHUTDOWN"
__slots__ = ["instance_name"]
-class OpAddMDDRBDComponent(OpCode):
- """Add a MD-DRBD component."""
- OP_ID = "OP_INSTANCE_ADD_MDDRBD"
- __slots__ = ["instance_name", "remote_node", "disk_name"]
-
-
-class OpRemoveMDDRBDComponent(OpCode):
- """Remove a MD-DRBD component."""
- OP_ID = "OP_INSTANCE_REMOVE_MDDRBD"
- __slots__ = ["instance_name", "disk_name", "disk_id"]
+class OpRebootInstance(OpCode):
+ """Reboot an instance."""
+ OP_ID = "OP_INSTANCE_REBOOT"
+ __slots__ = ["instance_name", "reboot_type", "extra_args",
+ "ignore_secondaries" ]
class OpReplaceDisks(OpCode):
- """Replace disks of an instance."""
+ """Replace the disks of an instance."""
OP_ID = "OP_INSTANCE_REPLACE_DISKS"
- __slots__ = ["instance_name", "remote_node"]
+ __slots__ = ["instance_name", "remote_node", "mode", "disks", "iallocator"]
class OpFailoverInstance(OpCode):
class OpConnectConsole(OpCode):
- """Failover an instance."""
+ """Connect to an instance's console."""
OP_ID = "OP_INSTANCE_CONSOLE"
__slots__ = ["instance_name"]
class OpActivateInstanceDisks(OpCode):
- """Remove an instance."""
+ """Activate an instance's disks."""
OP_ID = "OP_INSTANCE_ACTIVATE_DISKS"
__slots__ = ["instance_name"]
class OpDeactivateInstanceDisks(OpCode):
- """Remove an instance."""
+ """Deactivate an instance's disks."""
OP_ID = "OP_INSTANCE_DEACTIVATE_DISKS"
__slots__ = ["instance_name"]
class OpQueryInstances(OpCode):
"""Compute the list of instances."""
OP_ID = "OP_INSTANCE_QUERY"
- __slots__ = ["output_fields"]
+ __slots__ = ["output_fields", "names"]
class OpQueryInstanceData(OpCode):
__slots__ = ["instances"]
-class OpSetInstanceParms(OpCode):
+class OpSetInstanceParams(OpCode):
"""Change the parameters of an instance."""
- OP_ID = "OP_INSTANCE_SET_PARMS"
- __slots__ = ["instance_name", "mem", "vcpus", "ip", "bridge"]
+ OP_ID = "OP_INSTANCE_SET_PARAMS"
+ __slots__ = [
+ "instance_name", "mem", "vcpus", "ip", "bridge", "mac",
+ "kernel_path", "initrd_path", "hvm_boot_order",
+ ]
# OS opcodes
class OpDiagnoseOS(OpCode):
"""Compute the list of guest operating systems."""
OP_ID = "OP_OS_DIAGNOSE"
- __slots__ = []
+ __slots__ = ["output_fields", "names"]
+
# Exports opcodes
class OpQueryExports(OpCode):
OP_ID = "OP_BACKUP_QUERY"
__slots__ = ["nodes"]
+
class OpExportInstance(OpCode):
"""Export an instance."""
OP_ID = "OP_BACKUP_EXPORT"
__slots__ = ["instance_name", "target_node", "shutdown"]
+
+class OpRemoveExport(OpCode):
+ """Remove an instance's export."""
+ OP_ID = "OP_BACKUP_REMOVE"
+ __slots__ = ["instance_name"]
+
+# Tags opcodes
+class OpGetTags(OpCode):
+ """Returns the tags of the given object."""
+ OP_ID = "OP_TAGS_GET"
+ __slots__ = ["kind", "name"]
+
+
+class OpSearchTags(OpCode):
+ """Searches the tags in the cluster for a given pattern."""
+ OP_ID = "OP_TAGS_SEARCH"
+ __slots__ = ["pattern"]
+
+
+class OpAddTags(OpCode):
+ """Add a list of tags on a given object."""
+ OP_ID = "OP_TAGS_SET"
+ __slots__ = ["kind", "name", "tags"]
+
+
+class OpDelTags(OpCode):
+ """Remove a list of tags from a given object."""
+ OP_ID = "OP_TAGS_DEL"
+ __slots__ = ["kind", "name", "tags"]
+
+
+# Test opcodes
+class OpTestDelay(OpCode):
+ """Sleeps for a configured amount of time.
+
+ This is used just for debugging and testing.
+
+ Parameters:
+ - duration: the time to sleep
+ - on_master: if true, sleep on the master
+ - on_nodes: list of nodes in which to sleep
+
+ If the on_master parameter is true, it will execute a sleep on the
+ master (before any node sleep).
+
+ If the on_nodes list is not empty, it will sleep on those nodes
+ (after the sleep on the master, if that is enabled).
+
+ As an additional feature, the case of duration < 0 will be reported
+ as an execution error, so this opcode can be used as a failure
+ generator. The case of duration == 0 will not be treated specially.
+
+ """
+ OP_ID = "OP_TEST_DELAY"
+ __slots__ = ["duration", "on_master", "on_nodes"]
+
+
+class OpTestAllocator(OpCode):
+ """Allocator framework testing.
+
+ This opcode has two modes:
+ - gather and return allocator input for a given mode (allocate new
+ or replace secondary) and a given instance definition (direction
+ 'in')
+ - run a selected allocator for a given operation (as above) and
+ return the allocator output (direction 'out')
+
+ """
+ OP_ID = "OP_TEST_ALLOCATOR"
+ __slots__ = [
+ "direction", "mode", "allocator", "name",
+ "mem_size", "disks", "disk_template",
+ "os", "tags", "nics", "vcpus",
+ ]