mangoo.io.routing.handlers.WebSocketHandler Maven / Gradle / Ivy

Go to download
package mangoo.io.routing.handlers;

import io.undertow.websockets.WebSocketConnectionCallback;
import io.undertow.websockets.core.WebSocketChannel;
import io.undertow.websockets.spi.WebSocketHttpExchange;
import mangoo.io.core.Application;

import org.xnio.ChannelListener;

/**
 *
 * @author svenkubiak
 *
 */
@SuppressWarnings("unchecked")
public class WebSocketHandler implements WebSocketConnectionCallback {
    private Class controllerClass;

    public WebSocketHandler(Class controllerClass) {
        this.controllerClass = controllerClass;
    }

    @Override
    public void onConnect(WebSocketHttpExchange exchange, WebSocketChannel channel) {
        channel.getReceiveSetter().set((ChannelListener) Application.getInjector().getInstance(this.controllerClass));
        channel.resumeReceives();
    }
}