io.prestosql.testing.QueryRunner.getDefaultSession()方法的使用及代码示例

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

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

QueryRunner.getDefaultSession介绍

暂无

代码示例

代码示例来源：origin: prestosql/presto

@Test
public void testShowSchemas()
{
  MaterializedResult actualSchemas = computeActual("SHOW SCHEMAS").toTestTypes();
  MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(getQueryRunner().getDefaultSession(), VARCHAR)
      .row(getQueryRunner().getDefaultSession().getSchema().orElse("tpch"));
  assertContains(actualSchemas, resultBuilder.build());
}

代码示例来源：origin: prestosql/presto

public void assertFails(@Language("SQL") String sql, @Language("RegExp") String expectedMessageRegExp)
{
  try {
    runner.execute(runner.getDefaultSession(), sql).toTestTypes();
    fail(format("Expected query to fail: %s", sql));
  }
  catch (RuntimeException exception) {
    if (!nullToEmpty(exception.getMessage()).matches(expectedMessageRegExp)) {
      fail(format("Expected exception message '%s' to match '%s' for query: %s", exception.getMessage(), expectedMessageRegExp, sql), exception);
    }
  }
}

代码示例来源：origin: io.prestosql/presto-main

public void assertFails(@Language("SQL") String sql, @Language("RegExp") String expectedMessageRegExp)
{
  try {
    runner.execute(runner.getDefaultSession(), sql).toTestTypes();
    fail(format("Expected query to fail: %s", sql));
  }
  catch (RuntimeException exception) {
    if (!nullToEmpty(exception.getMessage()).matches(expectedMessageRegExp)) {
      fail(format("Expected exception message '%s' to match '%s' for query: %s", exception.getMessage(), expectedMessageRegExp, sql), exception);
    }
  }
}

代码示例来源：origin: prestosql/presto

@Test
public void testShowTables()
{
  MaterializedResult actualTables = computeActual("SHOW TABLES").toTestTypes();
  MaterializedResult expectedTables = MaterializedResult.resultBuilder(getQueryRunner().getDefaultSession(), VARCHAR)
      .row("orders")
      .build();
  assertContains(actualTables, expectedTables);
}

代码示例来源：origin: prestosql/presto

private static List<OptionalDouble> getEstimatedValues(List<Metric> metrics, String query, QueryRunner runner)
{
  return transaction(runner.getTransactionManager(), runner.getAccessControl())
      .singleStatement()
      .execute(runner.getDefaultSession(), (Session session) -> getEstimatedValuesInternal(metrics, query, runner, session));
}

代码示例来源：origin: prestosql/presto

public String getGraphvizExplainPlan(String query, ExplainType.Type planType)
{
  QueryExplainer explainer = getQueryExplainer();
  return transaction(queryRunner.getTransactionManager(), queryRunner.getAccessControl())
      .singleStatement()
      .execute(queryRunner.getDefaultSession(), session -> {
        return explainer.getGraphvizPlan(session, sqlParser.createStatement(query, createParsingOptions(session)), planType, emptyList(), WarningCollector.NOOP);
      });
}

代码示例来源：origin: prestosql/presto

private Session getSession(String user)
  {
    return testSessionBuilder()
        .setCatalog(queryRunner.getDefaultSession().getCatalog().get())
        .setSchema(queryRunner.getDefaultSession().getSchema().get())
        .setIdentity(new Identity(user, Optional.empty())).build();
  }
}

代码示例来源：origin: prestosql/presto

public String getExplainPlan(String query, ExplainType.Type planType)
{
  QueryExplainer explainer = getQueryExplainer();
  return transaction(queryRunner.getTransactionManager(), queryRunner.getAccessControl())
      .singleStatement()
      .execute(queryRunner.getDefaultSession(), session -> {
        return explainer.getPlan(session, sqlParser.createStatement(query, createParsingOptions(session)), planType, emptyList(), WarningCollector.NOOP);
      });
}

代码示例来源：origin: io.prestosql/presto-hive

private Session getSession(String user)
  {
    return testSessionBuilder()
        .setCatalog(queryRunner.getDefaultSession().getCatalog().get())
        .setSchema(queryRunner.getDefaultSession().getSchema().get())
        .setIdentity(new Identity(user, Optional.empty())).build();
  }
}

代码示例来源：origin: prestosql/presto

private Session getSession(String user)
  {
    return testSessionBuilder()
        .setCatalog(queryRunner.getDefaultSession().getCatalog().get())
        .setSchema(queryRunner.getDefaultSession().getSchema().get())
        .setIdentity(new Identity(user, Optional.empty())).build();
  }
}

