blob: c731a35051c113d52fcdda1a1babbdbd61a176d8 [file] [log] [blame]
/*
* 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;
import java.io.File;
import java.util.*;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.atomic.AtomicReference;
import com.google.common.base.Predicate;
import com.google.common.base.Predicates;
import com.google.common.collect.*;
import org.apache.cassandra.db.commitlog.ReplayPosition;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.apache.cassandra.config.DatabaseDescriptor;
import org.apache.cassandra.db.compaction.OperationType;
import org.apache.cassandra.dht.AbstractBounds;
import org.apache.cassandra.io.sstable.IndexSummary;
import org.apache.cassandra.io.sstable.SSTableReader;
import org.apache.cassandra.io.util.FileUtils;
import org.apache.cassandra.io.util.SegmentedFile;
import org.apache.cassandra.metrics.StorageMetrics;
import org.apache.cassandra.notifications.*;
import org.apache.cassandra.utils.IFilter;
import org.apache.cassandra.utils.Interval;
import org.apache.cassandra.utils.IntervalTree;
import org.apache.cassandra.utils.concurrent.OpOrder;
import org.apache.cassandra.utils.concurrent.Refs;
public class DataTracker
{
private static final Logger logger = LoggerFactory.getLogger(DataTracker.class);
public final Collection<INotificationConsumer> subscribers = new CopyOnWriteArrayList<>();
public final ColumnFamilyStore cfstore;
private final AtomicReference<View> view;
public DataTracker(ColumnFamilyStore cfstore)
{
this.cfstore = cfstore;
this.view = new AtomicReference<>();
this.init();
}
// get the Memtable that the ordered writeOp should be directed to
public Memtable getMemtableFor(OpOrder.Group opGroup, ReplayPosition replayPosition)
{
// since any new memtables appended to the list after we fetch it will be for operations started
// after us, we can safely assume that we will always find the memtable that 'accepts' us;
// if the barrier for any memtable is set whilst we are reading the list, it must accept us.
// there may be multiple memtables in the list that would 'accept' us, however we only ever choose
// the oldest such memtable, as accepts() only prevents us falling behind (i.e. ensures we don't
// assign operations to a memtable that was retired/queued before we started)
for (Memtable memtable : view.get().liveMemtables)
{
if (memtable.accepts(opGroup, replayPosition))
return memtable;
}
throw new AssertionError(view.get().liveMemtables.toString());
}
public Set<SSTableReader> getSSTables()
{
return view.get().sstables;
}
public Set<SSTableReader> getUncompactingSSTables()
{
return view.get().nonCompactingSStables();
}
public Iterable<SSTableReader> getUncompactingSSTables(Iterable<SSTableReader> candidates)
{
final View v = view.get();
return Iterables.filter(candidates, new Predicate<SSTableReader>()
{
public boolean apply(SSTableReader sstable)
{
return !v.compacting.contains(sstable);
}
});
}
public View getView()
{
return view.get();
}
/**
* Switch the current memtable. This atomically appends a new memtable to the end of the list of active memtables,
* returning the previously last memtable. It leaves the previous Memtable in the list of live memtables until
* discarding(memtable) is called. These two methods must be synchronized/paired, i.e. m = switchMemtable
* must be followed by discarding(m), they cannot be interleaved.
*
* @return the previously active memtable
*/
public Memtable switchMemtable(boolean truncating)
{
Memtable newMemtable = new Memtable(cfstore);
Memtable toFlushMemtable;
View currentView, newView;
do
{
currentView = view.get();
toFlushMemtable = currentView.getCurrentMemtable();
newView = currentView.switchMemtable(newMemtable);
}
while (!view.compareAndSet(currentView, newView));
if (truncating)
notifyRenewed(newMemtable);
return toFlushMemtable;
}
public void markFlushing(Memtable memtable)
{
View currentView, newView;
do
{
currentView = view.get();
newView = currentView.markFlushing(memtable);
}
while (!view.compareAndSet(currentView, newView));
}
public void replaceFlushed(Memtable memtable, SSTableReader sstable)
{
// sstable may be null if we flushed batchlog and nothing needed to be retained
if (!cfstore.isValid())
{
View currentView, newView;
do
{
currentView = view.get();
newView = currentView.replaceFlushed(memtable, sstable);
if (sstable != null)
newView = newView.replace(Arrays.asList(sstable), Collections.<SSTableReader>emptyList());
}
while (!view.compareAndSet(currentView, newView));
return;
}
// back up before creating a new View (which makes the new one eligible for compaction)
if (sstable != null)
maybeIncrementallyBackup(sstable);
View currentView, newView;
do
{
currentView = view.get();
newView = currentView.replaceFlushed(memtable, sstable);
}
while (!view.compareAndSet(currentView, newView));
if (sstable != null)
{
addNewSSTablesSize(Arrays.asList(sstable));
notifyAdded(sstable);
}
}
public void maybeIncrementallyBackup(final SSTableReader sstable)
{
if (!DatabaseDescriptor.isIncrementalBackupsEnabled())
return;
File backupsDir = Directories.getBackupsDirectory(sstable.descriptor);
sstable.createLinks(FileUtils.getCanonicalPath(backupsDir));
}
/**
* @return true if we are able to mark the given @param sstables as compacted, before anyone else
*
* Note that we could acquire references on the marked sstables and release them in
* unmarkCompacting, but since we will never call markObsolete on a sstable marked
* as compacting (unless there is a serious bug), we can skip this.
*/
public boolean markCompacting(Collection<SSTableReader> sstables)
{
return markCompacting(sstables, false, false);
}
public boolean markCompacting(Collection<SSTableReader> sstables, boolean newTables, boolean offline)
{
assert sstables != null && !Iterables.isEmpty(sstables);
while (true)
{
final View currentView = view.get();
if (Iterables.any(sstables, Predicates.in(currentView.compacting)))
return false;
Predicate live = new Predicate<SSTableReader>()
{
public boolean apply(SSTableReader sstable)
{
return currentView.sstablesMap.get(sstable) == sstable && !sstable.isMarkedCompacted();
}
};
if (newTables)
assert !Iterables.any(sstables, Predicates.in(currentView.sstables));
else if (!offline && !Iterables.all(sstables, live))
return false;
View newView = currentView.markCompacting(sstables);
if (view.compareAndSet(currentView, newView))
{
notifyCompacting(sstables, true);
return true;
}
}
}
/**
* Removes files from compacting status: this is different from 'markObsolete'
* because it should be run regardless of whether a compaction succeeded.
*/
public void unmarkCompacting(Iterable<SSTableReader> unmark)
{
View currentView, newView;
do
{
currentView = view.get();
newView = currentView.unmarkCompacting(unmark);
}
while (!view.compareAndSet(currentView, newView));
if (!cfstore.isValid())
{
// when the CFS is invalidated, it will call unreferenceSSTables(). However, unreferenceSSTables only deals
// with sstables that aren't currently being compacted. If there are ongoing compactions that finish or are
// interrupted after the CFS is invalidated, those sstables need to be unreferenced as well, so we do that here.
unreferenceSSTables();
}
notifyCompacting(unmark, false);
}
public void markObsolete(Collection<SSTableReader> sstables, OperationType compactionType)
{
removeSSTablesFromTracker(sstables);
releaseReferences(sstables, false);
notifySSTablesChanged(sstables, Collections.<SSTableReader>emptyList(), compactionType);
}
/**
*
* @param oldSSTables
* @param allReplacements
* @param compactionType
*/
// note that this DOES NOT insert the replacement sstables, it only removes the old sstables and notifies any listeners
// that they have been replaced by the provided sstables, which must have been performed by an earlier replaceReaders() call
public void markCompactedSSTablesReplaced(Collection<SSTableReader> oldSSTables, Collection<SSTableReader> allReplacements, OperationType compactionType)
{
removeSSTablesFromTracker(oldSSTables);
releaseReferences(oldSSTables, false);
notifySSTablesChanged(oldSSTables, allReplacements, compactionType);
addNewSSTablesSize(allReplacements);
}
public void addInitialSSTables(Collection<SSTableReader> sstables)
{
addSSTablesToTracker(sstables);
// no notifications or backup necessary
}
public void addSSTables(Collection<SSTableReader> sstables)
{
addSSTablesToTracker(sstables);
for (SSTableReader sstable : sstables)
{
maybeIncrementallyBackup(sstable);
notifyAdded(sstable);
}
}
/**
* Replaces existing sstables with new instances, makes sure compaction strategies have the correct instance
*
* @param toReplace
* @param replaceWith
*/
public void replaceWithNewInstances(Collection<SSTableReader> toReplace, Collection<SSTableReader> replaceWith)
{
replaceReaders(toReplace, replaceWith, true);
}
/**
* Adds the early opened files to the data tracker, but does not tell compaction strategies about it
*
* note that we dont track the live size of these sstables
* @param toReplace
* @param replaceWith
*/
public void replaceEarlyOpenedFiles(Collection<SSTableReader> toReplace, Collection<SSTableReader> replaceWith)
{
for (SSTableReader s : toReplace)
assert s.openReason.equals(SSTableReader.OpenReason.EARLY);
// note that we can replace an early opened file with a real one
replaceReaders(toReplace, replaceWith, false);
}
/**
* removes all sstables that are not busy compacting.
*/
public void unreferenceSSTables()
{
Set<SSTableReader> notCompacting;
View currentView, newView;
do
{
currentView = view.get();
if (!currentView.compacting.isEmpty())
logger.error("Set of compacting sstables is non-empty when invalidating sstables {}", currentView.compacting);
notCompacting = currentView.nonCompactingSStables();
newView = currentView.replace(notCompacting, Collections.<SSTableReader>emptySet());
}
while (!view.compareAndSet(currentView, newView));
if (notCompacting.isEmpty())
{
// notifySSTablesChanged -> LeveledManifest.promote doesn't like a no-op "promotion"
return;
}
notifySSTablesChanged(notCompacting, Collections.<SSTableReader>emptySet(), OperationType.UNKNOWN);
removeOldSSTablesSize(notCompacting);
releaseReferences(notCompacting, true);
}
/**
* Removes every SSTable in the directory from the DataTracker's view.
* @param directory the unreadable directory, possibly with SSTables in it, but not necessarily.
*/
void removeUnreadableSSTables(File directory)
{
View currentView, newView;
Set<SSTableReader> remaining = new HashSet<>();
do
{
currentView = view.get();
for (SSTableReader r : currentView.nonCompactingSStables())
if (!r.descriptor.directory.equals(directory))
remaining.add(r);
if (remaining.size() == currentView.nonCompactingSStables().size())
return;
newView = currentView.replace(currentView.sstables, remaining);
}
while (!view.compareAndSet(currentView, newView));
for (SSTableReader sstable : currentView.sstables)
if (!remaining.contains(sstable))
sstable.selfRef().release();
notifySSTablesChanged(remaining, Collections.<SSTableReader>emptySet(), OperationType.UNKNOWN);
}
/** (Re)initializes the tracker, purging all references. */
void init()
{
view.set(new View(
ImmutableList.of(new Memtable(cfstore)),
ImmutableList.<Memtable>of(),
Collections.<SSTableReader, SSTableReader>emptyMap(),
Collections.<SSTableReader>emptySet(),
Collections.<SSTableReader>emptySet(),
SSTableIntervalTree.empty()));
}
/**
* A special kind of replacement for SSTableReaders that were cloned with a new index summary sampling level (see
* SSTableReader.cloneWithNewSummarySamplingLevel and CASSANDRA-5519). This does not mark the old reader
* as compacted.
* @param oldSSTables replaced readers
* @param newSSTables replacement readers
*/
private void replaceReaders(Collection<SSTableReader> oldSSTables, Collection<SSTableReader> newSSTables, boolean notify)
{
View currentView, newView;
do
{
currentView = view.get();
newView = currentView.replace(oldSSTables, newSSTables);
}
while (!view.compareAndSet(currentView, newView));
if (!oldSSTables.isEmpty() && notify)
notifySSTablesChanged(oldSSTables, newSSTables, OperationType.UNKNOWN);
for (SSTableReader sstable : newSSTables)
sstable.setupKeyCache();
Refs.release(Refs.selfRefs(oldSSTables));
}
private void removeSSTablesFromTracker(Collection<SSTableReader> oldSSTables)
{
View currentView, newView;
do
{
currentView = view.get();
newView = currentView.replace(oldSSTables, Collections.<SSTableReader>emptyList());
}
while (!view.compareAndSet(currentView, newView));
removeOldSSTablesSize(oldSSTables);
}
private void addSSTablesToTracker(Collection<SSTableReader> sstables)
{
View currentView, newView;
do
{
currentView = view.get();
newView = currentView.replace(Collections.<SSTableReader>emptyList(), sstables);
}
while (!view.compareAndSet(currentView, newView));
addNewSSTablesSize(sstables);
}
private void addNewSSTablesSize(Iterable<SSTableReader> newSSTables)
{
for (SSTableReader sstable : newSSTables)
{
if (logger.isDebugEnabled())
logger.debug(String.format("adding %s to list of files tracked for %s.%s",
sstable.descriptor, cfstore.keyspace.getName(), cfstore.name));
long size = sstable.bytesOnDisk();
StorageMetrics.load.inc(size);
cfstore.metric.liveDiskSpaceUsed.inc(size);
cfstore.metric.totalDiskSpaceUsed.inc(size);
sstable.setupKeyCache();
}
}
private void removeOldSSTablesSize(Iterable<SSTableReader> oldSSTables)
{
for (SSTableReader sstable : oldSSTables)
{
if (logger.isDebugEnabled())
logger.debug(String.format("removing %s from list of files tracked for %s.%s",
sstable.descriptor, cfstore.keyspace.getName(), cfstore.name));
long size = sstable.bytesOnDisk();
StorageMetrics.load.dec(size);
cfstore.metric.liveDiskSpaceUsed.dec(size);
}
}
private void releaseReferences(Iterable<SSTableReader> oldSSTables, boolean tolerateCompacted)
{
for (SSTableReader sstable : oldSSTables)
{
boolean firstToCompact = sstable.markObsolete(this);
assert tolerateCompacted || firstToCompact : sstable + " was already marked compacted";
sstable.selfRef().release();
}
}
public long estimatedKeys()
{
long n = 0;
for (SSTableReader sstable : getSSTables())
n += sstable.estimatedKeys();
return n;
}
public int getMeanColumns()
{
long sum = 0;
long count = 0;
for (SSTableReader sstable : getSSTables())
{
long n = sstable.getEstimatedColumnCount().count();
sum += sstable.getEstimatedColumnCount().mean() * n;
count += n;
}
return count > 0 ? (int) (sum / count) : 0;
}
public double getDroppableTombstoneRatio()
{
double allDroppable = 0;
long allColumns = 0;
int localTime = (int)(System.currentTimeMillis()/1000);
for (SSTableReader sstable : getSSTables())
{
allDroppable += sstable.getDroppableTombstonesBefore(localTime - sstable.metadata.getGcGraceSeconds());
allColumns += sstable.getEstimatedColumnCount().mean() * sstable.getEstimatedColumnCount().count();
}
return allColumns > 0 ? allDroppable / allColumns : 0;
}
public void notifySSTablesChanged(Collection<SSTableReader> removed, Collection<SSTableReader> added, OperationType compactionType)
{
INotification notification = new SSTableListChangedNotification(added, removed, compactionType);
for (INotificationConsumer subscriber : subscribers)
subscriber.handleNotification(notification, this);
}
public void notifyCompacting(Iterable<SSTableReader> reader, boolean compacting)
{
INotification notification = new SSTableCompactingNotification(reader, compacting);
for (INotificationConsumer subscriber : subscribers)
subscriber.handleNotification(notification, this);
}
public void notifyAdded(SSTableReader added)
{
INotification notification = new SSTableAddedNotification(added);
for (INotificationConsumer subscriber : subscribers)
subscriber.handleNotification(notification, this);
}
public void notifySSTableRepairedStatusChanged(Collection<SSTableReader> repairStatusesChanged)
{
INotification notification = new SSTableRepairStatusChanged(repairStatusesChanged);
for (INotificationConsumer subscriber : subscribers)
subscriber.handleNotification(notification, this);
}
public void notifyDeleting(SSTableReader deleting)
{
INotification notification = new SSTableDeletingNotification(deleting);
for (INotificationConsumer subscriber : subscribers)
subscriber.handleNotification(notification, this);
}
public void notifyRenewed(Memtable renewed)
{
INotification notification = new MemtableRenewedNotification(renewed);
for (INotificationConsumer subscriber : subscribers)
subscriber.handleNotification(notification, this);
}
public void notifyTruncated(long truncatedAt)
{
INotification notification = new TruncationNotification(truncatedAt);
for (INotificationConsumer subscriber : subscribers)
subscriber.handleNotification(notification, this);
}
public void subscribe(INotificationConsumer consumer)
{
subscribers.add(consumer);
}
public void unsubscribe(INotificationConsumer consumer)
{
subscribers.remove(consumer);
}
public static SSTableIntervalTree buildIntervalTree(Iterable<SSTableReader> sstables)
{
List<Interval<RowPosition, SSTableReader>> intervals = new ArrayList<>(Iterables.size(sstables));
for (SSTableReader sstable : sstables)
intervals.add(Interval.<RowPosition, SSTableReader>create(sstable.first, sstable.last, sstable));
return new SSTableIntervalTree(intervals);
}
public Set<SSTableReader> getCompacting()
{
return getView().compacting;
}
public SSTableReader getCurrentVersion(SSTableReader sstable)
{
if (!sstable.isReplaced())
return sstable;
SSTableReader current = view.get().sstablesMap.get(sstable);
if (current == null)
current = Iterables.find(view.get().shadowed, Predicates.equalTo(sstable), null);
assert current != null : sstable + " not in live set";
return current;
}
public static class SSTableIntervalTree extends IntervalTree<RowPosition, SSTableReader, Interval<RowPosition, SSTableReader>>
{
private static final SSTableIntervalTree EMPTY = new SSTableIntervalTree(null);
private SSTableIntervalTree(Collection<Interval<RowPosition, SSTableReader>> intervals)
{
super(intervals, null);
}
public static SSTableIntervalTree empty()
{
return EMPTY;
}
}
/**
* An immutable structure holding the current memtable, the memtables pending
* flush, the sstables for a column family, and the sstables that are active
* in compaction (a subset of the sstables).
*/
public static class View
{
/**
* ordinarily a list of size 1, but when preparing to flush will contain both the memtable we will flush
* and the new replacement memtable, until all outstanding write operations on the old table complete.
* The last item in the list is always the "current" memtable.
*/
private final List<Memtable> liveMemtables;
/**
* contains all memtables that are no longer referenced for writing and are queued for / in the process of being
* flushed. In chronologically ascending order.
*/
private final List<Memtable> flushingMemtables;
public final Set<SSTableReader> compacting;
public final Set<SSTableReader> sstables;
// we use a Map here so that we can easily perform identity checks as well as equality checks.
// When marking compacting, we now indicate if we expect the sstables to be present (by default we do),
// and we then check that not only are they all present in the live set, but that the exact instance present is
// the one we made our decision to compact against.
public final Map<SSTableReader, SSTableReader> sstablesMap;
// all sstables that are still in the live set, but have been completely shadowed by a replacement sstable
public final Set<SSTableReader> shadowed;
public final SSTableIntervalTree intervalTree;
View(List<Memtable> liveMemtables, List<Memtable> flushingMemtables, Map<SSTableReader, SSTableReader> sstables, Set<SSTableReader> compacting, Set<SSTableReader> shadowed, SSTableIntervalTree intervalTree)
{
this.shadowed = shadowed;
assert liveMemtables != null;
assert flushingMemtables != null;
assert sstables != null;
assert compacting != null;
assert intervalTree != null;
this.liveMemtables = liveMemtables;
this.flushingMemtables = flushingMemtables;
this.sstablesMap = sstables;
this.sstables = sstablesMap.keySet();
this.compacting = compacting;
this.intervalTree = intervalTree;
}
public Memtable getOldestMemtable()
{
if (!flushingMemtables.isEmpty())
return flushingMemtables.get(0);
return liveMemtables.get(0);
}
public Memtable getCurrentMemtable()
{
return liveMemtables.get(liveMemtables.size() - 1);
}
public Iterable<Memtable> getMemtablesPendingFlush()
{
if (liveMemtables.size() == 1)
return flushingMemtables;
return Iterables.concat(liveMemtables.subList(0, 1), flushingMemtables);
}
/**
* @return the active memtable and all the memtables that are pending flush.
*/
public Iterable<Memtable> getAllMemtables()
{
return Iterables.concat(flushingMemtables, liveMemtables);
}
public Sets.SetView<SSTableReader> nonCompactingSStables()
{
return Sets.difference(ImmutableSet.copyOf(sstables), compacting);
}
View switchMemtable(Memtable newMemtable)
{
List<Memtable> newLiveMemtables = ImmutableList.<Memtable>builder().addAll(liveMemtables).add(newMemtable).build();
return new View(newLiveMemtables, flushingMemtables, sstablesMap, compacting, shadowed, intervalTree);
}
View markFlushing(Memtable toFlushMemtable)
{
List<Memtable> live = liveMemtables, flushing = flushingMemtables;
// since we can have multiple flushes queued, we may occasionally race and start a flush out of order,
// so must locate it in the list to remove, rather than just removing from the beginning
int i = live.indexOf(toFlushMemtable);
assert i < live.size() - 1;
List<Memtable> newLive = ImmutableList.<Memtable>builder()
.addAll(live.subList(0, i))
.addAll(live.subList(i + 1, live.size()))
.build();
// similarly, if we out-of-order markFlushing once, we may afterwards need to insert a memtable into the
// flushing list in a position other than the end, though this will be rare
i = flushing.size();
while (i > 0 && flushing.get(i - 1).creationTime() > toFlushMemtable.creationTime())
i--;
List<Memtable> newFlushing = ImmutableList.<Memtable>builder()
.addAll(flushing.subList(0, i))
.add(toFlushMemtable)
.addAll(flushing.subList(i, flushing.size()))
.build();
return new View(newLive, newFlushing, sstablesMap, compacting, shadowed, intervalTree);
}
View replaceFlushed(Memtable flushedMemtable, SSTableReader newSSTable)
{
int index = flushingMemtables.indexOf(flushedMemtable);
List<Memtable> newQueuedMemtables = ImmutableList.<Memtable>builder()
.addAll(flushingMemtables.subList(0, index))
.addAll(flushingMemtables.subList(index + 1, flushingMemtables.size()))
.build();
Map<SSTableReader, SSTableReader> newSSTables = sstablesMap;
SSTableIntervalTree intervalTree = this.intervalTree;
if (newSSTable != null)
{
assert !sstables.contains(newSSTable);
assert !shadowed.contains(newSSTable);
newSSTables = ImmutableMap.<SSTableReader, SSTableReader>builder()
.putAll(sstablesMap).put(newSSTable, newSSTable).build();
intervalTree = buildIntervalTree(newSSTables.keySet());
}
return new View(liveMemtables, newQueuedMemtables, newSSTables, compacting, shadowed, intervalTree);
}
View replace(Collection<SSTableReader> oldSSTables, Iterable<SSTableReader> replacements)
{
ImmutableSet<SSTableReader> oldSet = ImmutableSet.copyOf(oldSSTables);
int newSSTablesSize = shadowed.size() + sstables.size() - oldSSTables.size() + Iterables.size(replacements);
assert newSSTablesSize >= Iterables.size(replacements) : String.format("Incoherent new size %d replacing %s by %s in %s", newSSTablesSize, oldSSTables, replacements, this);
Map<SSTableReader, SSTableReader> newSSTables = new HashMap<>(newSSTablesSize);
Set<SSTableReader> newShadowed = new HashSet<>(shadowed.size());
for (SSTableReader sstable : sstables)
if (!oldSet.contains(sstable))
newSSTables.put(sstable, sstable);
for (SSTableReader sstable : shadowed)
if (!oldSet.contains(sstable))
newShadowed.add(sstable);
for (SSTableReader replacement : replacements)
{
if (replacement.openReason == SSTableReader.OpenReason.SHADOWED)
newShadowed.add(replacement);
else
newSSTables.put(replacement, replacement);
}
assert newSSTables.size() + newShadowed.size() == newSSTablesSize :
String.format("Expecting new size of %d, got %d while replacing %s by %s in %s",
newSSTablesSize, newSSTables.size() + newShadowed.size(), oldSSTables, replacements, this);
newShadowed = ImmutableSet.copyOf(newShadowed);
newSSTables = ImmutableMap.copyOf(newSSTables);
SSTableIntervalTree intervalTree = buildIntervalTree(newSSTables.keySet());
return new View(liveMemtables, flushingMemtables, newSSTables, compacting, newShadowed, intervalTree);
}
View markCompacting(Collection<SSTableReader> tomark)
{
Set<SSTableReader> compactingNew = ImmutableSet.<SSTableReader>builder().addAll(compacting).addAll(tomark).build();
return new View(liveMemtables, flushingMemtables, sstablesMap, compactingNew, shadowed, intervalTree);
}
View unmarkCompacting(Iterable<SSTableReader> tounmark)
{
Set<SSTableReader> compactingNew = ImmutableSet.copyOf(Sets.difference(compacting, ImmutableSet.copyOf(tounmark)));
return new View(liveMemtables, flushingMemtables, sstablesMap, compactingNew, shadowed, intervalTree);
}
private Set<SSTableReader> newSSTables(Collection<SSTableReader> oldSSTables, Iterable<SSTableReader> replacements)
{
ImmutableSet<SSTableReader> oldSet = ImmutableSet.copyOf(oldSSTables);
int newSSTablesSize = sstables.size() - oldSSTables.size() + Iterables.size(replacements);
assert newSSTablesSize >= Iterables.size(replacements) : String.format("Incoherent new size %d replacing %s by %s in %s", newSSTablesSize, oldSSTables, replacements, this);
Set<SSTableReader> newSSTables = new HashSet<>(newSSTablesSize);
for (SSTableReader sstable : sstables)
if (!oldSet.contains(sstable))
newSSTables.add(sstable);
Iterables.addAll(newSSTables, replacements);
assert newSSTables.size() == newSSTablesSize : String.format("Expecting new size of %d, got %d while replacing %s by %s in %s", newSSTablesSize, newSSTables.size(), oldSSTables, replacements, this);
return ImmutableSet.copyOf(newSSTables);
}
@Override
public String toString()
{
return String.format("View(pending_count=%d, sstables=%s, compacting=%s)", liveMemtables.size() + flushingMemtables.size() - 1, sstables, compacting);
}
public List<SSTableReader> sstablesInBounds(AbstractBounds<RowPosition> rowBounds)
{
if (intervalTree.isEmpty())
return Collections.emptyList();
RowPosition stopInTree = rowBounds.right.isMinimum(liveMemtables.get(0).cfs.partitioner) ? intervalTree.max() : rowBounds.right;
return intervalTree.search(Interval.<RowPosition, SSTableReader>create(rowBounds.left, stopInTree));
}
}
}