Create Attribute Type Rules for Interval Join, Scatter Plot, Sort Partitions, Type Casting

core/amber/src/main/scala/edu/uci/ics/texera/workflow/operators/aggregate/AggregationOperation.scala

@@ -22,7 +22,7 @@ import java.sql.Timestamp
             }
           },
           "then": {
-            "enum": ["integer", "long", "double"]
+            "enum": ["integer", "long", "double", "timestamp"]
           }
         },
         {

core/amber/src/main/scala/edu/uci/ics/texera/workflow/operators/intervalJoin/IntervalJoinOpDesc.scala

@@ -3,7 +3,7 @@ package edu.uci.ics.texera.workflow.operators.intervalJoin
 import com.fasterxml.jackson.annotation.{JsonProperty, JsonPropertyDescription}
 import com.fasterxml.jackson.databind.annotation.JsonDeserialize
 import com.google.common.base.Preconditions
-import com.kjetland.jackson.jsonSchema.annotations.JsonSchemaTitle
+import com.kjetland.jackson.jsonSchema.annotations.{JsonSchemaInject, JsonSchemaTitle}
 import edu.uci.ics.amber.engine.architecture.deploysemantics.layer.OpExecConfig
 import edu.uci.ics.texera.workflow.common.metadata.annotations.{
   AutofillAttributeName,
@@ -23,6 +23,20 @@ import edu.uci.ics.texera.workflow.common.workflow.HashPartition
   * 1. The tuples in both inputs come in ascending order
   * 2. The left input join key takes as points, join condition is: left key in the range of (right key, right key + constant)
   */
+@JsonSchemaInject(json = """
+{
+  "attributeTypeRules": {
+    "leftAttributeName": {
+      "enum": ["integer", "long", "double", "timestamp"]
+    },
+    "rightAttributeName": {
+      "const": {
+        "$data": "leftAttributeName"
+      }
+    }
+  }
+}
+""")
 class IntervalJoinOpDesc extends OperatorDescriptor {
 
   @JsonProperty(required = true)

core/amber/src/main/scala/edu/uci/ics/texera/workflow/operators/sortPartitions/SortPartitionsOpDesc.scala

@@ -2,7 +2,7 @@ package edu.uci.ics.texera.workflow.operators.sortPartitions
 
 import com.fasterxml.jackson.annotation.{JsonProperty, JsonPropertyDescription}
 import com.google.common.base.Preconditions
-import com.kjetland.jackson.jsonSchema.annotations.JsonSchemaTitle
+import com.kjetland.jackson.jsonSchema.annotations.{JsonSchemaInject, JsonSchemaTitle}
 import edu.uci.ics.amber.engine.architecture.deploysemantics.layer.OpExecConfig
 import edu.uci.ics.texera.workflow.common.metadata.annotations.AutofillAttributeName
 import edu.uci.ics.texera.workflow.common.metadata.{
@@ -15,6 +15,15 @@ import edu.uci.ics.texera.workflow.common.operators.OperatorDescriptor
 import edu.uci.ics.texera.workflow.common.tuple.schema.{OperatorSchemaInfo, Schema}
 import edu.uci.ics.texera.workflow.common.workflow.RangePartition
 
+@JsonSchemaInject(json = """
+{
+  "attributeTypeRules": {
+    "sortAttributeName":{
+      "enum": ["integer", "long", "double"]
+    }
+  }
+}
+""")
 class SortPartitionsOpDesc extends OperatorDescriptor {
 
   @JsonProperty(required = true)

core/amber/src/main/scala/edu/uci/ics/texera/workflow/operators/typecasting/TypeCastingUnit.java

@@ -2,10 +2,72 @@
 
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.fasterxml.jackson.annotation.JsonPropertyDescription;
+import com.kjetland.jackson.jsonSchema.annotations.JsonSchemaInject;
 import com.kjetland.jackson.jsonSchema.annotations.JsonSchemaTitle;
 import edu.uci.ics.texera.workflow.common.metadata.annotations.AutofillAttributeName;
 import edu.uci.ics.texera.workflow.common.tuple.schema.AttributeType;
 
+@JsonSchemaInject(json = 
+"{" +
+"  \"attributeTypeRules\": {" +
+"    \"attribute\": {" +
+"      \"allOf\": [" +
+"        {" +
+"          \"if\": {" +
+"            \"resultType\": {" +
+"              \"valEnum\": [\"integer\"]" +
+"            }" +
+"          }," +
+"          \"then\": {" +
+"            \"enum\": [\"string\", \"long\", \"double\", \"boolean\"]" +
+"          }" +
+"        }," +
+"        {" +
+"          \"if\": {" +
+"            \"resultType\": {" +
+"              \"valEnum\": [\"double\"]" +
+"            }" +
+"          }," +
+"          \"then\": {" +
+"            \"enum\": [\"string\", \"integer\", \"long\", \"boolean\"]" +
+"          }" +
+"        }," +
+"        {" +
+"          \"if\": {" +
+"            \"resultType\": {" +
+"              \"valEnum\": [\"boolean\"]" +
+"            }" +
+"          }," +
+"          \"then\": {" +
+"            \"enum\": [\"string\", \"integer\", \"long\", \"double\"]" +
+"          }" +
+"        }," +
+"        {" +
+"          \"if\": {" +
+"            \"resultType\": {" +
+"              \"valEnum\": [\"long\"]" +
+"            }" +
+"          }," +
+"          \"then\": {" +
+"            \"enum\": [\"string\", \"integer\", \"double\", \"boolean\", \"timestamp\"]" +
+"          }" +
+"        }," +
+"        {" +
+"          \"if\": {" +
+"            \"resultType\": {" +
+"              \"valEnum\": [\"timestamp\"]" +
+"            }" +
+"          }," +
+"          \"then\": {" +
+"            \"enum\": [\"string\", \"long\"]" +
+"          }" +
+"        }" +
+"        " +
+"      ]" +
+"    }" +
+"  }" +
+"}"
+)
 public class TypeCastingUnit {
     @JsonProperty(required = true)
     @JsonSchemaTitle("Attribute")

core/amber/src/main/scala/edu/uci/ics/texera/workflow/operators/visualization/scatterplot/ScatterplotOpDesc.java

@@ -2,6 +2,7 @@
 
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.fasterxml.jackson.annotation.JsonPropertyDescription;
+import com.kjetland.jackson.jsonSchema.annotations.JsonSchemaInject;
 import com.kjetland.jackson.jsonSchema.annotations.JsonSchemaTitle;
 import edu.uci.ics.amber.engine.architecture.deploysemantics.layer.OpExecConfig;
 import edu.uci.ics.amber.engine.architecture.deploysemantics.layer.OpExecFunc;
@@ -34,6 +35,17 @@
  * This is the description of the operator
  */
 
+@JsonSchemaInject(json =
+"{" +
+"  \"attributeTypeRules\": {" +
+"    \"xColumn\":{" +
+"      \"enum\": [\"integer\", \"double\"]" +
+"    }," +
+"    \"yColumn\":{" +
+"      \"enum\": [\"integer\", \"double\"]" +
+"    }" +
+"  }" +
+"}")
 public class ScatterplotOpDesc extends VisualizationOperator {
     @JsonProperty(required = true)
     @JsonSchemaTitle("X-Column")