0607cfd6fd
We are using Redis-based persistent name storage, so unrelated references should be cleaned when listener doesn't exist any more. Change-Id: I464a17d6de1d1679244db51f54f620017da071b3
82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
#
|
|
# 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 logging
|
|
|
|
from oslo_config import cfg
|
|
import redis
|
|
|
|
from oslo_messaging._drivers.zmq_driver.matchmaker import base
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
matchmaker_redis_opts = [
|
|
cfg.StrOpt('host',
|
|
default='127.0.0.1',
|
|
help='Host to locate redis.'),
|
|
cfg.IntOpt('port',
|
|
default=6379,
|
|
help='Use this port to connect to redis host.'),
|
|
cfg.StrOpt('password',
|
|
default='',
|
|
secret=True,
|
|
help='Password for Redis server (optional).'),
|
|
]
|
|
|
|
|
|
class RedisMatchMaker(base.MatchMakerBase):
|
|
|
|
def __init__(self, conf, *args, **kwargs):
|
|
super(RedisMatchMaker, self).__init__(conf, *args, **kwargs)
|
|
self.conf.register_opts(matchmaker_redis_opts, "matchmaker_redis")
|
|
|
|
self._redis = redis.StrictRedis(
|
|
host=self.conf.matchmaker_redis.host,
|
|
port=self.conf.matchmaker_redis.port,
|
|
password=self.conf.matchmaker_redis.password,
|
|
)
|
|
|
|
def _target_to_key(self, target):
|
|
attributes = ['topic', 'exchange', 'server']
|
|
prefix = "ZMQ-target"
|
|
key = ":".join((getattr(target, attr) or "*") for attr in attributes)
|
|
return "%s-%s" % (prefix, key)
|
|
|
|
def _get_keys_by_pattern(self, pattern):
|
|
return self._redis.keys(pattern)
|
|
|
|
def _get_hosts_by_key(self, key):
|
|
return self._redis.lrange(key, 0, -1)
|
|
|
|
def register(self, target, hostname):
|
|
key = self._target_to_key(target)
|
|
if hostname not in self._get_hosts_by_key(key):
|
|
self._redis.lpush(key, hostname)
|
|
|
|
def unregister(self, target, hostname):
|
|
key = self._target_to_key(target)
|
|
self._redis.lrem(key, 0, hostname)
|
|
|
|
def get_hosts(self, target):
|
|
pattern = self._target_to_key(target)
|
|
if "*" not in pattern:
|
|
# pattern have no placeholders, so this is valid key
|
|
return self._get_hosts_by_key(pattern)
|
|
|
|
hosts = []
|
|
for key in self._get_keys_by_pattern(pattern):
|
|
hosts.extend(self._get_hosts_by_key(key))
|
|
return hosts
|