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(); } }