DQ-165: Prevent user YAML override

atlanhq · Aug 5, 2024 · 0cb18dd · 0cb18dd
1 parent e50cc62
commit 0cb18dd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...rg/apache/atlas/repository/store/graph/v2/preprocessor/contract/ContractPreProcessor.java b/...rg/apache/atlas/repository/store/graph/v2/preprocessor/contract/ContractPreProcessor.java
@@ -110,7 +110,9 @@ private void processCreateContract(AtlasEntity entity, EntityMutationContext con
 
         boolean contractSync = syncContractCertificateStatus(entity, contract);
         contractString = DataContract.serialize(contract);
-        entity.setAttribute(ATTR_CONTRACT, contractString);
+        if (!isContractYaml(entity)) {
+            entity.setAttribute(ATTR_CONTRACT, contractString);
+        }
         String contractStringJSON = DataContract.serializeJSON(contract);
         entity.setAttribute(ATTR_CONTRACT_JSON, contractStringJSON);
 
@@ -298,4 +300,8 @@ private static String getContractString(AtlasEntity entity) {
         }
         return contractString;
     }
+
+    private static boolean isContractYaml(AtlasEntity entity) {
+        return !StringUtils.isEmpty((String) entity.getAttribute(ATTR_CONTRACT));
+    }
 }