valet/valet/api/common/hooks.py
Omar Rivera 125f5ccfab Fix erroneous Engine bootstrap
Use oslo_log
getting music out of api
updating valet.conf
Health Check for hanged Engine
Wrong python-requests version

Change-Id: Ibb2649c81dad94e51b579c0a99cfaf37b626095c
2017-05-01 21:46:30 -05:00

110 lines
3.6 KiB
Python

#
# Copyright 2014-2017 AT&T Intellectual Property
#
# 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.
"""Hooks."""
import json
from pecan import conf
from pecan.hooks import PecanHook
import threading
from valet import api
from valet.api.common.i18n import _
from valet.api.common import terminate_thread
from valet.api.v1.controllers import error
import webob
class MessageNotificationHook(PecanHook):
"""Send API request/responses out as Oslo msg notifications."""
def after(self, state):
"""Function sends valet notification."""
self.dummy = True
api.LOG.info('sending notification')
notifier = conf.messaging.notifier
status_code = state.response.status_code
status = webob.exc.status_map.get(status_code)
if issubclass(status, webob.exc.HTTPOk):
notifier_fn = notifier.info
else:
notifier_fn = notifier.error
# Not using this just yet.
ctxt = {}
request_path = state.request.path
event_type_parts = ['api']
api_version = state.request.path_info_pop()
if api_version:
event_type_parts.append(api_version)
api_subject = state.request.path_info_pop()
if api_subject:
event_type_parts.append(api_subject)
event_type = '.'.join(event_type_parts)
request_method = state.request.method
try:
request_body = json.loads(state.request.body)
except ValueError:
request_body = None
try:
response_body = json.loads(state.response.body)
except ValueError:
response_body = state.response.body
tenant_id = state.request.context.get('tenant_id', None)
user_id = state.request.context.get('user_id', None)
payload = {
'context': {
'tenant_id': tenant_id,
'user_id': user_id,
},
'request': {
'method': request_method,
'path': request_path,
'body': request_body,
},
'response': {
'status_code': status_code,
'body': response_body,
}
}
# notifier_fn blocks in case rabbit mq is down
# it prevents Valet API to return its response
# send the notification in a different thread
notifier_thread = threading.Thread(target=notifier_fn, args=(ctxt, event_type, payload))
notifier_thread.start()
# launch a timer to verify no hung threads are left behind
# (when timeout expires kill the notifier thread if it still alive)
watcher = threading.Timer(conf.messaging.timeout, terminate_thread, args=[notifier_thread])
watcher.start()
api.LOG.info('notification sent.')
class NotFoundHook(PecanHook):
"""Catchall 'not found' hook for API."""
def on_error(self, state, exc):
"""Redirect to app-specific not_found endpoint if 404 only."""
self.dummy = True
if isinstance(exc, webob.exc.WSGIHTTPException) and exc.code == 404:
message = _('The resource could not be found.')
error('/errors/not_found', message)