All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.springframework.web.reactive.socket.client.StandardWebSocketClient Maven / Gradle / Ivy
/*
* Copyright 2002-2017 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.web.reactive.socket.client;
import java.net.URI;
import java.util.List;
import java.util.Map;
import javax.websocket.ClientEndpointConfig;
import javax.websocket.ClientEndpointConfig.Configurator;
import javax.websocket.ContainerProvider;
import javax.websocket.Endpoint;
import javax.websocket.HandshakeResponse;
import javax.websocket.Session;
import javax.websocket.WebSocketContainer;
import reactor.core.publisher.Mono;
import reactor.core.publisher.MonoProcessor;
import reactor.core.scheduler.Schedulers;
import org.springframework.core.io.buffer.DataBufferFactory;
import org.springframework.core.io.buffer.DefaultDataBufferFactory;
import org.springframework.http.HttpHeaders;
import org.springframework.web.reactive.socket.HandshakeInfo;
import org.springframework.web.reactive.socket.WebSocketHandler;
import org.springframework.web.reactive.socket.adapter.StandardWebSocketHandlerAdapter;
import org.springframework.web.reactive.socket.adapter.StandardWebSocketSession;
/**
* {@link WebSocketClient} implementation for use with the Java WebSocket API.
*
* @author Violeta Georgieva
* @author Rossen Stoyanchev
* @since 5.0
* @see https://www.jcp.org/en/jsr/detail?id=356
*/
public class StandardWebSocketClient extends WebSocketClientSupport implements WebSocketClient {
private final DataBufferFactory bufferFactory = new DefaultDataBufferFactory();
private final WebSocketContainer webSocketContainer;
/**
* Default constructor that calls
* {@code ContainerProvider.getWebSocketContainer()} to obtain a (new)
* {@link WebSocketContainer} instance.
*/
public StandardWebSocketClient() {
this(ContainerProvider.getWebSocketContainer());
}
/**
* Constructor accepting an existing {@link WebSocketContainer} instance.
* @param webSocketContainer a web socket container
*/
public StandardWebSocketClient(WebSocketContainer webSocketContainer) {
this.webSocketContainer = webSocketContainer;
}
/**
* Return the configured {@link WebSocketContainer} to use.
*/
public WebSocketContainer getWebSocketContainer() {
return this.webSocketContainer;
}
@Override
public Mono execute(URI url, WebSocketHandler handler) {
return execute(url, new HttpHeaders(), handler);
}
@Override
public Mono execute(URI url, HttpHeaders headers, WebSocketHandler handler) {
return executeInternal(url, headers, handler);
}
private Mono executeInternal(URI url, HttpHeaders requestHeaders, WebSocketHandler handler) {
MonoProcessor completionMono = MonoProcessor.create();
return Mono.fromCallable(
() -> {
List protocols = beforeHandshake(url, requestHeaders, handler);
DefaultConfigurator configurator = new DefaultConfigurator(requestHeaders);
Endpoint endpoint = createEndpoint(url, handler, completionMono, configurator);
ClientEndpointConfig config = createEndpointConfig(configurator, protocols);
return this.webSocketContainer.connectToServer(endpoint, config, url);
})
.subscribeOn(Schedulers.elastic()) // connectToServer is blocking
.then(completionMono);
}
private StandardWebSocketHandlerAdapter createEndpoint(URI url, WebSocketHandler handler,
MonoProcessor completion, DefaultConfigurator configurator) {
return new StandardWebSocketHandlerAdapter(handler, session -> {
HttpHeaders responseHeaders = configurator.getResponseHeaders();
HandshakeInfo info = afterHandshake(url, responseHeaders);
return createWebSocketSession(session, info, completion);
});
}
protected StandardWebSocketSession createWebSocketSession(Session session, HandshakeInfo info,
MonoProcessor completion) {
return new StandardWebSocketSession(session, info, this.bufferFactory, completion);
}
private ClientEndpointConfig createEndpointConfig(Configurator configurator, List subProtocols) {
return ClientEndpointConfig.Builder.create()
.configurator(configurator)
.preferredSubprotocols(subProtocols)
.build();
}
protected DataBufferFactory bufferFactory() {
return this.bufferFactory;
}
private static final class DefaultConfigurator extends Configurator {
private final HttpHeaders requestHeaders;
private final HttpHeaders responseHeaders = new HttpHeaders();
public DefaultConfigurator(HttpHeaders requestHeaders) {
this.requestHeaders = requestHeaders;
}
public HttpHeaders getResponseHeaders() {
return this.responseHeaders;
}
@Override
public void beforeRequest(Map> requestHeaders) {
requestHeaders.putAll(this.requestHeaders);
}
@Override
public void afterResponse(HandshakeResponse response) {
response.getHeaders().forEach(this.responseHeaders::put);
}
}
}