Remove un-needed changes

apache · Dec 27, 2024 · 2fe117f · 2fe117f
1 parent d009bac
commit 2fe117f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/xtable-core/src/test/java/org/apache/xtable/TestSparkDeltaTable.java b/xtable-core/src/test/java/org/apache/xtable/TestSparkDeltaTable.java
@@ -134,7 +134,15 @@ public void upsertRows(List<Row> upsertRows) {
   }
 
   @SneakyThrows
+  @Override
   public void deleteRows(List<Row> deleteRows) {
+    String idsToDelete =
+        deleteRows.stream().map(row -> row.get(0).toString()).collect(Collectors.joining(", "));
+    deltaTable.delete("id in (" + idsToDelete + ")");
+  }
+
+  @SneakyThrows
+  public void mergeDeleteRows(List<Row> deleteRows) {
     List<Row> deletes = testDeltaHelper.transformForUpsertsOrDeletes(deleteRows, false);
     Dataset<Row> deleteDataset =
         sparkSession.createDataFrame(deletes, testDeltaHelper.getTableStructSchema());