return null; List<Integer> cols = tabToOriginColumns.get(inputRef.getTableRef()); if (cols == null) { cols = new ArrayList<>(); cols.add(inputRef.getIndex()); tabToOriginColumns.put(inputRef.getTableRef(), cols);
tRef.getTable().getRowType().getFieldList().get(foreignKeyPos).getType(); RexTableInputRef foreignKeyColumnRef = RexTableInputRef.of(tRef, foreignKeyPos, foreignKeyColumnType); int uniqueKeyPos = constraint.getColumnPairs().get(pos).target; RexTableInputRef uniqueKeyColumnRef = RexTableInputRef.of(nonFkTable, uniqueKeyPos, nonFkTable.getTable().getRowType().getFieldList().get(uniqueKeyPos).getType()); if (ecT.getEquivalenceClassesMap().containsKey(uniqueKeyColumnRef) &&
@Override public RexNode visitTableInputRef(RexTableInputRef inputRef) { if (ec != null) { Set<RexTableInputRef> s = ec.get(inputRef); if (s != null) { inputRef = s.iterator().next(); } } if (tableMapping != null) { inputRef = RexTableInputRef.of( tableMapping.get(inputRef.getTableRef()), inputRef.getIndex(), inputRef.getType()); } return inputRef; } };
@Test public void testExpressionLineageTwoColumns() { // mgr is column 3 in catalog.sales.emp // deptno is column 7 in catalog.sales.emp final RelNode rel = convertSql("select mgr, deptno from emp"); final RelMetadataQuery mq = RelMetadataQuery.instance(); final RexNode ref1 = RexInputRef.of(0, rel.getRowType().getFieldList()); final Set<RexNode> r1 = mq.getExpressionLineage(rel, ref1); assertThat(r1.size(), is(1)); final RexTableInputRef result1 = (RexTableInputRef) r1.iterator().next(); assertTrue(result1.getQualifiedName().equals(EMP_QNAME)); assertThat(result1.getIndex(), is(3)); final RexNode ref2 = RexInputRef.of(1, rel.getRowType().getFieldList()); final Set<RexNode> r2 = mq.getExpressionLineage(rel, ref2); assertThat(r2.size(), is(1)); final RexTableInputRef result2 = (RexTableInputRef) r2.iterator().next(); assertTrue(result2.getQualifiedName().equals(EMP_QNAME)); assertThat(result2.getIndex(), is(7)); assertThat(result1.getIdentifier(), is(result2.getIdentifier())); }
@Test public void testExpressionLineageInnerJoinRight() { // ename is column 0 in catalog.sales.bonus final RelNode rel = convertSql("select bonus.ename from emp join bonus using (ename)"); final RelMetadataQuery mq = RelMetadataQuery.instance(); final RexNode ref = RexInputRef.of(0, rel.getRowType().getFieldList()); final Set<RexNode> r = mq.getExpressionLineage(rel, ref); assertThat(r.size(), is(1)); final RexTableInputRef result = (RexTableInputRef) r.iterator().next(); assertTrue(result.getQualifiedName().equals(ImmutableList.of("CATALOG", "SALES", "BONUS"))); assertThat(result.getIndex(), is(0)); }
@Override public Void visitTableInputRef(RexTableInputRef ref) { occurrences.add(ref.getTableRef()); return super.visitTableInputRef(ref); } };
public static RexTableInputRef of(RelTableRef tableRef, int index, RelDataType type) { return new RexTableInputRef(tableRef, index, type); }
@Test public void testExpressionLineageUnion() { // sal is column 5 in catalog.sales.emp final RelNode rel = convertSql("select sal from (\n" + " select * from emp union all select * from emp) " + "where deptno = 10"); final RelNode tableRel = convertSql("select * from emp"); final RelMetadataQuery mq = RelMetadataQuery.instance(); final RexNode ref = RexInputRef.of(0, rel.getRowType().getFieldList()); final Set<RexNode> r = mq.getExpressionLineage(rel, ref); final String inputRef = RexInputRef.of(5, tableRel.getRowType().getFieldList()).toString(); assertThat(r.size(), is(2)); for (RexNode result : r) { final String resultString = result.toString(); assertThat(resultString, startsWith(EMP_QNAME.toString())); assertThat(resultString, endsWith(inputRef)); } Iterator<RexNode> it = r.iterator(); assertThat(((RexTableInputRef) it.next()).getIdentifier(), not(((RexTableInputRef) it.next()).getIdentifier())); }
@Override public RexNode visitTableInputRef(RexTableInputRef ref) { Collection<Integer> c = exprsLineage.get(ref.toString()); if (c.isEmpty()) { // Cannot map expression throw Util.FoundOne.NULL; } int pos = c.iterator().next(); if (rewritingMapping != null) { pos = rewritingMapping.getTargetOpt(pos); if (pos == -1) { // Cannot map expression throw Util.FoundOne.NULL; } } if (node != null) { return rexBuilder.makeInputRef(node, pos); } return rexBuilder.makeInputRef(ref.getType(), pos); }
@Override public RexNode visitTableInputRef(RexTableInputRef ref) { Collection<Integer> c = exprsLineage.get(ref); if (c.isEmpty()) { // Cannot map expression throw Util.FoundOne.NULL; } int pos = c.iterator().next(); if (rewritingMapping != null) { pos = rewritingMapping.getTargetOpt(pos); if (pos == -1) { // Cannot map expression throw Util.FoundOne.NULL; } } if (node != null) { return rexBuilder.makeInputRef(node, pos); } return rexBuilder.makeInputRef(ref.getType(), pos); }
@Override public RexNode visitTableInputRef(RexTableInputRef inputRef) { if (tableMapping != null) { inputRef = RexTableInputRef.of( tableMapping.get(inputRef.getTableRef()), inputRef.getIndex(), inputRef.getType()); } if (ec != null) { Set<RexTableInputRef> s = ec.get(inputRef); if (s != null) { inputRef = s.iterator().next(); } } return inputRef; } };
@Test public void testExpressionLineageTwoColumnsSwapped() { // deptno is column 7 in catalog.sales.emp // mgr is column 3 in catalog.sales.emp final RelNode rel = convertSql("select deptno, mgr from emp"); final RelMetadataQuery mq = RelMetadataQuery.instance(); final RexNode ref1 = RexInputRef.of(0, rel.getRowType().getFieldList()); final Set<RexNode> r1 = mq.getExpressionLineage(rel, ref1); assertThat(r1.size(), is(1)); final RexTableInputRef result1 = (RexTableInputRef) r1.iterator().next(); assertTrue(result1.getQualifiedName().equals(EMP_QNAME)); assertThat(result1.getIndex(), is(7)); final RexNode ref2 = RexInputRef.of(1, rel.getRowType().getFieldList()); final Set<RexNode> r2 = mq.getExpressionLineage(rel, ref2); assertThat(r2.size(), is(1)); final RexTableInputRef result2 = (RexTableInputRef) r2.iterator().next(); assertTrue(result2.getQualifiedName().equals(EMP_QNAME)); assertThat(result2.getIndex(), is(3)); assertThat(result1.getIdentifier(), is(result2.getIdentifier())); }
@Test public void testExpressionLineageInnerJoinLeft() { // ename is column 1 in catalog.sales.emp final RelNode rel = convertSql("select ename from emp,dept"); final RelMetadataQuery mq = RelMetadataQuery.instance(); final RexNode ref = RexInputRef.of(0, rel.getRowType().getFieldList()); final Set<RexNode> r = mq.getExpressionLineage(rel, ref); assertThat(r.size(), is(1)); final RexTableInputRef result = (RexTableInputRef) r.iterator().next(); assertTrue(result.getQualifiedName().equals(EMP_QNAME)); assertThat(result.getIndex(), is(1)); }
@Override public Void visitTableInputRef(RexTableInputRef ref) { occurrences.add(ref.getTableRef()); return super.visitTableInputRef(ref); } };
public static RexTableInputRef of(RelTableRef tableRef, int index, RelDataType type) { return new RexTableInputRef(tableRef, index, type); }
@Test public void testExpressionLineageUnion() { // sal is column 5 in catalog.sales.emp final RelNode rel = convertSql("select sal from (\n" + " select * from emp union all select * from emp) " + "where deptno = 10"); final RelNode tableRel = convertSql("select * from emp"); final RelMetadataQuery mq = RelMetadataQuery.instance(); final RexNode ref = RexInputRef.of(0, rel.getRowType().getFieldList()); final Set<RexNode> r = mq.getExpressionLineage(rel, ref); final String inputRef = RexInputRef.of(5, tableRel.getRowType().getFieldList()).toString(); assertThat(r.size(), is(2)); for (RexNode result : r) { final String resultString = result.toString(); assertThat(resultString, startsWith(EMP_QNAME.toString())); assertThat(resultString, endsWith(inputRef)); } Iterator<RexNode> it = r.iterator(); assertThat(((RexTableInputRef) it.next()).getIdentifier(), not(((RexTableInputRef) it.next()).getIdentifier())); }
@Override public RexNode visitTableInputRef(RexTableInputRef inputRef) { if (tableMapping != null) { inputRef = RexTableInputRef.of( tableMapping.get(inputRef.getTableRef()), inputRef.getIndex(), inputRef.getType()); } if (ec != null) { Set<RexTableInputRef> s = ec.get(inputRef); if (s != null) { inputRef = s.iterator().next(); } } return inputRef; } };
@Test public void testExpressionLineageTwoColumnsSwapped() { // deptno is column 7 in catalog.sales.emp // mgr is column 3 in catalog.sales.emp final RelNode rel = convertSql("select deptno, mgr from emp"); final RelMetadataQuery mq = RelMetadataQuery.instance(); final RexNode ref1 = RexInputRef.of(0, rel.getRowType().getFieldList()); final Set<RexNode> r1 = mq.getExpressionLineage(rel, ref1); assertThat(r1.size(), is(1)); final RexTableInputRef result1 = (RexTableInputRef) r1.iterator().next(); assertEquals(result1.getQualifiedName(), EMP_QNAME); assertThat(result1.getIndex(), is(7)); final RexNode ref2 = RexInputRef.of(1, rel.getRowType().getFieldList()); final Set<RexNode> r2 = mq.getExpressionLineage(rel, ref2); assertThat(r2.size(), is(1)); final RexTableInputRef result2 = (RexTableInputRef) r2.iterator().next(); assertEquals(result2.getQualifiedName(), EMP_QNAME); assertThat(result2.getIndex(), is(3)); assertThat(result1.getIdentifier(), is(result2.getIdentifier())); }
@Test public void testExpressionLineageInnerJoinRight() { // ename is column 0 in catalog.sales.bonus final RelNode rel = convertSql("select bonus.ename from emp join bonus using (ename)"); final RelMetadataQuery mq = RelMetadataQuery.instance(); final RexNode ref = RexInputRef.of(0, rel.getRowType().getFieldList()); final Set<RexNode> r = mq.getExpressionLineage(rel, ref); assertThat(r.size(), is(1)); final RexTableInputRef result = (RexTableInputRef) r.iterator().next(); assertEquals(result.getQualifiedName(), ImmutableList.of("CATALOG", "SALES", "BONUS")); assertThat(result.getIndex(), is(0)); }
tRef.getTable().getRowType().getFieldList().get(foreignKeyPos).getType(); RexTableInputRef foreignKeyColumnRef = RexTableInputRef.of(tRef, foreignKeyPos, foreignKeyColumnType); int uniqueKeyPos = constraint.getColumnPairs().get(pos).target; RexTableInputRef uniqueKeyColumnRef = RexTableInputRef.of(nonFkTable, uniqueKeyPos, nonFkTable.getTable().getRowType().getFieldList().get(uniqueKeyPos).getType()); if (ecT.getEquivalenceClassesMap().containsKey(uniqueKeyColumnRef) &&