X-Git-Url: https://code.grnet.gr/git/ganeti-local/blobdiff_plain/320e34df203a8f7dee05557f4235f6fc5c109b87..cfcea7efc17611a3d65e9b1d07d4ecae12e83f26:/lib/query.py diff --git a/lib/query.py b/lib/query.py index 1551f1d..9ffe794 100644 --- a/lib/query.py +++ b/lib/query.py @@ -1,7 +1,7 @@ # # -# Copyright (C) 2010, 2011 Google Inc. +# Copyright (C) 2010, 2011, 2012 Google Inc. # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by @@ -62,13 +62,19 @@ from ganeti import utils from ganeti import compat from ganeti import objects from ganeti import ht +from ganeti import runtime from ganeti import qlang +from ganeti import jstore from ganeti.constants import (QFT_UNKNOWN, QFT_TEXT, QFT_BOOL, QFT_NUMBER, QFT_UNIT, QFT_TIMESTAMP, QFT_OTHER, RS_NORMAL, RS_UNKNOWN, RS_NODATA, RS_UNAVAIL, RS_OFFLINE) +(NETQ_CONFIG, + NETQ_GROUP, + NETQ_STATS, + NETQ_INST) = range(300, 304) # Constants for requesting data from the caller/data provider. Each property # collected/computed separately by the data provider should have its own to @@ -92,13 +98,22 @@ from ganeti.constants import (QFT_UNKNOWN, QFT_TEXT, QFT_BOOL, QFT_NUMBER, (GQ_CONFIG, GQ_NODE, - GQ_INST) = range(200, 203) + GQ_INST, + GQ_DISKPARAMS) = range(200, 204) + +(CQ_CONFIG, + CQ_QUEUE_DRAINED, + CQ_WATCHER_PAUSE) = range(300, 303) + +(JQ_ARCHIVED, ) = range(400, 401) # Query field flags QFF_HOSTNAME = 0x01 QFF_IP_ADDRESS = 0x02 -# Next values: 0x04, 0x08, 0x10, 0x20, 0x40, 0x80, 0x100, 0x200 -QFF_ALL = (QFF_HOSTNAME | QFF_IP_ADDRESS) +QFF_JOB_ID = 0x04 +QFF_SPLIT_TIMESTAMP = 0x08 +# Next values: 0x10, 0x20, 0x40, 0x80, 0x100, 0x200 +QFF_ALL = (QFF_HOSTNAME | QFF_IP_ADDRESS | QFF_JOB_ID | QFF_SPLIT_TIMESTAMP) FIELD_NAME_RE = re.compile(r"^[a-z0-9/._]+$") TITLE_RE = re.compile(r"^[^\s]+$") @@ -260,13 +275,16 @@ class _FilterHints: if op != qlang.OP_OR: self._NeedAllNames() - def NoteUnaryOp(self, op): # pylint: disable=W0613 + def NoteUnaryOp(self, op, datakind): # pylint: disable=W0613 """Called when handling an unary operation. @type op: string @param op: Operator """ + if datakind is not None: + self._datakinds.add(datakind) + self._NeedAllNames() def NoteBinaryOp(self, op, datakind, name, value): @@ -333,6 +351,36 @@ def _PrepareRegex(pattern): raise errors.ParameterError("Invalid regex pattern (%s)" % err) +def _PrepareSplitTimestamp(value): + """Prepares a value for comparison by L{_MakeSplitTimestampComparison}. + + """ + if ht.TNumber(value): + return value + else: + return utils.MergeTime(value) + + +def _MakeSplitTimestampComparison(fn): + """Compares split timestamp values after converting to float. + + """ + return lambda lhs, rhs: fn(utils.MergeTime(lhs), rhs) + + +def _MakeComparisonChecks(fn): + """Prepares flag-specific comparisons using a comparison function. + + """ + return [ + (QFF_SPLIT_TIMESTAMP, _MakeSplitTimestampComparison(fn), + _PrepareSplitTimestamp), + (QFF_JOB_ID, lambda lhs, rhs: fn(jstore.ParseJobId(lhs), rhs), + jstore.ParseJobId), + (None, fn, None), + ] + + class _FilterCompilerHelper: """Converts a query filter to a callable usable for filtering. @@ -351,7 +399,7 @@ class _FilterCompilerHelper: List of tuples containing flags and a callable receiving the left- and right-hand side of the operator. The flags are an OR-ed value of C{QFF_*} - (e.g. L{QFF_HOSTNAME}). + (e.g. L{QFF_HOSTNAME} or L{QFF_SPLIT_TIMESTAMP}). Order matters. The first item with flags will be used. Flags are checked using binary AND. @@ -362,6 +410,8 @@ class _FilterCompilerHelper: lambda lhs, rhs: utils.MatchNameComponent(rhs, [lhs], case_sensitive=False), None), + (QFF_SPLIT_TIMESTAMP, _MakeSplitTimestampComparison(operator.eq), + _PrepareSplitTimestamp), (None, operator.eq, None), ] @@ -391,6 +441,10 @@ class _FilterCompilerHelper: qlang.OP_NOT_EQUAL: (_OPTYPE_BINARY, [(flags, compat.partial(_WrapNot, fn), valprepfn) for (flags, fn, valprepfn) in _EQUALITY_CHECKS]), + qlang.OP_LT: (_OPTYPE_BINARY, _MakeComparisonChecks(operator.lt)), + qlang.OP_LE: (_OPTYPE_BINARY, _MakeComparisonChecks(operator.le)), + qlang.OP_GT: (_OPTYPE_BINARY, _MakeComparisonChecks(operator.gt)), + qlang.OP_GE: (_OPTYPE_BINARY, _MakeComparisonChecks(operator.ge)), qlang.OP_REGEXP: (_OPTYPE_BINARY, [ (None, lambda lhs, rhs: rhs.search(lhs), _PrepareRegex), ]), @@ -512,19 +566,22 @@ class _FilterCompilerHelper: """ assert op_fn is None - if hints_fn: - hints_fn(op) - if len(operands) != 1: raise errors.ParameterError("Unary operator '%s' expects exactly one" " operand" % op) if op == qlang.OP_TRUE: - (_, _, _, retrieval_fn) = self._LookupField(operands[0]) + (_, datakind, _, retrieval_fn) = self._LookupField(operands[0]) + + if hints_fn: + hints_fn(op, datakind) op_fn = operator.truth arg = retrieval_fn elif op == qlang.OP_NOT: + if hints_fn: + hints_fn(op, None) + op_fn = operator.not_ arg = self._Compile(operands[0], level + 1) else: @@ -695,6 +752,7 @@ class Query: (status, name) = _ProcessResult(self._name_fn(ctx, item)) assert status == constants.RS_NORMAL # TODO: Are there cases where we wouldn't want to use NiceSort? + # Answer: if the name field is non-string... result.append((utils.NiceSortKey(name), idx, row)) else: result.append(row) @@ -763,7 +821,7 @@ def _VerifyResultRow(fields, row): elif value is not None: errs.append("abnormal field %s has a non-None value" % fdef.name) assert not errs, ("Failed validation: %s in row %s" % - (utils.CommaJoin(errors), row)) + (utils.CommaJoin(errs), row)) def _FieldDictKey((fdef, _, flags, fn)): @@ -868,6 +926,20 @@ def _MakeField(name, title, kind, doc): doc=doc) +def _StaticValueInner(value, ctx, _): # pylint: disable=W0613 + """Returns a static value. + + """ + return value + + +def _StaticValue(value): + """Prepares a function to return a static value. + + """ + return compat.partial(_StaticValueInner, value) + + def _GetNodeRole(node, master_name): """Determine node role. @@ -899,6 +971,36 @@ def _GetItemAttr(attr): return lambda _, item: getter(item) +def _GetNDParam(name): + """Return a field function to return an ND parameter out of the context. + + """ + def _helper(ctx, _): + if ctx.ndparams is None: + return _FS_UNAVAIL + else: + return ctx.ndparams.get(name, None) + return _helper + + +def _BuildNDFields(is_group): + """Builds all the ndparam fields. + + @param is_group: whether this is called at group or node level + + """ + if is_group: + field_kind = GQ_CONFIG + else: + field_kind = NQ_GROUP + return [(_MakeField("ndp/%s" % name, + constants.NDS_PARAMETER_TITLES.get(name, + "ndp/%s" % name), + _VTToQFT[kind], "The \"%s\" node parameter" % name), + field_kind, 0, _GetNDParam(name)) + for name, kind in constants.NDS_PARAMETER_TYPES.items()] + + def _ConvWrapInner(convert, fn, ctx, item): """Wrapper for converting values. @@ -982,6 +1084,7 @@ class NodeQueryData: # Used for individual rows self.curlive_data = None + self.ndparams = None def __iter__(self): """Iterate over all nodes. @@ -991,6 +1094,11 @@ class NodeQueryData: """ for node in self.nodes: + group = self.groups.get(node.group, None) + if group is None: + self.ndparams = None + else: + self.ndparams = self.cluster.FillND(node, group) if self.live_data: self.curlive_data = self.live_data.get(node.name, None) else: @@ -1140,6 +1248,32 @@ def _GetLiveNodeField(field, kind, ctx, node): return _FS_UNAVAIL +def _GetNodeHvState(_, node): + """Converts node's hypervisor state for query result. + + """ + hv_state = node.hv_state + + if hv_state is None: + return _FS_UNAVAIL + + return dict((name, value.ToDict()) for (name, value) in hv_state.items()) + + +def _GetNodeDiskState(_, node): + """Converts node's disk state for query result. + + """ + disk_state = node.disk_state + + if disk_state is None: + return _FS_UNAVAIL + + return dict((disk_kind, dict((name, value.ToDict()) + for (name, value) in kind_state.items())) + for (disk_kind, kind_state) in disk_state.items()) + + def _BuildNodeFields(): """Builds list of fields for node queries. @@ -1166,14 +1300,20 @@ def _BuildNodeFields(): (_MakeField("custom_ndparams", "CustomNodeParameters", QFT_OTHER, "Custom node parameters"), NQ_GROUP, 0, _GetItemAttr("ndparams")), + (_MakeField("hv_state", "HypervisorState", QFT_OTHER, "Hypervisor state"), + NQ_CONFIG, 0, _GetNodeHvState), + (_MakeField("disk_state", "DiskState", QFT_OTHER, "Disk state"), + NQ_CONFIG, 0, _GetNodeDiskState), ] + fields.extend(_BuildNDFields(False)) + # Node role role_values = (constants.NR_MASTER, constants.NR_MCANDIDATE, constants.NR_REGULAR, constants.NR_DRAINED, constants.NR_OFFLINE) role_doc = ("Node role; \"%s\" for master, \"%s\" for master candidate," - " \"%s\" for regular, \"%s\" for a drained, \"%s\" for offline" % + " \"%s\" for regular, \"%s\" for drained, \"%s\" for offline" % role_values) fields.append((_MakeField("role", "Role", QFT_TEXT, role_doc), NQ_CONFIG, 0, lambda ctx, node: _GetNodeRole(node, ctx.master_name))) @@ -1203,15 +1343,13 @@ def _BuildNodeFields(): # Add simple fields fields.extend([ (_MakeField(name, title, kind, doc), NQ_CONFIG, flags, _GetItemAttr(name)) - for (name, (title, kind, flags, doc)) in _NODE_SIMPLE_FIELDS.items() - ]) + for (name, (title, kind, flags, doc)) in _NODE_SIMPLE_FIELDS.items()]) # Add fields requiring live data fields.extend([ (_MakeField(name, title, kind, doc), NQ_LIVE, 0, compat.partial(_GetLiveNodeField, nfield, kind)) - for (name, (title, kind, nfield, doc)) in _NODE_LIVE_FIELDS.items() - ]) + for (name, (title, kind, nfield, doc)) in _NODE_LIVE_FIELDS.items()]) # Add timestamps fields.extend(_GetItemTimestampFields(NQ_CONFIG)) @@ -1348,15 +1486,17 @@ def _GetInstStatus(ctx, inst): if bool(ctx.live_data.get(inst.name)): if inst.name in ctx.wrongnode_inst: return constants.INSTST_WRONGNODE - elif inst.admin_up: + elif inst.admin_state == constants.ADMINST_UP: return constants.INSTST_RUNNING else: return constants.INSTST_ERRORUP - if inst.admin_up: + if inst.admin_state == constants.ADMINST_UP: return constants.INSTST_ERRORDOWN + elif inst.admin_state == constants.ADMINST_DOWN: + return constants.INSTST_ADMINDOWN - return constants.INSTST_ADMINDOWN + return constants.INSTST_ADMINOFFLINE def _GetInstDiskSize(index): @@ -1408,6 +1548,20 @@ def _GetInstNic(index, cb): return fn +def _GetInstNicNetwork(ctx, _, nic): # pylint: disable=W0613 + """Get a NIC's Network. + + @type ctx: L{InstanceQueryData} + @type nic: L{objects.NIC} + @param nic: NIC object + + """ + if nic.network is None: + return _FS_UNAVAIL + else: + return nic.network + + def _GetInstNicIp(ctx, _, nic): # pylint: disable=W0613 """Get a NIC's IP address. @@ -1519,6 +1673,9 @@ def _GetInstanceNetworkFields(): (_MakeField("nic.bridges", "NIC_bridges", QFT_OTHER, "List containing each network interface's bridge"), IQ_CONFIG, 0, _GetInstAllNicBridges), + (_MakeField("nic.networks", "NIC_networks", QFT_OTHER, + "List containing each interface's network"), IQ_CONFIG, 0, + lambda ctx, inst: [nic.network for nic in inst.nics]), ] # NICs by number @@ -1540,6 +1697,9 @@ def _GetInstanceNetworkFields(): (_MakeField("nic.bridge/%s" % i, "NicBridge/%s" % i, QFT_TEXT, "Bridge of %s network interface" % numtext), IQ_CONFIG, 0, _GetInstNic(i, _GetInstNicBridge)), + (_MakeField("nic.network/%s" % i, "NicNetwork/%s" % i, QFT_TEXT, + "Network of %s network interface" % numtext), + IQ_CONFIG, 0, _GetInstNic(i, _GetInstNicNetwork)), ]) aliases = [ @@ -1549,6 +1709,7 @@ def _GetInstanceNetworkFields(): ("bridge", "nic.bridge/0"), ("nic_mode", "nic.mode/0"), ("nic_link", "nic.link/0"), + ("nic_network", "nic.network/0"), ] return (fields, aliases) @@ -1609,8 +1770,7 @@ def _GetInstanceDiskFields(): (_MakeField("disk.size/%s" % i, "Disk/%s" % i, QFT_UNIT, "Disk size of %s disk" % utils.FormatOrdinal(i + 1)), IQ_CONFIG, 0, _GetInstDiskSize(i)) - for i in range(constants.MAX_DISKS) - ]) + for i in range(constants.MAX_DISKS)]) return fields @@ -1621,25 +1781,6 @@ def _GetInstanceParameterFields(): @return: List of field definitions used as input for L{_PrepareFieldList} """ - # TODO: Consider moving titles closer to constants - be_title = { - constants.BE_AUTO_BALANCE: "Auto_balance", - constants.BE_MEMORY: "ConfigMemory", - constants.BE_VCPUS: "ConfigVCPUs", - } - - hv_title = { - constants.HV_ACPI: "ACPI", - constants.HV_BOOT_ORDER: "Boot_order", - constants.HV_CDROM_IMAGE_PATH: "CDROM_image_path", - constants.HV_DISK_TYPE: "Disk_type", - constants.HV_INITRD_PATH: "Initrd_path", - constants.HV_KERNEL_PATH: "Kernel_path", - constants.HV_NIC_TYPE: "NIC_type", - constants.HV_PAE: "PAE", - constants.HV_VNC_BIND_ADDRESS: "VNC_bind_address", - } - fields = [ # Filled parameters (_MakeField("hvparams", "HypervisorParameters", QFT_OTHER, @@ -1672,23 +1813,23 @@ def _GetInstanceParameterFields(): return lambda ctx, _: ctx.inst_hvparams.get(name, _FS_UNAVAIL) fields.extend([ - (_MakeField("hv/%s" % name, hv_title.get(name, "hv/%s" % name), + (_MakeField("hv/%s" % name, + constants.HVS_PARAMETER_TITLES.get(name, "hv/%s" % name), _VTToQFT[kind], "The \"%s\" hypervisor parameter" % name), IQ_CONFIG, 0, _GetInstHvParam(name)) for name, kind in constants.HVS_PARAMETER_TYPES.items() - if name not in constants.HVC_GLOBALS - ]) + if name not in constants.HVC_GLOBALS]) # BE params def _GetInstBeParam(name): return lambda ctx, _: ctx.inst_beparams.get(name, None) fields.extend([ - (_MakeField("be/%s" % name, be_title.get(name, "be/%s" % name), + (_MakeField("be/%s" % name, + constants.BES_PARAMETER_TITLES.get(name, "be/%s" % name), _VTToQFT[kind], "The \"%s\" backend parameter" % name), IQ_CONFIG, 0, _GetInstBeParam(name)) - for name, kind in constants.BES_PARAMETER_TYPES.items() - ]) + for name, kind in constants.BES_PARAMETER_TYPES.items()]) return fields @@ -1775,10 +1916,12 @@ def _BuildInstanceFields(): IQ_NODES, 0, lambda ctx, inst: map(compat.partial(_GetInstNodeGroup, ctx, None), inst.secondary_nodes)), - (_MakeField("admin_state", "Autostart", QFT_BOOL, - "Desired state of instance (if set, the instance should be" - " up)"), - IQ_CONFIG, 0, _GetItemAttr("admin_up")), + (_MakeField("admin_state", "InstanceState", QFT_TEXT, + "Desired state of instance"), + IQ_CONFIG, 0, _GetItemAttr("admin_state")), + (_MakeField("admin_up", "Autostart", QFT_BOOL, + "Desired state of instance"), + IQ_CONFIG, 0, lambda ctx, inst: inst.admin_state == constants.ADMINST_UP), (_MakeField("tags", "Tags", QFT_OTHER, "Tags"), IQ_CONFIG, 0, lambda ctx, inst: list(inst.GetTags())), (_MakeField("console", "Console", QFT_OTHER, @@ -1789,8 +1932,7 @@ def _BuildInstanceFields(): # Add simple fields fields.extend([ (_MakeField(name, title, kind, doc), IQ_CONFIG, flags, _GetItemAttr(name)) - for (name, (title, kind, flags, doc)) in _INST_SIMPLE_FIELDS.items() - ]) + for (name, (title, kind, flags, doc)) in _INST_SIMPLE_FIELDS.items()]) # Fields requiring talking to the node fields.extend([ @@ -1808,15 +1950,16 @@ def _BuildInstanceFields(): status_values = (constants.INSTST_RUNNING, constants.INSTST_ADMINDOWN, constants.INSTST_WRONGNODE, constants.INSTST_ERRORUP, constants.INSTST_ERRORDOWN, constants.INSTST_NODEDOWN, - constants.INSTST_NODEOFFLINE) + constants.INSTST_NODEOFFLINE, constants.INSTST_ADMINOFFLINE) status_doc = ("Instance status; \"%s\" if instance is set to be running" " and actually is, \"%s\" if instance is stopped and" " is not running, \"%s\" if instance running, but not on its" " designated primary node, \"%s\" if instance should be" " stopped, but is actually running, \"%s\" if instance should" " run, but doesn't, \"%s\" if instance's primary node is down," - " \"%s\" if instance's primary node is marked offline" % - status_values) + " \"%s\" if instance's primary node is marked offline," + " \"%s\" if instance is offline and does not use dynamic" + " resources" % status_values) fields.append((_MakeField("status", "Status", QFT_TEXT, status_doc), IQ_LIVE, 0, _GetInstStatus)) assert set(status_values) == constants.INSTST_ALL, \ @@ -1831,6 +1974,7 @@ def _BuildInstanceFields(): aliases = [ ("vcpus", "be/vcpus"), + ("be/memory", "be/maxmem"), ("sda_size", "disk.size/0"), ("sdb_size", "disk.size/1"), ] + network_aliases @@ -1904,25 +2048,46 @@ class GroupQueryData: """Data container for node group data queries. """ - def __init__(self, groups, group_to_nodes, group_to_instances): + def __init__(self, cluster, groups, group_to_nodes, group_to_instances, + want_diskparams): """Initializes this class. + @param cluster: Cluster object @param groups: List of node group objects @type group_to_nodes: dict; group UUID as key @param group_to_nodes: Per-group list of nodes @type group_to_instances: dict; group UUID as key @param group_to_instances: Per-group list of (primary) instances + @type want_diskparams: bool + @param want_diskparams: Whether diskparamters should be calculated """ self.groups = groups self.group_to_nodes = group_to_nodes self.group_to_instances = group_to_instances + self.cluster = cluster + self.want_diskparams = want_diskparams + + # Used for individual rows + self.group_ipolicy = None + self.ndparams = None + self.group_dp = None def __iter__(self): """Iterate over all node groups. + This function has side-effects and only one instance of the resulting + generator should be used at a time. + """ - return iter(self.groups) + for group in self.groups: + self.group_ipolicy = self.cluster.SimpleFillIPolicy(group.ipolicy) + self.ndparams = self.cluster.SimpleFillND(group.ndparams) + if self.want_diskparams: + self.group_dp = self.cluster.SimpleFillDP(group.diskparams) + else: + self.group_dp = None + yield group _GROUP_SIMPLE_FIELDS = { @@ -1930,7 +2095,6 @@ _GROUP_SIMPLE_FIELDS = { "name": ("Group", QFT_TEXT, "Group name"), "serial_no": ("SerialNo", QFT_NUMBER, _SERIAL_NO_DOC % "Group"), "uuid": ("UUID", QFT_TEXT, "Group UUID"), - "ndparams": ("NDParams", QFT_OTHER, "Node parameters"), } @@ -1974,8 +2138,29 @@ def _BuildGroupFields(): fields.extend([ (_MakeField("tags", "Tags", QFT_OTHER, "Tags"), GQ_CONFIG, 0, lambda ctx, group: list(group.GetTags())), + (_MakeField("ipolicy", "InstancePolicy", QFT_OTHER, + "Instance policy limitations (merged)"), + GQ_CONFIG, 0, lambda ctx, _: ctx.group_ipolicy), + (_MakeField("custom_ipolicy", "CustomInstancePolicy", QFT_OTHER, + "Custom instance policy limitations"), + GQ_CONFIG, 0, _GetItemAttr("ipolicy")), + (_MakeField("custom_ndparams", "CustomNDParams", QFT_OTHER, + "Custom node parameters"), + GQ_CONFIG, 0, _GetItemAttr("ndparams")), + (_MakeField("ndparams", "NDParams", QFT_OTHER, + "Node parameters"), + GQ_CONFIG, 0, lambda ctx, _: ctx.ndparams), + (_MakeField("diskparams", "DiskParameters", QFT_OTHER, + "Disk parameters (merged)"), + GQ_DISKPARAMS, 0, lambda ctx, _: ctx.group_dp), + (_MakeField("custom_diskparams", "CustomDiskParameters", QFT_OTHER, + "Custom disk parameters"), + GQ_CONFIG, 0, _GetItemAttr("diskparams")), ]) + # ND parameters + fields.extend(_BuildNDFields(True)) + fields.extend(_GetItemTimestampFields(GQ_CONFIG)) return _PrepareFieldList(fields, []) @@ -2028,6 +2213,366 @@ def _BuildOsFields(): return _PrepareFieldList(fields, []) +def _JobUnavailInner(fn, ctx, (job_id, job)): # pylint: disable=W0613 + """Return L{_FS_UNAVAIL} if job is None. + + When listing specifc jobs (e.g. "gnt-job list 1 2 3"), a job may not be + found, in which case this function converts it to L{_FS_UNAVAIL}. + + """ + if job is None: + return _FS_UNAVAIL + else: + return fn(job) + + +def _JobUnavail(inner): + """Wrapper for L{_JobUnavailInner}. + + """ + return compat.partial(_JobUnavailInner, inner) + + +def _PerJobOpInner(fn, job): + """Executes a function per opcode in a job. + + """ + return map(fn, job.ops) + + +def _PerJobOp(fn): + """Wrapper for L{_PerJobOpInner}. + + """ + return _JobUnavail(compat.partial(_PerJobOpInner, fn)) + + +def _JobTimestampInner(fn, job): + """Converts unavailable timestamp to L{_FS_UNAVAIL}. + + """ + timestamp = fn(job) + + if timestamp is None: + return _FS_UNAVAIL + else: + return timestamp + + +def _JobTimestamp(fn): + """Wrapper for L{_JobTimestampInner}. + + """ + return _JobUnavail(compat.partial(_JobTimestampInner, fn)) + + +def _BuildJobFields(): + """Builds list of fields for job queries. + + """ + fields = [ + (_MakeField("id", "ID", QFT_NUMBER, "Job ID"), + None, QFF_JOB_ID, lambda _, (job_id, job): job_id), + (_MakeField("status", "Status", QFT_TEXT, "Job status"), + None, 0, _JobUnavail(lambda job: job.CalcStatus())), + (_MakeField("priority", "Priority", QFT_NUMBER, + ("Current job priority (%s to %s)" % + (constants.OP_PRIO_LOWEST, constants.OP_PRIO_HIGHEST))), + None, 0, _JobUnavail(lambda job: job.CalcPriority())), + (_MakeField("archived", "Archived", QFT_BOOL, "Whether job is archived"), + JQ_ARCHIVED, 0, lambda _, (job_id, job): job.archived), + (_MakeField("ops", "OpCodes", QFT_OTHER, "List of all opcodes"), + None, 0, _PerJobOp(lambda op: op.input.__getstate__())), + (_MakeField("opresult", "OpCode_result", QFT_OTHER, + "List of opcodes results"), + None, 0, _PerJobOp(operator.attrgetter("result"))), + (_MakeField("opstatus", "OpCode_status", QFT_OTHER, + "List of opcodes status"), + None, 0, _PerJobOp(operator.attrgetter("status"))), + (_MakeField("oplog", "OpCode_log", QFT_OTHER, + "List of opcode output logs"), + None, 0, _PerJobOp(operator.attrgetter("log"))), + (_MakeField("opstart", "OpCode_start", QFT_OTHER, + "List of opcode start timestamps (before acquiring locks)"), + None, 0, _PerJobOp(operator.attrgetter("start_timestamp"))), + (_MakeField("opexec", "OpCode_exec", QFT_OTHER, + "List of opcode execution start timestamps (after acquiring" + " locks)"), + None, 0, _PerJobOp(operator.attrgetter("exec_timestamp"))), + (_MakeField("opend", "OpCode_end", QFT_OTHER, + "List of opcode execution end timestamps"), + None, 0, _PerJobOp(operator.attrgetter("end_timestamp"))), + (_MakeField("oppriority", "OpCode_prio", QFT_OTHER, + "List of opcode priorities"), + None, 0, _PerJobOp(operator.attrgetter("priority"))), + (_MakeField("summary", "Summary", QFT_OTHER, + "List of per-opcode summaries"), + None, 0, _PerJobOp(lambda op: op.input.Summary())), + ] + + # Timestamp fields + for (name, attr, title, desc) in [ + ("received_ts", "received_timestamp", "Received", + "Timestamp of when job was received"), + ("start_ts", "start_timestamp", "Start", "Timestamp of job start"), + ("end_ts", "end_timestamp", "End", "Timestamp of job end"), + ]: + getter = operator.attrgetter(attr) + fields.extend([ + (_MakeField(name, title, QFT_OTHER, + "%s (tuple containing seconds and microseconds)" % desc), + None, QFF_SPLIT_TIMESTAMP, _JobTimestamp(getter)), + ]) + + return _PrepareFieldList(fields, []) + + +def _GetExportName(_, (node_name, expname)): # pylint: disable=W0613 + """Returns an export name if available. + + """ + if expname is None: + return _FS_UNAVAIL + else: + return expname + + +def _BuildExportFields(): + """Builds list of fields for exports. + + """ + fields = [ + (_MakeField("node", "Node", QFT_TEXT, "Node name"), + None, QFF_HOSTNAME, lambda _, (node_name, expname): node_name), + (_MakeField("export", "Export", QFT_TEXT, "Export name"), + None, 0, _GetExportName), + ] + + return _PrepareFieldList(fields, []) + + +_CLUSTER_VERSION_FIELDS = { + "software_version": ("SoftwareVersion", QFT_TEXT, constants.RELEASE_VERSION, + "Software version"), + "protocol_version": ("ProtocolVersion", QFT_NUMBER, + constants.PROTOCOL_VERSION, + "RPC protocol version"), + "config_version": ("ConfigVersion", QFT_NUMBER, constants.CONFIG_VERSION, + "Configuration format version"), + "os_api_version": ("OsApiVersion", QFT_NUMBER, max(constants.OS_API_VERSIONS), + "API version for OS template scripts"), + "export_version": ("ExportVersion", QFT_NUMBER, constants.EXPORT_VERSION, + "Import/export file format version"), + } + + +_CLUSTER_SIMPLE_FIELDS = { + "cluster_name": ("Name", QFT_TEXT, QFF_HOSTNAME, "Cluster name"), + "master_node": ("Master", QFT_TEXT, QFF_HOSTNAME, "Master node name"), + "volume_group_name": ("VgName", QFT_TEXT, 0, "LVM volume group name"), + } + + +class ClusterQueryData: + def __init__(self, cluster, drain_flag, watcher_pause): + """Initializes this class. + + @type cluster: L{objects.Cluster} + @param cluster: Instance of cluster object + @type drain_flag: bool + @param drain_flag: Whether job queue is drained + @type watcher_pause: number + @param watcher_pause: Until when watcher is paused (Unix timestamp) + + """ + self._cluster = cluster + self.drain_flag = drain_flag + self.watcher_pause = watcher_pause + + def __iter__(self): + return iter([self._cluster]) + + +def _ClusterWatcherPause(ctx, _): + """Returns until when watcher is paused (if available). + + """ + if ctx.watcher_pause is None: + return _FS_UNAVAIL + else: + return ctx.watcher_pause + + +def _BuildClusterFields(): + """Builds list of fields for cluster information. + + """ + fields = [ + (_MakeField("tags", "Tags", QFT_OTHER, "Tags"), CQ_CONFIG, 0, + lambda ctx, cluster: list(cluster.GetTags())), + (_MakeField("architecture", "ArchInfo", QFT_OTHER, + "Architecture information"), None, 0, + lambda ctx, _: runtime.GetArchInfo()), + (_MakeField("drain_flag", "QueueDrained", QFT_BOOL, + "Flag whether job queue is drained"), CQ_QUEUE_DRAINED, 0, + lambda ctx, _: ctx.drain_flag), + (_MakeField("watcher_pause", "WatcherPause", QFT_TIMESTAMP, + "Until when watcher is paused"), CQ_WATCHER_PAUSE, 0, + _ClusterWatcherPause), + ] + + # Simple fields + fields.extend([ + (_MakeField(name, title, kind, doc), CQ_CONFIG, flags, _GetItemAttr(name)) + for (name, (title, kind, flags, doc)) in _CLUSTER_SIMPLE_FIELDS.items() + ],) + + # Version fields + fields.extend([ + (_MakeField(name, title, kind, doc), None, 0, _StaticValue(value)) + for (name, (title, kind, value, doc)) in _CLUSTER_VERSION_FIELDS.items()]) + + # Add timestamps + fields.extend(_GetItemTimestampFields(CQ_CONFIG)) + + return _PrepareFieldList(fields, [ + ("name", "cluster_name")]) + + +class NetworkQueryData: + """Data container for network data queries. + + """ + def __init__(self, networks, network_to_groups, + network_to_instances, stats): + """Initializes this class. + + @param networks: List of network objects + @type network_to_groups: dict; network UUID as key + @param network_to_groups: Per-network list of groups + @type network_to_instances: dict; network UUID as key + @param network_to_instances: Per-network list of instances + @type stats: dict; network UUID as key + @param stats: Per-network usage statistics + + """ + self.networks = networks + self.network_to_groups = network_to_groups + self.network_to_instances = network_to_instances + self.stats = stats + + def __iter__(self): + """Iterate over all networks. + + """ + for net in self.networks: + if self.stats: + self.curstats = self.stats.get(net.uuid, None) + else: + self.curstats = None + yield net + + +_NETWORK_SIMPLE_FIELDS = { + "name": ("Network", QFT_TEXT, 0, "The network"), + "network": ("Subnet", QFT_TEXT, 0, "The subnet"), + "gateway": ("Gateway", QFT_OTHER, 0, "The gateway"), + "network6": ("IPv6Subnet", QFT_OTHER, 0, "The ipv6 subnet"), + "gateway6": ("IPv6Gateway", QFT_OTHER, 0, "The ipv6 gateway"), + "mac_prefix": ("MacPrefix", QFT_OTHER, 0, "The mac prefix"), + "network_type": ("NetworkType", QFT_OTHER, 0, "The network type"), + } + + +_NETWORK_STATS_FIELDS = { + "free_count": ("FreeCount", QFT_NUMBER, 0, "How many addresses are free"), + "reserved_count": ("ReservedCount", QFT_NUMBER, 0, + "How many addresses are reserved"), + "map": ("Map", QFT_TEXT, 0, "The actual mapping"), + "external_reservations": ("ExternalReservations", QFT_TEXT, 0, + "The external reservations"), + } + + +def _GetNetworkStatsField(field, kind, ctx): + """Gets the value of a "stats" field from L{NetworkQueryData}. + + @param field: Field name + @param kind: Data kind, one of L{constants.QFT_ALL} + @type ctx: L{NetworkQueryData} + + """ + + try: + value = ctx.curstats[field] + except KeyError: + return _FS_UNAVAIL + + if kind == QFT_TEXT: + return value + + assert kind in (QFT_NUMBER, QFT_UNIT) + + # Try to convert into number + try: + return int(value) + except (ValueError, TypeError): + logging.exception("Failed to convert network field '%s' (value %r) to int", + field, value) + return _FS_UNAVAIL + + +def _BuildNetworkFields(): + """Builds list of fields for network queries. + + """ + fields = [ + (_MakeField("tags", "Tags", QFT_OTHER, "Tags"), IQ_CONFIG, 0, + lambda ctx, inst: list(inst.GetTags())), + ] + + # Add simple fields + fields.extend([ + (_MakeField(name, title, kind, doc), + NETQ_CONFIG, 0, _GetItemAttr(name)) + for (name, (title, kind, _, doc)) in _NETWORK_SIMPLE_FIELDS.items()]) + + def _GetLength(getter): + return lambda ctx, network: len(getter(ctx)[network.uuid]) + + def _GetSortedList(getter): + return lambda ctx, network: utils.NiceSort(getter(ctx)[network.uuid]) + + network_to_groups = operator.attrgetter("network_to_groups") + network_to_instances = operator.attrgetter("network_to_instances") + + # Add fields for node groups + fields.extend([ + (_MakeField("group_cnt", "NodeGroups", QFT_NUMBER, "Number of nodegroups"), + NETQ_GROUP, 0, _GetLength(network_to_groups)), + (_MakeField("group_list", "GroupList", QFT_OTHER, "List of nodegroups"), + NETQ_GROUP, 0, _GetSortedList(network_to_groups)), + ]) + + # Add fields for instances + fields.extend([ + (_MakeField("inst_cnt", "Instances", QFT_NUMBER, "Number of instances"), + NETQ_INST, 0, _GetLength(network_to_instances)), + (_MakeField("inst_list", "InstanceList", QFT_OTHER, "List of instances"), + NETQ_INST, 0, _GetSortedList(network_to_instances)), + ]) + + # Add fields for usage statistics + fields.extend([ + (_MakeField(name, title, kind, doc), NETQ_STATS, 0, + compat.partial(_GetNetworkStatsField, name, kind)) + for (name, (title, kind, _, doc)) in _NETWORK_STATS_FIELDS.items()]) + + return _PrepareFieldList(fields, []) + +#: Fields for cluster information +CLUSTER_FIELDS = _BuildClusterFields() + #: Fields available for node queries NODE_FIELDS = _BuildNodeFields() @@ -2043,13 +2588,26 @@ GROUP_FIELDS = _BuildGroupFields() #: Fields available for operating system queries OS_FIELDS = _BuildOsFields() +#: Fields available for job queries +JOB_FIELDS = _BuildJobFields() + +#: Fields available for exports +EXPORT_FIELDS = _BuildExportFields() + +#: Fields available for network queries +NETWORK_FIELDS = _BuildNetworkFields() + #: All available resources ALL_FIELDS = { + constants.QR_CLUSTER: CLUSTER_FIELDS, constants.QR_INSTANCE: INSTANCE_FIELDS, constants.QR_NODE: NODE_FIELDS, constants.QR_LOCK: LOCK_FIELDS, constants.QR_GROUP: GROUP_FIELDS, constants.QR_OS: OS_FIELDS, + constants.QR_JOB: JOB_FIELDS, + constants.QR_EXPORT: EXPORT_FIELDS, + constants.QR_NETWORK: NETWORK_FIELDS, } #: All available field lists