ssl support for rabbitmq (#84)

* ssl support for rabbitmq * formatting
DARPA-ASKEM · Apr 11, 2024 · 1cfa5fd · 1cfa5fd
1 parent a927830
commit 1cfa5fd
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 4 deletions.
diff --git 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"]
-}
+}
diff --git 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
@@ -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():