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

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

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

QueryRunner.getTransactionManager介绍

暂无

代码示例

代码示例来源：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

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 Object getHiveTableProperty(String tableName, Function<HiveTableLayoutHandle, Object> propertyGetter)
{
  Session session = getSession();
  Metadata metadata = ((DistributedQueryRunner) getQueryRunner()).getCoordinator().getMetadata();
  return transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
      .readOnly()
      .execute(session, transactionSession -> {
        Optional<TableHandle> tableHandle = metadata.getTableHandle(transactionSession, new QualifiedObjectName(catalog, TPCH_SCHEMA, tableName));
        assertTrue(tableHandle.isPresent());
        List<TableLayoutResult> layouts = metadata.getLayouts(transactionSession, tableHandle.get(), Constraint.alwaysTrue(), Optional.empty());
        TableLayout layout = getOnlyElement(layouts).getLayout();
        return propertyGetter.apply((HiveTableLayoutHandle) layout.getHandle().getConnectorHandle());
      });
}

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

private Object getHiveTableProperty(String tableName, Function<HiveTableLayoutHandle, Object> propertyGetter)
{
  Session session = getSession();
  Metadata metadata = ((DistributedQueryRunner) getQueryRunner()).getCoordinator().getMetadata();
  return transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
      .readOnly()
      .execute(session, transactionSession -> {
        Optional<TableHandle> tableHandle = metadata.getTableHandle(transactionSession, new QualifiedObjectName(catalog, TPCH_SCHEMA, tableName));
        assertTrue(tableHandle.isPresent());
        List<TableLayoutResult> layouts = metadata.getLayouts(transactionSession, tableHandle.get(), Constraint.alwaysTrue(), Optional.empty());
        TableLayout layout = getOnlyElement(layouts).getLayout();
        return propertyGetter.apply((HiveTableLayoutHandle) layout.getHandle().getConnectorHandle());
      });
}

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

private TableMetadata getTableMetadata(String catalog, String schema, String tableName)
{
  Session session = getSession();
  Metadata metadata = ((DistributedQueryRunner) getQueryRunner()).getCoordinator().getMetadata();
  return transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
      .readOnly()
      .execute(session, transactionSession -> {
        Optional<TableHandle> tableHandle = metadata.getTableHandle(transactionSession, new QualifiedObjectName(catalog, schema, tableName));
        assertTrue(tableHandle.isPresent());
        return metadata.getTableMetadata(transactionSession, tableHandle.get());
      });
}

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

private TableMetadata getTableMetadata(String catalog, String schema, String tableName)
{
  Session session = getSession();
  Metadata metadata = ((DistributedQueryRunner) getQueryRunner()).getCoordinator().getMetadata();
  return transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
      .readOnly()
      .execute(session, transactionSession -> {
        Optional<TableHandle> tableHandle = metadata.getTableHandle(transactionSession, new QualifiedObjectName(catalog, schema, tableName));
        assertTrue(tableHandle.isPresent());
        return metadata.getTableMetadata(transactionSession, tableHandle.get());
      });
}

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

private HiveInsertTableHandle getHiveInsertTableHandle(Session session, String tableName)
{
  Metadata metadata = ((DistributedQueryRunner) getQueryRunner()).getCoordinator().getMetadata();
  return transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
      .execute(session, transactionSession -> {
        QualifiedObjectName objectName = new QualifiedObjectName(catalog, TPCH_SCHEMA, tableName);
        Optional<TableHandle> handle = metadata.getTableHandle(transactionSession, objectName);
        InsertTableHandle insertTableHandle = metadata.beginInsert(transactionSession, handle.get());
        HiveInsertTableHandle hiveInsertTableHandle = (HiveInsertTableHandle) insertTableHandle.getConnectorHandle();
        metadata.finishInsert(transactionSession, insertTableHandle, ImmutableList.of(), ImmutableList.of());
        return hiveInsertTableHandle;
      });
}

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

