利用websocket实现群聊以及单聊
- 项目结构
- 实现代码
- 运行截图
在这里提供一下思路，正常情况下我们登陆进去之后就应该打开一个ws连接，以便和服务器进行通信，将打开的管道用一个set容器进行存储，并将用户名或者其他能唯一标示用户的字段作为key，把与之对应的管道作为value存储到一个map集合中。
那么怎么实现广播的方法呢？其实很简单，这需要对set容器中的每一个管道进行遍历并执sendText()方法，就可以将消息传递给了集合中每个管道（前提是在每个用户进行登陆时将用户存储到一个集合中）。
接下来首先实现群聊，通过jquery获得输入框的内容，然后通过onmessage（）方法向后端传递消息，在调用广播方法就可以实现群聊了。
接下来就是单聊，单聊的实现原理就是得到被单聊的人的用户名，然后在map集合中的得到与之对应的管道，然后调用sendText（）方法就可以将消息发送给对应的人了。
话不多说，直接上图上代码
项目结构
实现代码
DemoConfig.java
import java.util.Set;import javax.websocket.Endpoint;
import javax.websocket.server.ServerApplicationConfig;
import javax.websocket.server.ServerEndpointConfig;public class DemoConfig implements ServerApplicationConfig {@Overridepublic Set<Class<?>> getAnnotatedEndpointClasses(Set<Class<?>> scan) {// TODO Auto-generated method stubSystem.out.println("config................"+scan.size());//可以提供过滤作用return scan;}@Overridepublic Set<ServerEndpointConfig> getEndpointConfigs(Set<Class<? extends Endpoint>> arg0) {// TODO Auto-generated method stubreturn null;}}index.jsp
<%@ page language="java" contentType="text/html; charset=utf-8"pageEncoding="utf-8"%>
<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
<html>
<head><meta http-equiv="Content-Type" content="text/html; charset=utf-8"><title>Insert title here</title>
</head>
<body>
<form action="LoginServlet">姓名:<input id="username" name="username" /><input type="submit" />
</form>
</body>
</html>
ChatRoom.jsp
<%@page import="java.io.PrintWriter"%>
<%@ page language="java" contentType="text/html; charset=utf-8"pageEncoding="utf-8"%>
<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
<html>
<head><meta http-equiv="Content-Type" content="text/html; charset=utf-8"><title>Insert title here</title><script type="text/javascript" src="jquery-1.4.2.min.js"></script><script type="text/javascript">var username="${sessionScope.username}";var ws;//一个ws对象就是一个管道var target="ws://localhost:8080/web_war_exploded/chatSocket?username="+username;window.onload=function(){//登录进入ChatRoom就打开socket通道if ('WebSocket' in window) {ws = new WebSocket(target);} else if ('MozWebSocket' in window) {ws = new MozWebSocket(target);} else {alert('WebSocket is not supported by this browser.');return;}ws.onmessage=function(event){eval("var msg="+event.data+";");if(undefined!=msg.welcome)$("#content").append(msg.welcome+"<br/>");if(undefined!=msg.usernames){$("#userList").html("");$(msg.usernames).each(function(){$("#userList").append("<input type=checkbox value='"+this+"'/>"+ this + "<br/>");});}if(undefined!=content){$("#content").append(msg.content+"<br/>");console.info(msg);}}}function subSend() {//判断是否选中var ss=$("#userList :checked");var val = $("#msg").val();console.info(ss.size());//如果未选中var obj=null;if(ss.size()==0){var obj={msg:val,type:1 //1广播 2单聊}}else {var to = $("#userList :checked").val();obj={to:to,msg:val,type:2 //1广播 2单聊}}var str = JSON.stringify(obj);/* $("msg").val("");var obj={to:to,msg:val,type:1 //1广播 2单聊} */ws.send(str);}</script>
</head>
<body>
<div id="container" style="border:1px solid black;width:400px;height:400px;float:left;"><div id="content" style="height:350px;"></div>
</div>
<div style="border-top:1px solid black;width:400px;height:50px;"><input id="msg" /><button onclick="subSend();">send</button>
</div><div id="userList" style="border:1px solid black;width:400px;height:400px;float:left;">
</div>
</body>
</html>
LoginServlet.java
package com.hys.servlet;import java.io.IOException;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;/*** Servlet implementation class LoginServlet*/
@WebServlet("/LoginServlet")
public class LoginServlet extends HttpServlet {private static final long serialVersionUID = 1L;/*** @see HttpServlet#HttpServlet()*/public LoginServlet() {super();// TODO Auto-generated constructor stub}/*** @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse response)*/protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {// TODO Auto-generated method stubrequest.setCharacterEncoding("utf-8");response.setContentType("text/html;charset=utf-8");String username = request.getParameter("username");request.getSession().setAttribute("username", username);/*System.out.println(username);*/response.sendRedirect("ChatRoom.jsp");}/*** @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse response)*/protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {// TODO Auto-generated method stubdoGet(request, response);}}ChatSocket.java
package com.hys.socket;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;import com.google.gson.Gson;
import com.hys.vo.ContentVo;
import com.hys.vo.Message;import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
@ServerEndpoint(value = "/chatSocket")
public class ChatSocket {
private String username;
private Session session;
private static Map<String, Session> map=new HashMap<String, Session>();/*将管道与用户名绑定*/
private static Set<ChatSocket> sockets=new HashSet<ChatSocket>();/*管道道的集合*/
private static List<String> names = new ArrayList<String>();/*用户的集合*/
private Gson gson = new Gson();@OnOpenpublic void open(Session session) {//当前的websocket的session不是之前的session//获取username//将当前的管道加入到管道的集合中去this.session = session;this.sockets.add(this);//将当前的username与session绑定String queryname = session.getQueryString();/*System.out.println(queryname);*/username=queryname.split("=")[1];this.names.add(username);this.map.put(this.username, this.session);/*System.out.println(username);*/String msg = "欢迎"+this.username+"进入聊天室";Message message=new Message();message.setWelcome(msg);message.setUsernames(this.names);this.broadcast(sockets,message.toJson());}@OnClosepublic void close(Session session) {this.sockets.remove(this);this.names.remove(this.username);String msg = this.username+"离开聊天室";Message message=new Message();message.setWelcome(msg);message.setUsernames(this.names);broadcast(sockets,message.toJson());}
// private static Gson gson=new Gson();@OnMessagepublic void onmessage(Session session,String json) {ContentVo vo=gson.fromJson(json, ContentVo.class);if(vo.getType()==1) {//广播Message message=new Message();message.setContent(this.username, vo.getMsg());this.broadcast(sockets,message.toJson());}else {//单聊//单聊过程与广播类似，/*分析前台传过来的消息的type如果是2的话就在map（存放的是key是用户名，value是socket）中通过getTo来取得接收信息的socket并通过to_session.getBasicRemote().sendText(gson.toJson(message))将消息传递给前台*//*举个不太恰当的例子，我想将一个苹果（消息）放到小明的冰箱（小明的前端页面），然后就我又不想动，我就想找小明然后我就喊小明的名字，小明（相当于服务器）听到我喊他的名字（从map集合中找到对应的session）然后小明就将苹果放入到了他的冰箱（对应的session调用sendText（）方法传递消息）放到冰箱里（前台）* */String to=vo.getTo();Session to_session = this.map.get(to);System.out.println(to);System.out.println(to_session);Message message=new Message();message.setContent(this.username, vo.getMsg());try {to_session.getBasicRemote().sendText(gson.toJson(message));} catch (IOException e) {// TODO Auto-generated catch blocke.printStackTrace();}}}//广播消息
public void broadcast(Set<ChatSocket>sockets ,String msg){//遍历当前所有的连接管道，将通知信息发送给每一个管道for(ChatSocket socket : sockets){try {//通过session发送信息socket.session.getBasicRemote().sendText(msg);} catch (IOException e) {e.printStackTrace();}}
}
}ContentVo.java
package com.hys.vo;public class ContentVo {
private String to;
private String msg;
private Integer type;
public String getTo() {return to;
}
public void setTo(String to) {this.to = to;
}
public String getMsg() {return msg;
}
public void setMsg(String msg) {this.msg = msg;
}
public Integer getType() {return type;
}
public void setType(Integer type) {this.type = type;
}
}Message.java
package com.hys.vo;import java.util.Date;
import java.util.List;import com.google.gson.Gson;public class Message {
private String welcome;
private List<String> usernames;
private String content;
public String getContent() {return content;
}
public void setContent(String content) {this.content = content;
}public void setContent(String name,String msg) {this.content = name+""+new Date().toLocaleString()+":<br/>"+msg+":<br/>";
}public String getWelcome() {return welcome;
}
public void setWelcome(String welcome) {this.welcome = welcome;
}
public List<String> getUsernames() {return usernames;
}
public void setUsernames(List<String> usernames) {this.usernames = usernames;
}
public String toJson() {return gson.toJson(this);
} public static Gson gson=new Gson();
}运行截图
##注意
如果使用eclipse的话，可以建立相应的目录结构将代码粘贴进去，导入jar包即可
