Commit 95f27fa0 authored by Lluis Gifre Renom's avatar Lluis Gifre Renom
Browse files

Common MessageBroker - NATS:

- added support for username and password
- added support for direct URI
parent 26ec4534
Loading
Loading
Loading
Loading
+15 −9
Original line number Diff line number Diff line
@@ -19,18 +19,24 @@ from common.message_broker.Message import Message
from .._Backend import _Backend
from .NatsBackendThread import NatsBackendThread

DEFAULT_NATS_URI = 'nats://127.0.0.1:4222'
#NATS_URI_TEMPLATE = 'nats://{:s}:{:s}@nats.{:s}.svc.cluster.local:{:s}' # with authentication
NATS_URI_TEMPLATE = 'nats://nats.{:s}.svc.cluster.local:{:s}'
NATS_URI_TEMPLATE_AUTH = 'nats://{:s}:{:s}@nats.{:s}.svc.cluster.local:{:s}'
NATS_URI_TEMPLATE_NOAUTH = 'nats://nats.{:s}.svc.cluster.local:{:s}'

class NatsBackend(_Backend):
    def __init__(self, **settings) -> None: # pylint: disable=super-init-not-called
        nats_uri = get_setting('NATS_URI', settings=settings, default=None)
        if nats_uri is None:
            nats_namespace   = get_setting('NATS_NAMESPACE', settings=settings)
            nats_client_port = get_setting('NATS_CLIENT_PORT', settings=settings)
        if nats_namespace is None or nats_client_port is None:
            nats_uri = get_setting('NATS_URI', settings=settings, default=DEFAULT_NATS_URI)
            nats_username    = get_setting('NATS_USERNAME', settings=settings, default=None)
            nats_password    = get_setting('NATS_PASSWORD', settings=settings, default=None)
            if nats_username is None or nats_password is None:
                nats_uri = NATS_URI_TEMPLATE_NOAUTH.format(
                    nats_namespace, nats_client_port)
            else:
            nats_uri = NATS_URI_TEMPLATE.format(nats_namespace, nats_client_port)
                nats_uri = NATS_URI_TEMPLATE_AUTH.format(
                    nats_username, nats_password, nats_namespace, nats_client_port)

        self._terminate = threading.Event()
        self._nats_backend_thread = NatsBackendThread(nats_uri)
        self._nats_backend_thread.start()