private HiveInsertTableHandle getHiveInsertTableHandle(Session session, String tableName)
{
  Metadata metadata = ((DistributedQueryRunner) getQueryRunner()).getCoordinator().getMetadata();
  return transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
      .execute(session, transactionSession -> {
        QualifiedObjectName objectName = new QualifiedObjectName(catalog, TPCH_SCHEMA, tableName);
        Optional<TableHandle> handle = metadata.getTableHandle(transactionSession, objectName);
        InsertTableHandle insertTableHandle = metadata.beginInsert(transactionSession, handle.get());
        HiveInsertTableHandle hiveInsertTableHandle = (HiveInsertTableHandle) insertTableHandle.getConnectorHandle();
        metadata.finishInsert(transactionSession, insertTableHandle, ImmutableList.of(), ImmutableList.of());
        return hiveInsertTableHandle;
      });
}

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

@Test
public void testCreateAndInsert()
{
  Session session = getSession();
  List<MaterializedRow> expected = MaterializedResult.resultBuilder(session, BIGINT, BIGINT)
      .row(101L, 1L)
      .row(201L, 2L)
      .row(202L, 2L)
      .row(301L, 3L)
      .row(302L, 3L)
      .build()
      .getMaterializedRows();
  transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
      .execute(session, transactionSession -> {
        assertUpdate(
            transactionSession,
            "CREATE TABLE tmp_create_insert WITH (partitioned_by=array ['z']) AS " +
                "SELECT * from (VALUES (CAST (101 AS BIGINT), CAST (1 AS BIGINT)), (201, 2), (202, 2)) t(a, z)",
            3);
        assertUpdate(transactionSession, "INSERT INTO tmp_create_insert VALUES (301, 3), (302, 3)", 2);
        MaterializedResult actualFromCurrentTransaction = computeActual(transactionSession, "SELECT * FROM tmp_create_insert");
        assertEqualsIgnoreOrder(actualFromCurrentTransaction, expected);
      });
  MaterializedResult actualAfterTransaction = computeActual(session, "SELECT * FROM tmp_create_insert");
  assertEqualsIgnoreOrder(actualAfterTransaction, expected);
}

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

@Test
public void testCreateAndInsert()
{
  Session session = getSession();
  List<MaterializedRow> expected = MaterializedResult.resultBuilder(session, BIGINT, BIGINT)
      .row(101L, 1L)
      .row(201L, 2L)
      .row(202L, 2L)
      .row(301L, 3L)
      .row(302L, 3L)
      .build()
      .getMaterializedRows();
  transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
      .execute(session, transactionSession -> {
        assertUpdate(
            transactionSession,
            "CREATE TABLE tmp_create_insert WITH (partitioned_by=array ['z']) AS " +
                "SELECT * from (VALUES (CAST (101 AS BIGINT), CAST (1 AS BIGINT)), (201, 2), (202, 2)) t(a, z)",
            3);
        assertUpdate(transactionSession, "INSERT INTO tmp_create_insert VALUES (301, 3), (302, 3)", 2);
        MaterializedResult actualFromCurrentTransaction = computeActual(transactionSession, "SELECT * FROM tmp_create_insert");
        assertEqualsIgnoreOrder(actualFromCurrentTransaction, expected);
      });
  MaterializedResult actualAfterTransaction = computeActual(session, "SELECT * FROM tmp_create_insert");
  assertEqualsIgnoreOrder(actualAfterTransaction, expected);
}

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

transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
      .execute(session, transactionSession -> {
        assertUpdate(transactionSession, "DELETE FROM tmp_delete_insert WHERE z >= 2");
assertEqualsIgnoreOrder(actualAfterRollback, expectedBefore);
transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
    .execute(session, transactionSession -> {
      assertUpdate(transactionSession, "DELETE FROM tmp_delete_insert WHERE z >= 2");

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

transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
      .execute(session, transactionSession -> {
        assertUpdate(transactionSession, "DELETE FROM tmp_delete_insert WHERE z >= 2");
assertEqualsIgnoreOrder(actualAfterRollback, expectedBefore);
transaction(getQueryRunner().getTransactionManager(), getQueryRunner().getAccessControl())
    .execute(session, transactionSession -> {
      assertUpdate(transactionSession, "DELETE FROM tmp_delete_insert WHERE z >= 2");

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

QueryRunner

相关文章

    查看更多
    微信公众号

    热门标签

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

    最新文章

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