committed

This commit is contained in:
2025-01-10 13:03:41 +05:30
parent 4ba496b8eb
commit 52f746308f
17 changed files with 1689 additions and 0 deletions
+106
View File
@@ -0,0 +1,106 @@
const User = require('../models/user');
const Notification = require('../models/notification');
const { checkThrottling } = require('../queries/throttling');
const { isInQuietHours } = require('../queries/quietHours');
const { checkDuplication } = require('../queries/deduplication');
const { aggregateLowPriorityNotifications } = require('../queries/aggregation');
const { getUrgentAlerts } = require('../queries/urgentAlerts');
const { MongoClient } = require('mongodb');
const mongoClient = new MongoClient('mongodb://localhost:27017');
async function deliverNotifications() {
const urgentAlerts = await getUrgentAlerts();
for (const alert of urgentAlerts) {
await processNotification(alert);
}
const pendingNotifications = await Notification.find({ status: 'pending', priority: { $ne: 'high' } });
for (const notification of pendingNotifications) {
await processNotification(notification);
}
}
async function processNotification(notification) {
await mongoClient.connect();
const user = await User.findOne({ userId: notification.userId });
if (!user) {
console.error(`User not found for notification: ${notification._id}`);
return;
}
if (!(await checkThrottling(user.userId))) {
await rescheduleNotification(notification);
return;
}
if (isInQuietHours(user, new Date())) {
await rescheduleNotification(notification);
return;
}
if (await checkDuplication(notification)) {
const db = mongoClient.db('test');
await Notification.updateOne({ _id: notification._id }, { status: 'suppressed' });
// await db.collection('scheduledNotifications').deleteOne({ _id: notification._id });
return;
}
if (notification.priority === 'low') {
const aggregatedNotification = await aggregateLowPriorityNotifications(user.userId);
if (aggregatedNotification) {
await sendNotification(aggregatedNotification, user);
return;
}
}
await sendNotification(notification, user);
}
async function sendNotification(notification, user) {
for (const channel of user.preferences.channels) {
try {
switch (channel) {
case 'email':
await sendEmail(notification, user);
break;
case 'sms':
await sendSMS(notification, user);
break;
case 'push':
await sendPushNotification(notification, user);
break;
}
await Notification.updateOne({ _id: notification._id }, { status: 'sent' });
return;
} catch (error) {
console.error(`Failed to send notification via ${channel}:`, error);
}
}
await Notification.updateOne({ _id: notification._id }, { status: 'failed' });
}
async function rescheduleNotification(notification) {
const newSendTime = new Date(Date.now() + 60 * 60 * 1000); // Reschedule for 1 hour later
const db = mongoClient.db('test');
await Notification.updateOne({ _id: notification._id }, { sendTime: newSendTime });
await db.collection('scheduledNotifications').updateOne({ _id: notification._id }, { sendTime: newSendTime });
}
// Mock delivery functions
async function sendEmail(notification, user) {
console.log(`Sending email to ${user.userId}: ${notification.message}`);
}
async function sendSMS(notification, user) {
console.log(`Sending SMS to ${user.userId}: ${notification.message}`);
}
async function sendPushNotification(notification, user) {
console.log(`Sending push notification to ${user.userId}: ${notification.message}`);
}
module.exports = { deliverNotifications,sendNotification };