花了将近一周的时间终于完成了利用WebSocket完成网页版聊天室这个小demo,期间还走过了一段“看似弯曲”的道路,但是我想其实也不算是弯路吧,因为你走过的路必将留下你的足迹。这个小demo看似简单,但是这一路走来自己也是花了不少心思才将其完成,今天就和大家一起分享一下。 刚刚接手这个任务时,我就想能不能使用Node.js来编写聊天室的服务器端,而后才发现这样做却存在“跨域问题”,因为你的前台是部署在Tomcat服务器上的,所以前台和后台的数据无法进行直接的交互,虽然可以间接的进行交互但是却不容易,在此不建议大家采用此种方法,可直接采用Java+html实现网页版聊天室。我将首先介绍客户端的构造而后介绍服务器端: 客户端
【1】 页面的编写(使用HTML编写)
<body> <div> <center><h1>欢迎光临聊天室 </h1></center> <div></div> 昵称: <input type="text"/> 内容: <input type="text"/> <button>发送</button> <button>退出聊天室</button> </div> </body>说明:
在此我首先放置了一个ID 名为message的div用来显示存放的消息
然后我放置了两个input标签:一个存放昵称,另一个存放用户要发出去的昵称
最后设置两个按钮用来发送消息和退出聊天
【2】页面的脚本编写
说明:以上几个方法例如onclose,onmessage均是websocket自带的方法,大家只需要在websocket文档上查看各个函数是什么功能。
服务器端 import java.io.IOException; import java.util.concurrent.CopyOnWriteArraySet; import javax.websocket.OnClose; import javax.websocket.OnError; import javax.websocket.OnMessage; import javax.websocket.OnOpen; import javax.websocket.Session; import javax.websocket.server.ServerEndpoint; //该注解用来指定一个URI,客户端可以通过这个URI来连接到WebSocket。类似Servlet的注解mapping。无需在web.xml中配置。 @ServerEndpoint("/websocket") public class MyWebSocket { // 静态变量,用来记录当前在线连接数。应该把它设计成线程安全的。 private static int onlineCount = 0; // concurrent包的线程安全Set,用来存放每个客户端对应的MyWebSocket对象。若要实现服务端与单一客户端通信的话,可以使用Map来存放,其中Key可以为用户标识 private static CopyOnWriteArraySet<MyWebSocket> webSocketSet = new CopyOnWriteArraySet<MyWebSocket>(); // 与某个客户端的连接会话,需要通过它来给客户端发送数据 private Session session; @OnOpen public void onOpen(Session session) { this.session = session; webSocketSet.add(this); // 加入set中 addOnlineCount(); // 在线数加1 System.out.println("有新连接加入!当前在线人数为" + getOnlineCount()); } /** * 连接关闭调用的方法 */ @OnClose public void onClose() { webSocketSet.remove(this); // 从set中删除 subOnlineCount(); // 在线数减1 System.out.println("有一连接关闭!当前在线人数为" + getOnlineCount()); } @OnMessage public void onMessage(String message, Session session) { System.out.println("来自客户端的消息:" + message); // 群发消息 for (MyWebSocket item : webSocketSet) { try { item.sendMessage(message); } catch (IOException e) { e.printStackTrace(); continue; } } } @OnError public void onError(Session session, Throwable error) { System.out.println("发生错误"); error.printStackTrace(); } public void sendMessage(String message) throws IOException { this.session.getBasicRemote().sendText(message); // this.session.getAsyncRemote().sendText(message); } public static synchronized int getOnlineCount() { return onlineCount; } public static synchronized void addOnlineCount() { MyWebSocket.onlineCount++; } public static synchronized void subOnlineCount() { MyWebSocket.onlineCount--; } }