131 lines
4.8 KiB
Python
131 lines
4.8 KiB
Python
# Copyright (c) 2015 Mirantis, Inc.
|
|
# 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.
|
|
|
|
from oslo_log import log
|
|
from oslo_utils import timeutils
|
|
|
|
from manila import exception
|
|
from manila.share import api
|
|
from manila.share import hook
|
|
from manila.share.hooks import zaqarclientwrapper # noqa
|
|
|
|
CONF = zaqarclientwrapper.CONF
|
|
LOG = log.getLogger(__name__)
|
|
ZAQARCLIENT = zaqarclientwrapper.ZAQARCLIENT
|
|
|
|
|
|
class ZaqarNotification(hook.HookBase):
|
|
share_api = api.API()
|
|
|
|
def _access_changed_trigger(self, context, func_name,
|
|
access_rules_ids, share_instance_id):
|
|
|
|
access = [self.db.share_access_get(context, rule_id)
|
|
for rule_id in access_rules_ids]
|
|
|
|
share_instance = self.db.share_instance_get(context, share_instance_id)
|
|
|
|
share = self.share_api.get(context, share_id=share_instance.share_id)
|
|
|
|
def rules_view(rules):
|
|
result = []
|
|
|
|
for rule in rules:
|
|
access_instance = None
|
|
|
|
for ins in rule.instance_mappings:
|
|
if ins.share_instance_id == share_instance_id:
|
|
access_instance = ins
|
|
break
|
|
else:
|
|
raise exception.InstanceNotFound(
|
|
instance_id=share_instance_id)
|
|
|
|
result.append({
|
|
'access_id': rule.id,
|
|
'access_instance_id': access_instance.id,
|
|
'access_type': rule.access_type,
|
|
'access_to': rule.access_to,
|
|
'access_level': rule.access_level,
|
|
})
|
|
return result
|
|
|
|
is_allow_operation = 'allow' in func_name
|
|
results = {
|
|
'share_id': share.share_id,
|
|
'share_instance_id': share_instance_id,
|
|
'export_locations': [
|
|
el.path for el in share_instance.export_locations],
|
|
'share_proto': share.share_proto,
|
|
'access_rules': rules_view(access),
|
|
'is_allow_operation': is_allow_operation,
|
|
'availability_zone': share_instance.availability_zone,
|
|
}
|
|
LOG.debug(results)
|
|
return results
|
|
|
|
def _execute_pre_hook(self, context, func_name, *args, **kwargs):
|
|
LOG.debug("\n PRE zaqar notification has been called for "
|
|
"method '%s'.\n", func_name)
|
|
if func_name == "deny_access":
|
|
LOG.debug("\nSending notification about denied access.\n")
|
|
data = self._access_changed_trigger(
|
|
context,
|
|
func_name,
|
|
kwargs.get('access_rules'),
|
|
kwargs.get('share_instance_id'),
|
|
)
|
|
self._send_notification(data)
|
|
|
|
def _execute_post_hook(self, context, func_name, pre_hook_data,
|
|
driver_action_results, *args, **kwargs):
|
|
LOG.debug("\n POST zaqar notification has been called for "
|
|
"method '%s'.\n", func_name)
|
|
if func_name == "allow_access":
|
|
LOG.debug("\nSending notification about allowed access.\n")
|
|
data = self._access_changed_trigger(
|
|
context,
|
|
func_name,
|
|
kwargs.get('access_rules'),
|
|
kwargs.get('share_instance_id'),
|
|
)
|
|
self._send_notification(data)
|
|
|
|
def _send_notification(self, data):
|
|
for queue_name in CONF.zaqar.zaqar_queues:
|
|
ZAQARCLIENT.queue_name = queue_name
|
|
message = {
|
|
"body": {
|
|
"example_message": (
|
|
"message generated at '%s'" % timeutils.utcnow()),
|
|
"data": data,
|
|
}
|
|
}
|
|
LOG.debug(
|
|
"\n Sending message %(m)s to '%(q)s' queue using '%(u)s' user "
|
|
"and '%(p)s' project.", {
|
|
'm': message,
|
|
'q': queue_name,
|
|
'u': CONF.zaqar.zaqar_username,
|
|
'p': CONF.zaqar.zaqar_project_name,
|
|
}
|
|
)
|
|
queue = ZAQARCLIENT.queue(queue_name)
|
|
queue.post(message)
|
|
|
|
def _execute_periodic_hook(self, context, periodic_hook_data,
|
|
*args, **kwargs):
|
|
LOG.debug("Periodic zaqar notification has been called. (Placeholder)")
|