BAEL-5084 raw-websocket testing for postman created
This commit is contained in:
parent
c5bada9f8b
commit
69c94f7850
|
@ -0,0 +1,23 @@
|
|||
package com.baeldung.raw_websocket;
|
||||
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.web.socket.WebSocketHandler;
|
||||
import org.springframework.web.socket.config.annotation.EnableWebSocket;
|
||||
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
|
||||
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
|
||||
|
||||
@Configuration
|
||||
@EnableWebSocket
|
||||
public class ServerWebSocketConfig implements WebSocketConfigurer {
|
||||
|
||||
@Override
|
||||
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
|
||||
registry.addHandler(webSocketHandler(), "/websocket");
|
||||
}
|
||||
|
||||
@Bean
|
||||
public WebSocketHandler webSocketHandler() {
|
||||
return new ServerWebSocketHandler();
|
||||
}
|
||||
}
|
|
@ -0,0 +1,72 @@
|
|||
package com.baeldung.raw_websocket;
|
||||
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.scheduling.annotation.Scheduled;
|
||||
import org.springframework.web.socket.CloseStatus;
|
||||
import org.springframework.web.socket.SubProtocolCapable;
|
||||
import org.springframework.web.socket.TextMessage;
|
||||
import org.springframework.web.socket.WebSocketSession;
|
||||
import org.springframework.web.socket.handler.TextWebSocketHandler;
|
||||
import org.springframework.web.util.HtmlUtils;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.time.LocalTime;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.Set;
|
||||
import java.util.concurrent.CopyOnWriteArraySet;
|
||||
|
||||
public class ServerWebSocketHandler extends TextWebSocketHandler implements SubProtocolCapable {
|
||||
|
||||
private static final Logger logger = LoggerFactory.getLogger(ServerWebSocketHandler.class);
|
||||
|
||||
private final Set<WebSocketSession> sessions = new CopyOnWriteArraySet<>();
|
||||
|
||||
@Override
|
||||
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
|
||||
logger.info("Server connection opened");
|
||||
sessions.add(session);
|
||||
|
||||
TextMessage message = new TextMessage("one-time message from server");
|
||||
logger.info("Server sends: {}", message);
|
||||
session.sendMessage(message);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) {
|
||||
logger.info("Server connection closed: {}", status);
|
||||
sessions.remove(session);
|
||||
}
|
||||
|
||||
@Scheduled(fixedRate = 10000)
|
||||
void sendPeriodicMessages() throws IOException {
|
||||
for (WebSocketSession session : sessions) {
|
||||
if (session.isOpen()) {
|
||||
String broadcast = "server periodic message " + LocalTime.now();
|
||||
logger.info("Server sends: {}", broadcast);
|
||||
session.sendMessage(new TextMessage(broadcast));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
|
||||
String request = message.getPayload();
|
||||
logger.info("Server received: {}", request);
|
||||
|
||||
String response = String.format("response from server to '%s'", HtmlUtils.htmlEscape(request));
|
||||
logger.info("Server sends: {}", response);
|
||||
session.sendMessage(new TextMessage(response));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void handleTransportError(WebSocketSession session, Throwable exception) {
|
||||
logger.info("Server transport error: {}", exception.getMessage());
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<String> getSubProtocols() {
|
||||
return Collections.singletonList("subprotocol.demo.websocket");
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue