fix: Put constraintName on the table, not column

erezrokah · erezrokah · commit 353673271984 · 2023-08-22T18:48:02.000+02:00
diff --git a/lib/src/main/java/io/cloudquery/helper/ArrowHelper.java b/lib/src/main/java/io/cloudquery/helper/ArrowHelper.java
@@ -149,9 +149,6 @@ public static Schema toArrowSchema(Table table) {
       Map<String, String> metadata = new HashMap<>();
       metadata.put("cq:extension:unique", column.isUnique() ? "true" : "false");
       metadata.put("cq:extension:primary_key", column.isPrimaryKey() ? "true" : "false");
-      if (column.getConstraintName() != null) {
-        metadata.put("cq:extension:constraint_name", column.getConstraintName());
-      }
       metadata.put("cq:extension:incremental", column.isIncrementalKey() ? "true" : "false");
       Field field =
           new Field(
@@ -171,6 +168,7 @@ public static Schema toArrowSchema(Table table) {
     if (table.getParent() != null) {
       metadata.put("cq:table_depends_on", table.getParent().getName());
     }
+    metadata.put("cq:extension:constraint_name", table.getConstraintName());
     return new Schema(asList(fields), metadata);
   }
 
diff --git a/lib/src/main/java/io/cloudquery/schema/Column.java b/lib/src/main/java/io/cloudquery/schema/Column.java
@@ -10,7 +10,6 @@
 public class Column {
   private String name;
   private String description;
-  private String constraintName;
   private ArrowType type;
   private ColumnResolver resolver;
   private boolean primaryKey;
diff --git a/lib/src/main/java/io/cloudquery/schema/Table.java b/lib/src/main/java/io/cloudquery/schema/Table.java
@@ -121,6 +121,7 @@ public static int maxDepth(List<Table> tables) {
   }
 
   @NonNull private String name;
+  @Builder.Default private String constraintName = "";
   private TableResolver resolver;
   private String title;
   private String description;

Original file line number	Diff line number	Diff line change
`@@ -121,6 +121,7 @@ public static int maxDepth(List<Table> tables) {`
`121`	`121`	`}`
`122`	`122`
`123`	`123`	`@NonNull private String name;`
	`124`	`+ @Builder.Default private String constraintName = "";`
`124`	`125`	`private TableResolver resolver;`
`125`	`126`	`private String title;`
`126`	`127`	`private String description;`