简易服务端之Servlet实现Websocket消息通信

Posted 2022-11-28 Android_小黑

本文地址：https://blog.csdn.net/qq_40785165/article/details/121622651，转载需附上此链接

坚持就是胜利

大家好，我是小黑，一个还没秃头的程序员~~~

本期写点服务端的内容，介绍一下大学时期经常用到的Servlet是如何实现WebSocket连接与通信的，编写Servlet不需要搭建复杂的框架，沿用大学期间学习的Web项目经验即可，适用于大学生开发毕业设计或者课程作业等场景。

话不多说，直接开始:
条件:

1. 需要Java-WebSocket-1.3.5.jar文件

链接：https://pan.baidu.com/s/1klmTVfjaqOBrCVRqsPbGNQ 
提取码：ijqw

3. 需要Tomcat8.0及以上

代码如下：

package websocket;

import java.io.IOException;
import java.util.concurrent.ConcurrentHashMap;

import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;

@ServerEndpoint(value = "/websocket/name")
public class WebSocket 
	  /**
     * 与某个客户端的连接对话，需要通过它来给客户端发送消息
     */
    private Session session;
    /**
     * 用于存所有的连接服务的客户端，这个对象存储是安全的
     */
    public static ConcurrentHashMap<String, WebSocket> webSocketSet = new ConcurrentHashMap<String, WebSocket>();
    /**
     * 标识当前连接客户端的用户名
     */
    private String name;

    public String getName() 
        return name;
    


    /**
     * @param session
     * @param name    可以试着用uId
     */
    @OnOpen
    public void OnOpen(Session session, @PathParam(value = "name") String name) 
        this.session = session;
        System.out.println(session.getId()+","+name);
        this.name = name; 
        // name是用来表示唯一客户端，如果需要指定发送，需要指定发送通过name来区分
        webSocketSet.put(name, this);
        System.out.println("[WebSocket] 连接成功，当前连接人数为：="+ webSocketSet.size()+name);
    


    @OnClose
    public void OnClose() 
        webSocketSet.remove(this.name);
        System.out.println("[WebSocket] 退出成功，当前连接人数为：="+webSocketSet.size()+webSocketSet.toString());
    

    @OnMessage
    public void OnMessage(String message) 
        System.out.println("[WebSocket] 收到消息："+ message);;
        try 
            sendMessage("接收的消息:"+message);
         catch (IOException e) 
            e.printStackTrace();
        

    

    @OnError
    public void onError(Throwable error) 
        System.out.println("[WebSocket] 发生错误："+ error.toString());
    

    /**
     * 群发
     *
     * @param message
     */
    public void GroupSending(String message) 
        for (String name : webSocketSet.keySet()) 
            try 
                webSocketSet.get(name).session.getBasicRemote().sendText(message);
             catch (Exception e) 
                e.printStackTrace();
            
        
    

    /**
     * 指定发送
     *
     * @param name
     * @param message
     */
    public static void appointSending(String name, String message) 
    	System.out.println(name);
        try 
            webSocketSet.get(name).session.getBasicRemote().sendText(message);
         catch (Exception e) 
            e.printStackTrace();
        
    

    public static ConcurrentHashMap<String, WebSocket> getWebSocketSet() 
        return webSocketSet;
    
    public void sendMessage(String message) throws IOException 
    	System.out.println("消息===="+message);
        this.session.getBasicRemote().sendText(message);
    

由于只是一次开发的记录，就不一一详解了，大家直接拿起来使用就行了，后面有机会我会继续分享关于服务端的开发学习过程的，大家晚安!