Repository: asterixdb Updated Branches: refs/heads/master a9ac670b8 -> 55a558f21
http://git-wip-us.apache.org/repos/asf/asterixdb/blob/55a558f2/asterixdb/asterix-metadata/src/main/java/org/apache/asterix/metadata/declared/MetadataManagerUtil.java ---------------------------------------------------------------------- diff --git a/asterixdb/asterix-metadata/src/main/java/org/apache/asterix/metadata/declared/MetadataManagerUtil.java b/asterixdb/asterix-metadata/src/main/java/org/apache/asterix/metadata/declared/MetadataManagerUtil.java new file mode 100644 index 0000000..0b22dab --- /dev/null +++ b/asterixdb/asterix-metadata/src/main/java/org/apache/asterix/metadata/declared/MetadataManagerUtil.java @@ -0,0 +1,178 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you 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 org.apache.asterix.metadata.declared; + +import java.util.ArrayList; +import java.util.List; + +import org.apache.asterix.common.config.DatasetConfig.DatasetType; +import org.apache.asterix.metadata.MetadataException; +import org.apache.asterix.metadata.MetadataManager; +import org.apache.asterix.metadata.MetadataTransactionContext; +import org.apache.asterix.metadata.declared.AqlDataSource.AqlDataSourceType; +import org.apache.asterix.metadata.entities.Dataset; +import org.apache.asterix.metadata.entities.DatasourceAdapter; +import org.apache.asterix.metadata.entities.Datatype; +import org.apache.asterix.metadata.entities.Feed; +import org.apache.asterix.metadata.entities.FeedPolicyEntity; +import org.apache.asterix.metadata.entities.Index; +import org.apache.asterix.metadata.entities.NodeGroup; +import org.apache.asterix.metadata.utils.MetadataConstants; +import org.apache.asterix.om.types.ARecordType; +import org.apache.asterix.om.types.IAType; +import org.apache.asterix.om.util.AsterixClusterProperties; +import org.apache.hyracks.algebricks.common.exceptions.AlgebricksException; +import org.apache.hyracks.algebricks.core.algebra.properties.DefaultNodeGroupDomain; +import org.apache.hyracks.algebricks.core.algebra.properties.INodeDomain; + +public class MetadataManagerUtil { + + private MetadataManagerUtil() { + throw new AssertionError("This util class should not be initialized."); + } + + public static IAType findType(MetadataTransactionContext mdTxnCtx, String dataverse, String typeName) + throws AlgebricksException { + if (dataverse == null || typeName == null) { + return null; + } + Datatype type; + try { + type = MetadataManager.INSTANCE.getDatatype(mdTxnCtx, dataverse, typeName); + } catch (MetadataException e) { + throw new AlgebricksException( + "Metadata exception while looking up type '" + typeName + "' in dataverse '" + dataverse + "'", e); + } + if (type == null) { + throw new AlgebricksException("Type name '" + typeName + "' unknown in dataverse '" + dataverse + "'"); + } + return type.getDatatype(); + } + + public static ARecordType findOutputRecordType(MetadataTransactionContext mdTxnCtx, String dataverse, + String outputRecordType) throws AlgebricksException { + if (outputRecordType == null) { + return null; + } + if (dataverse == null) { + throw new AlgebricksException("Cannot declare output-record-type with no dataverse!"); + } + IAType type = findType(mdTxnCtx, dataverse, outputRecordType); + if (!(type instanceof ARecordType)) { + throw new AlgebricksException("Type " + outputRecordType + " is not a record type!"); + } + return (ARecordType) type; + } + + public static DatasourceAdapter getAdapter(MetadataTransactionContext mdTxnCtx, String dataverseName, + String adapterName) throws MetadataException { + DatasourceAdapter adapter; + // search in default namespace (built-in adapter) + adapter = MetadataManager.INSTANCE.getAdapter(mdTxnCtx, MetadataConstants.METADATA_DATAVERSE_NAME, adapterName); + + // search in dataverse (user-defined adapter) + if (adapter == null) { + adapter = MetadataManager.INSTANCE.getAdapter(mdTxnCtx, dataverseName, adapterName); + } + return adapter; + } + + public static Dataset findDataset(MetadataTransactionContext mdTxnCtx, String dataverse, String dataset) + throws AlgebricksException { + try { + return MetadataManager.INSTANCE.getDataset(mdTxnCtx, dataverse, dataset); + } catch (MetadataException e) { + throw new AlgebricksException(e); + } + } + + public static Dataset findExistingDataset(MetadataTransactionContext mdTxnCtx, String dataverseName, + String datasetName) throws AlgebricksException { + Dataset dataset = findDataset(mdTxnCtx, dataverseName, datasetName); + if (dataset == null) { + throw new AlgebricksException("Unknown dataset " + datasetName + " in dataverse " + dataverseName); + } + return dataset; + } + + public static INodeDomain findNodeDomain(MetadataTransactionContext mdTxnCtx, String nodeGroupName) + throws AlgebricksException { + NodeGroup nodeGroup = MetadataManager.INSTANCE.getNodegroup(mdTxnCtx, nodeGroupName); + List<String> partitions = new ArrayList<>(); + for (String location : nodeGroup.getNodeNames()) { + int numPartitions = AsterixClusterProperties.INSTANCE.getNodePartitionsCount(location); + for (int i = 0; i < numPartitions; i++) { + partitions.add(location); + } + } + return new DefaultNodeGroupDomain(partitions); + } + + public static Feed findFeed(MetadataTransactionContext mdTxnCtx, String dataverse, String feedName) + throws AlgebricksException { + try { + return MetadataManager.INSTANCE.getFeed(mdTxnCtx, dataverse, feedName); + } catch (MetadataException e) { + throw new AlgebricksException(e); + } + } + + public static FeedPolicyEntity findFeedPolicy(MetadataTransactionContext mdTxnCtx, String dataverse, + String policyName) throws AlgebricksException { + try { + return MetadataManager.INSTANCE.getFeedPolicy(mdTxnCtx, dataverse, policyName); + } catch (MetadataException e) { + throw new AlgebricksException(e); + } + } + + public static List<Index> getDatasetIndexes(MetadataTransactionContext mdTxnCtx, String dataverseName, + String datasetName) throws AlgebricksException { + try { + return MetadataManager.INSTANCE.getDatasetIndexes(mdTxnCtx, dataverseName, datasetName); + } catch (MetadataException e) { + throw new AlgebricksException(e); + } + } + + public static AqlDataSource findDataSource(MetadataTransactionContext mdTxnCtx, AqlSourceId id) + throws AlgebricksException { + AqlSourceId aqlId = id; + try { + return lookupSourceInMetadata(mdTxnCtx, aqlId); + } catch (MetadataException e) { + throw new AlgebricksException(e); + } + } + + public static AqlDataSource lookupSourceInMetadata(MetadataTransactionContext mdTxnCtx, AqlSourceId aqlId) + throws AlgebricksException { + Dataset dataset = findDataset(mdTxnCtx, aqlId.getDataverseName(), aqlId.getDatasourceName()); + if (dataset == null) { + throw new AlgebricksException("Datasource with id " + aqlId + " was not found."); + } + IAType itemType = findType(mdTxnCtx, dataset.getItemTypeDataverseName(), dataset.getItemTypeName()); + IAType metaItemType = findType(mdTxnCtx, dataset.getMetaItemTypeDataverseName(), dataset.getMetaItemTypeName()); + INodeDomain domain = findNodeDomain(mdTxnCtx, dataset.getNodeGroupName()); + byte datasourceType = dataset.getDatasetType().equals(DatasetType.EXTERNAL) ? AqlDataSourceType.EXTERNAL_DATASET + : AqlDataSourceType.INTERNAL_DATASET; + return new DatasetDataSource(aqlId, dataset, itemType, metaItemType, datasourceType, + dataset.getDatasetDetails(), domain); + } +} \ No newline at end of file
