Add kafka settings

bite/bite/settings.py

@@ -158,6 +158,11 @@ MQTT_BROKER = {
     'PORT': '1883',
 }
 
+KAFKA_BROKER = {
+    'HOST': 'kafka',
+    'PORT': '9092',
+}
+
 # If no local_settings.py is availble in the current folder let's try to
 # load it from the application root
 try:

bite/telemetry/management/commands/dispatcher.py

@@ -39,6 +39,8 @@ class Command(BaseCommand):
 
     MQTT_HOST = settings.MQTT_BROKER['HOST']
     MQTT_PORT = int(settings.MQTT_BROKER['PORT'])
+    KAFKA_HOST = settings.KAFKA_BROKER['HOST']
+    KAFKA_PORT = int(settings.KAFKA_BROKER['PORT'])
     producer = None
 
     @sync_to_async
@@ -85,7 +87,9 @@ class Command(BaseCommand):
         while True:
             try:
                 self.producer = KafkaProducer(
-                    bootstrap_servers='localhost:9092',
+                    bootstrap_servers='{}:{}'.format(
+                        self.KAFKA_HOST, self.KAFKA_PORT
+                    ),
                     value_serializer=lambda v: json.dumps(v).encode('utf-8'),
                     retries=5
                 )

bite/telemetry/management/commands/handler.py

@@ -29,13 +29,13 @@ from django.core.exceptions import ObjectDoesNotExist
 from api.models import Device
 from telemetry.models import Telemetry
 
-MQTT_HOST = settings.MQTT_BROKER['HOST']
-MQTT_PORT = int(settings.MQTT_BROKER['PORT'])
-
 
 class Command(BaseCommand):
     help = 'MQTT to DB deamon'
 
+    KAFKA_HOST = settings.KAFKA_BROKER['HOST']
+    KAFKA_PORT = int(settings.KAFKA_BROKER['PORT'])
+
     def get_device(self, serial):
         try:
             return Device.objects.get(serial=serial)
@@ -56,7 +56,9 @@ class Command(BaseCommand):
                 consumer = KafkaConsumer(
                     "telemetry",
                     value_deserializer=lambda m: json.loads(m.decode('utf8')),
-                    bootstrap_servers='localhost:9092'
+                    bootstrap_servers='{}:{}'.format(
+                        self.KAFKA_HOST, self.KAFKA_PORT
+                    ),
                 )
                 break
             except NoBrokersAvailable: