protected internal void Finish()
{
// Ensure that command succeeds or fails, but not both.
int status = Interlocked.CompareExchange(ref state, SUCCESS, IN_PROGRESS);
if (status == IN_PROGRESS)
{
// Command finished successfully.
// Put connection back into pool.
conn.UpdateLastUsed();
node.PutAsyncConnection(conn);
// Do not put large buffers back into pool.
if (segment.size > BufferPool.BUFFER_CUTOFF)
{
// Put back original buffer instead.
segment = segmentOrig;
eventArgs.SetBuffer(segment.buffer, segment.offset, 0);
}
eventArgs.UserToken = segment;
cluster.PutEventArgs(eventArgs);
}
else if (status == FAIL_TOTAL_TIMEOUT || status == FAIL_SOCKET_TIMEOUT)
{
// Timeout thread closed connection, but transaction still completed.
// Do not connection put back into pool.
PutBackArgsOnError();
}
else
{
// Should never get here.
if (Log.WarnEnabled())
{
Log.Warn("AsyncCommand finished with unexpected return status: " + status);
}
}
try
{
OnSuccess();
}
catch (Exception e)
{
if (Log.WarnEnabled())
{
Log.Warn("OnSuccess() error: " + Util.GetErrorMessage(e));
}
}
}
protected internal void Finish()
{
// Ensure that command succeeds or fails, but not both.
int status = Interlocked.CompareExchange(ref complete, SUCCESS, 0);
if (status == 0)
{
conn.UpdateLastUsed();
node.PutAsyncConnection(conn);
// Do not put large buffers back into pool.
if (segment.size > BufferPool.BUFFER_CUTOFF)
{
// Put back original buffer instead.
segment = segmentOrig;
eventArgs.SetBuffer(segment.buffer, segment.offset, 0);
}
eventArgs.UserToken = segment;
cluster.PutEventArgs(eventArgs);
try
{
OnSuccess();
}
catch (AerospikeException ae)
{
// The user's OnSuccess() may have already been called which in turn generates this
// exception. It's important to call OnFailure() anyhow because the user's code
// may be waiting for completion notification which wasn't yet called in
// OnSuccess(). This is the only case where both OnSuccess() and OnFailure()
// gets called for the same command.
OnFailure(ae);
}
catch (Exception e)
{
OnFailure(new AerospikeException(e));
}
}
else
{
AlreadyCompleted(status);
}
}
private void ConnectionReady()
{
if (Interlocked.CompareExchange(ref state, 1, 0) == 0)
{
conn.UpdateLastUsed();
node.PutAsyncConnection(conn);
try
{
listener.OnSuccess(eventArgs, dataBuffer);
}
catch (Exception e)
{
if (Log.WarnEnabled())
{
Log.Warn("OnSuccess() error: " + Util.GetErrorMessage(e));
}
}
}
}
private void ReceiveEvent()
{
//Log.Info("Receive Event: " + eventArgs.BytesTransferred + "," + dataOffset + "," + dataLength + "," + inHeader);
if (usingSocketTimeout)
{
eventReceived = true;
}
if (eventArgs.BytesTransferred <= 0)
{
ConnectionFailed(new AerospikeException.Connection("Connection closed"));
return;
}
dataOffset += eventArgs.BytesTransferred;
if (dataOffset < dataLength)
{
eventArgs.SetBuffer(dataOffset, dataLength - dataOffset);
Receive();
return;
}
dataOffset = segment.offset;
if (inHeader)
{
long proto = ByteUtil.BytesToLong(dataBuffer, dataOffset);
int length = (int)(proto & 0xFFFFFFFFFFFFL);
if (length <= 0)
{
ReceiveBegin();
return;
}
compressed = ((proto >> 48) & 0xFF) == (long)Command.MSG_TYPE_COMPRESSED;
inHeader = false;
if (length > segment.size)
{
ResizeBuffer(length);
dataBuffer = segment.buffer;
dataOffset = segment.offset;
}
eventArgs.SetBuffer(dataBuffer, dataOffset, length);
dataLength = dataOffset + length;
Receive();
}
else
{
if (inAuthenticate)
{
inAuthenticate = false;
inHeader = true;
int resultCode = dataBuffer[dataOffset + 1];
if (resultCode != 0 && resultCode != ResultCode.SECURITY_NOT_ENABLED)
{
// Authentication failed. Session token probably expired.
// Signal tend thread to perform node login, so future
// transactions do not fail.
node.SignalLogin();
// This is a rare event because the client tracks session
// expiration and will relogin before session expiration.
// Do not try to login on same socket because login can take
// a long time and thousands of simultaneous logins could
// overwhelm server.
throw new AerospikeException(resultCode);
}
ConnectionReady();
return;
}
conn.UpdateLastUsed();
if (compressed)
{
int usize = (int)ByteUtil.BytesToLong(dataBuffer, dataOffset);
dataOffset += 8;
byte[] ubuf = new byte[usize];
ByteUtil.Decompress(dataBuffer, dataOffset, dataLength, ubuf, usize);
dataBuffer = ubuf;
dataOffset = 8;
dataLength = usize;
}
ParseCommand();
}
}