com.facebook.presto.testing.QueryRunner.execute()方法的使用及代码示例

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

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

QueryRunner.execute介绍

暂无

代码示例

代码示例来源:origin: prestodb/presto

@Override
  public void execute(String sql)
  {
    try {
      queryRunner.execute(queryRunner.getDefaultSession(), sql);
    }
    catch (Throwable e) {
      throw new RuntimeException("Error executing sql:\n" + sql, e);
    }
  }
}

代码示例来源:origin: prestodb/presto

protected static void assertQuerySucceeds(QueryRunner queryRunner, Session session, @Language("SQL") String sql)
{
  try {
    queryRunner.execute(session, sql);
  }
  catch (RuntimeException e) {
    fail(format("Expected query to succeed: %s", sql), e);
  }
}

代码示例来源:origin: prestodb/presto

protected static void assertQueryReturnsEmptyResult(QueryRunner queryRunner, Session session, @Language("SQL") String sql)
{
  try {
    MaterializedResult results = queryRunner.execute(session, sql).toTestTypes();
    assertNotNull(results);
    assertEquals(results.getRowCount(), 0);
  }
  catch (RuntimeException ex) {
    fail("Execution of query failed: " + sql, ex);
  }
}

代码示例来源:origin: prestodb/presto

protected static void assertQueryFails(QueryRunner queryRunner, Session session, @Language("SQL") String sql, @Language("RegExp") String expectedMessageRegExp)
{
  try {
    queryRunner.execute(session, sql);
    fail(format("Expected query to fail: %s", sql));
  }
  catch (RuntimeException ex) {
    assertExceptionMessage(sql, ex, expectedMessageRegExp);
  }
}

代码示例来源:origin: prestodb/presto

@AfterClass(alwaysRun = true)
  public final void destroy()
  {
    System.setProperty("kudu.schema-emulation.prefix", oldPrefix);
    if (queryRunner != null) {
      queryRunner.execute(DROP_TABLE);
      queryRunner.execute(DROP_SCHEMA);
      queryRunner.close();
      queryRunner = null;
    }
  }
}

代码示例来源:origin: prestodb/presto

@Test(timeOut = 60_000, expectedExceptions = RuntimeException.class, expectedExceptionsMessageRegExp = ".*Query exceeded distributed total memory limit of 2kB.*")
public void testQueryTotalMemoryLimit()
    throws Exception
{
  Map<String, String> properties = ImmutableMap.<String, String>builder()
      .put("query.max-memory", "1kB")
      .put("query.max-total-memory", "2kB")
      .build();
  try (QueryRunner queryRunner = createQueryRunner(SESSION, properties)) {
    queryRunner.execute(SESSION, "SELECT COUNT(*), repeat(orderstatus, 1000) FROM orders GROUP BY 2");
  }
}

代码示例来源:origin: prestodb/presto

@Benchmark
public MaterializedResult queryInformationSchema(BenchmarkData benchmarkData)
{
  return benchmarkData.queryRunner.execute(benchmarkData.query);
}

代码示例来源:origin: prestodb/presto

@Test
public void testAdminCanRead()
{
  Session admin = getSession("admin");
  queryRunner.execute(admin, "SELECT * FROM disks");
}

代码示例来源:origin: prestodb/presto

private MaterializedResult execute(String sql)
  {
    return getQueryRunner().execute(SESSION, sql);
  }
}

代码示例来源:origin: prestodb/presto

@Test(expectedExceptions = {SemanticException.class},
    expectedExceptionsMessageRegExp = ".*Function log not registered")
public void testLegacyLogFunctionDisabled()
{
  try (QueryRunner runner = createQueryRunner(false)) {
    runner.execute(QUERY);
  }
}

代码示例来源:origin: prestodb/presto

@Test(timeOut = 60_000, expectedExceptions = RuntimeException.class, expectedExceptionsMessageRegExp = ".*Query exceeded per-node user memory limit of 1kB.*")
public void testQueryMemoryPerNodeLimit()
    throws Exception
{
  Map<String, String> properties = ImmutableMap.<String, String>builder()
      .put("task.max-partial-aggregation-memory", "1B")
      .put("query.max-memory-per-node", "1kB")
      .build();
  try (QueryRunner queryRunner = createQueryRunner(SESSION, properties)) {
    queryRunner.execute(SESSION, "SELECT COUNT(*), repeat(orderstatus, 1000) FROM orders GROUP BY 2");
  }
}

