diff --git a/docker/docker-bake.hcl b/docker/docker-bake.hcl index 54498ba..8d10d66 100644 --- a/docker/docker-bake.hcl +++ b/docker/docker-bake.hcl @@ -59,4 +59,4 @@ target "pyciemss-worker-base" { target "pyciemss-worker" { inherits = ["_platforms", "pyciemss-worker-base"] -} \ No newline at end of file +} diff --git a/service/settings.py b/service/settings.py index 96cf7fa..7285b66 100644 --- a/service/settings.py +++ b/service/settings.py @@ -19,6 +19,7 @@ class Settings(BaseSettings): RABBITMQ_PORT: int = 5672 RABBITMQ_USERNAME: str = "guest" RABBITMQ_PASSWORD: str = "guest" + RABBITMQ_SSL: bool = False settings = Settings() diff --git a/service/utils/rabbitmq.py b/service/utils/rabbitmq.py index 39c5ad4..b8b0120 100644 --- a/service/utils/rabbitmq.py +++ b/service/utils/rabbitmq.py @@ -6,9 +6,23 @@ from settings import settings creds = pika.PlainCredentials(settings.RABBITMQ_USERNAME, settings.RABBITMQ_PASSWORD) -conn_config = pika.ConnectionParameters( - host=settings.RABBITMQ_HOST, port=settings.RABBITMQ_PORT, credentials=creds -) + +if settings.RABBITMQ_SSL: + conn_config = pika.URLParameters( + "amqps://" + + settings.RABBITMQ_USERNAME + + ":" + + settings.RABBITMQ_PASSWORD + + "@" + + settings.RABBITMQ_HOST + + ":" + + str(settings.RABBITMQ_PORT) + + "/" + ) +else: + conn_config = pika.ConnectionParameters( + host=settings.RABBITMQ_HOST, port=settings.RABBITMQ_PORT, credentials=creds + ) def mock_rabbitmq_consumer():