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

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

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

QueryContext.<init>介绍

暂无

代码示例

代码示例来源:origin: dremio/dremio-oss

@Override
 public QueryContext get() {
  final UserSession session = systemSession(getOptionManager());
  return new QueryContext(session, sabotContext.get(), new AttemptId().toQueryId());
 }
};

代码示例来源:origin: dremio/dremio-oss

protected AttemptManager newAttemptManager(SabotContext context, AttemptId attemptId, UserRequest queryRequest,
  AttemptObserver observer, UserSession session, OptionProvider options, CoordToExecTunnelCreator tunnelCreator,
  Cache<Long, PreparedPlan> plans, Predicate<DatasetConfig> datasetValidityChecker) {
 final QueryContext queryContext = new QueryContext(session, context, attemptId.toQueryId(),
   queryRequest.getPriority(), queryRequest.getMaxAllocation(), datasetValidityChecker);
 return new AttemptManager(context, attemptId, queryRequest, observer, options, tunnelCreator, plans,
  queryContext, queryResourceManager);
}

代码示例来源:origin: dremio/dremio-oss

@BeforeClass
public static void beforeClass() {
 context = new QueryContext(session(), getSabotContext(), QueryId.getDefaultInstance());
}

代码示例来源:origin: dremio/dremio-oss

private QueryContext newQueryContext(SqlQuery query) {
 try (TimedBlock b = time("initParser")) {
  QueryId queryId = QueryId.newBuilder().setPart1(ID_MAJOR).setPart2(ID_MINOR.incrementAndGet()).build();
  UserSession session = UserSession.Builder.newBuilder()
    .withCredentials(UserCredentials.newBuilder()
      .setUserName(query.getUsername())
      .build())
    .withUserProperties(UserProperties.getDefaultInstance())
    .withOptionManager(sabotContext.getOptionManager())
    .build();
  return new QueryContext(session, sabotContext, queryId);
 }
}

代码示例来源:origin: dremio/dremio-oss

@SuppressWarnings("static-method")
@Test
public void uncheckedInjection() {
 // set exceptions via a string
 final String exceptionDesc = "<<injected from descPassthroughMethod()>>";
 final String exceptionClassName = "java.lang.RuntimeException";
 final String jsonString = "{\"injections\":[{"
  + "\"type\":\"exception\"," +
  "\"siteClass\":\"com.dremio.exec.testing.TestExceptionInjection$DummyClass\","
  + "\"desc\":\"" + exceptionDesc + "\","
  + "\"nSkip\":0,"
  + "\"nFire\":1,"
  + "\"exceptionClass\":\"" + exceptionClassName + "\""
  + "}]}";
 ControlsInjectionUtil.setControls(session, jsonString);
 final QueryContext context = new QueryContext(session, nodes[0].getContext(), QueryId.getDefaultInstance());
 // test that the exception gets thrown
 final DummyClass dummyClass = new DummyClass(context);
 assertPassthroughThrows(dummyClass, exceptionClassName, exceptionDesc);
 try {
  context.close();
 } catch (Exception e) {
  fail();
 }
}

代码示例来源:origin: dremio/dremio-oss

@SuppressWarnings("static-method")
@Test
public void checkedInjection() {
 // set the injection via the parsing POJOs
 final String controls = Controls.newBuilder()
  .addException(DummyClass.class, DummyClass.THROWS_IOEXCEPTION, IOException.class, 0, 1)
  .build();
 ControlsInjectionUtil.setControls(session, controls);
 final QueryContext context = new QueryContext(session, nodes[0].getContext(), QueryId.getDefaultInstance());
 // test that the expected exception (checked) gets thrown
 final DummyClass dummyClass = new DummyClass(context);
 try {
  dummyClass.throwsIOException();
  fail(NO_THROW_FAIL);
 } catch (IOException e) {
  assertEquals(DummyClass.THROWS_IOEXCEPTION, e.getMessage());
 }
 try {
  context.close();
 } catch (Exception e) {
  fail();
 }
}

代码示例来源:origin: dremio/dremio-oss