代码示例来源:origin: prestodb/presto

@Test
public void testLegacyGroupBy()
{
  MaterializedResult result = runner.execute("select * from (values nan(), nan(), nan()) group by 1");
  List<MaterializedRow> rows = result.getMaterializedRows();
  assertEquals(rows.size(), 3);
  rows.stream()
      .forEach(row -> assertTrue(Double.isNaN((Double) row.getField(0))));
}

代码示例来源:origin: prestodb/presto

@Test
public void testLegacyLogFunctionEnabled()
{
  try (QueryRunner queryRunner = createQueryRunner(true)) {
    MaterializedResult result = queryRunner.execute(QUERY);
    assertEquals(result.getOnlyValue(), 2.0);
  }
}

代码示例来源:origin: prestodb/presto

@Test(timeOut = 60_000)
public void testTimeout()
{
  try {
    queryRunner.execute("SELECT * FROM disks");
    fail();
  }
  catch (PrestoException e) {
    assertEquals(e.getErrorCode(), ATOP_READ_TIMEOUT.toErrorCode(), e.getMessage());
  }
}

代码示例来源:origin: prestodb/presto

@Test
public void testSelectWithUnenforcedConstraint()
{
  createBlackholeAllTypesTable();
  MaterializedResult rows = queryRunner.execute("SELECT * FROM blackhole_all_types where _bigint > 10");
  assertEquals(rows.getRowCount(), 0);
  dropBlackholeAllTypesTable();
}

代码示例来源:origin: prestodb/presto

protected void assertAccessAllowed(Session session, @Language("SQL") String sql, TestingPrivilege... deniedPrivileges)
{
  executeExclusively(() -> {
    try {
      queryRunner.getAccessControl().deny(deniedPrivileges);
      queryRunner.execute(session, sql);
    }
    finally {
      queryRunner.getAccessControl().reset();
    }
  });
}

代码示例来源:origin: prestodb/presto

@Benchmark
public MaterializedResult benchmarkReorderJoins(BenchmarkInfo benchmarkInfo)
{
  return benchmarkInfo.getQueryRunner().execute(benchmarkInfo.getQuery());
}

代码示例来源:origin: prestodb/presto

private void assertOneNotNullResult(String query)
  {
    MaterializedResult results = getQueryRunner().execute(getSession(), query).toTestTypes();
    assertEquals(results.getRowCount(), 1);
    assertEquals(results.getMaterializedRows().get(0).getFieldCount(), 1);
    assertNotNull(results.getMaterializedRows().get(0).getField(0));
  }
}

代码示例来源:origin: prestodb/presto

@Test(timeOut = 240_000, expectedExceptions = RuntimeException.class, expectedExceptionsMessageRegExp = ".*Query exceeded per-query local spill limit of 10B")
public void testQueryMaxSpillPerNodeLimit()
{
  try (QueryRunner queryRunner = createLocalQueryRunner(new NodeSpillConfig().setQueryMaxSpillPerNode(DataSize.succinctBytes(10)))) {
    queryRunner.execute(queryRunner.getDefaultSession(), "SELECT COUNT(DISTINCT clerk) as count, orderdate FROM orders GROUP BY orderdate ORDER BY count, orderdate");
  }
}

代码示例来源:origin: prestodb/presto

@Test(timeOut = 240_000, expectedExceptions = RuntimeException.class, expectedExceptionsMessageRegExp = ".*Query exceeded local spill limit of 10B")
public void testMaxSpillPerNodeLimit()
{
  try (QueryRunner queryRunner = createLocalQueryRunner(new NodeSpillConfig().setMaxSpillPerNode(DataSize.succinctBytes(10)))) {
    queryRunner.execute(queryRunner.getDefaultSession(), "SELECT COUNT(DISTINCT clerk) as count, orderdate FROM orders GROUP BY orderdate ORDER BY count, orderdate");
  }
}

相关文章