Merge pull request #1093 from OneCommunityGlobal/huijie-recalculation…

…-tangible-hours-quick-fix

Huijie - fix recalculation for tangible hours

src/controllers/timeEntryController.js

@@ -1373,8 +1373,15 @@ const timeEntrycontroller = function (TimeEntry) {
   const recalculateHoursByCategoryAllUsers = async function (req, res) {
     const session = await mongoose.startSession();
     session.startTransaction();
+    let keepAliveInterval;
 
     try {
+      res.setHeader('Content-Type', 'text/plain');
+      res.setHeader('Transfer-Encoding', 'chunked');
+      keepAliveInterval = setInterval(() => {
+        res.write('Processing... keep connection alive\n');
+      }, 150 * 1000); // interval of 150 seconds
+
       const userprofiles = await UserProfile.find({}, '_id').lean();
 
       const recalculationPromises = userprofiles.map(async (userprofile) => {
@@ -1385,13 +1392,17 @@ const timeEntrycontroller = function (TimeEntry) {
       await Promise.all(recalculationPromises);
 
       await session.commitTransaction();
-      return res.status(200).send({
-        message: 'finished the recalculation for hoursByCategory for all users',
-      });
+      clearInterval(keepAliveInterval);
+      res.write('finished the recalculation for hoursByCategory for all users\n');
+      return res.end();
     } catch (err) {
       await session.abortTransaction();
+      if (keepAliveInterval) {
+        clearInterval(keepAliveInterval);
+      }
       logger.logException(err);
-      return res.status(500).send({ error: err.toString() });
+      res.write(`error: ${err.toString()}\n`);
+      return res.end();
     } finally {
       session.endSession();
     }