org.apache.phoenix.execute.PhoenixTxIndexMutationGenerator Maven / Gradle / Ivy
The newest version!
/*
* 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.phoenix.execute;
import java.io.IOException;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.ListIterator;
import java.util.Map;
import java.util.Set;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.Cell;
import org.apache.hadoop.hbase.CellScanner;
import org.apache.hadoop.hbase.HRegionLocation;
import org.apache.hadoop.hbase.KeyValue;
import org.apache.hadoop.hbase.KeyValueUtil;
import org.apache.hadoop.hbase.client.Mutation;
import org.apache.hadoop.hbase.client.Result;
import org.apache.hadoop.hbase.client.ResultScanner;
import org.apache.hadoop.hbase.client.Scan;
import org.apache.hadoop.hbase.client.Table;
import org.apache.hadoop.hbase.io.ImmutableBytesWritable;
import org.apache.hadoop.hbase.util.Bytes;
import org.apache.hadoop.hbase.util.Pair;
import org.apache.phoenix.cache.IndexMetaDataCache;
import org.apache.phoenix.compile.ScanRanges;
import org.apache.phoenix.coprocessor.MetaDataProtocol;
import org.apache.phoenix.filter.SkipScanFilter;
import org.apache.phoenix.hbase.index.MultiMutation;
import org.apache.phoenix.hbase.index.ValueGetter;
import org.apache.phoenix.hbase.index.covered.IndexMetaData;
import org.apache.phoenix.hbase.index.covered.IndexUpdate;
import org.apache.phoenix.hbase.index.covered.TableState;
import org.apache.phoenix.hbase.index.covered.update.ColumnReference;
import org.apache.phoenix.hbase.index.covered.update.ColumnTracker;
import org.apache.phoenix.hbase.index.covered.update.IndexedColumnGroup;
import org.apache.phoenix.hbase.index.util.ImmutableBytesPtr;
import org.apache.phoenix.index.IndexMaintainer;
import org.apache.phoenix.index.PhoenixIndexCodec;
import org.apache.phoenix.index.PhoenixIndexMetaData;
import org.apache.phoenix.jdbc.PhoenixConnection;
import org.apache.phoenix.query.ConnectionQueryServices;
import org.apache.phoenix.query.KeyRange;
import org.apache.phoenix.schema.PTable;
import org.apache.phoenix.schema.types.PVarbinary;
import org.apache.phoenix.transaction.PhoenixTransactionContext;
import org.apache.phoenix.transaction.PhoenixTransactionContext.PhoenixVisibilityLevel;
import org.apache.phoenix.util.ScanUtil;
import org.apache.phoenix.util.SchemaUtil;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
import com.google.common.primitives.Longs;
public class PhoenixTxIndexMutationGenerator {
private final PhoenixIndexCodec codec;
private final PhoenixIndexMetaData indexMetaData;
private final ConnectionQueryServices services;
private final byte[] regionStartKey;
private final byte[] regionEndKey;
private final byte[] tableName;
private PhoenixTxIndexMutationGenerator(ConnectionQueryServices services, Configuration conf, PhoenixIndexMetaData indexMetaData, byte[] tableName, byte[] regionStartKey, byte[] regionEndKey) {
this.services = services;
this.indexMetaData = indexMetaData;
this.regionStartKey = regionStartKey;
this.regionEndKey = regionEndKey;
this.tableName = tableName;
this.codec = new PhoenixIndexCodec(conf, tableName);
}
public PhoenixTxIndexMutationGenerator(Configuration conf, PhoenixIndexMetaData indexMetaData, byte[] tableName, byte[] regionStartKey, byte[] regionEndKey) {
this(null, conf, indexMetaData, tableName, regionStartKey, regionEndKey);
}
public PhoenixTxIndexMutationGenerator(ConnectionQueryServices services, PhoenixIndexMetaData indexMetaData, byte[] tableName) {
this(services, services.getConfiguration(), indexMetaData, tableName, null, null);
}
private static void addMutation(Map mutations, ImmutableBytesPtr row, Mutation m) {
MultiMutation stored = mutations.get(row);
// we haven't seen this row before, so add it
if (stored == null) {
stored = new MultiMutation(row);
mutations.put(row, stored);
}
stored.addAll(m);
}
public Collection> getIndexUpdates(Table htable, Iterator mutationIterator) throws IOException, SQLException {
if (!mutationIterator.hasNext()) {
return Collections.emptyList();
}
List indexMaintainers = indexMetaData.getIndexMaintainers();
ResultScanner currentScanner = null;
// Collect up all mutations in batch
Map mutations =
new HashMap();
// Collect the set of mutable ColumnReferences so that we can first
// run a scan to get the current state. We'll need this to delete
// the existing index rows.
int estimatedSize = indexMaintainers.size() * 10;
Set mutableColumns = Sets.newHashSetWithExpectedSize(estimatedSize);
for (IndexMaintainer indexMaintainer : indexMaintainers) {
// For transactional tables, we use an index maintainer
// to aid in rollback if there's a KeyValue column in the index. The alternative would be
// to hold on to all uncommitted index row keys (even ones already sent to HBase) on the
// client side.
Set allColumns = indexMaintainer.getAllColumns();
mutableColumns.addAll(allColumns);
}
Mutation m = mutationIterator.next();
Map updateAttributes = m.getAttributesMap();
byte[] txRollbackAttribute = updateAttributes.get(PhoenixTransactionContext.TX_ROLLBACK_ATTRIBUTE_KEY);
boolean isRollback = txRollbackAttribute!=null;
boolean isImmutable = indexMetaData.isImmutableRows();
Map findPriorValueMutations;
if (isImmutable && !isRollback) {
findPriorValueMutations = new HashMap();
} else {
findPriorValueMutations = mutations;
}
while (true) {
// add the mutation to the batch set
ImmutableBytesPtr row = new ImmutableBytesPtr(m.getRow());
// if we have no non PK columns, no need to find the prior values
if ( mutations != findPriorValueMutations && indexMetaData.requiresPriorRowState(m) ) {
addMutation(findPriorValueMutations, row, m);
}
addMutation(mutations, row, m);
if (!mutationIterator.hasNext()) {
break;
}
m = mutationIterator.next();
}
Collection> indexUpdates = new ArrayList>(mutations.size() * 2 * indexMaintainers.size());
// Track if we have row keys with Delete mutations (or Puts that are
// Tephra's Delete marker). If there are none, we don't need to do the scan for
// prior versions, if there are, we do. Since rollbacks always have delete mutations,
// this logic will work there too.
if (!findPriorValueMutations.isEmpty()) {
List keys = Lists.newArrayListWithExpectedSize(mutations.size());
for (ImmutableBytesPtr ptr : findPriorValueMutations.keySet()) {
keys.add(PVarbinary.INSTANCE.getKeyRange(ptr.copyBytesIfNecessary()));
}
Scan scan = new Scan();
// Project all mutable columns
for (ColumnReference ref : mutableColumns) {
scan.addColumn(ref.getFamily(), ref.getQualifier());
}
/*
* Indexes inherit the storage scheme of the data table which means all the indexes have the same
* storage scheme and empty key value qualifier. Note that this assumption would be broken if we start
* supporting new indexes over existing data tables to have a different storage scheme than the data
* table.
*/
byte[] emptyKeyValueQualifier = indexMaintainers.get(0).getEmptyKeyValueQualifier();
// Project empty key value column
scan.addColumn(indexMaintainers.get(0).getDataEmptyKeyValueCF(), emptyKeyValueQualifier);
ScanRanges scanRanges = ScanRanges.create(SchemaUtil.VAR_BINARY_SCHEMA, Collections.singletonList(keys), ScanUtil.SINGLE_COLUMN_SLOT_SPAN, null, true, -1);
scanRanges.initializeScan(scan);
Table txTable = indexMetaData.getTransactionContext().getTransactionalTable(htable, true);
// For rollback, we need to see all versions, including
// the last committed version as there may be multiple
// checkpointed versions.
SkipScanFilter filter = scanRanges.getSkipScanFilter();
if (isRollback) {
filter = new SkipScanFilter(filter,true);
indexMetaData.getTransactionContext().setVisibilityLevel(PhoenixVisibilityLevel.SNAPSHOT_ALL);
}
scan.setFilter(filter);
currentScanner = txTable.getScanner(scan);
}
if (isRollback) {
processRollback(indexMetaData, txRollbackAttribute, currentScanner, mutableColumns, indexUpdates, mutations);
} else {
processMutation(indexMetaData, txRollbackAttribute, currentScanner, mutableColumns, indexUpdates, mutations, findPriorValueMutations);
}
return indexUpdates;
}
private void processMutation(PhoenixIndexMetaData indexMetaData, byte[] txRollbackAttribute,
ResultScanner scanner,
Set upsertColumns,
Collection> indexUpdates,
Map mutations,
Map mutationsToFindPreviousValue) throws IOException {
List indexMaintainers = indexMetaData.getIndexMaintainers();
if (scanner != null) {
Result result;
ColumnReference emptyColRef = new ColumnReference(indexMaintainers.get(0)
.getDataEmptyKeyValueCF(), indexMaintainers.get(0).getEmptyKeyValueQualifier());
// Process existing data table rows by removing the old index row and adding the new index row
while ((result = scanner.next()) != null) {
Mutation m = mutationsToFindPreviousValue.remove(new ImmutableBytesPtr(result.getRow()));
TxTableState state = new TxTableState(upsertColumns, indexMetaData.getTransactionContext().getWritePointer(), m, emptyColRef, result);
generateDeletes(indexMetaData, indexUpdates, txRollbackAttribute, state);
generatePuts(indexMetaData, indexUpdates, state);
}
}
// Process new data table by adding new index rows
for (Mutation m : mutations.values()) {
TxTableState state = new TxTableState(upsertColumns, indexMetaData.getTransactionContext().getWritePointer(), m);
generatePuts(indexMetaData, indexUpdates, state);
generateDeletes(indexMetaData, indexUpdates, txRollbackAttribute, state);
}
}
private void processRollback(PhoenixIndexMetaData indexMetaData, byte[] txRollbackAttribute,
ResultScanner scanner,
Set mutableColumns,
Collection> indexUpdates,
Map mutations) throws IOException {
if (scanner != null) {
long readPtr = indexMetaData.getTransactionContext().getReadPointer();
Result result;
// Loop through last committed row state plus all new rows associated with current transaction
// to generate point delete markers for all index rows that were added. We don't have Tephra
// manage index rows in change sets because we don't want to be hit with the additional
// memory hit and do not need to do conflict detection on index rows.
ColumnReference emptyColRef = new ColumnReference(indexMetaData.getIndexMaintainers().get(0).getDataEmptyKeyValueCF(), indexMetaData.getIndexMaintainers().get(0).getEmptyKeyValueQualifier());
while ((result = scanner.next()) != null) {
Mutation m = mutations.remove(new ImmutableBytesPtr(result.getRow()));
// Sort by timestamp, type, cf, cq so we can process in time batches from oldest to newest
// (as if we're "replaying" them in time order).
List cells = result.listCells();
Collections.sort(cells, new Comparator() {
@Override
public int compare(Cell o1, Cell o2) {
int c = Longs.compare(o1.getTimestamp(), o2.getTimestamp());
if (c != 0) return c;
c = o1.getTypeByte() - o2.getTypeByte();
if (c != 0) return c;
c = Bytes.compareTo(o1.getFamilyArray(), o1.getFamilyOffset(), o1.getFamilyLength(), o1.getFamilyArray(), o1.getFamilyOffset(), o1.getFamilyLength());
if (c != 0) return c;
return Bytes.compareTo(o1.getQualifierArray(), o1.getQualifierOffset(), o1.getQualifierLength(), o1.getQualifierArray(), o1.getQualifierOffset(), o1.getQualifierLength());
}
});
int i = 0;
int nCells = cells.size();
Result oldResult = null, newResult;
do {
boolean hasPuts = false;
LinkedList singleTimeCells = Lists.newLinkedList();
long writePtr;
Cell cell = cells.get(i);
do {
hasPuts |= cell.getTypeByte() == KeyValue.Type.Put.getCode();
writePtr = cell.getTimestamp();
ListIterator it = singleTimeCells.listIterator();
do {
// Add at the beginning of the list to match the expected HBase
// newest to oldest sort order (which TxTableState relies on
// with the Result.getLatestColumnValue() calls). However, we
// still want to add Cells in the expected order for each time
// bound as otherwise we won't find it in our old state.
it.add(cell);
} while (++i < nCells && (cell=cells.get(i)).getTimestamp() == writePtr);
} while (i < nCells && cell.getTimestamp() <= readPtr);
// Generate point delete markers for the prior row deletion of the old index value.
// The write timestamp is the next timestamp, not the current timestamp,
// as the earliest cells are the current values for the row (and we don't
// want to delete the current row).
if (oldResult != null) {
TxTableState state = new TxTableState(mutableColumns, writePtr, m, emptyColRef, oldResult);
generateDeletes(indexMetaData, indexUpdates, txRollbackAttribute, state);
}
// Generate point delete markers for the new index value.
// If our time batch doesn't have Puts (i.e. we have only Deletes), then do not
// generate deletes. We would have generated the delete above based on the state
// of the previous row. The delete markers do not give us the state we need to
// delete.
if (hasPuts) {
newResult = Result.create(singleTimeCells);
// First row may represent the current state which we don't want to delete
if (writePtr > readPtr) {
TxTableState state = new TxTableState(mutableColumns, writePtr, m, emptyColRef, newResult);
generateDeletes(indexMetaData, indexUpdates, txRollbackAttribute, state);
}
oldResult = newResult;
} else {
oldResult = null;
}
} while (i < nCells);
}
}
}
private void generateDeletes(PhoenixIndexMetaData indexMetaData,
Collection> indexUpdates,
byte[] attribValue, TxTableState state) throws IOException {
byte[] regionStartKey = this.regionStartKey;
byte[] regionEndKey = this.regionEndKey;
if (services != null && indexMetaData.hasLocalIndexes()) {
try {
HRegionLocation tableRegionLocation = services.getTableRegionLocation(tableName, state.getCurrentRowKey());
regionStartKey = tableRegionLocation.getRegionInfo().getStartKey();
regionEndKey = tableRegionLocation.getRegionInfo().getEndKey();
} catch (SQLException e) {
throw new IOException(e);
}
}
Iterable deletes = codec.getIndexDeletes(state, indexMetaData, regionStartKey, regionEndKey);
for (IndexUpdate delete : deletes) {
if (delete.isValid()) {
delete.getUpdate().setAttribute(PhoenixTransactionContext.TX_ROLLBACK_ATTRIBUTE_KEY, attribValue);
indexUpdates.add(new Pair(delete.getUpdate(),delete.getTableName()));
}
}
}
private boolean generatePuts(
PhoenixIndexMetaData indexMetaData,
Collection> indexUpdates,
TxTableState state)
throws IOException {
state.applyMutation();
byte[] regionStartKey = this.regionStartKey;
byte[] regionEndKey = this.regionEndKey;
if (services != null && indexMetaData.hasLocalIndexes()) {
try {
HRegionLocation tableRegionLocation = services.getTableRegionLocation(tableName, state.getCurrentRowKey());
regionStartKey = tableRegionLocation.getRegionInfo().getStartKey();
regionEndKey = tableRegionLocation.getRegionInfo().getEndKey();
} catch (SQLException e) {
throw new IOException(e);
}
}
Iterable puts = codec.getIndexUpserts(state, indexMetaData, regionStartKey, regionEndKey);
boolean validPut = false;
for (IndexUpdate put : puts) {
if (put.isValid()) {
indexUpdates.add(new Pair(put.getUpdate(),put.getTableName()));
validPut = true;
}
}
return validPut;
}
private static class TxTableState implements TableState {
private final Mutation mutation;
private final long currentTimestamp;
private final List pendingUpdates;
private final Set indexedColumns;
private final Map valueMap;
private TxTableState(Set indexedColumns, long currentTimestamp, Mutation mutation) {
this.currentTimestamp = currentTimestamp;
this.indexedColumns = indexedColumns;
this.mutation = mutation;
int estimatedSize = indexedColumns.size();
this.valueMap = Maps.newHashMapWithExpectedSize(estimatedSize);
this.pendingUpdates = Lists.newArrayListWithExpectedSize(estimatedSize);
try {
CellScanner scanner = mutation.cellScanner();
while (scanner.advance()) {
Cell cell = scanner.current();
pendingUpdates.add(KeyValueUtil.ensureKeyValue(cell));
}
} catch (IOException e) {
throw new RuntimeException(e); // Impossible
}
}
public TxTableState(Set indexedColumns, long currentTimestamp, Mutation m, ColumnReference emptyColRef, Result r) {
this(indexedColumns, currentTimestamp, m);
for (ColumnReference ref : indexedColumns) {
Cell cell = r.getColumnLatestCell(ref.getFamily(), ref.getQualifier());
if (cell != null) {
ImmutableBytesWritable ptr = new ImmutableBytesWritable();
ptr.set(cell.getValueArray(), cell.getValueOffset(), cell.getValueLength());
valueMap.put(ref, ptr);
}
}
}
@Override
public long getCurrentTimestamp() {
return currentTimestamp;
}
@Override
public byte[] getCurrentRowKey() {
return mutation.getRow();
}
@Override
public List getIndexColumnHints() {
return Collections.emptyList();
}
private void applyMutation() {
for (Cell cell : pendingUpdates) {
if (cell.getTypeByte() == KeyValue.Type.Delete.getCode() || cell.getTypeByte() == KeyValue.Type.DeleteColumn.getCode()) {
ColumnReference ref = new ColumnReference(cell.getFamilyArray(), cell.getFamilyOffset(), cell.getFamilyLength(), cell.getQualifierArray(), cell.getQualifierOffset(), cell.getQualifierLength());
valueMap.remove(ref);
} else if (cell.getTypeByte() == KeyValue.Type.DeleteFamily.getCode() || cell.getTypeByte() == KeyValue.Type.DeleteFamilyVersion.getCode()) {
for (ColumnReference ref : indexedColumns) {
if (ref.matchesFamily(cell.getFamilyArray(), cell.getFamilyOffset(), cell.getFamilyLength())) {
valueMap.remove(ref);
}
}
} else if (cell.getTypeByte() == KeyValue.Type.Put.getCode()){
ColumnReference ref = new ColumnReference(cell.getFamilyArray(), cell.getFamilyOffset(), cell.getFamilyLength(), cell.getQualifierArray(), cell.getQualifierOffset(), cell.getQualifierLength());
if (indexedColumns.contains(ref)) {
ImmutableBytesWritable ptr = new ImmutableBytesWritable();
ptr.set(cell.getValueArray(), cell.getValueOffset(), cell.getValueLength());
valueMap.put(ref, ptr);
}
} else {
throw new IllegalStateException("Unexpected mutation type for " + cell);
}
}
}
@Override
public Collection getPendingUpdate() {
return pendingUpdates;
}
@Override
public Pair getIndexUpdateState(Collection indexedColumns, boolean ignoreNewerMutations, boolean returnNullScannerIfRowNotFound, IndexMetaData indexMetaData)
throws IOException {
// TODO: creating these objects over and over again is wasteful
ColumnTracker tracker = new ColumnTracker(indexedColumns);
ValueGetter getter = new ValueGetter() {
@Override
public ImmutableBytesWritable getLatestValue(ColumnReference ref, long ts) throws IOException {
return valueMap.get(ref);
}
@Override
public byte[] getRowKey() {
return mutation.getRow();
}
};
Pair pair = new Pair(getter, new IndexUpdate(tracker));
return pair;
}
}
public static PhoenixTxIndexMutationGenerator newGenerator(final PhoenixConnection connection, PTable table, List indexes,
Map attributes) {
final List indexMaintainers = Lists.newArrayListWithExpectedSize(indexes.size());
for (PTable index : indexes) {
IndexMaintainer maintainer = index.getIndexMaintainer(table, connection);
indexMaintainers.add(maintainer);
}
IndexMetaDataCache indexMetaDataCache = new IndexMetaDataCache() {
@Override
public void close() throws IOException {}
@Override
public List getIndexMaintainers() {
return indexMaintainers;
}
@Override
public PhoenixTransactionContext getTransactionContext() {
PhoenixTransactionContext context = connection.getMutationState().getPhoenixTransactionContext();
return context.newTransactionContext(context, true);
}
@Override
public int getClientVersion() {
return MetaDataProtocol.PHOENIX_VERSION;
}
};
try {
PhoenixIndexMetaData indexMetaData = new PhoenixIndexMetaData(indexMetaDataCache, attributes);
return new PhoenixTxIndexMutationGenerator(connection.getQueryServices(),connection.getQueryServices().getConfiguration(), indexMetaData,
table.getPhysicalName().getBytes(), null, null);
} catch (IOException e) {
throw new RuntimeException(e); // Impossible
}
}
}
| | | | © 2015 - 2024 Weber Informatics LLC | Privacy Policy