jffwang
/
hetu-core
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!778 when cache is enabled reuse exchange is supported 

Merge pull request !778 from sandeep/bugfix-reuseExchangeSupportWithCache
This commit is contained in:
i-robot 2021-03-27 12:18:59 +08:00 committed by Gitee
parent d8f284ca56 24a7b08cea
commit f413d70aa0
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
presto-main/src/main/java/io/prestosql/query/CachedSqlQueryExecution.java
View File

@ -365,6 +365,7 @@ public class CachedSqlQueryExecution
private final Metadata metadata;
private final Map<String, TableHandle> tables;
private final Map<ConnectorTransactionHandle, ConnectorTransactionHandle> connectorTransactionHandleMap; // Old ConnectorTransactionHandle from cached plan to new ConnectorTransactionHandle from metadata
private HashMap<UUID, UUID> reuseTableScanNewMappingIdMap;
TableHandleRewriter(Session session, Analysis analysis, Metadata metadata)
{
@ -383,6 +384,7 @@ public class CachedSqlQueryExecution
});
}
this.tables = tables;
this.reuseTableScanNewMappingIdMap = new HashMap();
}
@Override
@ -394,8 +396,14 @@ public class CachedSqlQueryExecution
// in other nodes if necessary
connectorTransactionHandleMap.put(tableHandle.getTransaction(), newTableHandle.getTransaction());
UUID newMappingID = node.getReuseTableScanMappingId();
if (node.getStrategy() != ReuseExchangeOperator.STRATEGY.REUSE_STRATEGY_DEFAULT) {
newMappingID = reuseTableScanNewMappingIdMap.computeIfAbsent(newMappingID, k -> UUID.randomUUID());
}
// Return a new table handle with the ID, output symbols, assignments, and enforced constraints of the cached table handle
return new TableScanNode(node.getId(), newTableHandle, node.getOutputSymbols(), node.getAssignments(), node.getEnforcedConstraint(), node.getPredicate(), ReuseExchangeOperator.STRATEGY.REUSE_STRATEGY_DEFAULT, new UUID(0, 0), 0, false);
return new TableScanNode(node.getId(), newTableHandle, node.getOutputSymbols(), node.getAssignments(), node.getEnforcedConstraint(),
node.getPredicate(), node.getStrategy(), newMappingID, node.getConsumerTableScanNodeCount(), false);
}
@Override