mirror of
https://github.com/virt-manager/virt-manager.git
synced 2024-10-27 01:55:20 +03:00
aefd4c4824
And clean up the API mess while we are at it. Treat the key as an opaque value that users shouldn't depend on. Besides the improved code clarity and API layout, this will help diagnose 'key error' issues, since we'll see an object name instead of UUID which is hard to trace back.
283 lines
9.0 KiB
Python
283 lines
9.0 KiB
Python
#
|
|
# Copyright (C) 2013 Red Hat, 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
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
# MA 02110-1301 USA.
|
|
#
|
|
|
|
import logging
|
|
|
|
|
|
# Debugging helper to force old style polling
|
|
_force_old_poll = False
|
|
|
|
|
|
def _new_poll_helper(origmap, typename, listfunc, buildfunc):
|
|
"""
|
|
Helper for new style listAll* APIs
|
|
"""
|
|
current = {}
|
|
new = {}
|
|
objs = []
|
|
|
|
try:
|
|
objs = listfunc()
|
|
except Exception, e:
|
|
logging.debug("Unable to list all %ss: %s", typename, e)
|
|
|
|
for obj in objs:
|
|
connkey = obj.name()
|
|
|
|
if connkey not in origmap:
|
|
# Object is brand new this period
|
|
current[connkey] = buildfunc(obj, connkey)
|
|
new[connkey] = current[connkey]
|
|
else:
|
|
# Previously known object
|
|
current[connkey] = origmap[connkey]
|
|
del(origmap[connkey])
|
|
|
|
return (origmap, new, current)
|
|
|
|
|
|
def _old_poll_helper(origmap, typename,
|
|
active_list, inactive_list,
|
|
lookup_func, build_func):
|
|
"""
|
|
Helper routine for old style split API libvirt polling.
|
|
@origmap: Pre-existing mapping of objects, with connkey->obj mapping.
|
|
objects must have an is_active and set_active API
|
|
@typename: string describing type of objects we are polling for use
|
|
in debug messages.
|
|
@active_list: Function that returns the list of active objects
|
|
@inactive_list: Function that returns the list of inactive objects
|
|
@lookup_func: Function to get an object handle for the passed name
|
|
@build_func: Function that builds a new object class. It is passed
|
|
args of (raw libvirt object, connkey)
|
|
"""
|
|
current = {}
|
|
new = {}
|
|
newActiveNames = []
|
|
newInactiveNames = []
|
|
|
|
try:
|
|
newActiveNames = active_list()
|
|
except Exception, e:
|
|
logging.debug("Unable to list active %ss: %s", typename, e)
|
|
try:
|
|
newInactiveNames = inactive_list()
|
|
except Exception, e:
|
|
logging.debug("Unable to list inactive %ss: %s", typename, e)
|
|
|
|
def check_obj(name):
|
|
obj = None
|
|
connkey = name
|
|
|
|
if connkey not in origmap:
|
|
if connkey not in origmap:
|
|
try:
|
|
obj = lookup_func(name)
|
|
except Exception, e:
|
|
logging.debug("Could not fetch %s '%s': %s",
|
|
typename, connkey, e)
|
|
return
|
|
|
|
# Object is brand new this period
|
|
current[connkey] = build_func(obj, connkey)
|
|
new[connkey] = current[connkey]
|
|
else:
|
|
# Previously known object
|
|
current[connkey] = origmap[connkey]
|
|
del(origmap[connkey])
|
|
|
|
for name in newActiveNames + newInactiveNames:
|
|
try:
|
|
check_obj(name)
|
|
except:
|
|
logging.exception("Couldn't fetch %s '%s'", typename, name)
|
|
|
|
return (origmap, new, current)
|
|
|
|
|
|
def fetch_nets(backend, origmap, build_func):
|
|
name = "network"
|
|
|
|
if backend.check_support(
|
|
backend.SUPPORT_CONN_LISTALLNETWORKS) and not _force_old_poll:
|
|
return _new_poll_helper(origmap, name,
|
|
backend.listAllNetworks, build_func)
|
|
else:
|
|
active_list = backend.listNetworks
|
|
inactive_list = backend.listDefinedNetworks
|
|
lookup_func = backend.networkLookupByName
|
|
|
|
return _old_poll_helper(origmap, name,
|
|
active_list, inactive_list,
|
|
lookup_func, build_func)
|
|
|
|
|
|
def fetch_pools(backend, origmap, build_func):
|
|
name = "pool"
|
|
|
|
if backend.check_support(
|
|
backend.SUPPORT_CONN_LISTALLSTORAGEPOOLS) and not _force_old_poll:
|
|
return _new_poll_helper(origmap, name,
|
|
backend.listAllStoragePools, build_func)
|
|
else:
|
|
active_list = backend.listStoragePools
|
|
inactive_list = backend.listDefinedStoragePools
|
|
lookup_func = backend.storagePoolLookupByName
|
|
|
|
return _old_poll_helper(origmap, name,
|
|
active_list, inactive_list,
|
|
lookup_func, build_func)
|
|
|
|
|
|
def fetch_volumes(backend, pool, origmap, build_func):
|
|
name = "volume"
|
|
|
|
if backend.check_support(
|
|
backend.SUPPORT_POOL_LISTALLVOLUMES, pool) and not _force_old_poll:
|
|
return _new_poll_helper(origmap, name,
|
|
pool.listAllVolumes, build_func)
|
|
else:
|
|
active_list = pool.listVolumes
|
|
inactive_list = lambda: []
|
|
lookup_func = pool.storageVolLookupByName
|
|
return _old_poll_helper(origmap, name,
|
|
active_list, inactive_list,
|
|
lookup_func, build_func)
|
|
|
|
|
|
def fetch_interfaces(backend, origmap, build_func):
|
|
name = "interface"
|
|
|
|
if backend.check_support(
|
|
backend.SUPPORT_CONN_LISTALLINTERFACES) and not _force_old_poll:
|
|
return _new_poll_helper(origmap, name,
|
|
backend.listAllInterfaces, build_func)
|
|
else:
|
|
active_list = backend.listInterfaces
|
|
inactive_list = backend.listDefinedInterfaces
|
|
lookup_func = backend.interfaceLookupByName
|
|
|
|
return _old_poll_helper(origmap, name,
|
|
active_list, inactive_list,
|
|
lookup_func, build_func)
|
|
|
|
|
|
def fetch_nodedevs(backend, origmap, build_func):
|
|
name = "nodedev"
|
|
if backend.check_support(
|
|
backend.SUPPORT_CONN_LISTALLDEVICES) and not _force_old_poll:
|
|
return _new_poll_helper(origmap, name,
|
|
backend.listAllDevices, build_func)
|
|
else:
|
|
active_list = lambda: backend.listDevices(None, 0)
|
|
inactive_list = lambda: []
|
|
lookup_func = backend.nodeDeviceLookupByName
|
|
return _old_poll_helper(origmap, name,
|
|
active_list, inactive_list,
|
|
lookup_func, build_func)
|
|
|
|
|
|
def _old_fetch_vms(backend, origmap, build_func):
|
|
# We can't easily use _old_poll_helper here because the domain API
|
|
# doesn't always return names like other objects, it returns
|
|
# IDs for active VMs
|
|
|
|
newActiveIDs = []
|
|
newInactiveNames = []
|
|
oldActiveIDs = {}
|
|
oldInactiveNames = {}
|
|
current = {}
|
|
new = {}
|
|
|
|
# Build list of previous vms with proper id/name mappings
|
|
for vm in origmap.values():
|
|
if vm.is_active():
|
|
oldActiveIDs[vm.get_id()] = vm
|
|
else:
|
|
oldInactiveNames[vm.get_name()] = vm
|
|
|
|
try:
|
|
newActiveIDs = backend.listDomainsID()
|
|
except Exception, e:
|
|
logging.debug("Unable to list active domains: %s", e)
|
|
|
|
try:
|
|
newInactiveNames = backend.listDefinedDomains()
|
|
except Exception, e:
|
|
logging.exception("Unable to list inactive domains: %s", e)
|
|
|
|
def add_vm(vm):
|
|
connkey = vm.get_name()
|
|
|
|
current[connkey] = vm
|
|
del(origmap[connkey])
|
|
|
|
def check_new(rawvm, connkey):
|
|
if connkey in origmap:
|
|
vm = origmap[connkey]
|
|
del(origmap[connkey])
|
|
else:
|
|
vm = build_func(rawvm, connkey)
|
|
new[connkey] = vm
|
|
|
|
current[connkey] = vm
|
|
|
|
for _id in newActiveIDs:
|
|
if _id in oldActiveIDs:
|
|
# No change, copy across existing VM object
|
|
vm = oldActiveIDs[_id]
|
|
add_vm(vm)
|
|
else:
|
|
# Check if domain is brand new, or old one that changed state
|
|
try:
|
|
vm = backend.lookupByID(_id)
|
|
connkey = vm.name()
|
|
|
|
check_new(vm, connkey)
|
|
except:
|
|
logging.exception("Couldn't fetch domain id '%s'", _id)
|
|
|
|
|
|
for name in newInactiveNames:
|
|
if name in oldInactiveNames:
|
|
# No change, copy across existing VM object
|
|
vm = oldInactiveNames[name]
|
|
add_vm(vm)
|
|
else:
|
|
# Check if domain is brand new, or old one that changed state
|
|
try:
|
|
vm = backend.lookupByName(name)
|
|
connkey = name
|
|
|
|
check_new(vm, connkey)
|
|
except:
|
|
logging.exception("Couldn't fetch domain '%s'", name)
|
|
|
|
return (origmap, new, current)
|
|
|
|
|
|
def fetch_vms(backend, origmap, build_func):
|
|
name = "domain"
|
|
if backend.check_support(
|
|
backend.SUPPORT_CONN_LISTALLDOMAINS):
|
|
return _new_poll_helper(origmap, name,
|
|
backend.listAllDomains, build_func)
|
|
else:
|
|
return _old_fetch_vms(backend, origmap, build_func)
|