Search in sources :

Example 1 with IsolationLevel

use of io.prestosql.spi.transaction.IsolationLevel in project hetu-core by openlookeng.

the class TpcdsConnectorFactory method create.

@Override
public Connector create(String catalogName, Map<String, String> config, ConnectorContext context) {
    int splitsPerNode = getSplitsPerNode(config);
    NodeManager nodeManager = context.getNodeManager();
    return new Connector() {

        @Override
        public ConnectorTransactionHandle beginTransaction(IsolationLevel isolationLevel, boolean readOnly) {
            return TpcdsTransactionHandle.INSTANCE;
        }

        @Override
        public ConnectorMetadata getMetadata(ConnectorTransactionHandle transactionHandle) {
            return new TpcdsMetadata();
        }

        @Override
        public ConnectorSplitManager getSplitManager() {
            return new TpcdsSplitManager(nodeManager, splitsPerNode, isWithNoSexism(config));
        }

        @Override
        public ConnectorRecordSetProvider getRecordSetProvider() {
            return new TpcdsRecordSetProvider();
        }

        @Override
        public ConnectorNodePartitioningProvider getNodePartitioningProvider() {
            return new TpcdsNodePartitioningProvider(nodeManager, splitsPerNode);
        }
    };
}
Also used : NodeManager(io.prestosql.spi.NodeManager) Connector(io.prestosql.spi.connector.Connector) IsolationLevel(io.prestosql.spi.transaction.IsolationLevel) ConnectorTransactionHandle(io.prestosql.spi.connector.ConnectorTransactionHandle)

Example 2 with IsolationLevel

use of io.prestosql.spi.transaction.IsolationLevel in project hetu-core by openlookeng.

the class IndexedTpchConnectorFactory method create.

@Override
public Connector create(String catalogName, Map<String, String> properties, ConnectorContext context) {
    int splitsPerNode = getSplitsPerNode(properties);
    TpchIndexedData indexedData = new TpchIndexedData(indexSpec);
    NodeManager nodeManager = context.getNodeManager();
    return new Connector() {

        @Override
        public ConnectorTransactionHandle beginTransaction(IsolationLevel isolationLevel, boolean readOnly) {
            return TpchTransactionHandle.INSTANCE;
        }

        @Override
        public ConnectorMetadata getMetadata(ConnectorTransactionHandle transactionHandle) {
            return new TpchIndexMetadata(indexedData);
        }

        @Override
        public ConnectorSplitManager getSplitManager() {
            return new TpchSplitManager(nodeManager, splitsPerNode);
        }

        @Override
        public ConnectorRecordSetProvider getRecordSetProvider() {
            return new TpchRecordSetProvider();
        }

        @Override
        public ConnectorIndexProvider getIndexProvider() {
            return new TpchIndexProvider(indexedData);
        }

        @Override
        public Set<SystemTable> getSystemTables() {
            return ImmutableSet.of(new ExampleSystemTable());
        }

        @Override
        public ConnectorNodePartitioningProvider getNodePartitioningProvider() {
            return new TpchNodePartitioningProvider(nodeManager, splitsPerNode);
        }
    };
}
Also used : Connector(io.prestosql.spi.connector.Connector) IsolationLevel(io.prestosql.spi.transaction.IsolationLevel) ConnectorTransactionHandle(io.prestosql.spi.connector.ConnectorTransactionHandle) TpchSplitManager(io.prestosql.plugin.tpch.TpchSplitManager) NodeManager(io.prestosql.spi.NodeManager) TpchNodePartitioningProvider(io.prestosql.plugin.tpch.TpchNodePartitioningProvider) SystemTable(io.prestosql.spi.connector.SystemTable) TpchRecordSetProvider(io.prestosql.plugin.tpch.TpchRecordSetProvider)

Example 3 with IsolationLevel

use of io.prestosql.spi.transaction.IsolationLevel in project hetu-core by openlookeng.

the class StartTransactionTask method execute.

