Examples with RequestContext org.glassfish.tyrus.core.RequestContext used on opensource projects

Example 1 with RequestContext

use of org.glassfish.tyrus.core.RequestContext in project spring-framework by spring-projects.

the class AbstractTyrusRequestUpgradeStrategy method upgradeInternal.

@Override
public void upgradeInternal(ServerHttpRequest request, ServerHttpResponse response, @Nullable String selectedProtocol, List<Extension> extensions, Endpoint endpoint) throws HandshakeFailureException {
    HttpServletRequest servletRequest = getHttpServletRequest(request);
    HttpServletResponse servletResponse = getHttpServletResponse(response);
    TyrusServerContainer serverContainer = (TyrusServerContainer) getContainer(servletRequest);
    TyrusWebSocketEngine engine = (TyrusWebSocketEngine) serverContainer.getWebSocketEngine();
    Object tyrusEndpoint = null;
    boolean success;
    try {
        // Shouldn't matter for processing but must be unique
        String path = "/" + random.nextLong();
        tyrusEndpoint = createTyrusEndpoint(endpoint, path, selectedProtocol, extensions, serverContainer, engine);
        register(engine, tyrusEndpoint);
        HttpHeaders headers = request.getHeaders();
        RequestContext requestContext = createRequestContext(servletRequest, path, headers);
        TyrusUpgradeResponse upgradeResponse = new TyrusUpgradeResponse();
        UpgradeInfo upgradeInfo = engine.upgrade(requestContext, upgradeResponse);
        success = SUCCESS.equals(upgradeInfo.getStatus());
        if (success) {
            if (logger.isTraceEnabled()) {
                logger.trace("Successful request upgrade: " + upgradeResponse.getHeaders());
            }
            handleSuccess(servletRequest, servletResponse, upgradeInfo, upgradeResponse);
        }
    } catch (Exception ex) {
        unregisterTyrusEndpoint(engine, tyrusEndpoint);
        throw new HandshakeFailureException("Error during handshake: " + request.getURI(), ex);
    }
    unregisterTyrusEndpoint(engine, tyrusEndpoint);
    if (!success) {
        throw new HandshakeFailureException("Unexpected handshake failure: " + request.getURI());
    }
}

Example 2 with RequestContext

use of org.glassfish.tyrus.core.RequestContext in project spring-framework by spring-projects.

the class AbstractTyrusRequestUpgradeStrategy method createRequestContext.

private RequestContext createRequestContext(HttpServletRequest request, String endpointPath, HttpHeaders headers) {
    RequestContext context = RequestContext.Builder.create().requestURI(URI.create(endpointPath)).userPrincipal(request.getUserPrincipal()).secure(request.isSecure()).remoteAddr(request.getRemoteAddr()).build();
    headers.forEach((header, value) -> context.getHeaders().put(header, value));
    return context;
}