Make Constraint serializable

Author: chenjian2664

core/trino-spi/src/main/java/io/trino/spi/connector/Constraint.java

@@ -13,6 +13,8 @@
  */
 package io.trino.spi.connector;
 
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonProperty;
 import io.trino.spi.expression.ConnectorExpression;
 import io.trino.spi.predicate.TupleDomain;
 
@@ -33,6 +35,19 @@ public class Constraint
     // variables referenced in the for ConnectorExpressionEvaluator to evaluate conjuncts of expression.
     private final Map<String, ColumnHandle> evaluationAssignments;
 
+    @JsonCreator
+    public Constraint(
+            @JsonProperty("summary") TupleDomain<ColumnHandle> summary,
+            @JsonProperty("expression") ConnectorExpression expression,
+            @JsonProperty("assignments") Map<String, ColumnHandle> assignments,
+            @JsonProperty("evaluationAssignments") Map<String, ColumnHandle> evaluationAssignments)
+    {
+        this.summary = requireNonNull(summary, "summary is null");
+        this.expression = requireNonNull(expression, "expression is null");
+        this.assignments = Map.copyOf(requireNonNull(assignments, "assignments is null"));
+        this.evaluationAssignments = Map.copyOf(requireNonNull(evaluationAssignments, "evaluationAssignments is null"));
+    }
+
     public static Constraint alwaysTrue()
     {
         return ALWAYS_TRUE;
@@ -53,17 +68,10 @@ public Constraint(TupleDomain<ColumnHandle> summary, ConnectorExpression express
         this(summary, expression, assignments, assignments);
     }
 
-    public Constraint(TupleDomain<ColumnHandle> summary, ConnectorExpression expression, Map<String, ColumnHandle> assignments, Map<String, ColumnHandle> evaluationAssignments)
-    {
-        this.summary = requireNonNull(summary, "summary is null");
-        this.expression = requireNonNull(expression, "expression is null");
-        this.assignments = Map.copyOf(requireNonNull(assignments, "assignments is null"));
-        this.evaluationAssignments = Map.copyOf(requireNonNull(evaluationAssignments, "evaluationAssignments is null"));
-    }
-
     /**
      * @return a predicate which is equivalent to, or looser than {@link #getExpression}, and should be AND-ed with, {@link #getExpression}.
      */
+    @JsonProperty("summary")
     public TupleDomain<ColumnHandle> getSummary()
     {
         return summary;
@@ -73,6 +81,7 @@ public TupleDomain<ColumnHandle> getSummary()
      * @return an expression predicate which is different from, and should be AND-ed with, {@link #getSummary}.
      * May include an engine-internal {@code $engine_predicate} conjunct that connectors should treat as opaque.
      */
+    @JsonProperty("expression")
     public ConnectorExpression getExpression()
     {
         return expression;
@@ -82,6 +91,7 @@ public ConnectorExpression getExpression()
      * @return mappings from variable names to column handles for the connector-translatable conjuncts of {@link #getExpression}.
      * Connectors may use {@code getAssignments().values()} to determine which columns are referenced by the predicate.
      */
+    @JsonProperty("assignments")
     public Map<String, ColumnHandle> getAssignments()
     {
         return assignments;
@@ -92,6 +102,7 @@ public Map<String, ColumnHandle> getAssignments()
      * Pass this to {@code ConnectorExpressionEvaluator} so it can resolve all variable bindings when evaluating
      * {@link #getExpression()} end-to-end. Empty means no $engine_predicate conjunct to evaluate.
      */
+    @JsonProperty("evaluationAssignments")
     public Map<String, ColumnHandle> getEvaluationAssignments()
     {
         return evaluationAssignments;