com.dremio.exec.ops.QueryContext.getOperatorTable()方法的使用及代码示例

x33g5p2x  于2022-01-28 转载在 其他  
字(5.6k)|赞(0)|评价(0)|浏览(116)

本文整理了Java中com.dremio.exec.ops.QueryContext.getOperatorTable方法的一些代码示例,展示了QueryContext.getOperatorTable的具体用法。这些代码示例主要来源于Github/Stackoverflow/Maven等平台,是从一些精选项目中提取出来的代码,具有较强的参考意义,能在一定程度帮忙到你。QueryContext.getOperatorTable方法的具体详情如下:
包路径:com.dremio.exec.ops.QueryContext
类名称:QueryContext
方法名:getOperatorTable

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(),

相关文章