net.openhft.chronicle.websocket.jetty.JettyServiceFactory Maven / Gradle / Ivy

Go to download
package net.openhft.chronicle.websocket.jetty;

import org.eclipse.jetty.websocket.servlet.*;

import java.util.function.Function;

/**
 * Created by peter on 23/04/16.
 */
public class JettyServiceFactory extends WebSocketServlet implements WebSocketCreator {
    private final Class responseClass;
    private final Function serviceFactory;

    public JettyServiceFactory(Class responseClass, Function serviceFactory) {
        this.responseClass = responseClass;
        this.serviceFactory = serviceFactory;
    }

    @Override
    public void configure(WebSocketServletFactory factory) {
        factory.setCreator(this);
    }

    @Override
    public Object createWebSocket(ServletUpgradeRequest servletUpgradeRequest, ServletUpgradeResponse servletUpgradeResponse) {
        return new JettyWebSocketServiceAdapter<>(responseClass, serviceFactory);
    }
}