ameba.container.grizzly.server.http.websocket.WebSocketAddOn Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ameba-container-grizzly Show documentation

A useful Java framework grizzly container!

There is a newer version: 0.1.12

Show newest version

package ameba.container.grizzly.server.http.websocket;

import ameba.container.Container;
import org.glassfish.grizzly.filterchain.FilterChainBuilder;
import org.glassfish.grizzly.http.server.AddOn;
import org.glassfish.grizzly.http.server.HttpServerFilter;
import org.glassfish.grizzly.http.server.NetworkListener;

/**
 * @author icode
 */
public class WebSocketAddOn implements AddOn {

    private final Container.WebSocketContainerProvider webSocketContainerProvider;
    private final String contextPath;

    public WebSocketAddOn(Container.WebSocketContainerProvider webSocketContainerProvider, String contextPath) {
        this.webSocketContainerProvider = webSocketContainerProvider;
        this.contextPath = contextPath;
    }

    @Override
    public void setup(NetworkListener networkListener, FilterChainBuilder builder) {
        // Get the index of HttpServerFilter in the HttpServer filter chain
        final int httpServerFilterIdx = builder.indexOfType(HttpServerFilter.class);

        if (httpServerFilterIdx >= 0) {
            // Insert the WebSocketFilter right before HttpServerFilter
            builder.add(httpServerFilterIdx, new GrizzlyServerFilter(webSocketContainerProvider, contextPath));
        }
    }
}