At the moment, if an instance is unexpectedly power off or paused, although the serial console worker threads are stopped, there is no cleanup performed. For example, the serial proxy will still be accepting connections. This patch lets the event handler manage serial console workers based on the recieved events. Change-Id: I154e5796698dc6a57defb588c5d47f4f095fe7de
		
			
				
	
	
		
			139 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			139 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
# Copyright 2015 Cloudbase Solutions Srl
 | 
						|
# All Rights Reserved.
 | 
						|
#
 | 
						|
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
 | 
						|
#    not use this file except in compliance with the License. You may obtain
 | 
						|
#    a copy of the License at
 | 
						|
#
 | 
						|
#         http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
#
 | 
						|
#    Unless required by applicable law or agreed to in writing, software
 | 
						|
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 | 
						|
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 | 
						|
#    License for the specific language governing permissions and limitations
 | 
						|
#    under the License.
 | 
						|
 | 
						|
import eventlet
 | 
						|
 | 
						|
import sys
 | 
						|
 | 
						|
if sys.platform == 'win32':
 | 
						|
    import wmi
 | 
						|
 | 
						|
from nova import exception
 | 
						|
from nova.i18n import _LW
 | 
						|
from nova.virt import event as virtevent
 | 
						|
from oslo_config import cfg
 | 
						|
from oslo_log import log as logging
 | 
						|
 | 
						|
from hyperv.nova import constants
 | 
						|
from hyperv.nova import serialconsoleops
 | 
						|
from hyperv.nova import utilsfactory
 | 
						|
 | 
						|
LOG = logging.getLogger(__name__)
 | 
						|
 | 
						|
hyperv_opts = [
 | 
						|
    cfg.IntOpt('power_state_check_timeframe',
 | 
						|
                default=60,
 | 
						|
                help='The timeframe to be checked for instance power '
 | 
						|
                     'state changes.'),
 | 
						|
    cfg.IntOpt('power_state_event_polling_interval',
 | 
						|
                default=2,
 | 
						|
                help='Instance power state change event polling frequency.'),
 | 
						|
 | 
						|
]
 | 
						|
 | 
						|
CONF = cfg.CONF
 | 
						|
CONF.register_opts(hyperv_opts, 'hyperv')
 | 
						|
 | 
						|
 | 
						|
class InstanceEventHandler(object):
 | 
						|
    # The event listener timeout is set to 0 in order to return immediately
 | 
						|
    # and avoid blocking the thread.
 | 
						|
    _WAIT_TIMEOUT = 0
 | 
						|
 | 
						|
    _TRANSITION_MAP = {
 | 
						|
        constants.HYPERV_VM_STATE_ENABLED: virtevent.EVENT_LIFECYCLE_STARTED,
 | 
						|
        constants.HYPERV_VM_STATE_DISABLED: virtevent.EVENT_LIFECYCLE_STOPPED,
 | 
						|
        constants.HYPERV_VM_STATE_PAUSED: virtevent.EVENT_LIFECYCLE_PAUSED,
 | 
						|
        # We ignore the suspended state transition because
 | 
						|
        # this cannot be handled by the manager at the moment.
 | 
						|
        #
 | 
						|
        # TODO(lpetrut): Uncomment this after this patch gets in:
 | 
						|
        # I0fad28025785e77ef32b88ff813f39000d9f9e35
 | 
						|
        #
 | 
						|
        # constants.HYPERV_VM_STATE_SUSPENDED:
 | 
						|
        #    virtevent.EVENT_LIFECYCLE_SUSPENDED
 | 
						|
    }
 | 
						|
 | 
						|
    def __init__(self, state_change_callback=None):
 | 
						|
        self._vmutils = utilsfactory.get_vmutils()
 | 
						|
        self._listener = self._vmutils.get_vm_power_state_change_listener(
 | 
						|
            timeframe=CONF.hyperv.power_state_check_timeframe,
 | 
						|
            filtered_states=self._TRANSITION_MAP.keys())
 | 
						|
 | 
						|
        self._serial_console_ops = serialconsoleops.SerialConsoleOps()
 | 
						|
 | 
						|
        self._polling_interval = CONF.hyperv.power_state_event_polling_interval
 | 
						|
        self._state_change_callback = state_change_callback
 | 
						|
 | 
						|
    def start_listener(self):
 | 
						|
        eventlet.spawn_n(self._poll_events)
 | 
						|
 | 
						|
    def _poll_events(self):
 | 
						|
        while True:
 | 
						|
            try:
 | 
						|
                # Retrieve one by one all the events that occured in
 | 
						|
                # the checked interval.
 | 
						|
                event = self._listener(self._WAIT_TIMEOUT)
 | 
						|
                self._dispatch_event(event)
 | 
						|
                continue
 | 
						|
            except wmi.x_wmi_timed_out:
 | 
						|
                # If no events were triggered in the checked interval,
 | 
						|
                # a timeout exception is raised. We'll just ignore it.
 | 
						|
                pass
 | 
						|
 | 
						|
            eventlet.sleep(self._polling_interval)
 | 
						|
 | 
						|
    def _dispatch_event(self, event):
 | 
						|
        instance_state = self._vmutils.get_vm_power_state(event.EnabledState)
 | 
						|
        instance_name = event.ElementName
 | 
						|
 | 
						|
        # Instance uuid set by Nova. If this is missing, we assume that
 | 
						|
        # the instance was not created by Nova and ignore the event.
 | 
						|
        instance_uuid = self._get_instance_uuid(instance_name)
 | 
						|
        if instance_uuid:
 | 
						|
            self._emit_event(instance_name, instance_uuid, instance_state)
 | 
						|
 | 
						|
    def _emit_event(self, instance_name, instance_uuid, instance_state):
 | 
						|
        virt_event = self._get_virt_event(instance_uuid,
 | 
						|
                                          instance_state)
 | 
						|
        eventlet.spawn_n(self._state_change_callback, virt_event)
 | 
						|
 | 
						|
        eventlet.spawn_n(self._handle_serial_console_workers,
 | 
						|
                         instance_name, instance_state)
 | 
						|
 | 
						|
    def _handle_serial_console_workers(self, instance_name, instance_state):
 | 
						|
        if instance_state == constants.HYPERV_VM_STATE_ENABLED:
 | 
						|
            self._serial_console_ops.start_console_handler(instance_name)
 | 
						|
        else:
 | 
						|
            self._serial_console_ops.stop_console_handler(instance_name)
 | 
						|
 | 
						|
    def _get_instance_uuid(self, instance_name):
 | 
						|
        try:
 | 
						|
            instance_uuid = self._vmutils.get_instance_uuid(instance_name)
 | 
						|
            if not instance_uuid:
 | 
						|
                LOG.warn(_LW("Instance uuid could not be retrieved for "
 | 
						|
                             "instance %s. Instance state change event "
 | 
						|
                             "will be ignored."),
 | 
						|
                         instance_name)
 | 
						|
            return instance_uuid
 | 
						|
        except exception.NotFound:
 | 
						|
            # The instance has been deleted.
 | 
						|
            pass
 | 
						|
 | 
						|
    def _get_virt_event(self, instance_uuid, instance_state):
 | 
						|
        transition = self._TRANSITION_MAP[instance_state]
 | 
						|
        return virtevent.LifecycleEvent(uuid=instance_uuid,
 | 
						|
                                        transition=transition)
 |