com.crawljax.web.LogWebSocketServlet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of crawljax-web Show documentation

The newest version!

package com.crawljax.web;

import javax.inject.Inject;
import javax.inject.Provider;

import java.util.Set;
import java.util.concurrent.CopyOnWriteArraySet;

import com.google.inject.Singleton;
import org.eclipse.jetty.websocket.servlet.ServletUpgradeRequest;
import org.eclipse.jetty.websocket.servlet.ServletUpgradeResponse;
import org.eclipse.jetty.websocket.servlet.WebSocketCreator;
import org.eclipse.jetty.websocket.servlet.WebSocketServlet;
import org.eclipse.jetty.websocket.servlet.WebSocketServletFactory;

/**
 * Socket that serves log entries.
 */
@Singleton
public class LogWebSocketServlet extends WebSocketServlet {

	public static final Set sockets = new CopyOnWriteArraySet();
	private static final long serialVersionUID = 4421543809294793344L;
	private final Provider logSocketProvider;

	@Inject
	public LogWebSocketServlet(Provider logSocketProvider) {
		this.logSocketProvider = logSocketProvider;
	}

	@Override
	public void configure(WebSocketServletFactory factory) {
		factory.setCreator(new WebSocketCreator() {

			@Override
			public Object createWebSocket(ServletUpgradeRequest servletUpgradeRequest,
			                              ServletUpgradeResponse servletUpgradeResponse) {
				return logSocketProvider.get();
			}
		});
	}

	public static void sendToAll(String text) {
		if (!sockets.isEmpty())
			for (LoggingSocket socket : sockets) {
				socket.sendText(text);
			}
	}
}