org.apache.calcite.rex.RexTableInputRef java code examples

Common ways to obtain RexTableInputRef

private void myMethod () {
 RexTableInputRef r =RexTableInputRef.RelTableRef tableRef;RelDataType type;RexTableInputRef.of(tableRef, index, type)
Smart code suggestions by Tabnine
}

 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) &&

Javadoc

Variable which references a column of a table occurrence in a relational plan.

This object is used by org.apache.calcite.rel.metadata.BuiltInMetadata.ExpressionLineageand org.apache.calcite.rel.metadata.BuiltInMetadata.AllPredicates.

Given a relational expression, its purpose is to be able to reference uniquely the provenance of a given expression. For that, it uses a unique table reference (contained in a RelTableRef) and an column index within the table.

For example, A.#0.$3 + 2 column $3 in the 0occurrence of table A in the plan.

Note that this kind of RexNode is an auxiliary data structure with a very specific purpose and should not be used in relational expressions.

Most used methods

Popular in Java

Updating database using SQL prepared statement
orElseThrow (Optional)
Return the contained value, if present, otherwise throw an exception to be created by the provided s
getOriginalFilename (MultipartFile)
Return the original filename in the client's filesystem.This may contain path information depending
scheduleAtFixedRate (Timer)
Point (java.awt)
A point representing a location in (x,y) coordinate space, specified in integer precision.
BigInteger (java.math)
An immutable arbitrary-precision signed integer.FAST CRYPTOGRAPHY This implementation is efficient f
HashMap (java.util)
HashMap is an implementation of Map. All optional operations are supported.All elements are permitte
Callable (java.util.concurrent)
A task that returns a result and may throw an exception. Implementors define a single method with no
Notification (javax.management)
Reflections (org.reflections)
Reflections one-stop-shop objectReflections scans your classpath, indexes the metadata, allows you t
PhpStorm for WordPress

How to useRexTableInputRef in org.apache.calcite.rex

Best Java code snippets using org.apache.calcite.rex.RexTableInputRef (Showing top 20 results out of 315)

How to use
RexTableInputRef
in
org.apache.calcite.rex