332 lines
12 KiB
Python
332 lines
12 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2011, OpenStack Foundation
|
|
# Copyright 2012, Red Hat, Inc.
|
|
# Copyright 2013 IBM Corp.
|
|
#
|
|
# 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 warnings
|
|
|
|
from oslo.config import cfg
|
|
from oslo import messaging
|
|
import webob
|
|
|
|
from glance.common import exception
|
|
import glance.domain.proxy
|
|
import glance.openstack.common.log as logging
|
|
from glance.openstack.common import timeutils
|
|
|
|
notifier_opts = [
|
|
cfg.StrOpt('notifier_strategy', default='default',
|
|
help=_('Notifications can be sent when images are create, '
|
|
'updated or deleted. There are three methods of sending '
|
|
'notifications, logging (via the log_file directive), '
|
|
'rabbit (via a rabbitmq queue), qpid (via a Qpid '
|
|
'message queue), or noop (no notifications sent, the '
|
|
'default). (DEPRECATED)')),
|
|
|
|
cfg.StrOpt('default_publisher_id', default="image.localhost",
|
|
help='Default publisher_id for outgoing notifications'),
|
|
]
|
|
|
|
CONF = cfg.CONF
|
|
CONF.register_opts(notifier_opts)
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
_STRATEGY_ALIASES = {
|
|
"logging": "log",
|
|
"rabbit": "messaging",
|
|
"qpid": "messaging",
|
|
"noop": "noop",
|
|
"default": "noop",
|
|
}
|
|
|
|
|
|
class Notifier(object):
|
|
"""Uses a notification strategy to send out messages about events."""
|
|
|
|
def __init__(self, strategy=None):
|
|
|
|
if CONF.notifier_strategy != 'default':
|
|
msg = _("notifier_strategy was deprecated in "
|
|
"favor of `notification_driver`")
|
|
warnings.warn(msg, DeprecationWarning)
|
|
|
|
# NOTE(flaper87): Use this to keep backwards
|
|
# compatibility. We'll try to get an oslo.messaging
|
|
# driver from the specified strategy.
|
|
_strategy = strategy or CONF.notifier_strategy
|
|
_driver = _STRATEGY_ALIASES.get(_strategy)
|
|
|
|
# NOTE(flaper87): The next 3 lines help
|
|
# with the migration to oslo.messaging.
|
|
# Without them, gate tests won't know
|
|
# what driver should be loaded.
|
|
# Once this patch lands, devstack will be
|
|
# updated and then these lines will be removed.
|
|
url = None
|
|
if _strategy in ['rabbit', 'qpid']:
|
|
url = _strategy + '://'
|
|
|
|
publisher_id = CONF.default_publisher_id
|
|
self._transport = messaging.get_transport(CONF, url)
|
|
self._notifier = messaging.Notifier(self._transport,
|
|
driver=_driver,
|
|
publisher_id=publisher_id)
|
|
|
|
def warn(self, event_type, payload):
|
|
self._notifier.warn({}, event_type, payload)
|
|
|
|
def info(self, event_type, payload):
|
|
self._notifier.info({}, event_type, payload)
|
|
|
|
def error(self, event_type, payload):
|
|
self._notifier.error({}, event_type, payload)
|
|
|
|
|
|
def format_image_notification(image):
|
|
"""
|
|
Given a glance.domain.Image object, return a dictionary of relevant
|
|
notification information. We purposely do not include 'location'
|
|
as it may contain credentials.
|
|
"""
|
|
return {
|
|
'id': image.image_id,
|
|
'name': image.name,
|
|
'status': image.status,
|
|
'created_at': timeutils.isotime(image.created_at),
|
|
'updated_at': timeutils.isotime(image.updated_at),
|
|
'min_disk': image.min_disk,
|
|
'min_ram': image.min_ram,
|
|
'protected': image.protected,
|
|
'checksum': image.checksum,
|
|
'owner': image.owner,
|
|
'disk_format': image.disk_format,
|
|
'container_format': image.container_format,
|
|
'size': image.size,
|
|
'is_public': image.visibility == 'public',
|
|
'properties': dict(image.extra_properties),
|
|
'tags': list(image.tags),
|
|
'deleted': False,
|
|
'deleted_at': None,
|
|
}
|
|
|
|
|
|
def format_task_notification(task):
|
|
# NOTE(nikhil): input is not passed to the notifier payload as it may
|
|
# contain sensitive info.
|
|
return {'id': task.task_id,
|
|
'type': task.type,
|
|
'status': task.status,
|
|
'result': None,
|
|
'owner': task.owner,
|
|
'message': None,
|
|
'expires_at': timeutils.isotime(task.expires_at),
|
|
'created_at': timeutils.isotime(task.created_at),
|
|
'updated_at': timeutils.isotime(task.updated_at),
|
|
'deleted': False,
|
|
'deleted_at': None,
|
|
}
|
|
|
|
|
|
class ImageRepoProxy(glance.domain.proxy.Repo):
|
|
|
|
def __init__(self, image_repo, context, notifier):
|
|
self.image_repo = image_repo
|
|
self.context = context
|
|
self.notifier = notifier
|
|
proxy_kwargs = {'context': self.context, 'notifier': self.notifier}
|
|
super(ImageRepoProxy, self).__init__(image_repo,
|
|
item_proxy_class=ImageProxy,
|
|
item_proxy_kwargs=proxy_kwargs)
|
|
|
|
def save(self, image):
|
|
super(ImageRepoProxy, self).save(image)
|
|
self.notifier.info('image.update',
|
|
format_image_notification(image))
|
|
|
|
def add(self, image):
|
|
super(ImageRepoProxy, self).add(image)
|
|
self.notifier.info('image.create',
|
|
format_image_notification(image))
|
|
|
|
def remove(self, image):
|
|
super(ImageRepoProxy, self).remove(image)
|
|
payload = format_image_notification(image)
|
|
payload['deleted'] = True
|
|
payload['deleted_at'] = timeutils.isotime()
|
|
self.notifier.info('image.delete', payload)
|
|
|
|
|
|
class ImageFactoryProxy(glance.domain.proxy.ImageFactory):
|
|
def __init__(self, factory, context, notifier):
|
|
kwargs = {'context': context, 'notifier': notifier}
|
|
super(ImageFactoryProxy, self).__init__(factory,
|
|
proxy_class=ImageProxy,
|
|
proxy_kwargs=kwargs)
|
|
|
|
|
|
class ImageProxy(glance.domain.proxy.Image):
|
|
|
|
def __init__(self, image, context, notifier):
|
|
self.image = image
|
|
self.context = context
|
|
self.notifier = notifier
|
|
super(ImageProxy, self).__init__(image)
|
|
|
|
def _format_image_send(self, bytes_sent):
|
|
return {
|
|
'bytes_sent': bytes_sent,
|
|
'image_id': self.image.image_id,
|
|
'owner_id': self.image.owner,
|
|
'receiver_tenant_id': self.context.tenant,
|
|
'receiver_user_id': self.context.user,
|
|
}
|
|
|
|
def get_data(self):
|
|
sent = 0
|
|
for chunk in self.image.get_data():
|
|
yield chunk
|
|
sent += len(chunk)
|
|
|
|
if sent != self.image.size:
|
|
notify = self.notifier.error
|
|
else:
|
|
notify = self.notifier.info
|
|
|
|
try:
|
|
notify('image.send',
|
|
self._format_image_send(sent))
|
|
except Exception as err:
|
|
msg = (_("An error occurred during image.send"
|
|
" notification: %(err)s") % {'err': err})
|
|
LOG.error(msg)
|
|
|
|
def set_data(self, data, size=None):
|
|
payload = format_image_notification(self.image)
|
|
self.notifier.info('image.prepare', payload)
|
|
try:
|
|
self.image.set_data(data, size)
|
|
except exception.StorageFull as e:
|
|
msg = (_("Image storage media is full: %s") % e)
|
|
self.notifier.error('image.upload', msg)
|
|
raise webob.exc.HTTPRequestEntityTooLarge(explanation=msg)
|
|
except exception.StorageWriteDenied as e:
|
|
msg = (_("Insufficient permissions on image storage media: %s")
|
|
% e)
|
|
self.notifier.error('image.upload', msg)
|
|
raise webob.exc.HTTPServiceUnavailable(explanation=msg)
|
|
except ValueError as e:
|
|
msg = (_("Cannot save data for image %s: %s")
|
|
% (self.image.image_id, e))
|
|
self.notifier.error('image.upload', msg)
|
|
raise webob.exc.HTTPBadRequest(explanation=unicode(e))
|
|
except exception.Duplicate as e:
|
|
msg = (_("Unable to upload duplicate image data for image %s: %s")
|
|
% (self.image.image_id, e))
|
|
self.notifier.error('image.upload', msg)
|
|
raise webob.exc.HTTPConflict(explanation=msg)
|
|
except exception.Forbidden as e:
|
|
msg = (_("Not allowed to upload image data for image %s: %s")
|
|
% (self.image.image_id, e))
|
|
self.notifier.error('image.upload', msg)
|
|
raise webob.exc.HTTPForbidden(explanation=msg)
|
|
except exception.NotFound as e:
|
|
msg = (_("Image %s could not be found after upload. The image may "
|
|
"have been deleted during the upload: %s")
|
|
% (self.image.image_id, e))
|
|
self.notifier.error('image.upload', msg)
|
|
raise webob.exc.HTTPNotFound(explanation=unicode(e))
|
|
except webob.exc.HTTPError as e:
|
|
msg = (_("Failed to upload image data for image %s"
|
|
" due to HTTP error: %s")
|
|
% (self.image.image_id, e))
|
|
self.notifier.error('image.upload', msg)
|
|
raise
|
|
except Exception as e:
|
|
msg = (_("Failed to upload image data for image %s "
|
|
"due to internal error: %s")
|
|
% (self.image.image_id, e))
|
|
self.notifier.error('image.upload', msg)
|
|
raise
|
|
else:
|
|
payload = format_image_notification(self.image)
|
|
self.notifier.info('image.upload', payload)
|
|
self.notifier.info('image.activate', payload)
|
|
|
|
|
|
class TaskRepoProxy(glance.domain.proxy.Repo):
|
|
|
|
def __init__(self, task_repo, context, notifier):
|
|
self.task_repo = task_repo
|
|
self.context = context
|
|
self.notifier = notifier
|
|
proxy_kwargs = {'context': self.context, 'notifier': self.notifier}
|
|
super(TaskRepoProxy, self).__init__(task_repo,
|
|
item_proxy_class=TaskProxy,
|
|
item_proxy_kwargs=proxy_kwargs)
|
|
|
|
def add(self, task):
|
|
self.notifier.info('task.create',
|
|
format_task_notification(task))
|
|
return super(TaskRepoProxy, self).add(task)
|
|
|
|
def remove(self, task):
|
|
payload = format_task_notification(task)
|
|
payload['deleted'] = True
|
|
payload['deleted_at'] = timeutils.isotime()
|
|
self.notifier.info('task.delete', payload)
|
|
return super(TaskRepoProxy, self).add(task)
|
|
|
|
|
|
class TaskFactoryProxy(glance.domain.proxy.TaskFactory):
|
|
def __init__(self, factory, context, notifier):
|
|
kwargs = {'context': context, 'notifier': notifier}
|
|
super(TaskFactoryProxy, self).__init__(factory,
|
|
proxy_class=TaskProxy,
|
|
proxy_kwargs=kwargs)
|
|
|
|
|
|
class TaskProxy(glance.domain.proxy.Task):
|
|
|
|
def __init__(self, task, context, notifier):
|
|
self.task = task
|
|
self.context = context
|
|
self.notifier = notifier
|
|
super(TaskProxy, self).__init__(task)
|
|
|
|
def run(self, executor):
|
|
self.notifier.info('task.run',
|
|
format_task_notification(self.task))
|
|
return super(TaskProxy, self).run(executor)
|
|
|
|
def begin_processing(self):
|
|
self.notifier.info(
|
|
'task.processing',
|
|
format_task_notification(self.task)
|
|
)
|
|
return super(TaskProxy, self).begin_processing()
|
|
|
|
def succeed(self, result):
|
|
self.notifier.info('task.success',
|
|
format_task_notification(self.task))
|
|
return super(TaskProxy, self).succeed(result)
|
|
|
|
def fail(self, message):
|
|
self.notifier.info('task.failure',
|
|
format_task_notification(self.task))
|
|
return super(TaskProxy, self).fail(message)
|