/**
* @param ctx Cache registry.
* @param implicit Implicit flag.
* @param implicitSingle Implicit with one key flag.
* @param concurrency Concurrency.
* @param isolation Isolation.
* @param timeout Timeout.
* @param invalidate Invalidation policy.
* @param syncCommit Synchronous commit flag.
* @param syncRollback Synchronous rollback flag.
* @param swapEnabled Whether to use swap storage.
* @param storeEnabled Whether to use read/write through.
*/
GridNearTxLocal(
GridCacheContext<K, V> ctx,
boolean implicit,
boolean implicitSingle,
GridCacheTxConcurrency concurrency,
GridCacheTxIsolation isolation,
long timeout,
boolean invalidate,
boolean syncCommit,
boolean syncRollback,
boolean swapEnabled,
boolean storeEnabled) {
super(
ctx,
ctx.versions().next(),
implicit,
implicitSingle,
concurrency,
isolation,
timeout,
invalidate,
swapEnabled,
storeEnabled);
assert ctx != null;
this.syncCommit = syncCommit;
this.syncRollback = syncRollback;
}
/** {@inheritDoc} */
@Override
public GridCacheEntry<K, V> wrap(boolean prjAware) {
GridCacheContext<K, V> nearCtx = cctx.dht().near().context();
GridCacheProjectionImpl<K, V> prjPerCall = nearCtx.projectionPerCall();
if (prjPerCall != null && prjAware)
return new GridPartitionedCacheEntryImpl<K, V>(prjPerCall, nearCtx, key, this);
GridCacheEntryImpl<K, V> wrapper = this.wrapper;
if (wrapper == null)
this.wrapper = wrapper = new GridPartitionedCacheEntryImpl<K, V>(null, nearCtx, key, this);
return wrapper;
}
/**
* Constructor.
*
* @param name Latch name.
* @param cnt Current count.
* @param initCnt Initial count.
* @param autoDel Auto delete flag.
* @param key Latch key.
* @param latchView Latch projection.
* @param ctx Cache context.
*/
public GridCacheCountDownLatchImpl(
String name,
int cnt,
int initCnt,
boolean autoDel,
GridCacheInternalKey key,
GridCacheProjection<GridCacheInternalKey, GridCacheCountDownLatchValue> latchView,
GridCacheContext ctx) {
assert name != null;
assert cnt >= 0;
assert initCnt >= 0;
assert key != null;
assert latchView != null;
assert ctx != null;
this.name = name;
this.cnt = cnt;
this.initCnt = initCnt;
this.autoDel = autoDel;
this.key = key;
this.latchView = latchView;
this.ctx = ctx;
log = ctx.gridConfig().getGridLogger().getLogger(getClass());
}
/**
* @param ctx Cache context.
* @param key Cache key.
* @param hash Key hash value.
* @param val Entry value.
* @param next Next entry in the linked list.
* @param ttl Time to live.
*/
public GridDhtCacheEntry(
GridCacheContext<K, V> ctx, K key, int hash, V val, GridCacheMapEntry<K, V> next, long ttl) {
super(ctx, key, hash, val, next, ttl);
// Record this entry with partition.
locPart = ctx.dht().topology().onAdded(this);
}
/** @param m Mapping. */
@SuppressWarnings({"unchecked"})
private void finish(GridDistributedTxMapping<K, V> m) {
GridRichNode n = m.node();
assert !m.empty();
GridNearTxFinishRequest req =
new GridNearTxFinishRequest<K, V>(
futId,
tx.xidVersion(),
tx.commitVersion(),
tx.threadId(),
commit,
tx.isInvalidate(),
m.explicitLock(),
tx.topologyVersion(),
null,
null,
null,
commit && tx.pessimistic() ? m.writes() : null,
tx.syncCommit() && commit || tx.syncRollback() && !commit);
// If this is the primary node for the keys.
if (n.isLocal()) {
req.miniId(GridUuid.randomUuid());
if (CU.DHT_ENABLED) {
GridFuture<GridCacheTx> fut =
commit ? dht().commitTx(n.id(), req) : dht().rollbackTx(n.id(), req);
// Add new future.
add(fut);
} else
// Add done future for testing.
add(new GridFinishedFuture<GridCacheTx>(ctx));
} else {
MiniFuture fut = new MiniFuture(m);
req.miniId(fut.futureId());
add(fut); // Append new future.
try {
cctx.io().send(n, req);
// If we don't wait for result, then mark future as done.
if (!isSync() && !m.explicitLock()) fut.onDone();
} catch (GridTopologyException e) {
// Remove previous mapping.
mappings.remove(m.node().id());
fut.onResult(e);
} catch (GridException e) {
// Fail the whole thing.
fut.onResult(e);
}
}
}
/**
* @param entry Transaction entry.
* @param nodes Nodes.
*/
private void map(GridCacheTxEntry<K, V> entry, Collection<GridRichNode> nodes) {
GridRichNode primary = CU.primary0(cctx.affinity(entry.key(), nodes));
GridDistributedTxMapping<K, V> t = mappings.get(primary.id());
if (t == null) mappings.put(primary.id(), t = new GridDistributedTxMapping<K, V>(primary));
t.add(entry);
}
/** {@inheritDoc} */
@Override
public boolean onDone(GridCacheTx tx, Throwable err) {
if (initialized() && super.onDone(tx, err)) {
// Don't forget to clean up.
cctx.mvcc().removeFuture(this);
return true;
}
return false;
}
/** {@inheritDoc} */
@Override
public GridFuture<Boolean> awaitAsync(final long timeout, final TimeUnit unit) {
return ctx.closures()
.callLocalSafe(
new Callable<Boolean>() {
@Override
public Boolean call() throws Exception {
return await(timeout, unit);
}
},
true);
}
/**
* Default constructor.
*
* @param name Sequence name.
* @param key Sequence key.
* @param seqView Sequence projection.
* @param ctx CacheContext.
* @param locVal Local counter.
* @param upBound Upper bound.
*/
public GridCacheAtomicSequenceImpl(
String name,
GridCacheInternalStorableKey key,
GridCacheProjection<GridCacheInternalStorableKey, GridCacheAtomicSequenceValue> seqView,
GridCacheContext ctx,
long locVal,
long upBound) {
assert key != null;
assert seqView != null;
assert ctx != null;
assert locVal <= upBound;
batchSize = ctx.config().getAtomicSequenceReserveSize();
this.ctx = ctx;
this.key = key;
this.seqView = seqView;
this.upBound = upBound;
this.locVal = locVal;
this.name = name;
log = ctx.gridConfig().getGridLogger().getLogger(getClass());
}
/** {@inheritDoc} */
@Override
public GridFuture<?> awaitAsync() {
return ctx.closures()
.callLocalSafe(
new Callable<Object>() {
@Nullable
@Override
public Object call() throws Exception {
await();
return null;
}
},
true);
}
/**
* @param cctx Context.
* @param tx Transaction.
* @param commit Commit flag.
*/
public GridNearTxFinishFuture(
GridCacheContext<K, V> cctx, GridNearTxLocal<K, V> tx, boolean commit) {
super(cctx.kernalContext(), F.<GridCacheTx>identityReducer(tx));
assert cctx != null;
this.cctx = cctx;
this.tx = tx;
this.commit = commit;
mappings = tx.mappings();
futId = GridUuid.randomUuid();
log = U.logger(ctx, logRef, GridNearTxFinishFuture.class);
}
/** @param ctx Context. */
public GridNearCache(GridCacheContext<K, V> ctx) {
super(ctx, ctx.config().getNearStartSize());
}
/** @param ctx Cache registry. */
public GridReplicatedCache(GridCacheContext<K, V> ctx) {
super(ctx, ctx.config().getStartSize());
}
/**
* Asynchronous sequence update operation. Will add given amount to the sequence value.
*
* @param l Increment amount.
* @param updateCall Cache call that will update sequence reservation count in accordance with l.
* @param updated If {@code true}, will return sequence value after update, otherwise will return
* sequence value prior to update.
* @return Future indicating sequence value.
* @throws GridException If update failed.
*/
private GridFuture<Long> internalUpdateAsync(
long l, @Nullable Callable<Long> updateCall, boolean updated) throws GridException {
checkRemoved();
A.ensure(l > 0, " Parameter mustn't be less then 1: " + l);
lock.lock();
try {
// If reserved range isn't exhausted.
if (locVal + l <= upBound) {
long curVal = locVal;
locVal += l;
return new GridFinishedFuture<Long>(ctx.kernalContext(), updated ? locVal : curVal);
}
} finally {
lock.unlock();
}
if (updateCall == null) updateCall = internalUpdate(l, updated);
while (true) {
if (updateGuard.compareAndSet(false, true)) {
try {
// This call must be outside lock.
return ctx.closures().callLocalSafe(updateCall, true);
} finally {
lock.lock();
try {
updateGuard.set(false);
cond.signalAll();
} finally {
lock.unlock();
}
}
} else {
lock.lock();
try {
while (locVal >= upBound && updateGuard.get()) {
try {
cond.await(500, MILLISECONDS);
} catch (InterruptedException e) {
throw new GridInterruptedException(e);
}
}
checkRemoved();
// If reserved range isn't exhausted.
if (locVal + l <= upBound) {
long curVal = locVal;
locVal += l;
return new GridFinishedFuture<Long>(ctx.kernalContext(), updated ? locVal : curVal);
}
} finally {
lock.unlock();
}
}
}
}
/** @return DHT cache. */
private GridDhtCache<K, V> dht() {
return cctx.near().dht();
}
/** {@inheritDoc} */
@Override
public GridFuture<?> countDownAllAsync() {
return ctx.closures().callLocalSafe(new CountDownCallable(0), true);
}
/** {@inheritDoc} */
@Override
public GridFuture<Integer> countDownAsync(int val) {
return ctx.closures().callLocalSafe(new CountDownCallable(val), true);
}