在java中使用WebSocket实现简易聊天室

WebSocket简介

WebSocket是一种在单个TCP连接上进行全双工通信的协议。使用WebScoket能够替代原始的轮询方式。

使用WebSocket实现简易聊天室

服务端：

pom.xml

 
    
      javax.websocket
      javax.websocket-api
      1.1
      provided
    
  

服务类：

使用@ServerEndpoint注解将一个类声明为webSocket服务器

使用@onOpen注解将一个方法声明为建立连接的回调方法

使用@onClose注解将一个方法声明为断开连接的回调方法

使用@onMessage注解将一个方法声明为收到消息的回调方法

使用@onError注解将一个方法声明为出现异常的回调方法

使用session.getBasicRemote().sendText()方法向指定连接发送字符消息

package com.llg;


import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.HashSet;
import java.util.Set;

@ServerEndpoint(value = "/MyServer")
public class WebSocketServer {

    //统计连接人数
    private static volatile int total = 0;
    //记录每个连接的session，以便发送消息
    private static Set sessions = new HashSet<>();

    //建立连接回调方法
    @OnOpen
    public void onOpen(Session session){
        total++;
        sessions.add(session);
        sendMessage("有人加入了聊天室,当前人数"+ total);
    }

    //连接断开回调
    @OnClose
    public void onClose(Session session){
        //移除已断开的连接
        total --;
        sessions.remove(session);
        sendMessage("有人离开了聊天室,当前人数"+ total);
    }

    //收到消息时回调
    @OnMessage
    public void onMessage(Session session,String message){
        //将消息广播给所有人
        sendMessage(message);
    }


    public void sendMessage(String message){
        //循环给每个连接的客户端发送消息
        for (Session session : sessions) {
            try {
                session.getBasicRemote().sendText(message);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

    }

}

页面：

    
    
    






发送