From deec503c92232d320e19ea1f2bbe88eb4faa1dda Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C5=81ukasz=20Osipiuk?= Date: Tue, 19 Oct 2021 19:02:59 +0200 Subject: [PATCH] Rename method --- .../planner/optimizations/BeginTableWrite.java | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/core/trino-main/src/main/java/io/trino/sql/planner/optimizations/BeginTableWrite.java b/core/trino-main/src/main/java/io/trino/sql/planner/optimizations/BeginTableWrite.java index bb9d59b7e68e..e0dfa7a820a5 100644 --- a/core/trino-main/src/main/java/io/trino/sql/planner/optimizations/BeginTableWrite.java +++ b/core/trino-main/src/main/java/io/trino/sql/planner/optimizations/BeginTableWrite.java @@ -195,14 +195,14 @@ public WriterTarget getWriterTarget(PlanNode node) DeleteNode deleteNode = (DeleteNode) node; DeleteTarget delete = deleteNode.getTarget(); return new DeleteTarget( - Optional.of(findTableScanHandle(deleteNode.getSource())), + Optional.of(findTableScanHandleForDeleteOrUpdate(deleteNode.getSource())), delete.getSchemaTableName()); } if (node instanceof UpdateNode) { UpdateNode updateNode = (UpdateNode) node; UpdateTarget update = updateNode.getTarget(); return new UpdateTarget( - Optional.of(findTableScanHandle(updateNode.getSource())), + Optional.of(findTableScanHandleForDeleteOrUpdate(updateNode.getSource())), update.getSchemaTableName(), update.getUpdatedColumns(), update.getUpdatedColumnHandles()); @@ -253,7 +253,7 @@ private WriterTarget createWriterTarget(WriterTarget target) throw new IllegalArgumentException("Unhandled target type: " + target.getClass().getSimpleName()); } - private TableHandle findTableScanHandle(PlanNode node) + private TableHandle findTableScanHandleForDeleteOrUpdate(PlanNode node) { if (node instanceof TableScanNode) { TableScanNode tableScanNode = (TableScanNode) node; @@ -261,23 +261,23 @@ private TableHandle findTableScanHandle(PlanNode node) return tableScanNode.getTable(); } if (node instanceof FilterNode) { - return findTableScanHandle(((FilterNode) node).getSource()); + return findTableScanHandleForDeleteOrUpdate(((FilterNode) node).getSource()); } if (node instanceof ProjectNode) { - return findTableScanHandle(((ProjectNode) node).getSource()); + return findTableScanHandleForDeleteOrUpdate(((ProjectNode) node).getSource()); } if (node instanceof SemiJoinNode) { - return findTableScanHandle(((SemiJoinNode) node).getSource()); + return findTableScanHandleForDeleteOrUpdate(((SemiJoinNode) node).getSource()); } if (node instanceof JoinNode) { JoinNode joinNode = (JoinNode) node; - return findTableScanHandle(joinNode.getLeft()); + return findTableScanHandleForDeleteOrUpdate(joinNode.getLeft()); } if (node instanceof AssignUniqueId) { - return findTableScanHandle(((AssignUniqueId) node).getSource()); + return findTableScanHandleForDeleteOrUpdate(((AssignUniqueId) node).getSource()); } if (node instanceof MarkDistinctNode) { - return findTableScanHandle(((MarkDistinctNode) node).getSource()); + return findTableScanHandleForDeleteOrUpdate(((MarkDistinctNode) node).getSource()); } throw new IllegalArgumentException("Invalid descendant for DeleteNode or UpdateNode: " + node.getClass().getName()); }