Fix primary key constraint for nullable keys

linkedin · Jun 21, 2024 · 2ec1a0d · 2ec1a0d
1 parent c3fa16a
commit 2ec1a0d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/hoptimator-catalog/src/main/java/com/linkedin/hoptimator/catalog/ScriptImplementor.java b/hoptimator-catalog/src/main/java/com/linkedin/hoptimator/catalog/ScriptImplementor.java
@@ -170,7 +170,7 @@ public SqlNode visit(SqlCall call) {
    * Implements a CREATE TABLE...WITH... DDL statement.
    *
    * N.B. the following magic:
-   *  - field 'KEY' is treated as a PRIMARY KEY
+   *  - non-nullable field 'KEY' is treated as a PRIMARY KEY
    */
   class ConnectorImplementor implements ScriptImplementor {
     private final String database;
@@ -192,7 +192,8 @@ public void implement(SqlWriter w) {
       (new CompoundIdentifierImplementor(database, name)).implement(w);
       SqlWriter.Frame frame1 = w.startList("(", ")");
       (new RowTypeSpecImplementor(rowType)).implement(w);
-      if (rowType.getField("KEY", true, false) != null) {
+      RelDataTypeField key = rowType.getField("KEY", true, false);
+      if (key != null && key.getType().isNullable() == false) {
         w.sep(",");
         w.literal("PRIMARY KEY (KEY) NOT ENFORCED");
       }