Merge branch 'cassandra-2.1' into trunk
Conflicts:
src/java/org/apache/cassandra/io/sstable/format/big/BigTableScanner.java
test/unit/org/apache/cassandra/io/sstable/SSTableScannerTest.java
Project: http://git-wip-us.apache.org/repos/asf/cassandra/repo
Commit: http://git-wip-us.apache.org/repos/asf/cassandra/commit/7aefd914
Tree: http://git-wip-us.apache.org/repos/asf/cassandra/tree/7aefd914
Diff: http://git-wip-us.apache.org/repos/asf/cassandra/diff/7aefd914
Branch: refs/heads/trunk
Commit: 7aefd914c79a7a5d652abcc1531fec89d80c0adc
Parents: c2dc31c 572ef50
Author: Benedict Elliott Smith <[email protected]>
Authored: Wed Mar 18 11:04:48 2015 +0000
Committer: Benedict Elliott Smith <[email protected]>
Committed: Wed Mar 18 11:04:48 2015 +0000
----------------------------------------------------------------------
CHANGES.txt | 1 +
.../apache/cassandra/dht/AbstractBounds.java | 74 ++++++++++
src/java/org/apache/cassandra/dht/Bounds.java | 10 ++
.../apache/cassandra/dht/ExcludingBounds.java | 10 ++
.../cassandra/dht/IncludingExcludingBounds.java | 10 ++
src/java/org/apache/cassandra/dht/Range.java | 10 ++
.../io/sstable/format/big/BigTableScanner.java | 42 ++++--
.../io/sstable/SSTableScannerTest.java | 143 ++++++++++++++++---
8 files changed, 270 insertions(+), 30 deletions(-)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/cassandra/blob/7aefd914/CHANGES.txt
----------------------------------------------------------------------
diff --cc CHANGES.txt
index 997bf04,36bdb39..62b1079
--- a/CHANGES.txt
+++ b/CHANGES.txt
@@@ -1,78 -1,5 +1,79 @@@
+3.0
+ * Partition intra-cluster message streams by size, not type (CASSANDRA-8789)
+ * Add nodetool command to validate all sstables in a node (CASSANDRA-5791)
+ * Add WriteFailureException to native protocol, notify coordinator of
+ write failures (CASSANDRA-8592)
+ * Convert SequentialWriter to nio (CASSANDRA-8709)
+ * Add role based access control (CASSANDRA-7653, 8650, 7216, 8760, 8849,
8761, 8850)
+ * Record client ip address in tracing sessions (CASSANDRA-8162)
+ * Indicate partition key columns in response metadata for prepared
+ statements (CASSANDRA-7660)
+ * Merge UUIDType and TimeUUIDType parse logic (CASSANDRA-8759)
+ * Avoid memory allocation when searching index summary (CASSANDRA-8793)
+ * Optimise (Time)?UUIDType Comparisons (CASSANDRA-8730)
+ * Make CRC32Ex into a separate maven dependency (CASSANDRA-8836)
+ * Use preloaded jemalloc w/ Unsafe (CASSANDRA-8714)
+ * Avoid accessing partitioner through StorageProxy (CASSANDRA-8244, 8268)
+ * Upgrade Metrics library and remove depricated metrics (CASSANDRA-5657)
+ * Serializing Row cache alternative, fully off heap (CASSANDRA-7438)
+ * Duplicate rows returned when in clause has repeated values (CASSANDRA-6707)
+ * Make CassandraException unchecked, extend RuntimeException (CASSANDRA-8560)
+ * Support direct buffer decompression for reads (CASSANDRA-8464)
+ * DirectByteBuffer compatible LZ4 methods (CASSANDRA-7039)
+ * Group sstables for anticompaction correctly (CASSANDRA-8578)
+ * Add ReadFailureException to native protocol, respond
+ immediately when replicas encounter errors while handling
+ a read request (CASSANDRA-7886)
+ * Switch CommitLogSegment from RandomAccessFile to nio (CASSANDRA-8308)
+ * Allow mixing token and partition key restrictions (CASSANDRA-7016)
+ * Support index key/value entries on map collections (CASSANDRA-8473)
+ * Modernize schema tables (CASSANDRA-8261)
+ * Support for user-defined aggregation functions (CASSANDRA-8053)
+ * Fix NPE in SelectStatement with empty IN values (CASSANDRA-8419)
+ * Refactor SelectStatement, return IN results in natural order instead
+ of IN value list order and ignore duplicate values in partition key IN
restrictions (CASSANDRA-7981)
+ * Support UDTs, tuples, and collections in user-defined
+ functions (CASSANDRA-7563)
+ * Fix aggregate fn results on empty selection, result column name,
+ and cqlsh parsing (CASSANDRA-8229)
+ * Mark sstables as repaired after full repair (CASSANDRA-7586)
+ * Extend Descriptor to include a format value and refactor reader/writer
+ APIs (CASSANDRA-7443)
+ * Integrate JMH for microbenchmarks (CASSANDRA-8151)
+ * Keep sstable levels when bootstrapping (CASSANDRA-7460)
+ * Add Sigar library and perform basic OS settings check on startup
(CASSANDRA-7838)
+ * Support for aggregation functions (CASSANDRA-4914)
+ * Remove cassandra-cli (CASSANDRA-7920)
+ * Accept dollar quoted strings in CQL (CASSANDRA-7769)
+ * Make assassinate a first class command (CASSANDRA-7935)
+ * Support IN clause on any partition key column (CASSANDRA-7855)
+ * Support IN clause on any clustering column (CASSANDRA-4762)
+ * Improve compaction logging (CASSANDRA-7818)
+ * Remove YamlFileNetworkTopologySnitch (CASSANDRA-7917)
+ * Do anticompaction in groups (CASSANDRA-6851)
+ * Support user-defined functions (CASSANDRA-7395, 7526, 7562, 7740, 7781,
7929,
+ 7924, 7812, 8063, 7813, 7708)
+ * Permit configurable timestamps with cassandra-stress (CASSANDRA-7416)
+ * Move sstable RandomAccessReader to nio2, which allows using the
+ FILE_SHARE_DELETE flag on Windows (CASSANDRA-4050)
+ * Remove CQL2 (CASSANDRA-5918)
+ * Add Thrift get_multi_slice call (CASSANDRA-6757)
+ * Optimize fetching multiple cells by name (CASSANDRA-6933)
+ * Allow compilation in java 8 (CASSANDRA-7028)
+ * Make incremental repair default (CASSANDRA-7250)
+ * Enable code coverage thru JaCoCo (CASSANDRA-7226)
+ * Switch external naming of 'column families' to 'tables' (CASSANDRA-4369)
+ * Shorten SSTable path (CASSANDRA-6962)
+ * Use unsafe mutations for most unit tests (CASSANDRA-6969)
+ * Fix race condition during calculation of pending ranges (CASSANDRA-7390)
+ * Fail on very large batch sizes (CASSANDRA-8011)
+ * Improve concurrency of repair (CASSANDRA-6455, 8208)
+ * Select optimal CRC32 implementation at runtime (CASSANDRA-8614)
+ * Evaluate MurmurHash of Token once per query (CASSANDRA-7096)
+ * Generalize progress reporting (CASSANDRA-8901)
+
2.1.4
+ * SSTableScanner enforces its bounds (CASSANDRA-8946)
* Cleanup cell equality (CASSANDRA-8947)
* Introduce intra-cluster message coalescing (CASSANDRA-8692)
* DatabaseDescriptor throws NPE when rpc_interface is used (CASSANDRA-8839)
http://git-wip-us.apache.org/repos/asf/cassandra/blob/7aefd914/src/java/org/apache/cassandra/dht/AbstractBounds.java
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/cassandra/blob/7aefd914/src/java/org/apache/cassandra/dht/Bounds.java
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/cassandra/blob/7aefd914/src/java/org/apache/cassandra/dht/ExcludingBounds.java
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/cassandra/blob/7aefd914/src/java/org/apache/cassandra/dht/IncludingExcludingBounds.java
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/cassandra/blob/7aefd914/src/java/org/apache/cassandra/dht/Range.java
----------------------------------------------------------------------
diff --cc src/java/org/apache/cassandra/dht/Range.java
index 224deac,505f1f3..936ad21
--- a/src/java/org/apache/cassandra/dht/Range.java
+++ b/src/java/org/apache/cassandra/dht/Range.java
@@@ -223,9 -230,20 +223,19 @@@ public class Range<T extends RingPositi
return Pair.create(lb, rb);
}
+ public boolean inclusiveLeft()
+ {
+ return false;
+ }
+
+ public boolean inclusiveRight()
+ {
+ return true;
+ }
+
public List<Range<T>> unwrap()
{
- @SuppressWarnings("unchecked")
- T minValue = (T) partitioner.minValue(right.getClass());
+ T minValue = right.minValue();
if (!isWrapAround() || right.equals(minValue))
return Arrays.asList(this);
List<Range<T>> unwrapped = new ArrayList<Range<T>>(2);
http://git-wip-us.apache.org/repos/asf/cassandra/blob/7aefd914/src/java/org/apache/cassandra/io/sstable/format/big/BigTableScanner.java
----------------------------------------------------------------------
diff --cc
src/java/org/apache/cassandra/io/sstable/format/big/BigTableScanner.java
index b270b2d,0000000..8247a2d
mode 100644,000000..100644
--- a/src/java/org/apache/cassandra/io/sstable/format/big/BigTableScanner.java
+++ b/src/java/org/apache/cassandra/io/sstable/format/big/BigTableScanner.java
@@@ -1,362 -1,0 +1,378 @@@
+/*
+ * 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.io.sstable.format.big;
+
+import java.io.IOException;
+import java.util.*;
+
+import com.google.common.collect.AbstractIterator;
+import com.google.common.collect.Ordering;
+import com.google.common.util.concurrent.RateLimiter;
+
+import org.apache.cassandra.db.DataRange;
+import org.apache.cassandra.db.DecoratedKey;
+import org.apache.cassandra.db.RowIndexEntry;
+import org.apache.cassandra.db.RowPosition;
+import org.apache.cassandra.db.columniterator.IColumnIteratorFactory;
+import org.apache.cassandra.db.columniterator.LazyColumnIterator;
+import org.apache.cassandra.db.columniterator.OnDiskAtomIterator;
+import org.apache.cassandra.dht.AbstractBounds;
++import org.apache.cassandra.dht.AbstractBounds.Boundary;
+import org.apache.cassandra.dht.Bounds;
+import org.apache.cassandra.dht.Range;
+import org.apache.cassandra.dht.Token;
+import org.apache.cassandra.io.sstable.CorruptSSTableException;
+import org.apache.cassandra.io.sstable.ISSTableScanner;
+import org.apache.cassandra.io.sstable.SSTableIdentityIterator;
+import org.apache.cassandra.io.sstable.format.SSTableReader;
+import org.apache.cassandra.io.util.FileUtils;
+import org.apache.cassandra.io.util.RandomAccessReader;
+import org.apache.cassandra.utils.ByteBufferUtil;
+import org.apache.cassandra.utils.Pair;
+
++import static org.apache.cassandra.dht.AbstractBounds.isEmpty;
++import static org.apache.cassandra.dht.AbstractBounds.maxLeft;
++import static org.apache.cassandra.dht.AbstractBounds.minRight;
++
+public class BigTableScanner implements ISSTableScanner
+{
+ protected final RandomAccessReader dfile;
+ protected final RandomAccessReader ifile;
+ public final SSTableReader sstable;
+
+ private final Iterator<AbstractBounds<RowPosition>> rangeIterator;
+ private AbstractBounds<RowPosition> currentRange;
+
+ private final DataRange dataRange;
+ private final RowIndexEntry.IndexSerializer rowIndexEntrySerializer;
+
+ protected Iterator<OnDiskAtomIterator> iterator;
+
+ public static ISSTableScanner getScanner(SSTableReader sstable, DataRange
dataRange, RateLimiter limiter)
+ {
+ return new BigTableScanner(sstable, dataRange, limiter);
+ }
+ public static ISSTableScanner getScanner(SSTableReader sstable,
Collection<Range<Token>> tokenRanges, RateLimiter limiter)
+ {
+ // We want to avoid allocating a SSTableScanner if the range don't
overlap the sstable (#5249)
+ List<Pair<Long, Long>> positions =
sstable.getPositionsForRanges(tokenRanges);
+ if (positions.isEmpty())
+ return new EmptySSTableScanner(sstable.getFilename());
+
+ return new BigTableScanner(sstable, tokenRanges, limiter);
+ }
+
+ /**
+ * @param sstable SSTable to scan; must not be null
+ * @param dataRange a single range to scan; must not be null
+ * @param limiter background i/o RateLimiter; may be null
+ */
+ private BigTableScanner(SSTableReader sstable, DataRange dataRange,
RateLimiter limiter)
+ {
+ assert sstable != null;
+
+ this.dfile = limiter == null ? sstable.openDataReader() :
sstable.openDataReader(limiter);
+ this.ifile = sstable.openIndexReader();
+ this.sstable = sstable;
+ this.dataRange = dataRange;
+ this.rowIndexEntrySerializer =
sstable.descriptor.version.getSSTableFormat().getIndexSerializer(sstable.metadata);
+
+ List<AbstractBounds<RowPosition>> boundsList = new ArrayList<>(2);
+ // we enforce the first/last keys of the sstablereader
+ if (dataRange.isWrapAround())
+ {
- if (dataRange.stopKey().isMinimum()
- || dataRange.stopKey().compareTo(sstable.last) >= 0
- || dataRange.startKey().compareTo(sstable.first) <= 0)
++ if (dataRange.stopKey().compareTo(sstable.first) >= 0)
+ {
- boundsList.add(new Bounds<RowPosition>(sstable.first,
sstable.last));
++ // since we wrap, we must contain the whole sstable prior to
stopKey()
++ Boundary<RowPosition> left = new
Boundary<RowPosition>(sstable.first, true);
++ Boundary<RowPosition> right;
++ right = dataRange.keyRange().rightBoundary();
++ right = minRight(right, sstable.last, true);
++ if (!isEmpty(left, right))
++ boundsList.add(AbstractBounds.bounds(left, right));
+ }
- else
++ if (dataRange.startKey().compareTo(sstable.last) <= 0)
+ {
- if (dataRange.startKey().compareTo(sstable.last) <= 0)
- boundsList.add(new Bounds<>(dataRange.startKey(),
sstable.last));
- if (dataRange.stopKey().compareTo(sstable.first) >= 0)
- boundsList.add(new Bounds<>(sstable.first,
dataRange.stopKey()));
++ // since we wrap, we must contain the whole sstable after
dataRange.startKey()
++ Boundary<RowPosition> right = new
Boundary<RowPosition>(sstable.last, true);
++ Boundary<RowPosition> left;
++ left = dataRange.keyRange().leftBoundary();
++ left = maxLeft(left, sstable.first, true);
++ if (!isEmpty(left, right))
++ boundsList.add(AbstractBounds.bounds(left, right));
+ }
+ }
+ else
+ {
+ assert dataRange.startKey().compareTo(dataRange.stopKey()) <= 0
|| dataRange.stopKey().isMinimum();
- RowPosition left = Ordering.natural().max(dataRange.startKey(),
sstable.first);
++ Boundary<RowPosition> left, right;
++ left = dataRange.keyRange().leftBoundary();
++ right = dataRange.keyRange().rightBoundary();
++ left = maxLeft(left, sstable.first, true);
+ // apparently isWrapAround() doesn't count Bounds that extend to
the limit (min) as wrapping
- RowPosition right = dataRange.stopKey().isMinimum() ?
sstable.last : Ordering.natural().min(dataRange.stopKey(), sstable.last);
- if (left.compareTo(right) <= 0)
- boundsList.add(new Bounds<>(left, right));
++ right = dataRange.stopKey().isMinimum() ? new
Boundary<RowPosition>(sstable.last, true)
++ : minRight(right,
sstable.last, true);
++ if (!isEmpty(left, right))
++ boundsList.add(AbstractBounds.bounds(left, right));
+ }
+ this.rangeIterator = boundsList.iterator();
+ }
+
+ /**
+ * @param sstable SSTable to scan; must not be null
+ * @param tokenRanges A set of token ranges to scan
+ * @param limiter background i/o RateLimiter; may be null
+ */
+ private BigTableScanner(SSTableReader sstable, Collection<Range<Token>>
tokenRanges, RateLimiter limiter)
+ {
+ assert sstable != null;
+
+ this.dfile = limiter == null ? sstable.openDataReader() :
sstable.openDataReader(limiter);
+ this.ifile = sstable.openIndexReader();
+ this.sstable = sstable;
+ this.dataRange = null;
+ this.rowIndexEntrySerializer =
sstable.descriptor.version.getSSTableFormat().getIndexSerializer(sstable.metadata);
+
+ List<Range<Token>> normalized = Range.normalize(tokenRanges);
+ List<AbstractBounds<RowPosition>> boundsList = new
ArrayList<>(normalized.size());
+ // we enforce the first/last keys of the sstablereader
+ for (Range<Token> range : normalized)
+ {
+ // cap our ranges by the start/end of the sstable
+ RowPosition right = range.right.maxKeyBound();
+ if (right.compareTo(sstable.last) > 0)
+ right = sstable.last;
+
+ RowPosition left = range.left.maxKeyBound();
+ if (left.compareTo(sstable.first) < 0)
+ {
+ if (sstable.first.compareTo(right) <= 0)
+ boundsList.add(new Bounds<>(sstable.first, right));
+ }
+ else if (left.compareTo(right) < 0)
+ boundsList.add(new Range<>(left, right));
+ }
+
+ this.rangeIterator = boundsList.iterator();
+ }
+
+ private void seekToCurrentRangeStart()
+ {
+ long indexPosition = sstable.getIndexScanPosition(currentRange.left);
+ ifile.seek(indexPosition);
+ try
+ {
+
+ while (!ifile.isEOF())
+ {
+ indexPosition = ifile.getFilePointer();
+ DecoratedKey indexDecoratedKey =
sstable.partitioner.decorateKey(ByteBufferUtil.readWithShortLength(ifile));
+ if (indexDecoratedKey.compareTo(currentRange.left) > 0 ||
currentRange.contains(indexDecoratedKey))
+ {
+ // Found, just read the dataPosition and seek into index
and data files
+ long dataPosition = ifile.readLong();
+ ifile.seek(indexPosition);
+ dfile.seek(dataPosition);
+ break;
+ }
+ else
+ {
+ RowIndexEntry.Serializer.skip(ifile);
+ }
+ }
+ }
+ catch (IOException e)
+ {
+ sstable.markSuspect();
+ throw new CorruptSSTableException(e, sstable.getFilename());
+ }
+ }
+
+ public void close() throws IOException
+ {
+ FileUtils.close(dfile, ifile);
+ }
+
+ public long getLengthInBytes()
+ {
+ return dfile.length();
+ }
+
+ public long getCurrentPosition()
+ {
+ return dfile.getFilePointer();
+ }
+
+ public String getBackingFiles()
+ {
+ return sstable.toString();
+ }
+
+ public boolean hasNext()
+ {
+ if (iterator == null)
+ iterator = createIterator();
+ return iterator.hasNext();
+ }
+
+ public OnDiskAtomIterator next()
+ {
+ if (iterator == null)
+ iterator = createIterator();
+ return iterator.next();
+ }
+
+ public void remove()
+ {
+ throw new UnsupportedOperationException();
+ }
+
+ private Iterator<OnDiskAtomIterator> createIterator()
+ {
+ return new KeyScanningIterator();
+ }
+
+ protected class KeyScanningIterator extends
AbstractIterator<OnDiskAtomIterator>
+ {
+ private DecoratedKey nextKey;
+ private RowIndexEntry nextEntry;
+ private DecoratedKey currentKey;
+ private RowIndexEntry currentEntry;
+
+ protected OnDiskAtomIterator computeNext()
+ {
+ try
+ {
+ if (nextEntry == null)
+ {
+ do
+ {
+ // we're starting the first range or we just passed
the end of the previous range
+ if (!rangeIterator.hasNext())
+ return endOfData();
+
+ currentRange = rangeIterator.next();
+ seekToCurrentRangeStart();
+
+ if (ifile.isEOF())
+ return endOfData();
+
+ currentKey =
sstable.partitioner.decorateKey(ByteBufferUtil.readWithShortLength(ifile));
+ currentEntry =
rowIndexEntrySerializer.deserialize(ifile, sstable.descriptor.version);
+ } while (!currentRange.contains(currentKey));
+ }
+ else
+ {
+ // we're in the middle of a range
+ currentKey = nextKey;
+ currentEntry = nextEntry;
+ }
+
+ if (ifile.isEOF())
+ {
+ nextEntry = null;
+ nextKey = null;
+ }
+ else
+ {
+ // we need the position of the start of the next key,
regardless of whether it falls in the current range
+ nextKey =
sstable.partitioner.decorateKey(ByteBufferUtil.readWithShortLength(ifile));
+ nextEntry = rowIndexEntrySerializer.deserialize(ifile,
sstable.descriptor.version);
+
+ if (!currentRange.contains(nextKey))
+ {
+ nextKey = null;
+ nextEntry = null;
+ }
+ }
+
+ if (dataRange == null ||
dataRange.selectsFullRowFor(currentKey.getKey()))
+ {
+ dfile.seek(currentEntry.position +
currentEntry.headerOffset());
+ ByteBufferUtil.readWithShortLength(dfile); // key
+ return new SSTableIdentityIterator(sstable, dfile,
currentKey);
+ }
+
+ return new LazyColumnIterator(currentKey, new
IColumnIteratorFactory()
+ {
+ public OnDiskAtomIterator create()
+ {
+ return
dataRange.columnFilter(currentKey.getKey()).getSSTableColumnIterator(sstable,
dfile, currentKey, currentEntry);
+ }
+ });
+
+ }
+ catch (IOException e)
+ {
+ sstable.markSuspect();
+ throw new CorruptSSTableException(e, sstable.getFilename());
+ }
+ }
+ }
+
+ @Override
+ public String toString()
+ {
+ return getClass().getSimpleName() + "(" +
+ "dfile=" + dfile +
+ " ifile=" + ifile +
+ " sstable=" + sstable +
+ ")";
+ }
+
+ public static class EmptySSTableScanner implements ISSTableScanner
+ {
+ private final String filename;
+
+ public EmptySSTableScanner(String filename)
+ {
+ this.filename = filename;
+ }
+
+ public long getLengthInBytes()
+ {
+ return 0;
+ }
+
+ public long getCurrentPosition()
+ {
+ return 0;
+ }
+
+ public String getBackingFiles()
+ {
+ return filename;
+ }
+
+ public boolean hasNext()
+ {
+ return false;
+ }
+
+ public OnDiskAtomIterator next()
+ {
+ return null;
+ }
+
+ public void close() throws IOException { }
+
+ public void remove() { }
+ }
+
+
+}
http://git-wip-us.apache.org/repos/asf/cassandra/blob/7aefd914/test/unit/org/apache/cassandra/io/sstable/SSTableScannerTest.java
----------------------------------------------------------------------
diff --cc test/unit/org/apache/cassandra/io/sstable/SSTableScannerTest.java
index 9da895e,91a820c..22eb5a0
--- a/test/unit/org/apache/cassandra/io/sstable/SSTableScannerTest.java
+++ b/test/unit/org/apache/cassandra/io/sstable/SSTableScannerTest.java
@@@ -18,42 -18,26 +18,40 @@@
*/
package org.apache.cassandra.io.sstable;
- import java.util.ArrayList;
- import java.util.Collection;
+ import java.util.*;
+import org.apache.cassandra.io.sstable.format.SSTableReader;
+import org.junit.BeforeClass;
+ import com.google.common.collect.Iterables;
import org.junit.Test;
import org.apache.cassandra.SchemaLoader;
import org.apache.cassandra.Util;
+import org.apache.cassandra.config.KSMetaData;
import org.apache.cassandra.db.*;
import org.apache.cassandra.db.columniterator.IdentityQueryFilter;
- import org.apache.cassandra.dht.Bounds;
- import org.apache.cassandra.dht.ByteOrderedPartitioner.BytesToken;
- import org.apache.cassandra.dht.Range;
- import org.apache.cassandra.dht.Token;
+ import org.apache.cassandra.dht.*;
+import org.apache.cassandra.locator.SimpleStrategy;
import org.apache.cassandra.utils.ByteBufferUtil;
+ import static org.apache.cassandra.dht.AbstractBounds.isEmpty;
import static org.junit.Assert.*;
-public class SSTableScannerTest extends SchemaLoader
+public class SSTableScannerTest
{
- public static final String KEYSPACE = "Keyspace1";
+ public static final String KEYSPACE = "SSTableScannerTest";
public static final String TABLE = "Standard1";
+ @BeforeClass
+ public static void defineSchema() throws Exception
+ {
+ SchemaLoader.prepareServer();
+ SchemaLoader.createKeyspace(KEYSPACE,
+ SimpleStrategy.class,
+ KSMetaData.optsWithRF(1),
+ SchemaLoader.standardCFMD(KEYSPACE,
TABLE));
+ }
+
private static String toKey(int key)
{
return String.format("%03d", key);
@@@ -86,15 -138,20 +152,20 @@@
DecoratedKey decoratedKey = Util.dk(toKey(key));
Mutation rm = new Mutation(KEYSPACE, decoratedKey.getKey());
rm.add(TABLE, Util.cellname("col"), ByteBufferUtil.EMPTY_BYTE_BUFFER,
timestamp, 1000);
- rm.apply();
+ rm.applyUnsafe();
}
- private static void assertScanMatches(SSTableReader sstable, int
scanStart, int scanEnd, int expectedStart, int expectedEnd)
+ private static void assertScanMatches(SSTableReader sstable, int
scanStart, int scanEnd, int ... boundaries)
{
- ISSTableScanner scanner = sstable.getScanner(new
DataRange(boundsFor(scanStart, scanEnd), new IdentityQueryFilter()));
- for (int i = expectedStart; i <= expectedEnd; i++)
- assertEquals(toKey(i), new
String(scanner.next().getKey().getKey().array()));
- assertFalse(scanner.hasNext());
+ assert boundaries.length % 2 == 0;
+ for (DataRange range : dataRanges(scanStart, scanEnd))
+ {
+ ISSTableScanner scanner = sstable.getScanner(range);
+ for (int b = 0 ; b < boundaries.length ; b += 2)
+ for (int i = boundaries[b] ; i <= boundaries[b + 1] ; i++)
+ assertEquals(toKey(i), new
String(scanner.next().getKey().getKey().array()));
+ assertFalse(scanner.hasNext());
+ }
}
private static void assertScanEmpty(SSTableReader sstable, int scanStart,
int scanEnd)
