notifyExpiredTokens fixes

backend/src/services/resource-cleanup/resource-cleanup-queue.ts

@@ -22,7 +22,7 @@ type TDailyResourceCleanUpQueueServiceFactoryDep = {
   secretFolderVersionDAL: Pick<TSecretFolderVersionDALFactory, "pruneExcessVersions">;
   snapshotDAL: Pick<TSnapshotDALFactory, "pruneExcessSnapshots">;
   secretSharingDAL: Pick<TSecretSharingDALFactory, "pruneExpiredSharedSecrets" | "pruneExpiredSecretRequests">;
-  serviceTokenService: Pick<TServiceTokenServiceFactory, "notifyExpiredTokens">;
+  serviceTokenService: Pick<TServiceTokenServiceFactory, "notifyExpiringTokens">;
   queueService: TQueueServiceFactory;
 };
 
@@ -53,7 +53,7 @@ export const dailyResourceCleanUpQueueServiceFactory = ({
     await secretVersionDAL.pruneExcessVersions();
     await secretVersionV2DAL.pruneExcessVersions();
     await secretFolderVersionDAL.pruneExcessVersions();
-    await serviceTokenService.notifyExpiredTokens();
+    await serviceTokenService.notifyExpiringTokens();
     logger.info(`${QueueName.DailyResourceCleanUp}: queue task completed`);
   });
 

backend/src/services/service-token/service-token-dal.ts

@@ -28,17 +28,9 @@ export const serviceTokenDALFactory = (db: TDbClient) => {
     }
   };
 
-  const findExpiredTokens = async (tx?: Knex, batchSize = 1000) => {
+  const findExpiringTokens = async (tx?: Knex, batchSize = 500, offset = 0) => {
     try {
-      const allDocs: { name: string; projectName: string; createdByEmail: string; id: string; projectId: string }[] =
-        [];
-      let offset = 0;
-      let hasMoreRecords = true;
-
-      while (hasMoreRecords) {
-        // eslint-disable-next-line
       const batch: { name: string; projectName: string; createdByEmail: string; id: string; projectId: string }[] =
-          // eslint-disable-next-line no-await-in-loop
         await (tx || db.replicaNode())(TableName.ServiceToken)
           .leftJoin<TUsers>(
             TableName.Users,
@@ -61,18 +53,11 @@ export const serviceTokenDALFactory = (db: TDbClient) => {
           .limit(batchSize)
           .offset(offset);
 
-        if (batch.length === 0) {
-          hasMoreRecords = false;
-        } else {
-          allDocs.push(...batch);
-          offset += batchSize;
-        }
-      }
-
-      return allDocs;
+      return batch;
     } catch (err) {
       throw new DatabaseError({ error: err, name: "FindExpiredTokens" });
     }
   };
-  return { ...stOrm, findById, findExpiredTokens };
+
+  return { ...stOrm, findById, findExpiringTokens };
 };

backend/src/services/service-token/service-token-service.ts

@@ -189,18 +189,29 @@ export const serviceTokenServiceFactory = ({
     return { ...serviceToken, lastUsed: new Date(), orgId: project.orgId };
   };
 
-  const notifyExpiredTokens = async () => {
+  const notifyExpiringTokens = async () => {
     const appCfg = getConfig();
+    let processedCount = 0;
+    let hasMoreRecords = true;
+    let offset = 0;
+    const batchSize = 500;
 
-    const expiredTokens = await serviceTokenDAL.findExpiredTokens();
-    if (expiredTokens.length === 0) return;
+    while (hasMoreRecords) {
+      // eslint-disable-next-line no-await-in-loop
+      const expiringTokens = await serviceTokenDAL.findExpiringTokens(undefined, batchSize, offset);
 
+      if (expiringTokens.length === 0) {
+        hasMoreRecords = false;
+        break;
+      }
+
+      // eslint-disable-next-line no-await-in-loop
       await Promise.all(
-      expiredTokens.map(async (token) => {
+        expiringTokens.map(async (token) => {
           try {
             await smtpService.sendMail({
               recipients: [token.createdByEmail],
-            subjectLine: "Your Service Token is about to expire",
+              subjectLine: "Service Token Expiry Notice",
               template: SmtpTemplates.ServiceTokenExpired,
               substitutions: {
                 tokenName: token.name,
@@ -214,6 +225,12 @@ export const serviceTokenServiceFactory = ({
           }
         })
       );
+
+      processedCount += expiringTokens.length;
+      offset += batchSize;
+    }
+
+    return processedCount;
   };
 
   return {
@@ -222,6 +239,6 @@ export const serviceTokenServiceFactory = ({
     getServiceToken,
     getProjectServiceTokens,
     fnValidateServiceToken,
-    notifyExpiredTokens
+    notifyExpiringTokens
   };
 };

backend/src/services/smtp/templates/serviceTokenExpired.handlebars

@@ -7,10 +7,10 @@
   </head>
 
   <body>
-    <h2>Your Service Token is about to expire</h2>
+    <h2>Service Token Expiry Notice</h2>
     <p>Your service token <strong>"{{tokenName}}"</strong> will expire within 24 hours.</p>
     
-    <p>This token is currently being on project "{{projectName}}". If this token is still needed for your workflow, please create a new one before it expires.</p>
+    <p>This token is currently being used on project "{{projectName}}". If this token is still needed for your workflow, please create a new one before it expires.</p>
     
     <a href="{{url}}">Create New Token</a>