//JAVA TO C# CONVERTER TODO TASK: Most Java annotations will not have direct .NET equivalent attributes:
//ORIGINAL LINE: @Test public void shouldReleaseSharedLock()
public virtual void ShouldReleaseSharedLock()
{
// Given
ForsetiClient clientA = mock(typeof(ForsetiClient));
SharedLock @lock = new SharedLock(clientA);
// When
assertTrue(@lock.Release(clientA));
// Then
assertThat(@lock.NumberOfHolders(), equalTo(0));
assertThat(@lock.UpdateLock, equalTo(false));
}
public override void ReleaseExclusive(ResourceType resourceType, params long[] resourceIds)
{
_stateHolder.incrementActiveClients(this);
try
{
ConcurrentMap <long, ForsetiLockManager.Lock> resourceTypeLocks = _lockMaps[resourceType.TypeId()];
MutableLongIntMap exclusiveLocks = _exclusiveLockCounts[resourceType.TypeId()];
MutableLongIntMap sharedLocks = _sharedLockCounts[resourceType.TypeId()];
foreach (long resourceId in resourceIds)
{
if (ReleaseLocalLock(resourceType, resourceId, exclusiveLocks))
{
continue;
}
if (sharedLocks.containsKey(resourceId))
{
// We are still holding a shared lock, so we will release it to be reused
ForsetiLockManager.Lock @lock = resourceTypeLocks.get(resourceId);
if (@lock is SharedLock)
{
SharedLock sharedLock = ( SharedLock )@lock;
if (sharedLock.UpdateLock)
{
sharedLock.ReleaseUpdateLock();
}
else
{
throw new System.InvalidOperationException("Incorrect state of exclusive lock. Lock should be updated " + "to exclusive before attempt to release it. Lock: " + this);
}
}
else
{
// in case if current lock is exclusive we swap it to new shared lock
SharedLock sharedLock = new SharedLock(this);
resourceTypeLocks.put(resourceId, sharedLock);
}
}
else
{
// we do not hold shared lock so we just releasing it
ReleaseGlobalLock(resourceTypeLocks, resourceId);
}
}
}
finally
{
_stateHolder.decrementActiveClients();
}
}
public override bool TryExclusiveLock(ResourceType resourceType, long resourceId)
{
_hasLocks = true;
_stateHolder.incrementActiveClients(this);
try
{
ConcurrentMap <long, ForsetiLockManager.Lock> lockMap = _lockMaps[resourceType.TypeId()];
MutableLongIntMap heldLocks = _exclusiveLockCounts[resourceType.TypeId()];
int heldCount = heldLocks.getIfAbsent(resourceId, -1);
if (heldCount != -1)
{
// We already have a lock on this, just increment our local reference counter.
heldLocks.put(resourceId, Math.incrementExact(heldCount));
return(true);
}
// Grab the global lock
ForsetiLockManager.Lock @lock;
if ((@lock = lockMap.putIfAbsent(resourceId, _myExclusiveLock)) != null)
{
if (@lock is SharedLock && _sharedLockCounts[resourceType.TypeId()].containsKey(resourceId))
{
SharedLock sharedLock = ( SharedLock )@lock;
if (sharedLock.TryAcquireUpdateLock(this))
{
if (sharedLock.NumberOfHolders() == 1)
{
heldLocks.put(resourceId, 1);
return(true);
}
else
{
sharedLock.ReleaseUpdateLock();
return(false);
}
}
}
return(false);
}
heldLocks.put(resourceId, 1);
return(true);
}
finally
{
_stateHolder.decrementActiveClients();
}
}
//JAVA TO C# CONVERTER TODO TASK: Most Java annotations will not have direct .NET equivalent attributes:
//ORIGINAL LINE: @Test public void shouldUpgradeToUpdateLock()
public virtual void ShouldUpgradeToUpdateLock()
{
// Given
ForsetiClient clientA = mock(typeof(ForsetiClient));
ForsetiClient clientB = mock(typeof(ForsetiClient));
SharedLock @lock = new SharedLock(clientA);
@lock.Acquire(clientB);
// When
assertTrue(@lock.TryAcquireUpdateLock(clientA));
// Then
assertThat(@lock.NumberOfHolders(), equalTo(2));
assertThat(@lock.UpdateLock, equalTo(true));
}
/// <summary>
/// Attempt to upgrade a share lock that we hold to an exclusive lock. </summary>
//JAVA TO C# CONVERTER WARNING: Method 'throws' clauses are not available in C#:
//ORIGINAL LINE: private boolean tryUpgradeToExclusiveWithShareLockHeld(org.neo4j.storageengine.api.lock.LockTracer tracer, org.neo4j.storageengine.api.lock.LockWaitEvent priorEvent, org.neo4j.storageengine.api.lock.ResourceType resourceType, long resourceId, SharedLock sharedLock, int tries, long waitStartMillis) throws org.neo4j.storageengine.api.lock.AcquireLockTimeoutException
private bool TryUpgradeToExclusiveWithShareLockHeld(LockTracer tracer, LockWaitEvent priorEvent, ResourceType resourceType, long resourceId, SharedLock sharedLock, int tries, long waitStartMillis)
{
if (sharedLock.TryAcquireUpdateLock(this))
{
LockWaitEvent waitEvent = null;
try
{
// Now we just wait for all clients to release the the share lock
while (sharedLock.NumberOfHolders() > 1)
{
AssertValid(waitStartMillis, resourceType, resourceId);
if (waitEvent == null && priorEvent == null)
{
waitEvent = tracer.WaitForLock(true, resourceType, resourceId);
}
WaitFor(sharedLock, resourceType, resourceId, true, tries++);
}
return(true);
}
catch (Exception e)
{
sharedLock.ReleaseUpdateLock();
if (e is DeadlockDetectedException || e is LockClientStoppedException)
{
throw ( Exception )e;
}
throw new TransactionFailureException("Failed to upgrade shared lock to exclusive: " + sharedLock, e);
}
finally
{
if (waitEvent != null)
{
waitEvent.Close();
}
ClearWaitList();
_waitingForLock = null;
}
}
return(false);
}
/// <summary>
/// Attempt to upgrade a share lock to an exclusive lock, grabbing the share lock if we don't hold it.
///
/// </summary>
//JAVA TO C# CONVERTER WARNING: Method 'throws' clauses are not available in C#:
//ORIGINAL LINE: private boolean tryUpgradeSharedToExclusive(org.neo4j.storageengine.api.lock.LockTracer tracer, org.neo4j.storageengine.api.lock.LockWaitEvent waitEvent, org.neo4j.storageengine.api.lock.ResourceType resourceType, java.util.concurrent.ConcurrentMap<long,ForsetiLockManager.Lock> lockMap, long resourceId, SharedLock sharedLock, long waitStartMillis) throws org.neo4j.storageengine.api.lock.AcquireLockTimeoutException
private bool TryUpgradeSharedToExclusive(LockTracer tracer, LockWaitEvent waitEvent, ResourceType resourceType, ConcurrentMap <long, ForsetiLockManager.Lock> lockMap, long resourceId, SharedLock sharedLock, long waitStartMillis)
{
int tries = 0;
bool holdsSharedLock = _sharedLockCounts[resourceType.TypeId()].containsKey(resourceId);
if (!holdsSharedLock)
{
// We don't hold the shared lock, we need to grab it to upgrade it to an exclusive one
if (!sharedLock.Acquire(this))
{
return(false);
}
try
{
if (TryUpgradeToExclusiveWithShareLockHeld(tracer, waitEvent, resourceType, resourceId, sharedLock, tries, waitStartMillis))
{
return(true);
}
else
{
ReleaseGlobalLock(lockMap, resourceId);
return(false);
}
}
catch (Exception e)
{
ReleaseGlobalLock(lockMap, resourceId);
throw e;
}
}
else
{
// We do hold the shared lock, so no reason to deal with the complexity in the case above.
return(TryUpgradeToExclusiveWithShareLockHeld(tracer, waitEvent, resourceType, resourceId, sharedLock, tries, waitStartMillis));
}
}
//JAVA TO C# CONVERTER WARNING: Method 'throws' clauses are not available in C#:
//ORIGINAL LINE: public void acquireExclusive(org.neo4j.storageengine.api.lock.LockTracer tracer, org.neo4j.storageengine.api.lock.ResourceType resourceType, long... resourceIds) throws org.neo4j.storageengine.api.lock.AcquireLockTimeoutException
public override void AcquireExclusive(LockTracer tracer, ResourceType resourceType, params long[] resourceIds)
{
_hasLocks = true;
_stateHolder.incrementActiveClients(this);
LockWaitEvent waitEvent = null;
try
{
ConcurrentMap <long, ForsetiLockManager.Lock> lockMap = _lockMaps[resourceType.TypeId()];
MutableLongIntMap heldLocks = _exclusiveLockCounts[resourceType.TypeId()];
foreach (long resourceId in resourceIds)
{
int heldCount = heldLocks.getIfAbsent(resourceId, -1);
if (heldCount != -1)
{
// We already have a lock on this, just increment our local reference counter.
heldLocks.put(resourceId, Math.incrementExact(heldCount));
continue;
}
// Grab the global lock
ForsetiLockManager.Lock existingLock;
int tries = 0;
long waitStartMillis = _clock.millis();
while ((existingLock = lockMap.putIfAbsent(resourceId, _myExclusiveLock)) != null)
{
AssertValid(waitStartMillis, resourceType, resourceId);
// If this is a shared lock:
// Given a grace period of tries (to try and not starve readers), grab an update lock and wait
// for it to convert to an exclusive lock.
if (tries > 50 && existingLock is SharedLock)
{
// Then we should upgrade that lock
SharedLock sharedLock = ( SharedLock )existingLock;
if (TryUpgradeSharedToExclusive(tracer, waitEvent, resourceType, lockMap, resourceId, sharedLock, waitStartMillis))
{
break;
}
}
if (waitEvent == null)
{
waitEvent = tracer.WaitForLock(true, resourceType, resourceId);
}
WaitFor(existingLock, resourceType, resourceId, true, tries++);
}
heldLocks.put(resourceId, 1);
}
}
finally
{
if (waitEvent != null)
{
waitEvent.Close();
}
ClearWaitList();
_waitingForLock = null;
_stateHolder.decrementActiveClients();
}
}
//JAVA TO C# CONVERTER WARNING: Method 'throws' clauses are not available in C#:
//ORIGINAL LINE: public void acquireShared(org.neo4j.storageengine.api.lock.LockTracer tracer, org.neo4j.storageengine.api.lock.ResourceType resourceType, long... resourceIds) throws org.neo4j.storageengine.api.lock.AcquireLockTimeoutException
public override void AcquireShared(LockTracer tracer, ResourceType resourceType, params long[] resourceIds)
{
_hasLocks = true;
_stateHolder.incrementActiveClients(this);
LockWaitEvent waitEvent = null;
try
{
// Grab the global lock map we will be using
ConcurrentMap <long, ForsetiLockManager.Lock> lockMap = _lockMaps[resourceType.TypeId()];
// And grab our local lock maps
MutableLongIntMap heldShareLocks = _sharedLockCounts[resourceType.TypeId()];
MutableLongIntMap heldExclusiveLocks = _exclusiveLockCounts[resourceType.TypeId()];
foreach (long resourceId in resourceIds)
{
// First, check if we already hold this as a shared lock
int heldCount = heldShareLocks.getIfAbsent(resourceId, -1);
if (heldCount != -1)
{
// We already have a lock on this, just increment our local reference counter.
heldShareLocks.put(resourceId, Math.incrementExact(heldCount));
continue;
}
// Second, check if we hold it as an exclusive lock
if (heldExclusiveLocks.containsKey(resourceId))
{
// We already have an exclusive lock, so just leave that in place.
// When the exclusive lock is released, it will be automatically downgraded to a shared lock,
// since we bumped the share lock reference count.
heldShareLocks.put(resourceId, 1);
continue;
}
// We don't hold the lock, so we need to grab it via the global lock map
int tries = 0;
SharedLock mySharedLock = null;
long waitStartMillis = _clock.millis();
// Retry loop
while (true)
{
AssertValid(waitStartMillis, resourceType, resourceId);
// Check if there is a lock for this entity in the map
ForsetiLockManager.Lock existingLock = lockMap.get(resourceId);
// No lock
if (existingLock == null)
{
// Try to create a new shared lock
if (mySharedLock == null)
{
mySharedLock = new SharedLock(this);
}
if (lockMap.putIfAbsent(resourceId, mySharedLock) == null)
{
// Success, we now hold the shared lock.
break;
}
else
{
continue;
}
}
// Someone holds shared lock on this entity, try and get in on that action
else if (existingLock is SharedLock)
{
if ((( SharedLock )existingLock).Acquire(this))
{
// Success!
break;
}
}
// Someone holds an exclusive lock on this entity
else if (existingLock is ExclusiveLock)
{
// We need to wait, just let the loop run.
}
else
{
throw new System.NotSupportedException("Unknown lock type: " + existingLock);
}
if (waitEvent == null)
{
waitEvent = tracer.WaitForLock(false, resourceType, resourceId);
}
// And take note of who we are waiting for. This is used for deadlock detection.
WaitFor(existingLock, resourceType, resourceId, false, tries++);
}
// Make a local note about the fact that we now hold this lock
heldShareLocks.put(resourceId, 1);
}
}
finally
{
if (waitEvent != null)
{
waitEvent.Close();
}
ClearWaitList();
_waitingForLock = null;
_stateHolder.decrementActiveClients();
}
}
