fix: immutability in some more classes (MINOR) (#4179)

confluentinc · Dec 20, 2019 · cbd3bab · cbd3bab
1 parent c239990
commit cbd3bab
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/ksql-execution/src/main/java/io/confluent/ksql/execution/codegen/CodeGenSpec.java b/ksql-execution/src/main/java/io/confluent/ksql/execution/codegen/CodeGenSpec.java
@@ -32,6 +32,7 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import org.apache.kafka.connect.data.ConnectSchema;
 import org.apache.kafka.connect.data.Schema;
 
 @Immutable
@@ -239,14 +240,14 @@ public String toString() {
   @Immutable
   public static final class SchemaArgumentSpec extends BaseArgumentSpec {
 
-    private final Schema schema;
+    private final ConnectSchema schema;
 
     SchemaArgumentSpec(
         String name,
         Schema schema
     ) {
       super(name, Schema.class);
-      this.schema = requireNonNull(schema, "schema");
+      this.schema = (ConnectSchema) requireNonNull(schema, "schema").schema();
     }
 
     @Override

diff --git a/...ion/src/main/java/io/confluent/ksql/execution/expression/tree/CreateStructExpression.java b/...ion/src/main/java/io/confluent/ksql/execution/expression/tree/CreateStructExpression.java
@@ -67,6 +67,7 @@ public int hashCode() {
     return Objects.hash(fields);
   }
 
+  @Immutable
   public static class Field {
     private final String name;
     private final Expression value;