/*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.facebook.presto.connector;
import com.facebook.presto.connector.informationSchema.InformationSchemaMetadata;
import com.facebook.presto.connector.informationSchema.InformationSchemaPageSourceProvider;
import com.facebook.presto.connector.informationSchema.InformationSchemaSplitManager;
import com.facebook.presto.index.IndexManager;
import com.facebook.presto.metadata.HandleResolver;
import com.facebook.presto.metadata.MetadataManager;
import com.facebook.presto.operator.RecordSinkManager;
import com.facebook.presto.spi.Connector;
import com.facebook.presto.spi.ConnectorFactory;
import com.facebook.presto.spi.ConnectorHandleResolver;
import com.facebook.presto.spi.ConnectorIndexResolver;
import com.facebook.presto.spi.ConnectorMetadata;
import com.facebook.presto.spi.ConnectorPageSourceProvider;
import com.facebook.presto.spi.ConnectorRecordSetProvider;
import com.facebook.presto.spi.ConnectorRecordSinkProvider;
import com.facebook.presto.spi.ConnectorSplitManager;
import com.facebook.presto.spi.NodeManager;
import com.facebook.presto.spi.classloader.ThreadContextClassLoader;
import com.facebook.presto.split.PageSourceManager;
import com.facebook.presto.split.RecordPageSourceProvider;
import com.facebook.presto.split.SplitManager;
import com.google.inject.Inject;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;
public class ConnectorManager
{
public static final String INFORMATION_SCHEMA_CONNECTOR_PREFIX = "$info_schema@";
private final MetadataManager metadataManager;
private final SplitManager splitManager;
private final PageSourceManager pageSourceManager;
private final IndexManager indexManager;
private final RecordSinkManager recordSinkManager;
private final HandleResolver handleResolver;
private final NodeManager nodeManager;
private final ConcurrentMap<String, ConnectorFactory> connectorFactories = new ConcurrentHashMap<>();
private final ConcurrentMap<String, Connector> connectors = new ConcurrentHashMap<>();
@Inject
public ConnectorManager(MetadataManager metadataManager,
SplitManager splitManager,
PageSourceManager pageSourceManager,
IndexManager indexManager,
RecordSinkManager recordSinkManager,
HandleResolver handleResolver,
Map<String, ConnectorFactory> connectorFactories,
NodeManager nodeManager)
{
this.metadataManager = metadataManager;
this.splitManager = splitManager;
this.pageSourceManager = pageSourceManager;
this.indexManager = indexManager;
this.recordSinkManager = recordSinkManager;
this.handleResolver = handleResolver;
this.nodeManager = nodeManager;
this.connectorFactories.putAll(connectorFactories);
}
public void addConnectorFactory(ConnectorFactory connectorFactory)
{
ConnectorFactory existingConnectorFactory = connectorFactories.putIfAbsent(connectorFactory.getName(), connectorFactory);
checkArgument(existingConnectorFactory == null, "Connector %s is already registered", connectorFactory.getName());
}
public synchronized void createConnection(String catalogName, String connectorName, Map<String, String> properties)
{
checkNotNull(catalogName, "catalogName is null");
checkNotNull(connectorName, "connectorName is null");
checkNotNull(properties, "properties is null");
ConnectorFactory connectorFactory = connectorFactories.get(connectorName);
checkArgument(connectorFactory != null, "No factory for connector %s", connectorName);
try (ThreadContextClassLoader ignored = new ThreadContextClassLoader(connectorFactory.getClass().getClassLoader())) {
createConnection(catalogName, connectorFactory, properties);
}
}
public synchronized void createConnection(String catalogName, ConnectorFactory connectorFactory, Map<String, String> properties)
{
checkNotNull(catalogName, "catalogName is null");
checkNotNull(properties, "properties is null");
checkNotNull(connectorFactory, "connectorFactory is null");
// for now connectorId == catalogName
String connectorId = catalogName;
checkState(!connectors.containsKey(connectorId), "A connector %s already exists", connectorId);
Connector connector = connectorFactory.create(connectorId, properties);
connectors.put(connectorId, connector);
addConnector(catalogName, connectorId, connector);
}
private void addConnector(String catalogName, String connectorId, Connector connector)
{
ConnectorMetadata connectorMetadata = connector.getMetadata();
checkState(connectorMetadata != null, "Connector %s can not provide metadata", connectorId);
ConnectorSplitManager connectorSplitManager = connector.getSplitManager();
checkState(connectorSplitManager != null, "Connector %s does not have a split manager", connectorId);
ConnectorPageSourceProvider connectorPageSourceProvider = null;
try {
connectorPageSourceProvider = connector.getPageSourceProvider();
checkNotNull(connectorPageSourceProvider, "Connector %s returned a null page source provider", connectorId);
}
catch (UnsupportedOperationException ignored) {
}
if (connectorPageSourceProvider == null) {
ConnectorRecordSetProvider connectorRecordSetProvider = null;
try {
connectorRecordSetProvider = connector.getRecordSetProvider();
checkNotNull(connectorRecordSetProvider, "Connector %s returned a null record set provider", connectorId);
}
catch (UnsupportedOperationException ignored) {
}
checkState(connectorRecordSetProvider != null, "Connector %s has neither a PageSource or RecordSet provider", connectorId);
connectorPageSourceProvider = new RecordPageSourceProvider(connectorRecordSetProvider);
}
ConnectorHandleResolver connectorHandleResolver = connector.getHandleResolver();
checkNotNull(connectorHandleResolver, "Connector %s does not have a handle resolver", connectorId);
ConnectorRecordSinkProvider connectorRecordSinkProvider = null;
try {
connectorRecordSinkProvider = connector.getRecordSinkProvider();
checkNotNull(connectorRecordSinkProvider, "Connector %s returned a null record sink provider", connectorId);
}
catch (UnsupportedOperationException ignored) {
}
ConnectorIndexResolver indexResolver = null;
try {
indexResolver = connector.getIndexResolver();
checkNotNull(indexResolver, "Connector %s returned a null index resolver", connectorId);
}
catch (UnsupportedOperationException ignored) {
}
// IMPORTANT: all the instances need to be fetched from the connector *before* we add them to the corresponding managers.
// Otherwise, a broken connector would leave the managers in an inconsistent state with respect to each other
metadataManager.addConnectorMetadata(connectorId, catalogName, connectorMetadata);
metadataManager.addInformationSchemaMetadata(makeInformationSchemaConnectorId(connectorId), catalogName, new InformationSchemaMetadata(catalogName));
splitManager.addConnectorSplitManager(makeInformationSchemaConnectorId(connectorId), new InformationSchemaSplitManager(nodeManager));
pageSourceManager.addConnectorPageSourceProvider(makeInformationSchemaConnectorId(connectorId), new InformationSchemaPageSourceProvider(metadataManager, splitManager));
splitManager.addConnectorSplitManager(connectorId, connectorSplitManager);
handleResolver.addHandleResolver(connectorId, connectorHandleResolver);
pageSourceManager.addConnectorPageSourceProvider(connectorId, connectorPageSourceProvider);
if (connectorRecordSinkProvider != null) {
recordSinkManager.addConnectorRecordSinkProvider(connectorId, connectorRecordSinkProvider);
}
if (indexResolver != null) {
indexManager.addIndexResolver(connectorId, indexResolver);
}
}
private static String makeInformationSchemaConnectorId(String connectorId)
{
return INFORMATION_SCHEMA_CONNECTOR_PREFIX + connectorId;
}
}