本文整理了Java中com.dremio.exec.ops.QueryContext.<init>
方法的一些代码示例,展示了QueryContext.<init>
的具体用法。这些代码示例主要来源于Github
/Stackoverflow
/Maven
等平台,是从一些精选项目中提取出来的代码,具有较强的参考意义,能在一定程度帮忙到你。QueryContext.<init>
方法的具体详情如下:
包路径:com.dremio.exec.ops.QueryContext
类名称: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);
内容来源于网络,如有侵权,请联系作者删除!