WebSocket消息推送

websocket握手过程(建立连接的过程)

使用网页端尝试和服务器建立websocket连接。 网页端会先给服务器发起—个HTTP请求，这个HTTP请求中会带有特殊的header。

Connection: Upgrade

Upgrade: Websocket

这两个header其实就是在告知服务器，我们要进行协议升级。 如果服务器支持websocket，就会返回一个特殊的HTTP响应，这个响应的状态码是101(切换协议)。 客户端和服务器之间就开始使用websocket来进行通信了。

websocket代码实例

 后端

1.引入依赖WebSocket

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

2.继承父类TextWebSocketHandler，重写API代码

import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

//将该类注入到spring
@Component
public class WebsocketTest extends TextWebSocketHandler {
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        System.out.println("连接建立成功");
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        //message.getPayload得到数据载荷
        System.out.println("收到的消息：" + message.getPayload());

        //服务器将消息返回给客户端
        session.sendMessage(message);
    }

    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        System.out.println("连接异常");
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        System.out.println("连接关闭");
    }
}

3.注册

将websocketTest进行注册与路径"/websocket"进行关联，最上面加入@EnableWebSocket注解，开启WebSocket服务。

import com.example.gobang.TestDemo.WebsocketTest;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

import javax.annotation.Resource;

@Configuration
@EnableWebSocket//开启WebSocket
public class RegWebSocket implements WebSocketConfigurer {

    @Resource
    private WebsocketTest websocketTest;

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        //将WebsocketTest类进行注册与路径"/websocket"进行关联
        registry.addHandler(websocketTest,"/websocket");
    }
}

前端

    提交
    

4.结果

 当点击提交时，客户端将“你好WebSocket”发给服务器，服务器handleTextMessage函数会收到message消息，然后将消息再推送给客户端，客户端触发实例WebSocket的onmessage函数从而接收服务器发送的消息