ssh框架配置websocket实现基本即时聊天
1。首先在pom.xml文件里添加maven的jar包
<!-- websocket -->
<dependency>
<groupId>javax.websocket</groupId>
<artifactId>javax.websocket-api</artifactId>
<version>1.1</version>
<scope>provided</scope>
</dependency>
2。新建一个class添加后端代码
//建立一个实体类用户获取前端信息
/**
* 客户端发送给服务端消息实体
* @author xiongzichao
*
*/
public class ContentVo {
//用户名称
private String nickName;
//用户头像
private String headImg;
//得到聊天内容
private String msg;
//判断直播房间
private String type;
public String getMsg() {
return msg;
}
public void setMsg(String msg) {
this.msg = msg;
}
public String getType() {
return type;
}
public void setType(String type) {
this.type = type;
}
public String getNickName() {
return nickName;
}
public void setNickName(String nickName) {
this.nickName = nickName;
}
public String getHeadImg() {
return headImg;
}
public void setHeadImg(String headImg) {
this.headImg = headImg;
}
}
//用4个ArrayList保存用户进来的session地址用于发送信息
public HashMap<String, List> updateList = new HashMap<String,List>();
ArrayList<Session> avaAmt = new ArrayList<Session>();
ArrayList<Session> avaAmt1 = new ArrayList<Session>();
ArrayList<Session> avaAmt2 = new ArrayList<Session>();
ArrayList<Session> avaAmt3 = new ArrayList<Session>();
//这里的注解为前端页面建立连接的地址
@ServerEndpoint("/chatSocket")
public class Websocket {
//定义一个全局变量集合sockets,用户存放每个登录用户的通信管道
private static Set<Websocket> sockets=new HashSet<Websocket>();
//定义一个全局变量Session,用于存放登录用户的用户名
private Session session;
//定义一个全局变量map，key为用户名，该用户对应的session为value
/*private static Map<String, Session> map=new HashMap<String, Session>();*/
//定义一个数组，用于存放所有的登录用户,显示在聊天页面的用户列表栏中
private static List<String>names=new ArrayList<String>();
//定义一个全局变量map，key为用户名，该用户对应的session为value
private static Map<String, Session> map=new HashMap<String, Session>();
private String username;
//提供一种像toString()和构造方法的很简单的机制，来实现Java 对象和Json之间的互相转换
//将字符串解析成相应的对象 or 将对象Json化 这个应该是最常用的吧
private Gson gson=new Gson();
@OnOpen
//连接建立完成时
public void onOpen(Session session) throws Throwable{
System.out.println("已经连接");
this.session = session;
//将当前连接上的用户session信息全部存到scokets中
sockets.add(this);
//getQueryString获取查询字符串。
String queryStrin = session.getQueryString();
//解决使用.getQueryString方法获取到路径信息后乱码问题
String queryString = URLDecoder.decode(queryStrin, "utf-8");
//这里是通过连接获取连接里的参数来匹配房间号码（我这里是做了个直播间有4个房间号）
String name = queryString.substring(queryString.indexOf("=")+1);
this.username=name;
/*this.username = queryString.substring(queryString.indexOf("=")+1);*/
//将当前登录用户以及对应的session存入到map中
//System.out.println("建立完成时的名称:"+this.username);
/*this.map.put(this.username, this.session);*/
//System.out.println("this.session:"+this.session);
switch(this.username.toString()) {
case "房间1": //if(num==1)
//把session存入list
Utils.avaAmt.add(this.session);
//System.out.println("no1:"+Utils.avaAmt.size());
Utils.updateList.put(this.username,Utils.avaAmt);
break;
case "房间2": //以此为入口
Utils.avaAmt1.add(this.session);
//System.out.println("no2:"+Utils.avaAmt1.size());
Utils.updateList.put(this.username,Utils.avaAmt1);
break; //跳出switch
case "房间3": //以此为入口
Utils.avaAmt2.add(this.session);
//System.out.println("no3:"+Utils.avaAmt2.size());
Utils.updateList.put(this.username,Utils.avaAmt2);
break; //跳出switch
case "房间4": //以此为入口
Utils.avaAmt3.add(this.session);
//System.out.println("no3:"+Utils.avaAmt2.size());
Utils.updateList.put(this.username,Utils.avaAmt3);
break; //跳出switch
default: //所有case都未匹配时执行
System.out.println("无匹配");
}
//获取当前房间里人数
count(this.username.toString());
/*//把session存入list
Utils.avaAmt.add(this.session);
System.out.println("Utils.avaAmt:"+Utils.avaAmt.size());
Utils.updateList.put(this.username,Utils.avaAmt);*/
//System.out.println("Utils.updateList:"+Utils.updateList);
}
@OnMessage
//收到客户端发送的消息时
public void onMessage(String message) throws Exception{
//System.out.print("有消息:");
//System.out.println("message:"+message);
//把json文件转实体类
ContentVo vo = gson.fromJson(message, ContentVo.class);
//System.out.println("聊天室:"+vo.getType());//哪个聊天室
//获取聊天内容 头像+昵称+内容
//System.out.println("头像:"+vo.getHeadImg());
// String msg = vo.getHeadImg()+" "+vo.getNickName()+":"+vo.getMsg();
List<Object> sj = new ArrayList<Object>();
sj.add(vo.getHeadImg());//头像
sj.add(vo.getNickName());//姓名
sj.add(vo.getMsg());//信息
//System.out.println("regid:"+sj);
broadcast2(sj, vo.getType());
}
@OnClose
//连接被关闭或者无法建立连接时
public void onClose()throws IOException{
System.out.println("连接已经关闭");
//移除退出登录用户的通信管道
sockets.remove(this);
//将用户名从names中剔除
names.remove(this.username);
switch(this.username.toString()) {
case "房间1": //if(num==1)
//把session存入list
Utils.avaAmt.remove(this.session);
break;
case "房间2": //以此为入口
Utils.avaAmt1.remove(this.session);
break; //跳出switch
case "房间3": //以此为入口
Utils.avaAmt2.remove(this.session);
break; //跳出switch
case "房间4": //以此为入口
Utils.avaAmt3.remove(this.session);
break; //跳出switch
default: //所有case都未匹配时执行
System.out.println("无匹配");
}
/* map.remove(this.username);*/
Collection<Session> values = map.values();
values.remove(this.session);
count(this.username.toString());
/*Utils.avaAmt.remove(this.session);*/
}
@OnError
//连接出错时
public void onError( Throwable error){
System.out.println("连接出错 ");
error.printStackTrace();
}
public void count(String number){
List<Session> results = (List<Session>)Utils.updateList.get(number);
Integer count = (results!=null) ? results.size() : 0;
Session to_session = map.get(this.username);
if(to_session!=null){
try {
to_session.getBasicRemote().sendText(gson.toJson(count.toString()));
} catch (IOException e) {
e.printStackTrace();
}
}
}
//向页面发送消息---
public void broadcast(String msg){
//遍历当前所有的连接管道，将通知信息发送给每一个管道
for(Websocket socket : sockets){
try {
//通过session发送信息
socket.session.getBasicRemote().sendText(msg);
} catch (IOException e) {
e.printStackTrace();
}
}
}
/**
* @param msg 需要发往客户端的消息
* @param number 比对map里的页面
* @throws Exception
*/
public void broadcast2(List<Object> sj,String number) throws Exception{
//遍历map里的list
List<Session> results = (List<Session>)Utils.updateList.get(number);
if(results!=null){
for(int i=0;i<results.size();i++){
//把java字符或者 数组转为json串 gson.toJson(sj)
results.get(i).getBasicRemote().sendText( gson.toJson(sj));
}
}
}
3。在你的项目里新建一个jsp（这里需要注意前端发送信息到后端需要把json转为javascript，后端信息发送前端是需要把java转为json 代码里有详细方法）
<script type="text/javascript">
var room = "对应后端房间号";
var webSocket;
if ('WebSocket' in window) {
webSocket = new WebSocket("ws://你的项目地址/chatSocket?username="+room);
} else {
alert("您的浏览器不支持");
}
webSocket.onerror = function(event) {
onError(event)
};
webSocket.onopen = function(event) {
onOpen(event)
};
webSocket.onmessage = function(event) {
onMessage(event)
};
function onMessage(event) {
//JSON.parse(event.data)数据转换为 JavaScript 对象
var fuImgUrl = JSON.parse(event.data);
console.log(fuImgUrl[0]);
console.log(fuImgUrl[1]);
console.log(fuImgUrl[2]);
document.getElementById('messages').innerHTML += '<br />' + fuImgUrl[1]+":"+fuImgUrl[2];
document.getElementById('img').src=fuImgUrl[0];
}
function onOpen(event) {
document.getElementById('messages').innerHTML = '连接完成';
}
function onError(event) {
alert(event.data);
}
function start() {
var object = {};
object['nickName'] = '马化腾';
object['headImg'] = 'http://5b0988e595225.cdn.sohucs.com/images/20170724/ad28da0d658b43aba84ce91df9cacdad.jpeg';
object['msg'] = '五弄坦克';
object['type'] = '房间号';
//将 JavaScript 对象转换为字符串。
var jsonStr = JSON.stringify(object);
webSocket.send(jsonStr);
return false;
}
//监听页面关闭事件
window.onbeforeunload = function(event) {
webSocket.onclose = function() {};
webSocket.close();
}
webSocket.onclose = function(event) {
alert("连接中断");
};
</script>
4。此时即可启动项目测试（如果你的项目里设有登录拦截还需要在struts.xml里设置不能拦截socket请求）我这里以ssh框架为例，在web.xml里添加不拦截你在后端代码注解里设置的地址比如我的：/chatSocket
<!--不拦截ws请求 -->
<constant name="struts.action.excludePattern" value="/chatSocket"></constant>
5。其实websocket我所遇到最多的弯路就是session的问题 这个session和httpSession不是一个，它是socket里的，建立连接时要注意你的项目是否会拦截socket请求，后端代码不会报错但是前端会出现地址出错的问题，其实这和你的前端代码没有关系，我暂时没有测试过传递大文件或者视频什么的，也没有测试过tomcat的并发测试，我看其他网上的最多好像是5000个会崩溃，有条件的朋友可以试一试。
扫一扫
9万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
