committed

services/notificationIngestion.js

@@ -0,0 +1,50 @@
+const express = require('express');
+const { Kafka } = require('kafkajs');
+const router = express.Router();
+const Notification = require('../models/notification');
+const { MongoClient } = require('mongodb');
+const kafka = new Kafka({ clientId: 'notification-app', brokers: ['localhost:9092'] });
+const producer = kafka.producer();
+const mongoClient = new MongoClient('mongodb://localhost:27017');
+
+router.post('/notify', async (req, res) => {
+  try {
+    const { message, userId, priority, sendTime } = req.body;
+    
+    if (!message || !userId) {
+      return res.status(400).json({ error: 'Missing required fields' });
+    }
+
+    const notification = new Notification({
+      message,
+      userId,
+      priority: priority || 'normal',
+      sendTime: sendTime,
+      createdAt: new Date(),
+      status: 'pending'
+    });
+
+    await notification.save(); 
+    // await mongoClient.connect();
+    // const db = mongoClient.db('notifications');
+    // await db.collection('notification').insertOne(notification);
+  
+
+    // Publish to Kafka
+    await producer.connect();
+    await producer.send({
+      topic: 'notifications',
+      messages: [{ value: JSON.stringify(notification) }],
+    });
+
+    res.status(200).json({ message: 'Notification received' });
+  } catch (error) {
+    console.error('Error processing notification:', error);
+    res.status(500).json({ error: 'Internal server error' });
+  } finally {
+    await producer.disconnect();
+  }
+});
+
+module.exports = router;
+