Default data element pointer type

rafaelbey · Nov 29, 2024 · f381189 · f381189
1 parent e9b21ac
commit f381189
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/...c/main/java/org/finos/legend/engine/protocol/pure/v1/model/data/DataElementReference.java b/...c/main/java/org/finos/legend/engine/protocol/pure/v1/model/data/DataElementReference.java
@@ -14,9 +14,21 @@
 
 package org.finos.legend.engine.protocol.pure.v1.model.data;
 
+import com.fasterxml.jackson.annotation.JsonSetter;
 import org.finos.legend.engine.protocol.pure.v1.model.context.PackageableElementPointer;
+import org.finos.legend.engine.protocol.pure.v1.model.context.PackageableElementType;
 
 public class DataElementReference extends EmbeddedData
 {
     public PackageableElementPointer dataElement;
+
+    @JsonSetter("dataElement")
+    public void setDataElement(PackageableElementPointer dataElement)
+    {
+        this.dataElement = dataElement;
+        if (this.dataElement.type == null)
+        {
+            this.dataElement.type = PackageableElementType.DATA;
+        }
+    }
 }