fix comments

akto-api-security · Aug 24, 2024 · d43833a · d43833a
1 parent 3caf0b0
commit d43833a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/apps/database-abstractor/src/main/java/com/akto/action/DbAction.java b/apps/database-abstractor/src/main/java/com/akto/action/DbAction.java
@@ -964,6 +964,7 @@ public String fetchDeactivatedCollections() {
         try {
             apiCollectionIds = DbLayer.fetchDeactivatedCollections();
         } catch (Exception e) {
+            loggerMaker.errorAndAddToDb(e, "Error in fetchDeactivatedCollections " + e.toString());
             return Action.ERROR.toUpperCase();
         }
         return Action.SUCCESS.toUpperCase();
@@ -974,6 +975,7 @@ public String updateUsage() {
             MetricTypes metric = MetricTypes.valueOf(metricType);
             DbLayer.updateUsage(metric, deltaUsage);
         } catch (Exception e) {
+            loggerMaker.errorAndAddToDb(e, "Error in updateUsage " + e.toString());
             return Action.ERROR.toUpperCase();
         }
         return Action.SUCCESS.toUpperCase();

diff --git a/libs/utils/src/main/java/com/akto/usage/UsageMetricCalculator.java b/libs/utils/src/main/java/com/akto/usage/UsageMetricCalculator.java
@@ -23,9 +23,12 @@
 import com.akto.dto.usage.UsageMetric;
 import com.akto.dto.usage.metadata.ActiveAccounts;
 import com.akto.log.LoggerMaker;
+import com.akto.util.Constants;
 import com.akto.util.enums.GlobalEnums.YamlTemplateSource;
 import com.google.gson.Gson;
 import com.mongodb.client.model.Filters;
+import com.mongodb.client.model.Projections;
+
 import org.bson.conversions.Bson;
 
 public class UsageMetricCalculator {
@@ -60,7 +63,7 @@ public static Set<Integer> getDeactivated() {
 
     public static Set<Integer> getDeactivatedLatest(){
         List<ApiCollection> deactivated = ApiCollectionsDao.instance
-                .findAll(Filters.eq(ApiCollection._DEACTIVATED, true));
+                .findAll(Filters.eq(ApiCollection._DEACTIVATED, true), Projections.include(Constants.ID));
         Set<Integer> deactivatedIds = new HashSet<>(
                 deactivated.stream().map(apiCollection -> apiCollection.getId()).collect(Collectors.toList()));