Spring Boot에서 WebSocket을 사용하여 푸시 및 알림 기능 구현-java지도 시간-php.cn

Spring Boot에서 WebSocket을 사용하여 푸시 및 알림 기능 구현

PHPz

풀어 주다： 2023-06-23 11:47:50

원래의

1707명이 탐색했습니다.

현대 웹 애플리케이션 개발에서 WebSocket은 즉각적인 통신과 실시간 데이터 전송을 위한 일반적인 기술입니다. Spring Boot 프레임워크는 통합 WebSocket에 대한 지원을 제공하므로 개발자가 푸시 및 알림 기능을 구현하는 것이 매우 편리합니다.

이 기사에서는 WebSocket을 사용하여 Spring Boot에서 푸시 및 알림 기능을 구현하는 방법을 소개하고 간단한 실시간 온라인 채팅방 구현을 시연합니다.

Create Spring Boot 프로젝트

먼저 Spring Boot 프로젝트를 생성해야 합니다. Spring Initializr 웹 사이트에서 웹 및 WebSocket 종속성을 사용하여 새 프로젝트를 빠르게 생성할 수 있습니다. 코드는 다음과 같습니다.

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-websocket</artifactId>
    </dependency>
</dependencies>

로그인 후 복사

WebSocket 구성

다음으로 WebSocket을 Spring Boot 애플리케이션에서 실행할 수 있도록 구성해야 합니다.

먼저 애플리케이션의 구성 파일에 다음 속성을 추가해야 합니다.

spring.websocket.enabled=true

로그인 후 복사

다음으로 WebSocket 지원을 활성화하려면 Spring Boot 구성 클래스에 @EnableWebSocket 주석을 추가하세요. 동시에 WebSocketConfigurer 인터페이스를 구현하고 여기에 핸들러와 메시지 인터셉터를 등록하여 WebSocket 요청과 메시지를 처리해야 합니다. @EnableWebSocket注解，启用WebSocket支持。同时，我们需要实现一个WebSocketConfigurer接口，在其中注册处理程序和消息拦截器，以便处理WebSocket请求和消息。

代码如下：

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new WebSocketHandler(), "/chat").setAllowedOrigins("*").withSockJS();
    }
}

로그인 후 복사

在上面的代码中，我们注册了一个处理程序（即WebSocketHandler），该处理程序在客户端连接到“/ chat”端点时被调用。使用setAllowedOrigins方法来指定允许的来源，以便进行跨域请求，使用withSockJS方法以启用SockJS支持，以便与不支持WebSocket的浏览器兼容。

编写WebSocket处理程序

现在我们需要编写处理程序类，这个类会处理所有的WebSocket请求和消息。

在Spring Boot中，这个类只需要实现WebSocketHandler接口即可。我们将使用SimpeTextWebSocketHandler类，该类提供了处理WebSocket消息的基本功能，并且我们可以在其基础上扩展自己的处理程序。

代码如下：

public class WebSocketHandler extends TextWebSocketHandler {

    private final List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        sessions.add(session);
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        sessions.remove(session);
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        for (WebSocketSession s : sessions) {
            if (s.isOpen() && !s.equals(session)) {
                s.sendMessage(message);
            }
        }
    }
}

로그인 후 복사

在上面的代码中，afterConnectionEstablished方法在新的WebSocket会话建立时被调用，afterConnectionClosed方法在WebSocket会话关闭时被调用。handleTextMessage方法处理所有的WebSocket消息，并发送给所有当前连接的客户端。

创建WebSocket客户端

现在，我们需要创建一个WebSocket客户端来测试实现的推送和通知功能。可以使用JavaScript的WebSocket API来创建一个WebSocket客户端。

代码如下：

const socket = new WebSocket('ws://localhost:8080/chat');

socket.onopen = function() {
  console.log('连接已建立');
};

socket.onmessage = function(event) {
  console.log('收到消息:', event.data);
};

socket.onerror = function(error) {
  console.log('发生错误:', error);
};

socket.onclose = function() {
  console.log('连接已关闭');
};

로그인 후 복사

在上面的代码中，我们创建了一个WebSocket实例并尝试连接到ws：// localhost：8080 / chat。然后，我们监听WebSocket事件以获取打开，关闭，错误和消息事件。

运行项目

现在，我们可以启动Spring Boot应用并测试推送和通知功能。我们可以使用两个或多个WebSocket客户端来模拟不同的用户，并在一个客户端输入消息并将其广播到所有会话中的其他客户端。

使用Maven运行spring-boot：run

코드는 다음과 같습니다.

rrreee

위 코드에서는 클라이언트가 "/chat" 엔드포인트에 연결할 때 호출되는 핸들러(예: WebSocketHandler)를 등록했습니다. 교차 출처 요청에 허용되는 출처를 지정하려면 setAllowedOrigins 메서드를 사용하고, WebSocket을 지원하지 않는 브라우저와의 호환성을 위해 SockJS 지원을 활성화하려면 withSockJS 메서드를 사용하세요.