Examples with Session org.eclipse.jetty.websocket.api.Session used on opensource projects

Example 66 with Session

use of org.eclipse.jetty.websocket.api.Session in project jetty.project by eclipse.

the class WebSocketScopeSessionTest method testMultiSession_Sequential.

@Test
public void testMultiSession_Sequential() throws Exception {
    ScopedInstance<WebSocketScopeContext> wsScope1Bean = newInstance(WebSocketScopeContext.class);
    WebSocketScopeContext wsScope1 = wsScope1Bean.instance;
    ScopedInstance<WebSocketScopeContext> wsScope2Bean = newInstance(WebSocketScopeContext.class);
    WebSocketScopeContext wsScope2 = wsScope2Bean.instance;
    wsScope1.create();
    try {
        // Scope 1
        wsScope1.begin();
        BogusSession sess = new BogusSession("1");
        wsScope1.setSession(sess);
        ScopedInstance<BogusSocket> sock1Bean = newInstance(BogusSocket.class);
        BogusSocket sock1 = sock1Bean.instance;
        assertThat("Socket 1 Session", sock1.getSession(), sameInstance((Session) sess));
        sock1Bean.destroy();
    } finally {
        wsScope1.end();
    }
    wsScope1.destroy();
    wsScope1Bean.destroy();
    wsScope2.create();
    try {
        // Scope 2
        wsScope2.begin();
        BogusSession sess = new BogusSession("2");
        wsScope2.setSession(sess);
        ScopedInstance<BogusSocket> sock2Bean = newInstance(BogusSocket.class);
        BogusSocket sock2 = sock2Bean.instance;
        assertThat("Socket 2 Session", sock2.getSession(), sameInstance((Session) sess));
        sock2Bean.destroy();
    } finally {
        wsScope2.end();
    }
    wsScope2.destroy();
    wsScope2Bean.destroy();
}

Example 67 with Session

use of org.eclipse.jetty.websocket.api.Session in project spring-framework by spring-projects.

the class JettyWebSocketClient method doHandshakeInternal.

@Override
public ListenableFuture<WebSocketSession> doHandshakeInternal(WebSocketHandler wsHandler, HttpHeaders headers, final URI uri, List<String> protocols, List<WebSocketExtension> extensions, Map<String, Object> attributes) {
    final ClientUpgradeRequest request = new ClientUpgradeRequest();
    request.setSubProtocols(protocols);
    for (WebSocketExtension e : extensions) {
        request.addExtensions(new WebSocketToJettyExtensionConfigAdapter(e));
    }
    for (String header : headers.keySet()) {
        request.setHeader(header, headers.get(header));
    }
    Principal user = getUser();
    final JettyWebSocketSession wsSession = new JettyWebSocketSession(attributes, user);
    final JettyWebSocketHandlerAdapter listener = new JettyWebSocketHandlerAdapter(wsHandler, wsSession);
    Callable<WebSocketSession> connectTask = new Callable<WebSocketSession>() {

        @Override
        public WebSocketSession call() throws Exception {
            Future<Session> future = client.connect(listener, uri, request);
            future.get();
            return wsSession;
        }
    };
    if (this.taskExecutor != null) {
        return this.taskExecutor.submitListenable(connectTask);
    } else {
        ListenableFutureTask<WebSocketSession> task = new ListenableFutureTask<>(connectTask);
        task.run();
        return task;
    }
}

Example 68 with Session

use of org.eclipse.jetty.websocket.api.Session in project spring-framework by spring-projects.

the class JettyWebSocketSessionTests method getPrincipalNotAvailable.

@Test
@SuppressWarnings("resource")
public void getPrincipalNotAvailable() {
    UpgradeRequest request = Mockito.mock(UpgradeRequest.class);
    given(request.getUserPrincipal()).willReturn(null);
    UpgradeResponse response = Mockito.mock(UpgradeResponse.class);
    given(response.getAcceptedSubProtocol()).willReturn(null);
    Session nativeSession = Mockito.mock(Session.class);
    given(nativeSession.getUpgradeRequest()).willReturn(request);
    given(nativeSession.getUpgradeResponse()).willReturn(response);
    JettyWebSocketSession session = new JettyWebSocketSession(attributes);
    session.initializeNativeSession(nativeSession);
    reset(nativeSession);
    assertNull(session.getPrincipal());
    verifyNoMoreInteractions(nativeSession);
}

Example 69 with Session

use of org.eclipse.jetty.websocket.api.Session in project spring-framework by spring-projects.

the class JettyWebSocketSessionTests method getPrincipalFromNativeSession.

@Test
@SuppressWarnings("resource")
public void getPrincipalFromNativeSession() {
    TestPrincipal user = new TestPrincipal("joe");
    UpgradeRequest request = Mockito.mock(UpgradeRequest.class);
    given(request.getUserPrincipal()).willReturn(user);
    UpgradeResponse response = Mockito.mock(UpgradeResponse.class);
    given(response.getAcceptedSubProtocol()).willReturn(null);
    Session nativeSession = Mockito.mock(Session.class);
    given(nativeSession.getUpgradeRequest()).willReturn(request);
    given(nativeSession.getUpgradeResponse()).willReturn(response);
    JettyWebSocketSession session = new JettyWebSocketSession(attributes);
    session.initializeNativeSession(nativeSession);
    reset(nativeSession);
    assertSame(user, session.getPrincipal());
    verifyNoMoreInteractions(nativeSession);
}

Example 70 with Session

use of org.eclipse.jetty.websocket.api.Session in project pulsar by yahoo.

the class ProxyAuthenticationTest method socketTest.

@Test
public void socketTest() throws InterruptedException {
    URI consumeUri = URI.create(CONSUME_URI);
    URI produceUri = URI.create(PRODUCE_URI);
    WebSocketClient consumeClient = new WebSocketClient();
    SimpleConsumerSocket consumeSocket = new SimpleConsumerSocket();
    WebSocketClient produceClient = new WebSocketClient();
    SimpleProducerSocket produceSocket = new SimpleProducerSocket();
    try {
        consumeClient.start();
        ClientUpgradeRequest consumeRequest = new ClientUpgradeRequest();
        Future<Session> consumerFuture = consumeClient.connect(consumeSocket, consumeUri, consumeRequest);
        log.info("Connecting to : {}", consumeUri);
        ClientUpgradeRequest produceRequest = new ClientUpgradeRequest();
        produceClient.start();
        Future<Session> producerFuture = produceClient.connect(produceSocket, produceUri, produceRequest);
        // let it connect
        Thread.sleep(1000);
        Assert.assertTrue(consumerFuture.get().isOpen());
        Assert.assertTrue(producerFuture.get().isOpen());
        consumeSocket.awaitClose(1, TimeUnit.SECONDS);
        produceSocket.awaitClose(1, TimeUnit.SECONDS);
        Assert.assertTrue(produceSocket.getBuffer().size() > 0);
        Assert.assertEquals(produceSocket.getBuffer(), consumeSocket.getBuffer());
    } catch (Throwable t) {
        log.error(t.getMessage());
    } finally {
        ExecutorService executor = newFixedThreadPool(1);
        try {
            executor.submit(() -> {
                try {
                    consumeClient.stop();
                    produceClient.stop();
                    log.info("proxy clients are stopped successfully");
                } catch (Exception e) {
                    log.error(e.getMessage());
                }
            }).get(2, TimeUnit.SECONDS);
        } catch (Exception e) {
            log.error("failed to close clients ", e);
        }
        executor.shutdownNow();
    }
}