h4ckx0r/engine.io-client-java
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

compatible with engine.io-client v1.4.2

Browse Source
This commit is contained in:
Naoyuki Kanezawa
2014-11-03 22:45:45 +09:00
parent ba8755384e
commit 96fba4e20a
2 changed files with 181 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
src/main/java/com/github/nkzawa/engineio/client/Socket.java
View File

@@ -115,7 +115,7 @@ public class Socket extends Emitter {
private List<String> transports;
private List<String> upgrades;
private Map<String, String> query;
private LinkedList<Packet> writeBuffer = new LinkedList<Packet>();
/*package*/ LinkedList<Packet> writeBuffer = new LinkedList<Packet>();
private LinkedList<Runnable> callbackBuffer = new LinkedList<Runnable>();
/*package*/ Transport transport;
private Future pingTimeoutTimer;
@@ -325,6 +325,7 @@ public class Socket extends Emitter {
logger.fine(String.format("probe transport '%s' pong", name));
self.upgrading = true;
self.emit(EVENT_UPGRADING, transport[0]);
if (null == transport) return;
Socket.priorWebsocketSuccess = WebSocket.NAME.equals(transport[0].name);
logger.fine(String.format("pausing current transport '%s'", self.transport.name));
@@ -332,9 +333,7 @@ public class Socket extends Emitter {
@Override
public void run() {
if (failed[0]) return;
if (ReadyState.CLOSED == self.readyState || ReadyState.CLOSING == self.readyState) {
return;
}
if (ReadyState.CLOSED == self.readyState) return;
logger.fine("changing transport and sending upgrade packet");
@@ -667,6 +666,10 @@ public class Socket extends Emitter {
}
private void sendPacket(Packet packet, Runnable fn) {
if (ReadyState.CLOSING == this.readyState || ReadyState.CLOSED == this.readyState) {
return;
}
if (fn == null) {
// ConcurrentLinkedList does not permit `null`.
fn = noop;
@@ -688,11 +691,55 @@ public class Socket extends Emitter {
@Override
public void run() {
if (Socket.this.readyState == ReadyState.OPENING || Socket.this.readyState == ReadyState.OPEN) {
Socket.this.onClose("forced close");
logger.fine("socket closing - telling transport to close");
Socket.this.transport.close();
}
Socket.this.readyState = ReadyState.CLOSING;
final Socket self = Socket.this;
final Runnable close = new Runnable() {
@Override
public void run() {
self.onClose("forced close");
logger.fine("socket closing - telling transport to close");
self.transport.close();
}
};
final Listener[] cleanupAndClose = new Listener[1];
cleanupAndClose[0] = new Listener() {
@Override
public void call(Object ...args) {
self.off(EVENT_UPGRADE, cleanupAndClose[0]);
self.off(EVENT_UPGRADE_ERROR, cleanupAndClose[0]);
close.run();
}
};
final Runnable waitForUpgrade = new Runnable() {
@Override
public void run() {
// wait for updade to finish since we can't send packets while pausing a transport
self.once(EVENT_UPGRADE, cleanupAndClose[0]);
self.once(EVENT_UPGRADE_ERROR, cleanupAndClose[0]);
}
};
if (Socket.this.writeBuffer.size() > 0) {
Socket.this.once(EVENT_DRAIN, new Listener() {
@Override
public void call(Object... args) {
if (Socket.this.upgrading) {
waitForUpgrade.run();
} else {
close.run();
}
}
});
} else if (Socket.this.upgrading) {
waitForUpgrade.run();
} else {
close.run();
}
}
}
});
return this;
@@ -710,7 +757,7 @@ public class Socket extends Emitter {
}
private void onClose(String reason, Exception desc) {
if (this.readyState == ReadyState.OPENING || this.readyState == ReadyState.OPEN) {
if (ReadyState.OPENING == this.readyState || ReadyState.OPEN == this.readyState || ReadyState.CLOSING == this.readyState) {
logger.fine(String.format("socket close with reason: %s", reason));
final Socket self = this;