本文整理了Java中com.dremio.exec.ops.QueryContext.getOperatorTable
方法的一些代码示例,展示了QueryContext.getOperatorTable
的具体用法。这些代码示例主要来源于Github
/Stackoverflow
/Maven
等平台,是从一些精选项目中提取出来的代码,具有较强的参考意义,能在一定程度帮忙到你。QueryContext.getOperatorTable
方法的具体详情如下:
包路径:com.dremio.exec.ops.QueryContext
类名称:QueryContext
方法名:getOperatorTable
暂无
代码示例来源:origin: dremio/dremio-oss
ExpansionHelper(QueryContext context) {
this.context = Preconditions.checkNotNull(context, "query context required");
converter = new SqlConverter(
context.getPlannerSettings(),
context.getOperatorTable(),
context,
MaterializationDescriptorProvider.EMPTY,
context.getFunctionRegistry(),
context.getSession(),
AbstractAttemptObserver.NOOP,
context.getCatalog(),
context.getSubstitutionProviderFactory(),
context.getConfig(),
context.getScanResult());
}
代码示例来源:origin: dremio/dremio-oss
private static RelNode preprocessNode(SqlHandlerConfig config, RelNode rel) throws SqlUnsupportedException {
/*
* Traverse the tree to do the following pre-processing tasks: 1. replace the convert_from, convert_to function to
* actual implementations Eg: convert_from(EXPR, 'JSON') be converted to convert_fromjson(EXPR); TODO: Ideally all
* function rewrites would move here instead of RexToExpr.
*
* 2. see where the tree contains unsupported functions; throw SqlUnsupportedException if there is any.
*/
PreProcessRel visitor = PreProcessRel.createVisitor(
config.getContext().getOperatorTable(),
rel.getCluster().getRexBuilder());
try {
rel = rel.accept(visitor);
} catch (UnsupportedOperationException ex) {
visitor.convertException();
throw ex;
}
return rel;
}
代码示例来源:origin: dremio/dremio-oss
protected QueryContext mockQueryContext(SabotContext dbContext) throws Exception {
final UserSession userSession = UserSession.Builder.newBuilder().withOptionManager(dbContext.getOptionManager()).build();
final SessionOptionManager sessionOptions = (SessionOptionManager) userSession.getOptions();
final QueryOptionManager queryOptions = new QueryOptionManager(sessionOptions);
final ExecutionControls executionControls = new ExecutionControls(queryOptions, NodeEndpoint.getDefaultInstance());
final OperatorTable table = new OperatorTable(FUNCTIONS());
final LogicalPlanPersistence lp = dbContext.getLpPersistence();
final CatalogService registry = dbContext.getCatalogService();
final QueryContext context = Mockito.mock(QueryContext.class);
when(context.getSession()).thenReturn(userSession);
when(context.getLpPersistence()).thenReturn(lp);
when(context.getCatalogService()).thenReturn(registry);
when(context.getFunctionRegistry()).thenReturn(FUNCTIONS());
when(context.getSession()).thenReturn(UserSession.Builder.newBuilder().setSupportComplexTypes(true).build());
when(context.getCurrentEndpoint()).thenReturn(NodeEndpoint.getDefaultInstance());
when(context.getActiveEndpoints()).thenReturn(ImmutableList.of(NodeEndpoint.getDefaultInstance()));
when(context.getPlannerSettings()).thenReturn(new PlannerSettings(dbContext.getConfig(), queryOptions, dbContext.getClusterResourceInformation()));
when(context.getOptions()).thenReturn(queryOptions);
when(context.getConfig()).thenReturn(DEFAULT_SABOT_CONFIG);
when(context.getOperatorTable()).thenReturn(table);
when(context.getAllocator()).thenReturn(allocator);
when(context.getExecutionControls()).thenReturn(executionControls);
when(context.getMaterializationProvider()).thenReturn(Mockito.mock(MaterializationDescriptorProvider.class));
return context;
}
代码示例来源:origin: dremio/dremio-oss
context.getOperatorTable(),
context.getFunctionRegistry()),
null);
代码示例来源:origin: dremio/dremio-oss
private SqlConverter getNewConverter(QueryContext context, SqlQuery query, AttemptObserver observerForSubstitution) {
Catalog catalog = context.getCatalog();
final List<String> sqlContext = query.getContext();
if(sqlContext != null){
NamespaceKey path = new NamespaceKey(sqlContext);
try {
catalog = catalog.resolveCatalog(path);
} catch (Exception e) {
throw UserException.validationError(e)
.message("Unable to resolve schema path [%s]. Failure resolving [%s] portion of path.", sqlContext, path)
.build(logger);
}
}
return new SqlConverter(
context.getPlannerSettings(),
context.getOperatorTable(),
context,
context.getMaterializationProvider(),
context.getFunctionRegistry(),
context.getSession(),
observerForSubstitution,
catalog,
context.getSubstitutionProviderFactory(),
context.getConfig(),
context.getScanResult());
}
代码示例来源:origin: dremio/dremio-oss
final SqlConverter converter = new SqlConverter(
queryContext.getPlannerSettings(),
queryContext.getOperatorTable(),
queryContext,
queryContext.getMaterializationProvider(),
代码示例来源:origin: dremio/dremio-oss
protected ElasticsearchGroupScan generate(String sql) throws Exception {
AttemptObserver observer = new PassthroughQueryObserver(ExecTest.mockUserClientConnection(null));
SqlConverter converter = new SqlConverter(context.getPlannerSettings(),
context.getOperatorTable(), context, context.getMaterializationProvider(), context.getFunctionRegistry(),
context.getSession(), observer, context.getCatalog(), context.getSubstitutionProviderFactory(), context.getConfig(),
context.getScanResult());
SqlNode node = converter.parse(sql);
SqlHandlerConfig config = new SqlHandlerConfig(context, converter, observer, null);
NormalHandler handler = new NormalHandler();
PhysicalPlan plan = handler.getPlan(config, sql, node);
List<PhysicalOperator> operators = plan.getSortedOperators();
ElasticsearchGroupScan scan = find(operators);
assertNotNull("Physical plan does not contain an elasticsearch scan for query: " + sql, scan);
return scan;
}
代码示例来源:origin: dremio/dremio-oss
final SqlConverter converter = new SqlConverter(
queryContext.getPlannerSettings(),
queryContext.getOperatorTable(),
queryContext,
queryContext.getMaterializationProvider(),
代码示例来源:origin: dremio/dremio-oss
final SqlConverter parser = new SqlConverter(
context.getPlannerSettings(),
context.getOperatorTable(),
context,
context.getMaterializationProvider(),
内容来源于网络,如有侵权,请联系作者删除!