Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Dev/issue 10 #18

Merged
merged 11 commits into from
Jul 26, 2022
Prev Previous commit
Next Next commit
Refactor: add CatalogReader to TrainDBPlanner
  • Loading branch information
taewhi committed Jul 20, 2022
commit d6c616d2905eb5d25f667be44c24905f99b17d9e
14 changes: 12 additions & 2 deletions traindb-core/src/main/java/traindb/planner/TrainDBPlanner.java
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,11 @@
package traindb.planner;

import java.util.Collection;
import java.util.List;
import org.apache.calcite.plan.Context;
import org.apache.calcite.plan.ConventionTraitDef;
import org.apache.calcite.plan.RelOptCostFactory;
import org.apache.calcite.plan.RelOptTable;
import org.apache.calcite.plan.RelOptUtil;
import org.apache.calcite.plan.volcano.VolcanoPlanner;
import org.apache.calcite.rel.RelCollationTraitDef;
Expand All @@ -27,20 +29,24 @@
import traindb.catalog.CatalogException;
import traindb.catalog.pm.MSynopsis;
import traindb.planner.rules.TrainDBRules;
import traindb.prepare.TrainDBCatalogReader;

public class TrainDBPlanner extends VolcanoPlanner {

private CatalogContext catalogContext;
private TrainDBCatalogReader catalogReader;

public TrainDBPlanner(CatalogContext catalogContext) {
this(catalogContext, null, null);
public TrainDBPlanner(CatalogContext catalogContext, TrainDBCatalogReader catalogReader) {
this(catalogContext, catalogReader, null, null);
}

public TrainDBPlanner(CatalogContext catalogContext,
TrainDBCatalogReader catalogReader,
@Nullable RelOptCostFactory costFactory,
@Nullable Context externalContext) {
super(costFactory, externalContext);
this.catalogContext = catalogContext;
this.catalogReader = catalogReader;
initPlanner();
}

Expand All @@ -64,4 +70,8 @@ public Collection<MSynopsis> getAvailableSynopses(String baseSchema, String base
} catch (CatalogException e) {}
return null;
}

public RelOptTable getTable(List<String> names) {
return catalogReader.getTable(names);
}
}
12 changes: 10 additions & 2 deletions traindb-core/src/main/java/traindb/prepare/TrainDBPrepareImpl.java
Original file line number Diff line number Diff line change
Expand Up @@ -196,7 +196,7 @@ private ParseResult convert_(Context context, String sql, boolean analyze,
// Use the Volcano because it can handle the traits.
CatalogContext catalogContext =
((TrainDBConnectionImpl.TrainDBContextImpl) context).getCatalogContext();
final VolcanoPlanner planner = new TrainDBPlanner(catalogContext);
final VolcanoPlanner planner = new TrainDBPlanner(catalogContext, catalogReader);

final SqlToRelConverter.Config config =
SqlToRelConverter.config().withTrimUnusedFields(true)
Expand Down Expand Up @@ -420,7 +420,15 @@ protected RelOptPlanner createPlanner(
}
CatalogContext catalogContext =
((TrainDBConnectionImpl.TrainDBContextImpl) prepareContext).getCatalogContext();
final VolcanoPlanner planner = new TrainDBPlanner(catalogContext, costFactory, externalContext);

TrainDBCatalogReader catalogReader =
new TrainDBCatalogReader(
prepareContext.getRootSchema(),
prepareContext.getDefaultSchemaPath(),
prepareContext.getTypeFactory(),
prepareContext.config());
final VolcanoPlanner planner = new TrainDBPlanner(
catalogContext, catalogReader, costFactory, externalContext);
return planner;
}

Expand Down