@Override
public ListenableFuture<?> execute(StartTransaction statement, TransactionManager transactionManager, Metadata metadata, AccessControl accessControl, QueryStateMachine stateMachine, List<Expression> parameters, HeuristicIndexerManager heuristicIndexerManager) {
    Session session = stateMachine.getSession();
    if (!session.isClientTransactionSupport()) {
        throw new PrestoException(StandardErrorCode.INCOMPATIBLE_CLIENT, "Client does not support transactions");
    }
    if (session.getTransactionId().isPresent()) {
        throw new PrestoException(StandardErrorCode.NOT_SUPPORTED, "Nested transactions not supported");
    }
    Optional<IsolationLevel> isolationLevel = extractIsolationLevel(statement);
    Optional<Boolean> readOnly = extractReadOnly(statement);
    TransactionId transactionId = transactionManager.beginTransaction(isolationLevel.orElse(TransactionManager.DEFAULT_ISOLATION), readOnly.orElse(TransactionManager.DEFAULT_READ_ONLY), false);
    stateMachine.setStartedTransactionId(transactionId);
    // Since the current session does not contain this new transaction ID, we need to manually mark it as inactive
    // when this statement completes.
    transactionManager.trySetInactive(transactionId);
    return immediateFuture(null);
}
Also used : IsolationLevel(io.prestosql.spi.transaction.IsolationLevel) PrestoException(io.prestosql.spi.PrestoException) Session(io.prestosql.Session) TransactionId(io.prestosql.transaction.TransactionId)

Example 4 with IsolationLevel

use of io.prestosql.spi.transaction.IsolationLevel in project hetu-core by openlookeng.

the class TpchConnectorFactory method create.

@Override
public Connector create(String catalogName, Map<String, String> properties, ConnectorContext context) {
    int splitsPerNode = getSplitsPerNode(properties);
    ColumnNaming columnNaming = ColumnNaming.valueOf(properties.getOrDefault(TPCH_COLUMN_NAMING_PROPERTY, ColumnNaming.SIMPLIFIED.name()).toUpperCase());
    NodeManager nodeManager = context.getNodeManager();
    return new Connector() {

        @Override
        public ConnectorTransactionHandle beginTransaction(IsolationLevel isolationLevel, boolean readOnly) {
            return TpchTransactionHandle.INSTANCE;
        }

        @Override
        public ConnectorMetadata getMetadata(ConnectorTransactionHandle transaction) {
            return new TpchMetadata(columnNaming, predicatePushdownEnabled, partitioningEnabled);
        }

        @Override
        public ConnectorSplitManager getSplitManager() {
            return new TpchSplitManager(nodeManager, splitsPerNode);
        }

        @Override
        public ConnectorRecordSetProvider getRecordSetProvider() {
            return new TpchRecordSetProvider();
        }

        @Override
        public ConnectorNodePartitioningProvider getNodePartitioningProvider() {
            return new TpchNodePartitioningProvider(nodeManager, splitsPerNode);
        }
    };
}
Also used : NodeManager(io.prestosql.spi.NodeManager) Connector(io.prestosql.spi.connector.Connector) IsolationLevel(io.prestosql.spi.transaction.IsolationLevel) ConnectorTransactionHandle(io.prestosql.spi.connector.ConnectorTransactionHandle)

Aggregations

IsolationLevel (io.prestosql.spi.transaction.IsolationLevel)4 NodeManager (io.prestosql.spi.NodeManager)3 Connector (io.prestosql.spi.connector.Connector)3 ConnectorTransactionHandle (io.prestosql.spi.connector.ConnectorTransactionHandle)3 Session (io.prestosql.Session)1 TpchNodePartitioningProvider (io.prestosql.plugin.tpch.TpchNodePartitioningProvider)1 TpchRecordSetProvider (io.prestosql.plugin.tpch.TpchRecordSetProvider)1 TpchSplitManager (io.prestosql.plugin.tpch.TpchSplitManager)1 PrestoException (io.prestosql.spi.PrestoException)1 SystemTable (io.prestosql.spi.connector.SystemTable)1 TransactionId (io.prestosql.transaction.TransactionId)1