diff options
author | Ashlee Young <ashlee@onosfw.com> | 2015-09-09 22:15:21 -0700 |
---|---|---|
committer | Ashlee Young <ashlee@onosfw.com> | 2015-09-09 22:15:21 -0700 |
commit | 13d05bc8458758ee39cb829098241e89616717ee (patch) | |
tree | 22a4d1ce65f15952f07a3df5af4b462b4697cb3a /framework/src/onos/web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocketServlet.java | |
parent | 6139282e1e93c2322076de4b91b1c85d0bc4a8b3 (diff) |
ONOS checkin based on commit tag e796610b1f721d02f9b0e213cf6f7790c10ecd60
Change-Id: Ife8810491034fe7becdba75dda20de4267bd15cd
Diffstat (limited to 'framework/src/onos/web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocketServlet.java')
-rw-r--r-- | framework/src/onos/web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocketServlet.java | 95 |
1 files changed, 95 insertions, 0 deletions
diff --git a/framework/src/onos/web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocketServlet.java b/framework/src/onos/web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocketServlet.java new file mode 100644 index 00000000..ffc558da --- /dev/null +++ b/framework/src/onos/web/gui/src/main/java/org/onosproject/ui/impl/UiWebSocketServlet.java @@ -0,0 +1,95 @@ +/* + * Copyright 2015 Open Networking Laboratory + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.onosproject.ui.impl; + +import org.eclipse.jetty.websocket.WebSocket; +import org.eclipse.jetty.websocket.WebSocketServlet; +import org.onlab.osgi.DefaultServiceDirectory; +import org.onlab.osgi.ServiceDirectory; + +import javax.servlet.ServletException; +import javax.servlet.http.HttpServletRequest; +import java.util.HashSet; +import java.util.Iterator; +import java.util.Set; +import java.util.Timer; +import java.util.TimerTask; + +/** + * Web socket servlet capable of creating web sockets for the user interface. + */ +public class UiWebSocketServlet extends WebSocketServlet { + + private static final long PING_DELAY_MS = 5000; + + private static UiWebSocketServlet instance; + + private ServiceDirectory directory = new DefaultServiceDirectory(); + + private final Set<UiWebSocket> sockets = new HashSet<>(); + private final Timer timer = new Timer(); + private final TimerTask pruner = new Pruner(); + private boolean isStopped = false; + + /** + * Closes all currently open UI web-sockets. + */ + public static void closeAll() { + if (instance != null) { + instance.isStopped = true; + instance.sockets.forEach(UiWebSocket::close); + instance.sockets.clear(); + instance.pruner.cancel(); + instance.timer.cancel(); + } + } + + @Override + public void init() throws ServletException { + super.init(); + instance = this; + timer.schedule(pruner, PING_DELAY_MS, PING_DELAY_MS); + } + + @Override + public WebSocket doWebSocketConnect(HttpServletRequest request, String protocol) { + if (isStopped) { + return null; + } + UiWebSocket socket = new UiWebSocket(directory); + synchronized (sockets) { + sockets.add(socket); + } + return socket; + } + + // Task for pruning web-sockets that are idle. + private class Pruner extends TimerTask { + @Override + public void run() { + synchronized (sockets) { + Iterator<UiWebSocket> it = sockets.iterator(); + while (it.hasNext()) { + UiWebSocket socket = it.next(); + if (socket.isIdle()) { + it.remove(); + socket.close(); + } + } + } + } + } +} |