Added backward compatibilty to mark summaries as new

Author: Ark2307

apps/dashboard/src/main/java/com/akto/listener/InitializerListener.java

@@ -2970,11 +2970,32 @@ private static void moveOktaOidcSSO(BackwardCompatibility backwardCompatibility)
         }
     }
 
+    private static void markSummariesAsVulnerable(BackwardCompatibility backwardCompatibility){
+        // case for the customers where vulnerable are stored in new collection and only testing runs are marked as new.
+
+        if(backwardCompatibility.getMarkSummariesVulnerable() == 0){
+
+            List<ObjectId> summaryIds = VulnerableTestingRunResultDao.instance.summaryIdsStoredForVulnerableTests();
+            if(!summaryIds.isEmpty()){
+                TestingRunResultSummariesDao.instance.updateMany(
+                    Filters.in(Constants.ID, summaryIds), 
+                    Updates.set(TestingRunResultSummary.IS_NEW_TESTING_RUN_RESULT_SUMMARY, true)
+                );
+            }
+
+            BackwardCompatibilityDao.instance.updateOne(
+                Filters.eq("_id", backwardCompatibility.getId()),
+                Updates.set(BackwardCompatibility.MARK_SUMMARIES_NEW_FOR_VULNERABLE, Context.now())
+            );
+        }
+    }
+
     public static void setBackwardCompatibilities(BackwardCompatibility backwardCompatibility){
         if (DashboardMode.isMetered()) {
             initializeOrganizationAccountBelongsTo(backwardCompatibility);
             setOrganizationsInBilling(backwardCompatibility);
         }
+        markSummariesAsVulnerable(backwardCompatibility);
         setAktoDefaultNewUI(backwardCompatibility);
         dropLastCronRunInfoField(backwardCompatibility);
         fetchIntegratedConnections(backwardCompatibility);

libs/dao/src/main/java/com/akto/dao/testing/VulnerableTestingRunResultDao.java

@@ -1,5 +1,6 @@
 package com.akto.dao.testing;
 
+import java.util.ArrayList;
 import java.util.List;
 
 import org.bson.conversions.Bson;
@@ -11,6 +12,10 @@
 import com.akto.dto.testing.TestingRunResult;
 import com.akto.dto.testing.TestingRunResultSummary;
 import com.akto.util.Constants;
+import com.mongodb.BasicDBObject;
+import com.mongodb.client.MongoCursor;
+import com.mongodb.client.model.Accumulators;
+import com.mongodb.client.model.Aggregates;
 import com.mongodb.client.model.CreateCollectionOptions;
 import com.mongodb.client.model.Filters;
 import com.mongodb.client.model.Projections;
@@ -51,15 +56,7 @@ public boolean isStoredInVulnerableCollection(ObjectId objectId, boolean isSumma
                     Filters.eq(TestingRunResultSummary.IS_NEW_TESTING_RUN_RESULT_SUMMARY, true)
                 );
                 boolean isNew = TestingRunResultSummariesDao.instance.count(filter) > 0;
-                if(!isNew){
-                    TestingRunResultSummary trrs = TestingRunResultSummariesDao.instance.findOne(
-                        Filters.eq(Constants.ID, objectId),
-                        Projections.include(TestingRunResultSummary.TESTING_RUN_ID)
-                    );
-                    return TestingRunDao.instance.isStoredInVulnerableCollection(trrs.getTestingRunId());
-                }else{
-                    return isNew;
-                }
+                return isNew;
             } catch (Exception e) {
                 e.printStackTrace();
                 return false;
@@ -100,6 +97,27 @@ public List<TestingRunResult> findAll(Bson q, Bson projection, boolean isStoredI
         return TestingRunResultDao.instance.findAll(q, projection);
     }
 
+    public List<ObjectId> summaryIdsStoredForVulnerableTests(){
+        String groupedId = "summaries";
+        List<Bson> pipeLine = new ArrayList<>();
+        pipeLine.add(
+            Aggregates.group(groupedId, Accumulators.addToSet("summaryIds", "$" + TestingRunResult.TEST_RUN_RESULT_SUMMARY_ID))
+        );
+        try {
+            MongoCursor<BasicDBObject> cursor = instance.getMCollection().aggregate(pipeLine, BasicDBObject.class).cursor();
+            List<ObjectId> uniqueSummaries = new ArrayList<>();
+            while (cursor.hasNext()) {
+                BasicDBObject dbObject = cursor.next();
+                uniqueSummaries = (List<ObjectId>) dbObject.get("summaryIds");
+            }
+            return uniqueSummaries;
+        } catch (Exception e) {
+            e.printStackTrace();
+            return new ArrayList<>();
+        }
+        
+    }
+
     @Override
     public String getCollName() {
         return "vulnerable_testing_run_results";

libs/dao/src/main/java/com/akto/dto/BackwardCompatibility.java

@@ -103,6 +103,9 @@ public class BackwardCompatibility {
     public static final String MOVE_OKTA_OIDC_SSO = "moveOktaOidcSSO";
     private int moveOktaOidcSSO;
 
+    public static final String MARK_SUMMARIES_NEW_FOR_VULNERABLE = "markSummariesVulnerable";
+    private int markSummariesVulnerable;
+
     public BackwardCompatibility(int id, int dropFilterSampleData, int resetSingleTypeInfoCount, int dropWorkflowTestResult,
                                  int readyForNewTestingFramework,int addAktoDataTypes, boolean deploymentStatusUpdated,
                                  int authMechanismData, boolean mirroringLambdaTriggered, int deleteAccessListFromApiToken,
@@ -112,7 +115,7 @@ public BackwardCompatibility(int id, int dropFilterSampleData, int resetSingleTy
                                  int loginSignupGroups, int vulnerableApiUpdationVersionV1, int riskScoreGroups,
                                  int deactivateCollections, int disableAwsSecretPii, int apiCollectionAutomatedField, 
                                  int automatedApiGroups, int addAdminRoleIfAbsent, int dropSpecialCharacterApiCollections, int fixApiAccessType,
-                                 int addDefaultFilters, int moveAzureSamlToNormalSaml, int deleteOptionsAPIs, int moveOktaOidcSSO) {
+                                 int addDefaultFilters, int moveAzureSamlToNormalSaml, int deleteOptionsAPIs, int moveOktaOidcSSO, int markSummariesVulnerable) {
         this.id = id;
         this.dropFilterSampleData = dropFilterSampleData;
         this.resetSingleTypeInfoCount = resetSingleTypeInfoCount;
@@ -145,6 +148,7 @@ public BackwardCompatibility(int id, int dropFilterSampleData, int resetSingleTy
         this.moveAzureSamlToNormalSaml = moveAzureSamlToNormalSaml;
         this.deleteOptionsAPIs = deleteOptionsAPIs;
         this.moveOktaOidcSSO = moveOktaOidcSSO;
+        this.markSummariesVulnerable = markSummariesVulnerable;
     }
 
     public BackwardCompatibility() {
@@ -437,4 +441,12 @@ public int getMoveOktaOidcSSO() {
     public void setMoveOktaOidcSSO(int moveOktaOidcSSO) {
         this.moveOktaOidcSSO = moveOktaOidcSSO;
     }
+
+    public int getMarkSummariesVulnerable() {
+        return markSummariesVulnerable;
+    }
+
+    public void setMarkSummariesVulnerable(int markSummariesVulnerable) {
+        this.markSummariesVulnerable = markSummariesVulnerable;
+    }
 }