lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Browse Source
This commit is contained in:
Redkale
2019-01-18 16:55:48 +08:00
parent cf2ab617c2
commit d605045858
1 changed files with 46 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
src/org/redkale/net/TcpAioAsyncConnection.java
View File

@@ -112,20 +112,28 @@ public class TcpAioAsyncConnection extends AsyncConnection {
private <A> void nextWrite(Throwable exc, A attachment) {
BlockingQueue<WriteEntry> queue = this.writeQueue;
WriteEntry entry = queue == null ? null : queue.poll();
if (entry != null) {
if (exc == null) {
if (exc != null && !isOpen()) {
WriteEntry entry;
while ((entry = queue.poll()) != null) {
try {
if (entry.writeOneBuffer == null) {
write(false, entry.writeBuffers, entry.writeOffset, entry.writeLength, entry.writeAttachment, entry.writeHandler);
} else {
write(false, entry.writeOneBuffer, entry.writeAttachment, entry.writeHandler);
}
} catch (Exception e) {
entry.writeHandler.failed(e, entry.writeAttachment);
entry.writeHandler.failed(exc, entry.writeAttachment);
} catch (Throwable e) {
e.printStackTrace(System.err);
}
} else { //当连接已经关掉了，不需要调用write方法，直接报异常
entry.writeHandler.failed(exc, entry.writeAttachment);
}
return;
}
WriteEntry entry = queue == null ? null : queue.poll();
if (entry != null) {
try {
if (entry.writeOneBuffer == null) {
write(false, entry.writeBuffers, entry.writeOffset, entry.writeLength, entry.writeAttachment, entry.writeHandler);
} else {
write(false, entry.writeOneBuffer, entry.writeAttachment, entry.writeHandler);
}
} catch (Exception e) {
entry.writeHandler.failed(e, entry.writeAttachment);
}
} else {
semaphore.release();
@@ -316,18 +324,27 @@ public class TcpAioAsyncConnection extends AsyncConnection {
failed(e, attachment);
return;
}
nextWrite(null, attachment);
writeHandler.completed(writeCount, attachment);
try {
writeHandler.completed(writeCount, attachment);
} finally {
nextWrite(null, attachment);
}
} else {
nextWrite(null, attachment);
writeHandler.completed(result.intValue(), attachment);
try {
writeHandler.completed(result.intValue(), attachment);
} finally {
nextWrite(null, attachment);
}
}
}
@Override
public void failed(Throwable exc, A attachment) {
nextWrite(isOpen() ? null : exc, attachment);
writeHandler.failed(exc, attachment);
try {
writeHandler.failed(exc, attachment);
} finally {
nextWrite(exc, attachment);
}
}
}
@@ -354,14 +371,21 @@ public class TcpAioAsyncConnection extends AsyncConnection {
failed(e, attachment);
return;
}
nextWrite(null, attachment);
writeHandler.completed(result, attachment);
try {
writeHandler.completed(result, attachment);
} finally {
nextWrite(null, attachment);
}
}
@Override
public void failed(Throwable exc, A attachment) {
nextWrite(isOpen() ? null : exc, attachment);
writeHandler.failed(exc, attachment);
try {
writeHandler.failed(exc, attachment);
} finally {
nextWrite(exc, attachment);
}
}
}