How to use

beginTransaction

method

in

io.prestosql.spi.connector.Connector

private ConnectorTransactionHandle beginTransaction(Connector connector)
{
  if (connector instanceof InternalConnector) {
    return ((InternalConnector) connector).beginTransaction(transactionId, isolationLevel, readOnly);
  }
  else {
    return connector.beginTransaction(isolationLevel, readOnly);
  }
}

private ConnectorTransactionHandle beginTransaction(Connector connector)
{
  if (connector instanceof InternalConnector) {
    return ((InternalConnector) connector).beginTransaction(transactionId, isolationLevel, readOnly);
  }
  else {
    return connector.beginTransaction(isolationLevel, readOnly);
  }
}

private static void assertCreateConnector(String metastoreUri)
{
  HiveConnectorFactory connectorFactory = new HiveConnectorFactory(
      "hive-test",
      HiveConnector.class.getClassLoader(),
      Optional.empty());
  Map<String, String> config = ImmutableMap.<String, String>builder()
      .put("hive.metastore.uri", metastoreUri)
      .build();
  Connector connector = connectorFactory.create("hive-test", config, new TestingConnectorContext());
  ConnectorTransactionHandle transaction = connector.beginTransaction(READ_UNCOMMITTED, true);
  assertInstanceOf(connector.getMetadata(transaction), ClassLoaderSafeConnectorMetadata.class);
  assertInstanceOf(connector.getSplitManager(), ClassLoaderSafeConnectorSplitManager.class);
  assertInstanceOf(connector.getPageSourceProvider(), ConnectorPageSourceProvider.class);
  connector.commit(transaction);
}

private static void assertCreateConnector(String metastoreUri)
{
  HiveConnectorFactory connectorFactory = new HiveConnectorFactory(
      "hive-test",
      HiveConnector.class.getClassLoader(),
      Optional.empty());
  Map<String, String> config = ImmutableMap.<String, String>builder()
      .put("hive.metastore.uri", metastoreUri)
      .build();
  Connector connector = connectorFactory.create("hive-test", config, new TestingConnectorContext());
  ConnectorTransactionHandle transaction = connector.beginTransaction(READ_UNCOMMITTED, true);
  assertInstanceOf(connector.getMetadata(transaction), ClassLoaderSafeConnectorMetadata.class);
  assertInstanceOf(connector.getSplitManager(), ClassLoaderSafeConnectorSplitManager.class);
  assertInstanceOf(connector.getPageSourceProvider(), ConnectorPageSourceProvider.class);
  connector.commit(transaction);
}