entity_auditx index Update field limits only if needed

(cherry picked from commit fc2bf65)
atlanhq · Sep 29, 2023 · 889025c · 889025c
1 parent 1e2a94e
commit 889025c
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/repository/src/main/java/org/apache/atlas/repository/audit/ESBasedAuditRepository.java b/repository/src/main/java/org/apache/atlas/repository/audit/ESBasedAuditRepository.java
@@ -294,7 +294,7 @@ void createSession() throws AtlasException {
                 LOG.info("Create ES index for entity audits in ES Based Audit Repo");
                 createAuditIndex();
             }
-            if (isFieldLimitDifferent()) {
+            if (shouldUpdateFieldLimitSetting()) {
                 LOG.info("Updating ES total field limit");
                 updateFieldLimit();
             }
@@ -328,24 +328,25 @@ private boolean createAuditIndex() throws IOException {
         return isSuccess(response);
     }
 
-    private boolean isFieldLimitDifferent() {
-        JsonNode fieldLimit;
+    private boolean shouldUpdateFieldLimitSetting() {
+        JsonNode currentFieldLimit;
         try {
-            fieldLimit = getIndexFieldLimit();
+            currentFieldLimit = getIndexFieldLimit();
         } catch (IOException e) {
             LOG.error("Problem while retrieving the index field limit!", e);
             return false;
         }
         Integer fieldLimitFromConfigurationFile = configuration.getInt(TOTAL_FIELD_LIMIT);
-        return fieldLimit == null || fieldLimitFromConfigurationFile.equals(fieldLimit.intValue());
+        return currentFieldLimit == null || fieldLimitFromConfigurationFile > currentFieldLimit.asInt();
     }
 
     private JsonNode getIndexFieldLimit() throws IOException {
         Request request = new Request("GET", INDEX_NAME + "/_settings");
         Response response = lowLevelClient.performRequest(request);
         ObjectMapper objectMapper = new ObjectMapper();
-        String fieldName = INDEX_NAME + ".settings.index.mapping.total_fields.limit";
-        return objectMapper.readTree(copyToString(response.getEntity().getContent(), Charset.defaultCharset())).get(fieldName);
+        String fieldPath = String.format("/%s/settings/index/mapping/total_fields/limit", INDEX_NAME);
+
+        return objectMapper.readTree(copyToString(response.getEntity().getContent(), Charset.defaultCharset())).at(fieldPath);
     }
 
     private void updateFieldLimit() {