Examples with RelOptPlanner org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.plan.RelOptPlanner used on opensource projects

Example 26 with RelOptPlanner

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.plan.RelOptPlanner in project drill by apache.

the class IndexSelector method getBestIndexNoFilter.

/**
 * we assume all the indexes added in indexPropList are all applicable (and covering).
 * For now this function is used and tested only in IndexScanWithSortOnlyPrule
 */
public IndexProperties getBestIndexNoFilter() {
    if (indexPropList.size() == 0) {
        return null;
    }
    RelOptPlanner planner = indexContext.getCall().getPlanner();
    List<IndexGroup> candidateIndexes = Lists.newArrayList();
    for (IndexProperties p : indexPropList) {
        p.setSatisfiesCollation(buildAndCheckCollation(p));
        IndexGroup index = new IndexGroup();
        index.addIndexProp(p);
        candidateIndexes.add(index);
    }
    Collections.sort(candidateIndexes, new IndexComparator(planner, builder));
    return candidateIndexes.get(0).getIndexProps().get(0);
}

Example 27 with RelOptPlanner

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.plan.RelOptPlanner in project herddb by diennea.

the class CalcitePlanner method runPlanner.

private PlannerResult runPlanner(String defaultTableSpace, String query) throws RelConversionException, SqlParseException, ValidationException, MetadataStorageManagerException, StatementExecutionException {
    try {
        SchemaPlus subSchema = getSchemaForTableSpace(defaultTableSpace);
        if (subSchema == null) {
            clearCache();
            throw new StatementExecutionException("tablespace " + defaultTableSpace + " is not available");
        }
        Properties props = new Properties();
        props.put(CalciteConnectionProperty.TIME_ZONE.camelName(), TimeZone.getDefault().getID());
        props.put(CalciteConnectionProperty.LOCALE.camelName(), Locale.ROOT.toString());
        props.put(CalciteConnectionProperty.DEFAULT_NULL_COLLATION.camelName(), NullCollation.LAST.toString());
        final CalciteConnectionConfigImpl calciteRuntimeContextConfig = new CalciteConnectionConfigImpl(props);
        final FrameworkConfig config = Frameworks.newConfigBuilder().parserConfig(SQL_PARSER_CONFIG).defaultSchema(subSchema).traitDefs(TRAITS).context(new Context() {

            @Override
            public <C> C unwrap(Class<C> aClass) {
                if (aClass == CalciteConnectionConfigImpl.class || aClass == CalciteConnectionConfig.class) {
                    return (C) calciteRuntimeContextConfig;
                }
                return null;
            }
        }).programs(Programs.ofRules(Programs.RULE_SET)).build();
        Planner planner = new PlannerImpl(config);
        if (LOG.isLoggable(Level.FINER)) {
            LOG.log(Level.FINER, "Query: {0}", query);
        }
        try {
            SqlNode n = planner.parse(query);
            n = planner.validate(n);
            RelNode logicalPlan = planner.rel(n).project();
            if (LOG.isLoggable(DUMP_QUERY_LEVEL)) {
                LOG.log(DUMP_QUERY_LEVEL, "Query: {0} {1}", new Object[] { query, RelOptUtil.dumpPlan("-- Logical Plan", logicalPlan, SqlExplainFormat.TEXT, SqlExplainLevel.ALL_ATTRIBUTES) });
            }
            RelDataType originalRowType = logicalPlan.getRowType();
            RelOptCluster cluster = logicalPlan.getCluster();
            final RelOptPlanner optPlanner = cluster.getPlanner();
            optPlanner.addRule(CoreRules.FILTER_REDUCE_EXPRESSIONS);
            RelTraitSet desiredTraits = cluster.traitSet().replace(EnumerableConvention.INSTANCE);
            final RelCollation collation = logicalPlan instanceof Sort ? ((Sort) logicalPlan).collation : null;
            if (collation != null) {
                desiredTraits = desiredTraits.replace(collation);
            }
            final RelNode newRoot = optPlanner.changeTraits(logicalPlan, desiredTraits);
            optPlanner.setRoot(newRoot);
            RelNode bestExp = optPlanner.findBestExp();
            if (LOG.isLoggable(DUMP_QUERY_LEVEL)) {
                LOG.log(DUMP_QUERY_LEVEL, "Query: {0} {1}", new Object[] { query, RelOptUtil.dumpPlan("-- Best  Plan", bestExp, SqlExplainFormat.TEXT, SqlExplainLevel.ALL_ATTRIBUTES) });
            }
            return new PlannerResult(bestExp, originalRowType, logicalPlan, n);
        } catch (AssertionError err) {
            throw new StatementExecutionException("Internal Calcite error " + err, err);
        }
    } catch (java.lang.LinkageError err) {
        LOG.log(Level.SEVERE, "Error on Java Classpath", err);
        throw new StatementExecutionException("Internal Calcite error " + err, err);
    }
}