Java ConnectorSplitManager类代码示例

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
private static void registerBogusConnector(TransactionManager transactionManager, String connectorId)
{
    transactionManager.addConnector(connectorId, new Connector()
    {
        @Override
        public ConnectorTransactionHandle beginTransaction(IsolationLevel isolationLevel, boolean readOnly)
        {
            // Just return something
            return new ConnectorTransactionHandle() {};
        }

        @Override
        public ConnectorMetadata getMetadata(ConnectorTransactionHandle transactionHandle)
        {
            throw new UnsupportedOperationException();
        }

        @Override
        public ConnectorSplitManager getSplitManager()
        {
            throw new UnsupportedOperationException();
        }
    });
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
@Override
public Connector create(String connectorId, Map<String, String> properties)
{
    return new Connector()
    {
        @Override
        public ConnectorTransactionHandle beginTransaction(IsolationLevel isolationLevel, boolean readOnly)
        {
            checkConnectorSupports(READ_COMMITTED, isolationLevel);
            return JmxTransactionHandle.INSTANCE;
        }

        @Override
        public ConnectorMetadata getMetadata(ConnectorTransactionHandle transactionHandle)
        {
            return new JmxMetadata(connectorId, mbeanServer);
        }

        @Override
        public ConnectorSplitManager getSplitManager()
        {
            return new JmxSplitManager(connectorId, nodeManager);
        }

        @Override
        public ConnectorRecordSetProvider getRecordSetProvider()
        {
            return new JmxRecordSetProvider(mbeanServer, nodeManager.getCurrentNode().getNodeIdentifier());
        }
    };
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
public SplitSource getSplits(Session session, TableLayoutHandle layout)
{
    String connectorId = layout.getConnectorId();
    ConnectorSplitManager splitManager = getConnectorSplitManager(connectorId);

    // assumes connectorId and catalog are the same
    ConnectorSession connectorSession = session.toConnectorSession(connectorId);

    ConnectorSplitSource source = splitManager.getSplits(layout.getTransactionHandle(), connectorSession, layout.getConnectorHandle());

    return new ConnectorAwareSplitSource(connectorId, layout.getTransactionHandle(), source);
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
@Override
public Connector create(String connectorId, Map<String, String> properties)
{
    int splitsPerNode = getSplitsPerNode(properties);

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

        @Override
        public ConnectorMetadata getMetadata(ConnectorTransactionHandle transaction)
        {
            return new TpchMetadata(connectorId);
        }

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

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

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
@Override
public Connector create(String connectorId, Map<String, String> properties)
{
    requireNonNull(properties, "properties is null");
    int splitsPerNode = getSplitsPerNode(properties);

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

        @Override
        public ConnectorMetadata getMetadata(ConnectorTransactionHandle transaction)
        {
            return new SampledTpchMetadata(connectorId);
        }

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

        @Override
        public ConnectorRecordSetProvider getRecordSetProvider()
        {
            return new SampledTpchRecordSetProvider(connectorId, sampleWeight);
        }
    };
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
@Override
public ConnectorSplitManager getSplitManager()
{
    return hdfsSplitManager;
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
@Override
public ConnectorSplitManager getSplitManager()
{
    return new RestSplitManager(nodeManager);
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
@Override
public ConnectorSplitManager getSplitManager() {
    return splitManager;
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
@Override
public ConnectorSplitManager getSplitManager()
{
    return splitManager;
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
@Override
public Connector create(String connectorId, Map<String, String> properties)
{
    int splitsPerNode = getSplitsPerNode(properties);
    TpchIndexedData indexedData = new TpchIndexedData(connectorId, indexSpec);

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

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

        @Override
        public ConnectorSplitManager getSplitManager()
        {
            return new TpchSplitManager(connectorId, 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());
        }
    };
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
@Override
public ConnectorSplitManager getSplitManager()
{
    return new LegacyConnectorSplitManager(connector.getSplitManager());
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
public LegacyConnectorSplitManager(com.facebook.presto.spi.ConnectorSplitManager splitManager)
{
    this.splitManager = requireNonNull(splitManager, "splitManager is null");
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
@Override
public ConnectorSplitManager getSplitManager()
{
    return new ConnectorSplitManager() {};
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
public void addConnectorSplitManager(String connectorId, ConnectorSplitManager connectorSplitManager)
{
    checkState(splitManagers.putIfAbsent(connectorId, connectorSplitManager) == null, "SplitManager for connector '%s' is already registered", connectorId);
}
 

import com.facebook.presto.spi.connector.ConnectorSplitManager; //导入依赖的package包/类
private ConnectorSplitManager getConnectorSplitManager(String connectorId)
{
    ConnectorSplitManager result = splitManagers.get(connectorId);
    checkArgument(result != null, "No split manager for connector '%s'", connectorId);
    return result;
}