use of com.cinchapi.concourse.server.concurrent.RangeToken in project concourse by cinchapi.
the class AtomicOperation method grabLocks.
/**
* Check each one of the {@link #intentions} against the
* {@link #durable} and grab the appropriate locks along the way. This
* method will return {@code true} if all expectations are met and all
* necessary locks are grabbed. Otherwise it will return {@code false}, in
* which case this operation should be aborted immediately.
*
* @return {@code true} if all expectations are met and all necessary locks
* are grabbed.
*/
private boolean grabLocks() {
if (isReadOnly()) {
return true;
} else {
// NOTE: If we can't grab a lock immediately because it is held by
// someone else, then we must fail immediately because the
// AtomicOperation can't properly commit.
locks = Maps.newHashMap();
try {
// intentions
for (Token token : writes2Lock) {
if (notifiedAboutVersionChange) {
return false;
}
LockType type;
if (token instanceof RangeToken) {
RangeToken rangeToken = (RangeToken) token;
if (!rangeReads2Lock.isEmpty(rangeToken.getKey())) {
Range<Value> containing = rangeReads2Lock.get(rangeToken.getKey(), rangeToken.getValues()[0]);
if (containing != null) {
rangeReads2Lock.remove(rangeToken.getKey(), containing);
Iterable<Range<Value>> xor = Ranges.xor(Range.singleton(rangeToken.getValues()[0]), containing);
for (Range<Value> range : xor) {
rangeReads2Lock.put(rangeToken.getKey(), range);
}
}
}
type = LockType.RANGE_WRITE;
} else {
reads2Lock.remove(token);
type = LockType.WRITE;
}
LockDescription lock = LockDescription.forToken(token, lockService, rangeLockService, type);
if (lock.getLock().tryLock()) {
locks.put(lock.getToken(), lock);
} else {
return false;
}
}
// grabbed previously.
for (Token token : reads2Lock) {
if (notifiedAboutVersionChange) {
return false;
}
LockDescription lock = LockDescription.forToken(token, lockService, rangeLockService, LockType.READ);
if (lock.getLock().tryLock()) {
locks.put(lock.getToken(), lock);
} else {
return false;
}
}
// grabbed previously.
for (Entry<Text, RangeSet<Value>> entry : rangeReads2Lock.ranges.entrySet()) {
/* (Authorized) */
if (notifiedAboutVersionChange) {
return false;
}
Text key = entry.getKey();
for (Range<Value> range : entry.getValue().asRanges()) {
RangeToken rangeToken = Ranges.convertToRangeToken(key, range);
LockDescription lock = LockDescription.forToken(rangeToken, lockService, rangeLockService, LockType.RANGE_READ);
if (lock.getLock().tryLock()) {
locks.put(lock.getToken(), lock);
} else {
return false;
}
}
}
} catch (NullPointerException e) {
// abort immediately which means that #locks will become null.
return false;
}
return true;
}
}
use of com.cinchapi.concourse.server.concurrent.RangeToken in project concourse by cinchapi.
the class AtomicOperation method browse.
@Override
public final Map<TObject, Set<Long>> browse(String key) throws AtomicStateException {
checkState();
Text key0 = Text.wrapCached(key);
RangeToken rangeToken = RangeToken.forReading(key0, Operator.BETWEEN, Value.NEGATIVE_INFINITY, Value.POSITIVE_INFINITY);
source.addVersionChangeListener(rangeToken, this);
Iterable<Range<Value>> ranges = RangeTokens.convertToRange(rangeToken);
for (Range<Value> range : ranges) {
rangeReads2Lock.put(key0, range);
}
return super.browse(key);
}
use of com.cinchapi.concourse.server.concurrent.RangeToken in project concourse by cinchapi.
the class AtomicOperation method explore.
@Override
public Map<Long, Set<TObject>> explore(String key, Aliases aliases) {
checkState();
Operator operator = aliases.operator();
TObject[] values = aliases.values();
Text key0 = Text.wrapCached(key);
RangeToken rangeToken = RangeToken.forReading(key0, operator, Transformers.transformArray(values, Value::wrap, Value.class));
source.addVersionChangeListener(rangeToken, this);
Iterable<Range<Value>> ranges = RangeTokens.convertToRange(rangeToken);
for (Range<Value> range : ranges) {
rangeReads2Lock.put(key0, range);
}
return super.explore(key, aliases);
}
use of com.cinchapi.concourse.server.concurrent.RangeToken in project concourse by cinchapi.
the class AtomicOperation method set.
@Override
public final void set(String key, TObject value, long record) throws AtomicStateException {
checkState();
Token token = Token.wrap(key, record);
RangeToken rangeToken = RangeToken.forWriting(Text.wrapCached(key), Value.wrap(value));
Token wide = wideReads.get(record);
if (wide != null) {
wide.upgrade();
writes2Lock.add(wide);
} else {
source.addVersionChangeListener(token, this);
writes2Lock.add(token);
// CON-669: Prevent a
writes2Lock.add(Token.shareable(record));
// conflicting wide read,
// but don't listen for
// wide version change
}
writes2Lock.add(rangeToken);
super.set(key, value, record);
}
use of com.cinchapi.concourse.server.concurrent.RangeToken in project concourse by cinchapi.
the class Engine method set.
@Override
public void set(String key, TObject value, long record) {
Token sharedToken = Token.shareable(record);
Token writeToken = Token.wrap(key, record);
RangeToken rangeToken = RangeToken.forWriting(Text.wrap(key), Value.wrap(value));
Lock shared = lockService.getWriteLock(sharedToken);
Lock write = lockService.getWriteLock(writeToken);
Lock range = rangeLockService.getWriteLock(rangeToken);
shared.lock();
write.lock();
range.lock();
try {
super.set(key, value, record);
notifyVersionChange(writeToken);
notifyVersionChange(sharedToken);
notifyVersionChange(rangeToken);
} finally {
shared.unlock();
write.unlock();
range.unlock();
}
}
Aggregations