Added Cron Job Scheduling

EventBus/index.js

@@ -28,23 +28,21 @@ app.post("/events", async (req, res) => {
       });
     }
 
-    axios.post("http://localhost:4001/events", formData, {
-      headers: formData.getHeaders(),
-    }).then(() => {
-      res.send({ status: "OK" });
-    });
+    const headers = formData.getHeaders();
 
-    axios.post("http://localhost:4002/events", formData, {
-      headers: formData.getHeaders(),
-    }).then(() => {
-      res.send({ status: "OK" });
-    });
+    await axios.post("http://localhost:4001/events", formData, {
+      headers,
+    })
 
-    axios.post("http://localhost:4003/events", formData, {
-      headers: formData.getHeaders(),
-    }).then(() => {
-      res.send({ status: "OK" });
-    });
+    await axios.post("http://localhost:4002/events", formData, {
+      headers,
+    })
+
+    await axios.post("http://localhost:4003/events", formData, {
+      headers,
+    })
+
+    res.send({ status: "OK" });
 
   } catch (err) {
     console.log(err.message);

UsageMntrServ/controllers/DailyUsageController.js

@@ -41,8 +41,30 @@ const addNewDailyUsageInstance = async (req, res) => {
         res.status(500).json({ message: "Internal server error." });
     }
 
+}
+
+const resetDailyLimit = async (req, res) => {
+    try {
+        const dailyUsages = await DailyUsage.find();
+
+        if (!dailyUsages) {
+            return res.status(404).json({ message: "Daily usage not found." });
+        }
+
+        dailyUsages.forEach(async (dailyUsage) => {
+            dailyUsage.usedBandwidth = 0;
+            await dailyUsage.save();
+        });
+
+        res.status(200).json({ message: "Daily usage reset." });
+    } catch (error) {
+        console.error(error);
+        res.status(500).json({ message: "Internal server error." });
+    }
+
 }
 module.exports = {
     getDailyUsageById,
-    addNewDailyUsageInstance
+    addNewDailyUsageInstance,
+    resetDailyLimit
 }

UsageMntrServ/index.js

@@ -1,15 +1,31 @@
+require("dotenv").config();
+
 const express = require("express");
 const bodyParser = require("body-parser");
 const cors = require("cors");
 const axios = require("axios");
-const { getDailyUsageById, addNewDailyUsageInstance } = require("./controllers/DailyUsageController");
+const cron = require('node-cron');
+const db = require("mongoose");
+const { getDailyUsageById, addNewDailyUsageInstance, resetDailyLimit } = require("./controllers/DailyUsageController");
+
 
 const app = express();
 app.use(bodyParser.json());
 app.use(cors());
 
+
 app.get("/usage", getDailyUsageById);
 app.post("/createUser", addNewDailyUsageInstance);
+app.post("/resetDailyLimit", resetDailyLimit);
+
+
+cron.schedule('0 0 * * *', async () => {
+  console.log("UsageMntrServ: Resetting daily limit...");
+  await axios.post("http://localhost:4001/resetDailyLimit").catch((err) => {
+    console.log(err.message);
+  });
+});
+
 
 app.post("/events", async (req, res) => {
   const { type } = req.body;
@@ -42,6 +58,14 @@ app.post("/events", async (req, res) => {
 });
 
 
-app.listen(4002, () => {
-  console.log("UsageMntrServ: Listening on 4002");
-});
+db.connect(process.env.MONGO_URI)
+  .then((result) => {
+    console.log("Connected to the Database...");
+    // Listen for Requests
+    app.listen(4002, () => {
+      console.log("UsageMntrServ: Listening on 4002");
+    });
+  })
+  .catch((err) => {
+    console.error(err);
+  });

UsageMntrServ/package.json

@@ -12,8 +12,10 @@
     "axios": "^1.6.2",
     "body-parser": "^1.20.2",
     "cors": "^2.8.5",
+    "dotenv": "^16.3.1",
     "express": "^4.18.2",
     "mongoose": "^8.0.3",
+    "node-cron": "^3.0.3",
     "nodemon": "^3.0.2"
   }
 }