@SuppressWarnings("static-method")
@Test
public void skipAndLimit() {
 final String passthroughDesc = "<<injected from descPassthrough>>";
 final int nSkip = 7;
 final int nFire = 3;
 final Class<? extends Throwable> exceptionClass = RuntimeException.class;
 final String controls = Controls.newBuilder()
  .addException(DummyClass.class, passthroughDesc, exceptionClass, nSkip, nFire)
  .build();
 ControlsInjectionUtil.setControls(session, controls);
 final QueryContext context = new QueryContext(session, nodes[0].getContext(), QueryId.getDefaultInstance());
 final DummyClass dummyClass = new DummyClass(context);
 // these shouldn't throw
 for (int i = 0; i < nSkip; ++i) {
  dummyClass.descPassthroughMethod(passthroughDesc);
 }
 // these should throw
 for (int i = 0; i < nFire; ++i) {
  assertPassthroughThrows(dummyClass, exceptionClass.getName(), passthroughDesc);
 }
 // this shouldn't throw
 dummyClass.descPassthroughMethod(passthroughDesc);
 try {
  context.close();
 } catch (Exception e) {
  fail();
 }
}

代码示例来源:origin: dremio/dremio-oss

@Test
public void pauseInjected() {
 final long expectedDuration = 1000L;
 final ExtendedLatch trigger = new ExtendedLatch(1);
 final Pointer<Exception> ex = new Pointer<>();
 final String controls = Controls.newBuilder()
  .addPause(DummyClass.class, DummyClass.PAUSES)
  .build();
 ControlsInjectionUtil.setControls(session, controls);
 final QueryContext queryContext = new QueryContext(session, nodes[0].getContext(), QueryId.getDefaultInstance());
 (new ResumingThread(queryContext, trigger, ex, expectedDuration)).start();
 // test that the pause happens
 final DummyClass dummyClass = new DummyClass(queryContext, trigger);
 final long actualDuration = dummyClass.pauses();
 assertTrue(String.format("Test should stop for at least %d milliseconds.", expectedDuration),
  expectedDuration <= actualDuration);
 assertTrue("No exception should be thrown.", ex.value == null);
 try {
  queryContext.close();
 } catch (final Exception e) {
  fail("Failed to close query context: " + e);
 }
}

代码示例来源:origin: dremio/dremio-oss

);
final QueryContext queryContext = new QueryContext(session(), context, UserBitShared.QueryId.getDefaultInstance());
final AttemptObserver observer = new PassthroughQueryObserver(ExecTest.mockUserClientConnection(null));
final SqlConverter converter = new SqlConverter(

代码示例来源:origin: dremio/dremio-oss

OptionValue.createBoolean(OptionValue.OptionType.SYSTEM, ExecConstants.USE_NEW_MEMORY_BOUNDED_BEHAVIOR.getOptionName(), false)
 );
final QueryContext queryContext = new QueryContext(session(), context, UserBitShared.QueryId.getDefaultInstance());
final AttemptObserver observer = new PassthroughQueryObserver(ExecTest.mockUserClientConnection(null));
final SqlConverter converter = new SqlConverter(

代码示例来源:origin: dremio/dremio-oss

final ExtendedLatch trigger = new ExtendedLatch(1);
final Pointer<Exception> ex = new Pointer<>();
final QueryContext queryContext = new QueryContext(session, nodeContext1, QueryId.getDefaultInstance());
(new ResumingThread(queryContext, trigger, ex, expectedDuration)).start();
final QueryContext queryContext = new QueryContext(session, nodeContext2, QueryId.getDefaultInstance());

代码示例来源:origin: dremio/dremio-oss

final QueryContext queryContext = new QueryContext(session, nodes[0].getContext(), QueryId.getDefaultInstance());

代码示例来源:origin: dremio/dremio-oss

final QueryContext queryContext1 = new QueryContext(session, nodeContext1, QueryId.getDefaultInstance());
final DummyClass class1 = new DummyClass(queryContext1);
final QueryContext queryContext2 = new QueryContext(session, nodeContext2, QueryId.getDefaultInstance());
final DummyClass class2 = new DummyClass(queryContext2);

相关文章