| /* |
| * 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.cassandra.db.index.composites; |
| |
| import java.nio.ByteBuffer; |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| import org.apache.cassandra.config.CFMetaData; |
| import org.apache.cassandra.config.ColumnDefinition; |
| import org.apache.cassandra.db.*; |
| import org.apache.cassandra.db.composites.CBuilder; |
| import org.apache.cassandra.db.composites.CellName; |
| import org.apache.cassandra.db.composites.CellNameType; |
| import org.apache.cassandra.db.composites.Composite; |
| import org.apache.cassandra.db.composites.CompoundDenseCellNameType; |
| import org.apache.cassandra.db.index.SecondaryIndex; |
| import org.apache.cassandra.db.marshal.*; |
| |
| /** |
| * Common superclass for indexes that capture collection keys, including |
| * indexes on such keys themselves. |
| * |
| * A cell indexed by this index will have the general form: |
| * ck_0 ... ck_n c_name [col_elt] : v |
| * where ck_i are the cluster keys, c_name the CQL3 column name, col_elt the |
| * collection element that we want to index (which may or may not be there depending |
| * on whether c_name is the collection we're indexing), and v the cell value. |
| * |
| * Such a cell is indexed if c_name is the indexed collection (in which case we are guaranteed to have |
| * col_elt). The index entry can be viewed in the following way: |
| * - the row key is determined by subclasses of this type. |
| * - the cell name will be 'rk ck_0 ... ck_n' where rk is the row key of the initial cell. |
| */ |
| public abstract class CompositesIndexIncludingCollectionKey extends CompositesIndex |
| { |
| public static CellNameType buildIndexComparator(CFMetaData baseMetadata, ColumnDefinition columnDef) |
| { |
| int count = 1 + baseMetadata.clusteringColumns().size(); // row key + clustering prefix |
| List<AbstractType<?>> types = new ArrayList<AbstractType<?>>(count); |
| types.add(SecondaryIndex.keyComparator); |
| for (int i = 0; i < count - 1; i++) |
| types.add(baseMetadata.comparator.subtype(i)); |
| return new CompoundDenseCellNameType(types); |
| } |
| |
| protected Composite makeIndexColumnPrefix(ByteBuffer rowKey, Composite cellName) |
| { |
| int count = 1 + baseCfs.metadata.clusteringColumns().size(); |
| CBuilder builder = getIndexComparator().builder(); |
| builder.add(rowKey); |
| for (int i = 0; i < Math.min(cellName.size(), count - 1); i++) |
| builder.add(cellName.get(i)); |
| return builder.build(); |
| } |
| |
| public IndexedEntry decodeEntry(DecoratedKey indexedValue, Cell indexEntry) |
| { |
| int count = 1 + baseCfs.metadata.clusteringColumns().size(); |
| CBuilder builder = baseCfs.getComparator().builder(); |
| for (int i = 0; i < count - 1; i++) |
| builder.add(indexEntry.name().get(i + 1)); |
| return new IndexedEntry(indexedValue, indexEntry.name(), indexEntry.timestamp(), indexEntry.name().get(0), builder.build()); |
| } |
| |
| @Override |
| public boolean indexes(CellName name) |
| { |
| // We index if the CQL3 column name is the one of the collection we index |
| AbstractType<?> comp = baseCfs.metadata.getColumnDefinitionComparator(columnDef); |
| return name.size() > columnDef.position() |
| && comp.compare(name.get(columnDef.position()), columnDef.name.bytes) == 0; |
| } |
| } |