progress on kafka

This commit is contained in:
lennyaiko
2025-04-10 13:59:04 +01:00
parent 51480d664a
commit 34cba74580
6 changed files with 102 additions and 50 deletions
+15 -1
View File
@@ -1,6 +1,20 @@
from app import create_app from app import create_app
from app.integrations import KafkaIntegration
app = create_app() app = create_app()
if __name__ == "__main__": if __name__ == "__main__":
app.run(host="0.0.0.0", port=5000, debug=True) kafka = KafkaIntegration()
if kafka.is_kafka_running():
print("Kafka is running")
else:
print("Kafka is not running")
exit(1)
try:
message = kafka.receive_messages(
topic=
)
app.run(host="0.0.0.0", port=5000, debug=True)
Binary file not shown.
+4 -1
View File
@@ -9,5 +9,8 @@ class Config:
JWT_SECRET_KEY = os.getenv("JWT_SECRET_KEY", "your_jwt_secret") JWT_SECRET_KEY = os.getenv("JWT_SECRET_KEY", "your_jwt_secret")
DEBUG = True DEBUG = True
KAFKA_BROKER = "dev-events.simbrellang.net:9085" KAFKA_BROKER = "dev-events.simbrellang.net:9084"
KAFKA_PAYMENT_TOPIC = "PROCESS_PAYMENT" KAFKA_PAYMENT_TOPIC = "PROCESS_PAYMENT"
settings = Config()
+77 -46
View File
@@ -1,32 +1,31 @@
from confluent_kafka import Producer from confluent_kafka import Consumer
import json import json
import logging import logging
from app.config import settings from app.config import settings
import requests
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class KafkaIntegration: class KafkaIntegration:
_producer = None _consumer = None
_config = {
_consumer_config = {
"bootstrap.servers": settings.KAFKA_BROKER, "bootstrap.servers": settings.KAFKA_BROKER,
"client.id": "loan-service-producer", "group.id": "loan-service-consumer",
"acks": "all", "auto.offset.reset": "earliest",
"retries": 3, "enable.auto.commit": True,
"debug": "broker,topic,msg",
} }
@staticmethod @staticmethod
def _get_producer(): def _get_consumer():
"""Kafka producer""" """Kafka consumer"""
if not KafkaIntegration._producer: if not KafkaIntegration._consumer:
KafkaIntegration._producer = Producer(KafkaIntegration._config) KafkaIntegration._consumer = Consumer(KafkaIntegration._consumer_config)
logger.info( logger.info(
f"Connected to Kafka broker at {KafkaIntegration._config['bootstrap.servers']}" f"Consumer connected to Kafka broker at {KafkaIntegration._consumer_config['bootstrap.servers']}"
) )
return KafkaIntegration._consumer
return KafkaIntegration._producer
@staticmethod @staticmethod
def delivery_report(err, msg): def delivery_report(err, msg):
@@ -42,44 +41,76 @@ class KafkaIntegration:
) )
@staticmethod @staticmethod
def send_disbursement_request(disbursement_data, request_id): def receive_messages(topic, timeout=1.0):
""" """
Send disbursement request to PROCESS_PAYMENT topic Receive messages from a Kafka topic.
Args: :param topic: The Kafka topic to subscribe to
disbursement_data (dict): Disbursement data request payload :param timeout: Time to wait for a message (in seconds)
request_id (str): Unique request identifier (used as Kafka key) :return: The message value (decoded) or None if no message is received
Returns:
None
""" """
consumer = KafkaIntegration._get_consumer()
consumer.subscribe([topic])
try: try:
# proceed to send the disbursement request to Kafka msg = consumer.poll(timeout=timeout)
producer = KafkaIntegration._get_producer() if msg is None:
logger.debug(f"No message received from topic {topic} within timeout")
return None
if msg.error():
logger.error(f"Consumer error: {msg.error()}")
raise RuntimeError(f"Consumer error: {msg.error()}")
# sending disbursement request to Kafka # Decode and return the message value
producer.produce( message_value = msg.value().decode("utf-8")
"PROCESS_PAYMENT", logger.debug(
key=str(request_id), f"Received message from {msg.topic()} [{msg.partition()}] @ offset {msg.offset()}: {message_value}"
value=json.dumps(disbursement_data).encode("utf-8"),
callback=KafkaIntegration.delivery_report,
) )
return json.loads(message_value) if message_value else None
producer.poll(0)
logger.info(
f"Disbursement request {request_id} sent to Kafka: {disbursement_data}"
)
except Exception as e: except Exception as e:
logger.error( logger.error(f"Error while receiving message: {e}")
f"Failed to send disbursement request {str(request_id)} to Kafka: {str(e)}" raise
)
raise Exception(
f"Failed to send disbursement request {str(request_id)} to Kafka: {str(e)}"
)
@staticmethod @staticmethod
def flush(): def close_consumer():
"""Shutdown""" """Shutdown consumer"""
producer = KafkaIntegration._get_producer() consumer = KafkaIntegration._get_consumer()
producer.flush() consumer.clonse()
logger.info("Kafka consumer closed")
@staticmethod
def is_kafka_running():
"""
Check if Kafka broker is reachable by attempting to fetch metadata.
:return: True if Kafka is running, False otherwise
"""
try:
producer = KafkaIntegration._get_producer()
# Attempt to fetch metadata for a dummy topic
metadata = producer.list_topics(timeout=5)
if metadata.brokers: # If brokers are returned, Kafka is running
logger.info(f"Kafka broker at {settings.KAFKA_BROKER} is running")
return True
else:
logger.warning(f"No brokers found for {settings.KAFKA_BROKER}")
return False
except Exception as e:
logger.error(f"Kafka broker check failed: {e}")
return False
@staticmethod
def _call_disbursement_endpoint(
message, endpoint_url="http://localhost:8000/loans/disbursement"
):
"""Call an HTTP endpoint with the received message"""
try:
response = requests.post(endpoint_url, json=message, timeout=5)
response.raise_for_status()
logger.info(
f"Successfully sent message to {endpoint_url}: {response.status_code}"
)
print(response.json())
except requests.exceptions.RequestException as e:
logger.error(f"Failed to call endpoint {endpoint_url}: {e}")
raise
+6 -2
View File
@@ -1,6 +1,8 @@
from flask import Blueprint, request, jsonify, current_app from flask import Blueprint, request, jsonify, current_app
import requests import requests
from app.utils.auth import get_headers from app.utils.auth import get_headers
from app.integrations import KafkaIntegration
from threading import Thread
loan_bp = Blueprint("loan", __name__) loan_bp = Blueprint("loan", __name__)
@@ -154,8 +156,10 @@ def disbursement():
data = request.json data = request.json
api_url = f"{current_app.config['API_BASE_URL']}/Disbursement" api_url = f"{current_app.config['API_BASE_URL']}/Disbursement"
response = requests.post(api_url, json=data, headers=get_headers()) return jsonify({"requestId": data["requestId"]}), 200
return jsonify(response.json()), response.status_code
# response = requests.post(api_url, json=data, headers=get_headers())
# return jsonify(response.json()), response.status_code
@loan_bp.route("/collect-loan", methods=["POST"]) @loan_bp.route("/collect-loan", methods=["POST"])