代码示例来源：origin: prestosql/presto

private Session getSession(String user)
  {
    return testSessionBuilder()
        .setCatalog(queryRunner.getDefaultSession().getCatalog().get())
        .setSchema(queryRunner.getDefaultSession().getSchema().get())
        .setIdentity(new Identity(user, Optional.empty())).build();
  }
}

代码示例来源：origin: io.prestosql/presto-main

public void assertQueryAndPlan(
    @Language("SQL") String actual,
    @Language("SQL") String expected,
    PlanMatchPattern pattern,
    Consumer<Plan> planValidator)
{
  assertQuery(actual, expected);
  Plan plan = runner.createPlan(runner.getDefaultSession(), actual, WarningCollector.NOOP);
  PlanAssert.assertPlan(runner.getDefaultSession(), runner.getMetadata(), runner.getStatsCalculator(), plan, pattern);
  planValidator.accept(plan);
}

代码示例来源：origin: prestosql/presto

private Session createAdminSession()
{
  return Session.builder(getQueryRunner().getDefaultSession())
      .setIdentity(new Identity("admin", Optional.empty(), ImmutableMap.of("hive", new SelectedRole(SelectedRole.Type.ROLE, Optional.of("admin")))))
      .build();
}

代码示例来源：origin: io.prestosql/presto-hive

private Session createAdminSession()
{
  return Session.builder(getQueryRunner().getDefaultSession())
      .setIdentity(new Identity("admin", Optional.empty(), ImmutableMap.of("hive", new SelectedRole(SelectedRole.Type.ROLE, Optional.of("admin")))))
      .build();
}

代码示例来源：origin: prestosql/presto

private Session createUserSession(String user)
  {
    return Session.builder(getQueryRunner().getDefaultSession())
        .setIdentity(new Identity(user, Optional.empty()))
        .build();
  }
}

代码示例来源：origin: prestosql/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: io.prestosql/presto-hive

private Session createUserSession(String user)
  {
    return Session.builder(getQueryRunner().getDefaultSession())
        .setIdentity(new Identity(user, Optional.empty()))
        .build();
  }
}

代码示例来源：origin: io.prestosql/presto-hive

@Test
public void testSchemaOperations()
{
  Session admin = Session.builder(getQueryRunner().getDefaultSession())
      .setIdentity(new Identity("hive", Optional.empty(), ImmutableMap.of("hive", new SelectedRole(SelectedRole.Type.ROLE, Optional.of("admin")))))
      .build();
  assertUpdate(admin, "CREATE SCHEMA new_schema");
  assertUpdate(admin, "CREATE TABLE new_schema.test (x bigint)");
  assertQueryFails(admin, "DROP SCHEMA new_schema", "Schema not empty: new_schema");
  assertUpdate(admin, "DROP TABLE new_schema.test");
  assertUpdate(admin, "DROP SCHEMA new_schema");
}

代码示例来源：origin: prestosql/presto

@Test
public void testSchemaOperations()
{
  Session admin = Session.builder(getQueryRunner().getDefaultSession())
      .setIdentity(new Identity("hive", Optional.empty(), ImmutableMap.of("hive", new SelectedRole(SelectedRole.Type.ROLE, Optional.of("admin")))))
      .build();
  assertUpdate(admin, "CREATE SCHEMA new_schema");
  assertUpdate(admin, "CREATE TABLE new_schema.test (x bigint)");
  assertQueryFails(admin, "DROP SCHEMA new_schema", "Schema not empty: new_schema");
  assertUpdate(admin, "DROP TABLE new_schema.test");
  assertUpdate(admin, "DROP SCHEMA new_schema");
}

代码示例来源：origin: prestosql/presto

@Override
  @Test
  public void testShowSchemas()
  {
    MaterializedResult actualSchemas = computeActual("SHOW SCHEMAS").toTestTypes();
    MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(getQueryRunner().getDefaultSession(), VARCHAR)
        .row("tiny")
        .row("sf1");
    assertContains(actualSchemas, resultBuilder.build());
  }
}

内容来源于网络，如有侵权，请联系作者删除！

QueryRunner

相关文章

    查看更多
    微信公众号

    热门标签

    更多
    JavaquerypythonNode开发语言requestUtil数据库Table后端算法LoggerMessageElementParser

    最新文章

    更多
    • 蜀ICP备13028337号-1 大数据知识库 https://www.saoniuhuo.com © All rights reserved
    • 本站内容来源互联网,如果侵犯您的权益请联系我们删除, 联系方式：448109455@qq.com