#
#
-# Copyright (C) 2010 Google Inc.
+# Copyright (C) 2010, 2011 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
# 02110-1301, USA.
-"""Module for query operations"""
+"""Module for query operations
+
+How it works:
+
+ - Add field definitions
+ - See how L{NODE_FIELDS} is built
+ - Each field gets:
+ - Query field definition (L{objects.QueryFieldDefinition}, use
+ L{_MakeField} for creating), containing:
+ - Name, must be lowercase and match L{FIELD_NAME_RE}
+ - Title for tables, must not contain whitespace and match
+ L{TITLE_RE}
+ - Value data type, e.g. L{constants.QFT_NUMBER}
+ - Data request type, see e.g. C{NQ_*}
+ - A retrieval function, see L{Query.__init__} for description
+ - Pass list of fields through L{_PrepareFieldList} for preparation and
+ checks
+ - Instantiate L{Query} with prepared field list definition and selected fields
+ - Call L{Query.RequestedData} to determine what data to collect/compute
+ - Call L{Query.Query} or L{Query.OldStyleQuery} with collected data and use
+ result
+ - Data container must support iteration using C{__iter__}
+ - Items are passed to retrieval functions and can have any format
+ - Call L{Query.GetFields} to get list of definitions for selected fields
+
+@attention: Retrieval functions must be idempotent. They can be called multiple
+ times, in any order and any number of times. This is important to keep in
+ mind for implementing filters in the future.
+
+"""
import logging
import operator
from ganeti import objects
from ganeti import ht
+from ganeti.constants import (QFT_UNKNOWN, QFT_TEXT, QFT_BOOL, QFT_NUMBER,
+ QFT_UNIT, QFT_TIMESTAMP, QFT_OTHER,
+ QRFS_NORMAL, QRFS_UNKNOWN, QRFS_NODATA,
+ QRFS_UNAVAIL, QRFS_OFFLINE)
+
+
+# Constants for requesting data from the caller/data provider. Each property
+# collected/computed separately by the data provider should have its own to
+# only collect the requested data and not more.
(NQ_CONFIG,
NQ_INST,
NQ_LIVE,
- NQ_GROUP) = range(1, 5)
+ NQ_GROUP,
+ NQ_OOB) = range(1, 6)
(IQ_CONFIG,
IQ_LIVE,
LQ_OWNER,
LQ_PENDING) = range(10, 13)
+(GQ_CONFIG,
+ GQ_NODE,
+ GQ_INST) = range(200, 203)
+
+
FIELD_NAME_RE = re.compile(r"^[a-z0-9/._]+$")
TITLE_RE = re.compile(r"^[^\s]+$")
#: Verification function for each field type
_VERIFY_FN = {
- constants.QFT_UNKNOWN: ht.TNone,
- constants.QFT_TEXT: ht.TString,
- constants.QFT_BOOL: ht.TBool,
- constants.QFT_NUMBER: ht.TInt,
- constants.QFT_UNIT: ht.TInt,
- constants.QFT_TIMESTAMP: ht.TOr(ht.TInt, ht.TFloat),
- constants.QFT_OTHER: lambda _: True,
+ QFT_UNKNOWN: ht.TNone,
+ QFT_TEXT: ht.TString,
+ QFT_BOOL: ht.TBool,
+ QFT_NUMBER: ht.TInt,
+ QFT_UNIT: ht.TInt,
+ QFT_TIMESTAMP: ht.TOr(ht.TInt, ht.TFloat),
+ QFT_OTHER: lambda _: True,
}
+# Unique objects for special field statuses
+_FS_UNKNOWN = object()
+_FS_NODATA = object()
+_FS_UNAVAIL = object()
+_FS_OFFLINE = object()
+
def _GetUnknownField(ctx, item): # pylint: disable-msg=W0613
"""Gets the contents of an unknown field.
"""
- return (constants.QRFS_UNKNOWN, None)
+ return _FS_UNKNOWN
def _GetQueryFields(fielddefs, selected):
try:
fdef = fielddefs[name]
except KeyError:
- fdef = (_MakeField(name, name, constants.QFT_UNKNOWN),
- None, _GetUnknownField)
+ fdef = (_MakeField(name, name, QFT_UNKNOWN), None, _GetUnknownField)
assert len(fdef) == 3
support iteration using C{__iter__}
"""
- result = [[fn(ctx, item) for (_, _, fn) in self._fields]
+ result = [[_ProcessResult(fn(ctx, item)) for (_, _, fn) in self._fields]
for item in ctx]
# Verify result
if __debug__:
- for (idx, row) in enumerate(result):
- assert _VerifyResultRow(self._fields, row), \
- ("Inconsistent result for fields %s in row %s: %r" %
- (GetAllFields(self._fields), idx, row))
+ for row in result:
+ _VerifyResultRow(self._fields, row)
return result
"""
unknown = set(fdef.name
- for (fdef, _, _) in self._fields
- if fdef.kind == constants.QFT_UNKNOWN)
+ for (fdef, _, _) in self._fields if fdef.kind == QFT_UNKNOWN)
if unknown:
raise errors.OpPrereqError("Unknown output fields selected: %s" %
(utils.CommaJoin(unknown), ),
for row in self.Query(ctx)]
+def _ProcessResult(value):
+ """Converts result values into externally-visible ones.
+
+ """
+ if value is _FS_UNKNOWN:
+ return (QRFS_UNKNOWN, None)
+ elif value is _FS_NODATA:
+ return (QRFS_NODATA, None)
+ elif value is _FS_UNAVAIL:
+ return (QRFS_UNAVAIL, None)
+ elif value is _FS_OFFLINE:
+ return (QRFS_OFFLINE, None)
+ else:
+ return (QRFS_NORMAL, value)
+
+
def _VerifyResultRow(fields, row):
"""Verifies the contents of a query result row.
@param row: Row data
"""
- return (len(row) == len(fields) and
- compat.all((status == constants.QRFS_NORMAL and
- _VERIFY_FN[fdef.kind](value)) or
- # Value for an abnormal status must be None
- (status != constants.QRFS_NORMAL and value is None)
- for ((status, value), (fdef, _, _)) in zip(row, fields)))
+ assert len(row) == len(fields)
+ errs = []
+ for ((status, value), (fdef, _, _)) in zip(row, fields):
+ if status == QRFS_NORMAL:
+ if not _VERIFY_FN[fdef.kind](value):
+ errs.append("normal field %s fails validation (value is %s)" %
+ (fdef.name, value))
+ 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))
def _PrepareFieldList(fields):
@type fields: list of tuples; (L{objects.QueryFieldDefinition}, data kind,
retrieval function)
- @param fields: List of fields
+ @param fields: List of fields, see L{Query.__init__} for a better description
@rtype: dict
@return: Field dictionary for L{Query}
"""
getter = operator.attrgetter(attr)
- return lambda _, item: (constants.QRFS_NORMAL, getter(item))
+ return lambda _, item: getter(item)
def _GetItemTimestamp(getter):
timestamp = getter(item)
if timestamp is None:
# Old configs might not have all timestamps
- return (constants.QRFS_UNAVAIL, None)
+ return _FS_UNAVAIL
else:
- return (constants.QRFS_NORMAL, timestamp)
+ return timestamp
return fn
"""
return [
- (_MakeField("ctime", "CTime", constants.QFT_TIMESTAMP), datatype,
+ (_MakeField("ctime", "CTime", QFT_TIMESTAMP), datatype,
_GetItemTimestamp(operator.attrgetter("ctime"))),
- (_MakeField("mtime", "MTime", constants.QFT_TIMESTAMP), datatype,
+ (_MakeField("mtime", "MTime", QFT_TIMESTAMP), datatype,
_GetItemTimestamp(operator.attrgetter("mtime"))),
]
"""
def __init__(self, nodes, live_data, master_name, node_to_primary,
- node_to_secondary, groups):
+ node_to_secondary, groups, oob_support, cluster):
"""Initializes this class.
"""
self.node_to_primary = node_to_primary
self.node_to_secondary = node_to_secondary
self.groups = groups
+ self.oob_support = oob_support
+ self.cluster = cluster
# Used for individual rows
self.curlive_data = None
#: Fields that are direct attributes of an L{objects.Node} object
_NODE_SIMPLE_FIELDS = {
- "drained": ("Drained", constants.QFT_BOOL),
- "master_candidate": ("MasterC", constants.QFT_BOOL),
- "master_capable": ("MasterCapable", constants.QFT_BOOL),
- "name": ("Node", constants.QFT_TEXT),
- "offline": ("Offline", constants.QFT_BOOL),
- "serial_no": ("SerialNo", constants.QFT_NUMBER),
- "uuid": ("UUID", constants.QFT_TEXT),
- "vm_capable": ("VMCapable", constants.QFT_BOOL),
+ "drained": ("Drained", QFT_BOOL),
+ "master_candidate": ("MasterC", QFT_BOOL),
+ "master_capable": ("MasterCapable", QFT_BOOL),
+ "name": ("Node", QFT_TEXT),
+ "offline": ("Offline", QFT_BOOL),
+ "serial_no": ("SerialNo", QFT_NUMBER),
+ "uuid": ("UUID", QFT_TEXT),
+ "vm_capable": ("VMCapable", QFT_BOOL),
}
#: Fields requiring talking to the node
_NODE_LIVE_FIELDS = {
- "bootid": ("BootID", constants.QFT_TEXT, "bootid"),
- "cnodes": ("CNodes", constants.QFT_NUMBER, "cpu_nodes"),
- "csockets": ("CSockets", constants.QFT_NUMBER, "cpu_sockets"),
- "ctotal": ("CTotal", constants.QFT_NUMBER, "cpu_total"),
- "dfree": ("DFree", constants.QFT_UNIT, "vg_free"),
- "dtotal": ("DTotal", constants.QFT_UNIT, "vg_size"),
- "mfree": ("MFree", constants.QFT_UNIT, "memory_free"),
- "mnode": ("MNode", constants.QFT_UNIT, "memory_dom0"),
- "mtotal": ("MTotal", constants.QFT_UNIT, "memory_total"),
+ "bootid": ("BootID", QFT_TEXT, "bootid"),
+ "cnodes": ("CNodes", QFT_NUMBER, "cpu_nodes"),
+ "csockets": ("CSockets", QFT_NUMBER, "cpu_sockets"),
+ "ctotal": ("CTotal", QFT_NUMBER, "cpu_total"),
+ "dfree": ("DFree", QFT_UNIT, "vg_free"),
+ "dtotal": ("DTotal", QFT_UNIT, "vg_size"),
+ "mfree": ("MFree", QFT_UNIT, "memory_free"),
+ "mnode": ("MNode", QFT_UNIT, "memory_dom0"),
+ "mtotal": ("MTotal", QFT_UNIT, "memory_total"),
}
-def _GetNodeGroup(ctx, node):
+def _GetGroup(cb):
+ """Build function for calling another function with an node group.
+
+ @param cb: The callback to be called with the nodegroup
+
+ """
+ def fn(ctx, node):
+ """Get group data for a node.
+
+ @type ctx: L{NodeQueryData}
+ @type inst: L{objects.Node}
+ @param inst: Node object
+
+ """
+ ng = ctx.groups.get(node.group, None)
+ if ng is None:
+ # Nodes always have a group, or the configuration is corrupt
+ return _FS_UNAVAIL
+
+ return cb(ctx, node, ng)
+
+ return fn
+
+
+def _GetNodeGroup(ctx, node, ng): # pylint: disable-msg=W0613
"""Returns the name of a node's group.
@type ctx: L{NodeQueryData}
@type node: L{objects.Node}
@param node: Node object
+ @type ng: L{objects.NodeGroup}
+ @param ng: The node group this node belongs to
"""
- ng = ctx.groups.get(node.group, None)
- if ng is None:
- # Nodes always have a group, or the configuration is corrupt
- return (constants.QRFS_UNAVAIL, None)
+ return ng.name
+
+
+def _GetNodePower(ctx, node):
+ """Returns the node powered state
- return (constants.QRFS_NORMAL, ng.name)
+ @type ctx: L{NodeQueryData}
+ @type node: L{objects.Node}
+ @param node: Node object
+
+ """
+ if ctx.oob_support[node.name]:
+ return node.powered
+
+ return _FS_UNAVAIL
+
+
+def _GetNdParams(ctx, node, ng):
+ """Returns the ndparams for this node.
+
+ @type ctx: L{NodeQueryData}
+ @type node: L{objects.Node}
+ @param node: Node object
+ @type ng: L{objects.NodeGroup}
+ @param ng: The node group this node belongs to
+
+ """
+ return ctx.cluster.SimpleFillND(ng.FillND(node))
def _GetLiveNodeField(field, kind, ctx, node):
"""
if node.offline:
- return (constants.QRFS_OFFLINE, None)
+ return _FS_OFFLINE
if not ctx.curlive_data:
- return (constants.QRFS_NODATA, None)
+ return _FS_NODATA
try:
value = ctx.curlive_data[field]
except KeyError:
- return (constants.QRFS_UNAVAIL, None)
+ return _FS_UNAVAIL
- if kind == constants.QFT_TEXT:
- return (constants.QRFS_NORMAL, value)
+ if kind == QFT_TEXT:
+ return value
- assert kind in (constants.QFT_NUMBER, constants.QFT_UNIT)
+ assert kind in (QFT_NUMBER, QFT_UNIT)
# Try to convert into number
try:
- return (constants.QRFS_NORMAL, int(value))
+ return int(value)
except (ValueError, TypeError):
logging.exception("Failed to convert node field '%s' (value %r) to int",
value, field)
- return (constants.QRFS_UNAVAIL, None)
+ return _FS_UNAVAIL
def _BuildNodeFields():
"""
fields = [
- (_MakeField("pip", "PrimaryIP", constants.QFT_TEXT), NQ_CONFIG,
- lambda ctx, node: (constants.QRFS_NORMAL, node.primary_ip)),
- (_MakeField("sip", "SecondaryIP", constants.QFT_TEXT), NQ_CONFIG,
- lambda ctx, node: (constants.QRFS_NORMAL, node.secondary_ip)),
- (_MakeField("tags", "Tags", constants.QFT_OTHER), NQ_CONFIG,
- lambda ctx, node: (constants.QRFS_NORMAL, list(node.GetTags()))),
- (_MakeField("master", "IsMaster", constants.QFT_BOOL), NQ_CONFIG,
- lambda ctx, node: (constants.QRFS_NORMAL, node.name == ctx.master_name)),
- (_MakeField("role", "Role", constants.QFT_TEXT), NQ_CONFIG,
- lambda ctx, node: (constants.QRFS_NORMAL,
- _GetNodeRole(node, ctx.master_name))),
- (_MakeField("group", "Group", constants.QFT_TEXT), NQ_GROUP, _GetNodeGroup),
- (_MakeField("group.uuid", "GroupUUID", constants.QFT_TEXT),
- NQ_CONFIG, lambda ctx, node: (constants.QRFS_NORMAL, node.group)),
+ (_MakeField("pip", "PrimaryIP", QFT_TEXT), NQ_CONFIG,
+ _GetItemAttr("primary_ip")),
+ (_MakeField("sip", "SecondaryIP", QFT_TEXT), NQ_CONFIG,
+ _GetItemAttr("secondary_ip")),
+ (_MakeField("tags", "Tags", QFT_OTHER), NQ_CONFIG,
+ lambda ctx, node: list(node.GetTags())),
+ (_MakeField("master", "IsMaster", QFT_BOOL), NQ_CONFIG,
+ lambda ctx, node: node.name == ctx.master_name),
+ (_MakeField("role", "Role", QFT_TEXT), NQ_CONFIG,
+ lambda ctx, node: _GetNodeRole(node, ctx.master_name)),
+ (_MakeField("group", "Group", QFT_TEXT), NQ_GROUP,
+ _GetGroup(_GetNodeGroup)),
+ (_MakeField("group.uuid", "GroupUUID", QFT_TEXT),
+ NQ_CONFIG, _GetItemAttr("group")),
+ (_MakeField("powered", "Powered", QFT_BOOL), NQ_OOB, _GetNodePower),
+ (_MakeField("ndparams", "NodeParameters", QFT_OTHER), NQ_GROUP,
+ _GetGroup(_GetNdParams)),
+ (_MakeField("custom_ndparams", "CustomNodeParameters", QFT_OTHER),
+ NQ_GROUP, _GetItemAttr("ndparams")),
]
def _GetLength(getter):
- return lambda ctx, node: (constants.QRFS_NORMAL,
- len(getter(ctx)[node.name]))
+ return lambda ctx, node: len(getter(ctx)[node.name])
def _GetList(getter):
- return lambda ctx, node: (constants.QRFS_NORMAL,
- list(getter(ctx)[node.name]))
+ return lambda ctx, node: list(getter(ctx)[node.name])
# Add fields operating on instance lists
for prefix, titleprefix, getter in \
[("p", "Pri", operator.attrgetter("node_to_primary")),
("s", "Sec", operator.attrgetter("node_to_secondary"))]:
fields.extend([
- (_MakeField("%sinst_cnt" % prefix, "%sinst" % prefix.upper(),
- constants.QFT_NUMBER),
+ (_MakeField("%sinst_cnt" % prefix, "%sinst" % prefix.upper(), QFT_NUMBER),
NQ_INST, _GetLength(getter)),
(_MakeField("%sinst_list" % prefix, "%sInstances" % titleprefix,
- constants.QFT_OTHER),
+ QFT_OTHER),
NQ_INST, _GetList(getter)),
])
@param inst: Instance object
"""
- # Can't use QRFS_OFFLINE here as it would describe the instance to be offline
- # when we actually don't know due to missing data
+ # Can't use QRFS_OFFLINE here as it would describe the instance to
+ # be offline when we actually don't know due to missing data
if inst.primary_node in ctx.bad_nodes:
- return (constants.QRFS_NODATA, None)
+ return _FS_NODATA
else:
- return (constants.QRFS_NORMAL, bool(ctx.live_data.get(inst.name)))
+ return bool(ctx.live_data.get(inst.name))
def _GetInstLiveData(name):
inst.primary_node in ctx.offline_nodes):
# Can't use QRFS_OFFLINE here as it would describe the instance to be
# offline when we actually don't know due to missing data
- return (constants.QRFS_NODATA, None)
+ return _FS_NODATA
if inst.name in ctx.live_data:
data = ctx.live_data[inst.name]
if name in data:
- return (constants.QRFS_NORMAL, data[name])
+ return data[name]
- return (constants.QRFS_UNAVAIL, None)
+ return _FS_UNAVAIL
return fn
"""
if inst.primary_node in ctx.offline_nodes:
- return (constants.QRFS_NORMAL, "ERROR_nodeoffline")
+ return "ERROR_nodeoffline"
if inst.primary_node in ctx.bad_nodes:
- return (constants.QRFS_NORMAL, "ERROR_nodedown")
+ return "ERROR_nodedown"
if bool(ctx.live_data.get(inst.name)):
if inst.admin_up:
- return (constants.QRFS_NORMAL, "running")
+ return "running"
else:
- return (constants.QRFS_NORMAL, "ERROR_up")
+ return "ERROR_up"
if inst.admin_up:
- return (constants.QRFS_NORMAL, "ERROR_down")
+ return "ERROR_down"
- return (constants.QRFS_NORMAL, "ADMIN_down")
+ return "ADMIN_down"
def _GetInstDiskSize(index):
"""
try:
- return (constants.QRFS_NORMAL, inst.disks[index].size)
+ return inst.disks[index].size
except IndexError:
- return (constants.QRFS_UNAVAIL, None)
+ return _FS_UNAVAIL
return fn
try:
nic = inst.nics[index]
except IndexError:
- return (constants.QRFS_UNAVAIL, None)
+ return _FS_UNAVAIL
return cb(ctx, index, nic)
"""
if nic.ip is None:
- return (constants.QRFS_UNAVAIL, None)
+ return _FS_UNAVAIL
else:
- return (constants.QRFS_NORMAL, nic.ip)
+ return nic.ip
def _GetInstNicBridge(ctx, index, _):
nicparams = ctx.inst_nicparams[index]
if nicparams[constants.NIC_MODE] == constants.NIC_MODE_BRIDGED:
- return (constants.QRFS_NORMAL, nicparams[constants.NIC_LINK])
+ return nicparams[constants.NIC_LINK]
else:
- return (constants.QRFS_UNAVAIL, None)
+ return _FS_UNAVAIL
def _GetInstAllNicBridges(ctx, inst):
assert len(result) == len(inst.nics)
- return (constants.QRFS_NORMAL, result)
+ return result
def _GetInstNicParam(name):
"""
assert len(ctx.inst_nicparams) >= index
- return (constants.QRFS_NORMAL, ctx.inst_nicparams[index][name])
+ return ctx.inst_nicparams[index][name]
return fn
@return: List of field definitions used as input for L{_PrepareFieldList}
"""
- nic_mac_fn = lambda ctx, _, nic: (constants.QRFS_NORMAL, nic.mac)
+ nic_mac_fn = lambda ctx, _, nic: nic.mac
nic_mode_fn = _GetInstNicParam(constants.NIC_MODE)
nic_link_fn = _GetInstNicParam(constants.NIC_LINK)
fields = [
# First NIC (legacy)
- (_MakeField("ip", "IP_address", constants.QFT_TEXT), IQ_CONFIG,
+ (_MakeField("ip", "IP_address", QFT_TEXT), IQ_CONFIG,
_GetInstNic(0, _GetInstNicIp)),
- (_MakeField("mac", "MAC_address", constants.QFT_TEXT), IQ_CONFIG,
+ (_MakeField("mac", "MAC_address", QFT_TEXT), IQ_CONFIG,
_GetInstNic(0, nic_mac_fn)),
- (_MakeField("bridge", "Bridge", constants.QFT_TEXT), IQ_CONFIG,
+ (_MakeField("bridge", "Bridge", QFT_TEXT), IQ_CONFIG,
_GetInstNic(0, _GetInstNicBridge)),
- (_MakeField("nic_mode", "NIC_Mode", constants.QFT_TEXT), IQ_CONFIG,
+ (_MakeField("nic_mode", "NIC_Mode", QFT_TEXT), IQ_CONFIG,
_GetInstNic(0, nic_mode_fn)),
- (_MakeField("nic_link", "NIC_Link", constants.QFT_TEXT), IQ_CONFIG,
+ (_MakeField("nic_link", "NIC_Link", QFT_TEXT), IQ_CONFIG,
_GetInstNic(0, nic_link_fn)),
# All NICs
- (_MakeField("nic.count", "NICs", constants.QFT_NUMBER), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL, len(inst.nics))),
- (_MakeField("nic.macs", "NIC_MACs", constants.QFT_OTHER), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL, [nic.mac for nic in inst.nics])),
- (_MakeField("nic.ips", "NIC_IPs", constants.QFT_OTHER), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL, [nic.ip for nic in inst.nics])),
- (_MakeField("nic.modes", "NIC_modes", constants.QFT_OTHER), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL,
- [nicp[constants.NIC_MODE]
- for nicp in ctx.inst_nicparams])),
- (_MakeField("nic.links", "NIC_links", constants.QFT_OTHER), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL,
- [nicp[constants.NIC_LINK]
- for nicp in ctx.inst_nicparams])),
- (_MakeField("nic.bridges", "NIC_bridges", constants.QFT_OTHER), IQ_CONFIG,
+ (_MakeField("nic.count", "NICs", QFT_NUMBER), IQ_CONFIG,
+ lambda ctx, inst: len(inst.nics)),
+ (_MakeField("nic.macs", "NIC_MACs", QFT_OTHER), IQ_CONFIG,
+ lambda ctx, inst: [nic.mac for nic in inst.nics]),
+ (_MakeField("nic.ips", "NIC_IPs", QFT_OTHER), IQ_CONFIG,
+ lambda ctx, inst: [nic.ip for nic in inst.nics]),
+ (_MakeField("nic.modes", "NIC_modes", QFT_OTHER), IQ_CONFIG,
+ lambda ctx, inst: [nicp[constants.NIC_MODE]
+ for nicp in ctx.inst_nicparams]),
+ (_MakeField("nic.links", "NIC_links", QFT_OTHER), IQ_CONFIG,
+ lambda ctx, inst: [nicp[constants.NIC_LINK]
+ for nicp in ctx.inst_nicparams]),
+ (_MakeField("nic.bridges", "NIC_bridges", QFT_OTHER), IQ_CONFIG,
_GetInstAllNicBridges),
]
# NICs by number
for i in range(constants.MAX_NICS):
fields.extend([
- (_MakeField("nic.ip/%s" % i, "NicIP/%s" % i, constants.QFT_TEXT),
+ (_MakeField("nic.ip/%s" % i, "NicIP/%s" % i, QFT_TEXT),
IQ_CONFIG, _GetInstNic(i, _GetInstNicIp)),
- (_MakeField("nic.mac/%s" % i, "NicMAC/%s" % i, constants.QFT_TEXT),
+ (_MakeField("nic.mac/%s" % i, "NicMAC/%s" % i, QFT_TEXT),
IQ_CONFIG, _GetInstNic(i, nic_mac_fn)),
- (_MakeField("nic.mode/%s" % i, "NicMode/%s" % i, constants.QFT_TEXT),
+ (_MakeField("nic.mode/%s" % i, "NicMode/%s" % i, QFT_TEXT),
IQ_CONFIG, _GetInstNic(i, nic_mode_fn)),
- (_MakeField("nic.link/%s" % i, "NicLink/%s" % i, constants.QFT_TEXT),
+ (_MakeField("nic.link/%s" % i, "NicLink/%s" % i, QFT_TEXT),
IQ_CONFIG, _GetInstNic(i, nic_link_fn)),
- (_MakeField("nic.bridge/%s" % i, "NicBridge/%s" % i, constants.QFT_TEXT),
+ (_MakeField("nic.bridge/%s" % i, "NicBridge/%s" % i, QFT_TEXT),
IQ_CONFIG, _GetInstNic(i, _GetInstNicBridge)),
])
if usage is None:
usage = 0
- return (constants.QRFS_NORMAL, usage)
+ return usage
def _GetInstanceDiskFields():
"""
fields = [
- (_MakeField("disk_usage", "DiskUsage", constants.QFT_UNIT), IQ_DISKUSAGE,
+ (_MakeField("disk_usage", "DiskUsage", QFT_UNIT), IQ_DISKUSAGE,
_GetInstDiskUsage),
- (_MakeField("sda_size", "LegacyDisk/0", constants.QFT_UNIT), IQ_CONFIG,
+ (_MakeField("sda_size", "LegacyDisk/0", QFT_UNIT), IQ_CONFIG,
_GetInstDiskSize(0)),
- (_MakeField("sdb_size", "LegacyDisk/1", constants.QFT_UNIT), IQ_CONFIG,
+ (_MakeField("sdb_size", "LegacyDisk/1", QFT_UNIT), IQ_CONFIG,
_GetInstDiskSize(1)),
- (_MakeField("disk.count", "Disks", constants.QFT_NUMBER), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL, len(inst.disks))),
- (_MakeField("disk.sizes", "Disk_sizes", constants.QFT_OTHER), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL,
- [disk.size for disk in inst.disks])),
+ (_MakeField("disk.count", "Disks", QFT_NUMBER), IQ_CONFIG,
+ lambda ctx, inst: len(inst.disks)),
+ (_MakeField("disk.sizes", "Disk_sizes", QFT_OTHER), IQ_CONFIG,
+ lambda ctx, inst: [disk.size for disk in inst.disks]),
]
# Disks by number
fields.extend([
- (_MakeField("disk.size/%s" % i, "Disk/%s" % i, constants.QFT_UNIT),
+ (_MakeField("disk.size/%s" % i, "Disk/%s" % i, QFT_UNIT),
IQ_CONFIG, _GetInstDiskSize(i))
for i in range(constants.MAX_DISKS)
])
# TODO: Consider moving titles closer to constants
be_title = {
constants.BE_AUTO_BALANCE: "Auto_balance",
- constants.BE_MEMORY: "Configured_memory",
- constants.BE_VCPUS: "VCPUs",
+ constants.BE_MEMORY: "ConfigMemory",
+ constants.BE_VCPUS: "ConfigVCPUs",
}
hv_title = {
fields = [
# Filled parameters
- (_MakeField("hvparams", "HypervisorParameters", constants.QFT_OTHER),
- IQ_CONFIG, lambda ctx, _: (constants.QRFS_NORMAL, ctx.inst_hvparams)),
- (_MakeField("beparams", "BackendParameters", constants.QFT_OTHER),
- IQ_CONFIG, lambda ctx, _: (constants.QRFS_NORMAL, ctx.inst_beparams)),
- (_MakeField("vcpus", "LegacyVCPUs", constants.QFT_NUMBER), IQ_CONFIG,
- lambda ctx, _: (constants.QRFS_NORMAL,
- ctx.inst_beparams[constants.BE_VCPUS])),
+ (_MakeField("hvparams", "HypervisorParameters", QFT_OTHER),
+ IQ_CONFIG, lambda ctx, _: ctx.inst_hvparams),
+ (_MakeField("beparams", "BackendParameters", QFT_OTHER),
+ IQ_CONFIG, lambda ctx, _: ctx.inst_beparams),
+ (_MakeField("vcpus", "LegacyVCPUs", QFT_NUMBER), IQ_CONFIG,
+ lambda ctx, _: ctx.inst_beparams[constants.BE_VCPUS]),
# Unfilled parameters
- (_MakeField("custom_hvparams", "CustomHypervisorParameters",
- constants.QFT_OTHER),
- IQ_CONFIG, lambda ctx, inst: (constants.QRFS_NORMAL, inst.hvparams)),
- (_MakeField("custom_beparams", "CustomBackendParameters",
- constants.QFT_OTHER),
- IQ_CONFIG, lambda ctx, inst: (constants.QRFS_NORMAL, inst.beparams)),
- (_MakeField("custom_nicparams", "CustomNicParameters",
- constants.QFT_OTHER),
- IQ_CONFIG, lambda ctx, inst: (constants.QRFS_NORMAL,
- [nic.nicparams for nic in inst.nics])),
+ (_MakeField("custom_hvparams", "CustomHypervisorParameters", QFT_OTHER),
+ IQ_CONFIG, _GetItemAttr("hvparams")),
+ (_MakeField("custom_beparams", "CustomBackendParameters", QFT_OTHER),
+ IQ_CONFIG, _GetItemAttr("beparams")),
+ (_MakeField("custom_nicparams", "CustomNicParameters", QFT_OTHER),
+ IQ_CONFIG, lambda ctx, inst: [nic.nicparams for nic in inst.nics]),
]
# HV params
def _GetInstHvParam(name):
- return lambda ctx, _: (constants.QRFS_NORMAL,
- ctx.inst_hvparams.get(name, None))
+ return lambda ctx, _: ctx.inst_hvparams.get(name, None)
fields.extend([
# For now all hypervisor parameters are exported as QFT_OTHER
- (_MakeField("hv/%s" % name, hv_title.get(name, "hv/%s" % name),
- constants.QFT_OTHER),
+ (_MakeField("hv/%s" % name, hv_title.get(name, "hv/%s" % name), QFT_OTHER),
IQ_CONFIG, _GetInstHvParam(name))
for name in constants.HVS_PARAMETERS
if name not in constants.HVC_GLOBALS
# BE params
def _GetInstBeParam(name):
- return lambda ctx, _: (constants.QRFS_NORMAL,
- ctx.inst_beparams.get(name, None))
+ return lambda ctx, _: ctx.inst_beparams.get(name, None)
fields.extend([
# For now all backend parameters are exported as QFT_OTHER
- (_MakeField("be/%s" % name, be_title.get(name, "be/%s" % name),
- constants.QFT_OTHER),
+ (_MakeField("be/%s" % name, be_title.get(name, "be/%s" % name), QFT_OTHER),
IQ_CONFIG, _GetInstBeParam(name))
for name in constants.BES_PARAMETERS
])
_INST_SIMPLE_FIELDS = {
- "disk_template": ("Disk_template", constants.QFT_TEXT),
- "hypervisor": ("Hypervisor", constants.QFT_TEXT),
- "name": ("Node", constants.QFT_TEXT),
+ "disk_template": ("Disk_template", QFT_TEXT),
+ "hypervisor": ("Hypervisor", QFT_TEXT),
+ "name": ("Node", QFT_TEXT),
# Depending on the hypervisor, the port can be None
- "network_port": ("Network_port", constants.QFT_OTHER),
- "os": ("OS", constants.QFT_TEXT),
- "serial_no": ("SerialNo", constants.QFT_NUMBER),
- "uuid": ("UUID", constants.QFT_TEXT),
+ "network_port": ("Network_port", QFT_OTHER),
+ "os": ("OS", QFT_TEXT),
+ "serial_no": ("SerialNo", QFT_NUMBER),
+ "uuid": ("UUID", QFT_TEXT),
}
"""
fields = [
- (_MakeField("pnode", "Primary_node", constants.QFT_TEXT), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL, inst.primary_node)),
- (_MakeField("snodes", "Secondary_Nodes", constants.QFT_OTHER), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL, list(inst.secondary_nodes))),
- (_MakeField("admin_state", "Autostart", constants.QFT_BOOL), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL, inst.admin_up)),
- (_MakeField("tags", "Tags", constants.QFT_OTHER), IQ_CONFIG,
- lambda ctx, inst: (constants.QRFS_NORMAL, list(inst.GetTags()))),
+ (_MakeField("pnode", "Primary_node", QFT_TEXT), IQ_CONFIG,
+ _GetItemAttr("primary_node")),
+ (_MakeField("snodes", "Secondary_Nodes", QFT_OTHER), IQ_CONFIG,
+ lambda ctx, inst: list(inst.secondary_nodes)),
+ (_MakeField("admin_state", "Autostart", QFT_BOOL), IQ_CONFIG,
+ _GetItemAttr("admin_up")),
+ (_MakeField("tags", "Tags", QFT_OTHER), IQ_CONFIG,
+ lambda ctx, inst: list(inst.GetTags())),
]
# Add simple fields
# Fields requiring talking to the node
fields.extend([
- (_MakeField("oper_state", "Running", constants.QFT_BOOL), IQ_LIVE,
+ (_MakeField("oper_state", "Running", QFT_BOOL), IQ_LIVE,
_GetInstOperState),
- (_MakeField("oper_ram", "RuntimeMemory", constants.QFT_UNIT), IQ_LIVE,
+ (_MakeField("oper_ram", "Memory", QFT_UNIT), IQ_LIVE,
_GetInstLiveData("memory")),
- (_MakeField("oper_vcpus", "RuntimeVCPUs", constants.QFT_NUMBER), IQ_LIVE,
+ (_MakeField("oper_vcpus", "VCPUs", QFT_NUMBER), IQ_LIVE,
_GetInstLiveData("vcpus")),
- (_MakeField("status", "Status", constants.QFT_TEXT), IQ_LIVE,
- _GetInstStatus),
+ (_MakeField("status", "Status", QFT_TEXT), IQ_LIVE, _GetInstStatus),
])
fields.extend(_GetInstanceParameterFields())
if owners:
owners = utils.NiceSort(owners)
- return (constants.QRFS_NORMAL, owners)
+ return owners
def _GetLockPending(_, data):
pending = [(mode, utils.NiceSort(names))
for (mode, names) in pending]
- return (constants.QRFS_NORMAL, pending)
+ return pending
def _BuildLockFields():
"""
return _PrepareFieldList([
- (_MakeField("name", "Name", constants.QFT_TEXT), None,
- lambda ctx, (name, mode, owners, pending): (constants.QRFS_NORMAL, name)),
- (_MakeField("mode", "Mode", constants.QFT_OTHER), LQ_MODE,
- lambda ctx, (name, mode, owners, pending): (constants.QRFS_NORMAL, mode)),
- (_MakeField("owner", "Owner", constants.QFT_OTHER), LQ_OWNER,
- _GetLockOwners),
- (_MakeField("pending", "Pending", constants.QFT_OTHER), LQ_PENDING,
- _GetLockPending),
+ (_MakeField("name", "Name", QFT_TEXT), None,
+ lambda ctx, (name, mode, owners, pending): name),
+ (_MakeField("mode", "Mode", QFT_OTHER), LQ_MODE,
+ lambda ctx, (name, mode, owners, pending): mode),
+ (_MakeField("owner", "Owner", QFT_OTHER), LQ_OWNER, _GetLockOwners),
+ (_MakeField("pending", "Pending", QFT_OTHER), LQ_PENDING, _GetLockPending),
])
+class GroupQueryData:
+ """Data container for node group data queries.
+
+ """
+ def __init__(self, groups, group_to_nodes, group_to_instances):
+ """Initializes this class.
+
+ @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
+
+ """
+ self.groups = groups
+ self.group_to_nodes = group_to_nodes
+ self.group_to_instances = group_to_instances
+
+ def __iter__(self):
+ """Iterate over all node groups.
+
+ """
+ return iter(self.groups)
+
+
+_GROUP_SIMPLE_FIELDS = {
+ "alloc_policy": ("AllocPolicy", QFT_TEXT),
+ "name": ("Group", QFT_TEXT),
+ "serial_no": ("SerialNo", QFT_NUMBER),
+ "uuid": ("UUID", QFT_TEXT),
+ "ndparams": ("NDParams", QFT_OTHER),
+ }
+
+
+def _BuildGroupFields():
+ """Builds list of fields for node group queries.
+
+ """
+ # Add simple fields
+ fields = [(_MakeField(name, title, kind), GQ_CONFIG, _GetItemAttr(name))
+ for (name, (title, kind)) in _GROUP_SIMPLE_FIELDS.items()]
+
+ def _GetLength(getter):
+ return lambda ctx, group: len(getter(ctx)[group.uuid])
+
+ def _GetSortedList(getter):
+ return lambda ctx, group: utils.NiceSort(getter(ctx)[group.uuid])
+
+ group_to_nodes = operator.attrgetter("group_to_nodes")
+ group_to_instances = operator.attrgetter("group_to_instances")
+
+ # Add fields for nodes
+ fields.extend([
+ (_MakeField("node_cnt", "Nodes", QFT_NUMBER),
+ GQ_NODE, _GetLength(group_to_nodes)),
+ (_MakeField("node_list", "NodeList", QFT_OTHER),
+ GQ_NODE, _GetSortedList(group_to_nodes)),
+ ])
+
+ # Add fields for instances
+ fields.extend([
+ (_MakeField("pinst_cnt", "Instances", QFT_NUMBER),
+ GQ_INST, _GetLength(group_to_instances)),
+ (_MakeField("pinst_list", "InstanceList", QFT_OTHER),
+ GQ_INST, _GetSortedList(group_to_instances)),
+ ])
+
+ fields.extend(_GetItemTimestampFields(GQ_CONFIG))
+
+ return _PrepareFieldList(fields)
+
+
#: Fields available for node queries
NODE_FIELDS = _BuildNodeFields()
#: Fields available for lock queries
LOCK_FIELDS = _BuildLockFields()
+
+#: Fields available for node group queries
+GROUP_FIELDS = _BuildGroupFields()
+
+#: All available field lists
+ALL_FIELD_LISTS = [NODE_FIELDS, INSTANCE_FIELDS, LOCK_FIELDS, GROUP_FIELDS]