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

Browse Source
This commit is contained in:
Redkale
2016-08-24 13:48:52 +08:00
parent 30b6ddc4f4
commit 5a1586bc6a
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/org/redkale/net/http/WebSocketServlet.java
View File

@@ -13,6 +13,7 @@ import java.security.*;
import java.util.*; import java.util.*;
import java.util.logging.*; import java.util.logging.*;
import javax.annotation.*; import javax.annotation.*;
import org.redkale.service.WebSocketNodeService;
import org.redkale.util.*; import org.redkale.util.*;
/** /**
@@ -67,6 +68,10 @@ public abstract class WebSocketServlet extends HttpServlet {
public final void preInit(HttpContext context, AnyValue conf) { public final void preInit(HttpContext context, AnyValue conf) {
InetSocketAddress addr = context.getServerAddress(); InetSocketAddress addr = context.getServerAddress();
this.engine = new WebSocketEngine(addr.getHostString() + ":" + addr.getPort() + "-[" + name() + "]", this.node, logger); this.engine = new WebSocketEngine(addr.getHostString() + ":" + addr.getPort() + "-[" + name() + "]", this.node, logger);
if (this.node == null) {
this.node = new WebSocketNodeService();
if (logger.isLoggable(Level.INFO)) logger.info("not found WebSocketNode, create a default value for " + getClass().getName());
}
this.node.putWebSocketEngine(engine); this.node.putWebSocketEngine(engine);
this.node.init(conf); this.node.init(conf);
this.engine.init(conf); this.engine.init(conf);
@@ -86,8 +91,8 @@ public abstract class WebSocketServlet extends HttpServlet {
public final void execute(final HttpRequest request, final HttpResponse response) throws IOException { public final void execute(final HttpRequest request, final HttpResponse response) throws IOException {
final boolean debug = logger.isLoggable(Level.FINEST); final boolean debug = logger.isLoggable(Level.FINEST);
if (!"GET".equalsIgnoreCase(request.getMethod()) if (!"GET".equalsIgnoreCase(request.getMethod())
|| !request.getConnection().contains("Upgrade") || !request.getConnection().contains("Upgrade")
|| !"websocket".equalsIgnoreCase(request.getHeader("Upgrade"))) { || !"websocket".equalsIgnoreCase(request.getHeader("Upgrade"))) {
if (debug) logger.finest("WebSocket connect abort, (Not GET Method) or (Connection != Upgrade) or (Upgrade != websocket). request=" + request); if (debug) logger.finest("WebSocket connect abort, (Not GET Method) or (Connection != Upgrade) or (Upgrade != websocket). request=" + request);
response.finish(true); response.finish(true);
return; return;