public static RelDataType createTypeFromProjection(
RelDataType type,
List<String> columnNameList,
RelDataTypeFactory typeFactory,
boolean caseSensitive) {
// If the names in columnNameList and type have case-sensitive differences,
// the resulting type will use those from type. These are presumably more
// canonical.
final List<RelDataTypeField> fields = new ArrayList<RelDataTypeField>(columnNameList.size());
for (String name : columnNameList) {
RelDataTypeField field = type.getField(name, caseSensitive);
fields.add(type.getFieldList().get(field.getIndex()));
}
return typeFactory.createStructType(fields);
}
/**
* Creates a record type with specified field names.
*
* <p>The array of field names may be null, or any of the names within it can be null. We
* recommend using explicit names where possible, because it makes it much easier to figure out
* the intent of fields when looking at planner output.
*
* @param typeFactory Type factory
* @param exprs Expressions
* @param names Field names, may be null, or elements may be null
* @return Record type
*/
public static RelDataType createStructType(
RelDataTypeFactory typeFactory, final RexNode[] exprs, final String[] names) {
return typeFactory.createStructType(
new RelDataTypeFactory.FieldInfo() {
public int getFieldCount() {
return exprs.length;
}
public String getFieldName(int index) {
if (names == null) {
return "$f" + index;
}
final String name = names[index];
if (name == null) {
return "$f" + index;
}
return name;
}
public RelDataType getFieldType(int index) {
return exprs[index].getType();
}
});
}
public void onRegister(RelDataTypeFactory typeFactory) {
rowType = typeFactory.createStructType(columnList);
collationList = deduceMonotonicity(this);
}
/** Initializes this catalog reader. */
protected void init() {
final RelDataType intType = typeFactory.createSqlType(SqlTypeName.INTEGER);
final RelDataType varchar10Type = typeFactory.createSqlType(SqlTypeName.VARCHAR, 10);
final RelDataType varchar20Type = typeFactory.createSqlType(SqlTypeName.VARCHAR, 20);
final RelDataType timestampType = typeFactory.createSqlType(SqlTypeName.TIMESTAMP);
final RelDataType booleanType = typeFactory.createSqlType(SqlTypeName.BOOLEAN);
final RelDataType rectilinearCoordType =
typeFactory.createStructType(new RelDataType[] {intType, intType}, new String[] {"X", "Y"});
// TODO jvs 12-Feb-2005: register this canonical instance with type
// factory
addressType =
new ObjectSqlType(
SqlTypeName.STRUCTURED,
new SqlIdentifier("ADDRESS", SqlParserPos.ZERO),
false,
Arrays.asList(
new RelDataTypeFieldImpl("STREET", 0, varchar20Type),
new RelDataTypeFieldImpl("CITY", 1, varchar20Type),
new RelDataTypeFieldImpl("ZIP", 1, intType),
new RelDataTypeFieldImpl("STATE", 1, varchar20Type)),
RelDataTypeComparability.None);
// Register "SALES" schema.
MockSchema salesSchema = new MockSchema("SALES");
registerSchema(salesSchema);
// Register "EMP" table.
MockTable empTable = new MockTable(this, salesSchema, "EMP");
empTable.addColumn("EMPNO", intType);
empTable.addColumn("ENAME", varchar20Type);
empTable.addColumn("JOB", varchar10Type);
empTable.addColumn("MGR", intType);
empTable.addColumn("HIREDATE", timestampType);
empTable.addColumn("SAL", intType);
empTable.addColumn("COMM", intType);
empTable.addColumn("DEPTNO", intType);
empTable.addColumn("SLACKER", booleanType);
registerTable(empTable);
// Register "DEPT" table.
MockTable deptTable = new MockTable(this, salesSchema, "DEPT");
deptTable.addColumn("DEPTNO", intType);
deptTable.addColumn("NAME", varchar10Type);
registerTable(deptTable);
// Register "BONUS" table.
MockTable bonusTable = new MockTable(this, salesSchema, "BONUS");
bonusTable.addColumn("ENAME", varchar20Type);
bonusTable.addColumn("JOB", varchar10Type);
bonusTable.addColumn("SAL", intType);
bonusTable.addColumn("COMM", intType);
registerTable(bonusTable);
// Register "SALGRADE" table.
MockTable salgradeTable = new MockTable(this, salesSchema, "SALGRADE");
salgradeTable.addColumn("GRADE", intType);
salgradeTable.addColumn("LOSAL", intType);
salgradeTable.addColumn("HISAL", intType);
registerTable(salgradeTable);
// Register "EMP_ADDRESS" table
MockTable contactAddressTable = new MockTable(this, salesSchema, "EMP_ADDRESS");
contactAddressTable.addColumn("EMPNO", intType);
contactAddressTable.addColumn("HOME_ADDRESS", addressType);
contactAddressTable.addColumn("MAILING_ADDRESS", addressType);
registerTable(contactAddressTable);
// Register "CUSTOMER" schema.
MockSchema customerSchema = new MockSchema("CUSTOMER");
registerSchema(customerSchema);
// Register "CONTACT" table.
MockTable contactTable = new MockTable(this, customerSchema, "CONTACT");
contactTable.addColumn("CONTACTNO", intType);
contactTable.addColumn("FNAME", varchar10Type);
contactTable.addColumn("LNAME", varchar10Type);
contactTable.addColumn("EMAIL", varchar20Type);
contactTable.addColumn("COORD", rectilinearCoordType);
registerTable(contactTable);
// Register "ACCOUNT" table.
MockTable accountTable = new MockTable(this, customerSchema, "ACCOUNT");
accountTable.addColumn("ACCTNO", intType);
accountTable.addColumn("TYPE", varchar20Type);
accountTable.addColumn("BALANCE", intType);
registerTable(accountTable);
}
private static RelDataType makeStruct(RelDataTypeFactory typeFactory, RelDataType type) {
if (type.isStruct()) {
return type;
}
return typeFactory.createStructType(RelDataTypeFactory.FieldInfoBuilder.of("$0", type));
}
