Search in sources :

Example 1 with WebSocketUpgradeRequestFuture

use of com.generallycloud.baseio.codec.http11.future.WebSocketUpgradeRequestFuture in project baseio by generallycloud.

the class TestSimpleWebSocketClient method main.

public static void main(String[] args) throws Exception {
    IoEventHandleAdaptor eventHandleAdaptor = new IoEventHandleAdaptor() {

        @Override
        public void accept(SocketSession session, Future future) throws Exception {
            if (future instanceof ClientHttpFuture) {
                ClientHttpFuture f = (ClientHttpFuture) future;
                if (f.getRequestHeader("Sec-WebSocket-Accept") != null) {
                    f.updateWebSocketProtocol();
                    WebSocketFuture f2 = new WebSocketFutureImpl(session.getContext());
                    f2.write("{action: \"add-user\", username: \"火星人\"}");
                    // f2.write("{\"action\":999}");
                    session.flush(f2);
                }
                System.out.println(f.getRequestHeaders());
            } else {
                WebSocketFuture f = (WebSocketFuture) future;
                System.out.println(f.getReadText());
            }
        }
    };
    ServerConfiguration configuration = new ServerConfiguration();
    configuration.setSERVER_HOST("47.89.30.77");
    // configuration.setSERVER_HOST("120.76.222.210");
    // configuration.setSERVER_HOST("115.29.193.48");
    // configuration.setSERVER_HOST("workerman.net");
    configuration.setSERVER_PORT(7680);
    // configuration.setSERVER_PORT(30005);
    // configuration.setSERVER_PORT(29000);
    // configuration.setSERVER_PORT(8280);
    SocketChannelContext context = new NioSocketChannelContext(configuration);
    SocketChannelConnector connector = new SocketChannelConnector(context);
    context.setIoEventHandleAdaptor(eventHandleAdaptor);
    context.setProtocolFactory(new ProtobaseProtocolFactory());
    context.addSessionEventListener(new LoggerSocketSEListener());
    connector.getContext().setBeatFutureFactory(new WebSocketBeatFutureFactory());
    connector.getContext().setProtocolFactory(new ClientHTTPProtocolFactory());
    connector.getContext().setSslContext(SSLUtil.initClient(true));
    SocketSession session = connector.connect();
    String url = "/web-socket-chat";
    url = "/c1020";
    HttpFuture future = new WebSocketUpgradeRequestFuture(session.getContext(), url);
    // future.setRequestURL("ws://120.76.222.210:30005/");
    // future.setResponseHeader("Host", "120.76.222.210:30005");
    // future.setResponseHeader("Pragma", "no-cache");
    // future.setResponseHeader("Cache-Control", "no-cache");
    // future.setResponseHeader("User-Agent",
    // "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/53.0.2785.143 Safari/537.36");
    // future.setResponseHeader("Accept-Encoding", "gzip, deflate, sdch");
    // future.setResponseHeader("Accept-Language", "zh-CN,zh;q=0.8");
    // future.setRequestHeader("", "");
    session.flush(future);
    // ThreadUtil.sleep(1000);
    // WebSocketReadFuture f2 = new WebSocketReadFutureImpl();
    // f2.write("test");
    // session.flush(f2);
    ThreadUtil.sleep(999999999);
    CloseUtil.close(connector);
}
Also used : SocketChannelConnector(com.generallycloud.baseio.connector.SocketChannelConnector) LoggerSocketSEListener(com.generallycloud.baseio.component.LoggerSocketSEListener) ServerConfiguration(com.generallycloud.baseio.configuration.ServerConfiguration) WebSocketFutureImpl(com.generallycloud.baseio.codec.http11.future.WebSocketFutureImpl) WebSocketUpgradeRequestFuture(com.generallycloud.baseio.codec.http11.future.WebSocketUpgradeRequestFuture) WebSocketFuture(com.generallycloud.baseio.codec.http11.future.WebSocketFuture) ClientHTTPProtocolFactory(com.generallycloud.baseio.codec.http11.ClientHTTPProtocolFactory) NioSocketChannelContext(com.generallycloud.baseio.component.NioSocketChannelContext) SocketChannelContext(com.generallycloud.baseio.component.SocketChannelContext) ClientHttpFuture(com.generallycloud.baseio.codec.http11.future.ClientHttpFuture) HttpFuture(com.generallycloud.baseio.codec.http11.future.HttpFuture) NioSocketChannelContext(com.generallycloud.baseio.component.NioSocketChannelContext) ProtobaseProtocolFactory(com.generallycloud.baseio.codec.protobase.ProtobaseProtocolFactory) ClientHttpFuture(com.generallycloud.baseio.codec.http11.future.ClientHttpFuture) SocketSession(com.generallycloud.baseio.component.SocketSession) IoEventHandleAdaptor(com.generallycloud.baseio.component.IoEventHandleAdaptor) Future(com.generallycloud.baseio.protocol.Future) WebSocketUpgradeRequestFuture(com.generallycloud.baseio.codec.http11.future.WebSocketUpgradeRequestFuture) ClientHttpFuture(com.generallycloud.baseio.codec.http11.future.ClientHttpFuture) HttpFuture(com.generallycloud.baseio.codec.http11.future.HttpFuture) WebSocketFuture(com.generallycloud.baseio.codec.http11.future.WebSocketFuture) WebSocketBeatFutureFactory(com.generallycloud.baseio.codec.http11.future.WebSocketBeatFutureFactory)

Aggregations

ClientHTTPProtocolFactory (com.generallycloud.baseio.codec.http11.ClientHTTPProtocolFactory)1 ClientHttpFuture (com.generallycloud.baseio.codec.http11.future.ClientHttpFuture)1 HttpFuture (com.generallycloud.baseio.codec.http11.future.HttpFuture)1 WebSocketBeatFutureFactory (com.generallycloud.baseio.codec.http11.future.WebSocketBeatFutureFactory)1 WebSocketFuture (com.generallycloud.baseio.codec.http11.future.WebSocketFuture)1 WebSocketFutureImpl (com.generallycloud.baseio.codec.http11.future.WebSocketFutureImpl)1 WebSocketUpgradeRequestFuture (com.generallycloud.baseio.codec.http11.future.WebSocketUpgradeRequestFuture)1 ProtobaseProtocolFactory (com.generallycloud.baseio.codec.protobase.ProtobaseProtocolFactory)1 IoEventHandleAdaptor (com.generallycloud.baseio.component.IoEventHandleAdaptor)1 LoggerSocketSEListener (com.generallycloud.baseio.component.LoggerSocketSEListener)1 NioSocketChannelContext (com.generallycloud.baseio.component.NioSocketChannelContext)1 SocketChannelContext (com.generallycloud.baseio.component.SocketChannelContext)1 SocketSession (com.generallycloud.baseio.component.SocketSession)1 ServerConfiguration (com.generallycloud.baseio.configuration.ServerConfiguration)1 SocketChannelConnector (com.generallycloud.baseio.connector.SocketChannelConnector)1 Future (com.generallycloud.baseio.protocol.Future)1