springboot整合websocket实现客户端与服务端通信

springboot整合websocket实现客户端与服务端通信

weixin_30492047 2019-05-06 17:15:00  712  收藏

文章标签： java 网络 后端

版权

工程师小C的小店我也想开通小店

Python编程三剑客：Python编程从入门到实践第2版+快速上手第2版+极客编程（套装共3册）

作者：[美] 埃里克·马瑟斯（Eric Matthes）出版社：人民邮电出版社

好评：100.0% 销售量：20￥149

更多

定义

 WebSocket是通过单个TCP连接提供全双工（双向通信）通信信道的计算机通信协议。此WebSocket API可在用户的浏览器和服务器之间进行双向通信。用户可以向服务器发送消息并接收事件驱动的响应，而无需轮询服务器。 它可以让多个用户连接到同一个实时服务器，并通过API进行通信并立即获得响应。

案例介绍

  后端在接收到用户新下的订单后，通知到后台系统

服务端代码

pom.xml

            org.springframework.boot
            spring-boot-starter-websocket

websocket配置类

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;
 
/**
 * websocket配置
 * @author : wang zns
 * @date : 2019-05-06
 */
@Configuration
public class WebSocketConfig {
 
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }
 
}

websocket核心业务类

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
 
import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.concurrent.CopyOnWriteArraySet;
import java.util.concurrent.atomic.LongAdder;
 
/**
 * websocket服务端核心类
 * @author : wang zns
 * @date : 2019-05-06
 */
@ServerEndpoint("/websocket")
@Component
@Slf4j
public class WebSocketService {
 
    /**
     * 记录当前websocket的连接数（保证线程安全）
     */
    private static LongAdder connectAccount = new LongAdder();
 
    /**
     *存放每个客户端对应的websocketServer对象(需保证线程安全)
     */
    private static CopyOnWriteArraySet webSocketSet = new CopyOnWriteArraySet<>();
    /**
     * 与客户端的连接对象
     */
    private Session session;
 
    /**
     * 连接成功调用的方法
     * @param session
     */
    @OnOpen
    public void onOpen(Session session) {
        this.session = session;
        webSocketSet.add(this);
        connectAccount.increment();
        log.info("有新的连接接入，当前连接数为{}", connectAccount);
    }
 
    /**
     * 连接关闭时调用
     */
    @OnClose
    public void onClose() {
        webSocketSet.remove(this);
        connectAccount.decrement();
        log.info("有连接关闭，当前连接数为{}", connectAccount);
    }
 
    /**
     * 收到客户端消息时调用
     * @param message
     */
    @OnMessage
    public void onMessage(String message) {
        log.info("收到客户端发来的消息,message -> {}", message);
    }
 
    /**
     * 服务端向客户端发送消息
     * @param message
     */
    public void sendMessage(String message) {
        for (WebSocketService webSocketService : webSocketSet) {
            try {
                log.info("【websocket消息】 广播消息, message={}", message);
                webSocketService.session.getBasicRemote().sendText(message);
            } catch (IOException e) {
                log.info("向客户端发送消息失败, {}", e.getMessage());
            }
        }
    }
 
}

客户端代码

在此案例中，我们的客户端为浏览器,所以直接通过js进行连接

11

效果演示

服务端

 

客户端

 

写在最后

  websocket的作用远不止这些，本文只是一种较为常见的用法; 在最后附上参考的blog -> Spring Boot整合websocket实现群聊，点对点聊天

转载于:https://www.cnblogs.com/devise/p/10820810.html