clarify deleteColumn doc

apache · Oct 12, 2023 · 6daf2cc · 6daf2cc
1 parent 7d87fd3
commit 6daf2cc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/api/src/main/java/com/datastrato/graviton/rel/TableChange.java b/api/src/main/java/com/datastrato/graviton/rel/TableChange.java
@@ -184,10 +184,11 @@ static TableChange updateColumnPosition(String[] fieldNames, ColumnPosition newP
   /**
    * Create a TableChange for deleting a field.
    *
-   * <p>If the field does not exist, the change will result in an {@link IllegalArgumentException}.
+   * <p>If the field does not exist, the change will result in an {@link IllegalArgumentException}
+   * unless {@code ifExists} is true.
    *
    * @param fieldNames Field names of the column to delete.
-   * @param ifExists Silence the error if column doesn't exist during drop.
+   * @param ifExists If true, the change will succeed even if the field does not exist.
    * @return A TableChange for the delete.
    */
   static TableChange deleteColumn(String[] fieldNames, Boolean ifExists) {

diff --git a/catalogs/catalog-hive/src/test/java/com/datastrato/graviton/catalog/hive/TestHiveTable.java b/catalogs/catalog-hive/src/test/java/com/datastrato/graviton/catalog/hive/TestHiveTable.java
@@ -472,6 +472,12 @@ public void testAlterHiveTable() throws IOException {
     Assertions.assertArrayEquals(expected, alteredTable.columns());
 
     // test delete column change
+    hiveCatalog
+        .asTableCatalog()
+        .alterTable(
+            NameIdentifier.of(tableIdentifier.namespace(), "test_hive_table_new"),
+            TableChange.deleteColumn(new String[] {"not_exist_col"}, true));
+
     hiveCatalog
         .asTableCatalog()
         .alterTable(