super(config);
}
@Override
public PhysicalPlan getPlan(SqlNode sqlNode) throws ValidationException, RelConversionException, IOException {
SqlCreateTable sqlCreateTable = unwrap(sqlNode, SqlCreateTable.class);
try {
// Convert the query in CTAS statement into a RelNode
SqlNode validatedQuery = validateNode(sqlCreateTable.getQuery());
RelNode relQuery = convertToRel(validatedQuery);
List<String> tblFiledNames = sqlCreateTable.getFieldNames();
RelDataType queryRowType = relQuery.getRowType();
if (tblFiledNames.size() > 0) {
// Field count should match.
if (tblFiledNames.size() != queryRowType.getFieldCount()) {
return DirectPlan.createDirectPlan(context, false,
"Table's field list and the table's query field list have different counts.");
}
// CTAS's query field list shouldn't have "*" when table's field list is specified.
for (String field : queryRowType.getFieldNames()) {
if (field.equals("*")) {
return DirectPlan.createDirectPlan(context, false,
"Table's query field list has a '*', which is invalid when table's field list is specified.");
}
}
}
// if the CTAS statement has table fields lists (ex. below), add a project rel to rename the query fields.
// Ex. CREATE TABLE tblname(col1, medianOfCol2, avgOfCol3) AS
// SELECT col1, median(col2), avg(col3) FROM sourcetbl GROUP BY col1 ;
if (tblFiledNames.size() > 0) {
// create rowtype to which the select rel needs to be casted.
RelDataType rowType = new DrillFixedRelDataTypeImpl(planner.getTypeFactory(), tblFiledNames);
relQuery = RelOptUtil.createCastRel(relQuery, rowType, true);
}
SchemaPlus schema = findSchema(context.getRootSchema(), context.getNewDefaultSchema(),
sqlCreateTable.getSchemaPath());
AbstractSchema drillSchema = getDrillSchema(schema);
if (!drillSchema.isMutable()) {
return DirectPlan.createDirectPlan(context, false, String.format("Current schema '%s' is not a mutable schema. " +
"Can't create tables in this schema.", drillSchema.getFullSchemaName()));
}
String newTblName = sqlCreateTable.getName();
if (schema.getTable(newTblName) != null) {
return DirectPlan.createDirectPlan(context, false, String.format("Table '%s' already exists.", newTblName));
}
log("Optiq Logical", relQuery);
// Convert the query to Drill Logical plan and insert a writer operator on top.
DrillRel drel = convertToDrel(relQuery, drillSchema, newTblName);
log("Drill Logical", drel);
Prel prel = convertToPrel(drel);
log("Drill Physical", prel);
PhysicalOperator pop = convertToPop(prel);
PhysicalPlan plan = convertToPlan(pop);
log("Drill Plan", plan);
return plan;
} catch(Exception e) {
logger.error("Failed to create table '{}'", sqlCreateTable.getName(), e);
return DirectPlan.createDirectPlan(context, false, String.format("Error: %s", e.getMessage()));
}
}