Salvatore Orlando 819c74ef30 Replace occurrences of registry.notify
This change replaces remaining occurences of the notify method with
calls to the publish method.
As NSX admin utilities heavily rely on callbacks, this change also
ensures that all callbacks are now accepting event payloads rather
thank kwargs.

Change-Id: I0450fff486898d6ab74086b7952dc27134cb77e2
2021-10-18 03:24:34 -07:00

66 lines
2.3 KiB
Python

# Copyright 2016 VMware, 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 neutron_lib.callbacks import registry
from oslo_log import log as logging
from vmware_nsx.shell.admin.plugins.common import constants
from vmware_nsx.shell.admin.plugins.common import utils as admin_utils
from vmware_nsx.shell.admin.plugins.nsxv3.resources import utils
import vmware_nsx.shell.resources as shell
LOG = logging.getLogger(__name__)
neutron_client = utils.NeutronDbClient()
@admin_utils.output_header
@admin_utils.unpack_payload
def nsx_rate_limit_show(resource, event, trigger, **kwargs):
"""Show the current NSX rate limit."""
nsxlib = utils.get_connected_nsxlib()
rate_limit = nsxlib.http_services.get_rate_limit()
LOG.info("Current NSX rate limit is %s", rate_limit)
@admin_utils.output_header
@admin_utils.unpack_payload
def nsx_rate_limit_update(resource, event, trigger, **kwargs):
"""Set the NSX rate limit
The default value is 40. 0 means no limit
"""
nsxlib = utils.get_connected_nsxlib()
rate_limit = None
if kwargs.get('property'):
properties = admin_utils.parse_multi_keyval_opt(kwargs['property'])
rate_limit = properties.get('value', None)
if rate_limit is None or not rate_limit.isdigit():
usage = ("nsxadmin -r rate-limit -o nsx-update "
"--property value=<new limit>")
LOG.error("Missing parameters. Usage: %s", usage)
return
nsxlib.http_services.update_rate_limit(rate_limit)
LOG.info("NSX rate limit was updated to %s", rate_limit)
registry.subscribe(nsx_rate_limit_show,
constants.RATE_LIMIT,
shell.Operations.SHOW.value)
registry.subscribe(nsx_rate_limit_update,
constants.RATE_LIMIT,
shell.Operations.NSX_UPDATE.value)