本文不介绍websocket的理论知识,直接进行代码的实现,并且只对服务端进行实现,没有前端页面代码,通过在线工具进行测试
开发环境:eclipse Mars Release(4.5.0) + jdk1.7.0_79 + apache-maven-3.3.9 + tomcat7.0.82
1.项目模块
2.pom.xml中maven坐标的导入
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<groupId>com.lsy</groupId>
<artifactId>websocket-tomcat</artifactId>
<version>0.0.1-SNAPSHOT</version>
<packaging>war</packaging>
<dependencies>
<dependency>
<groupId>javax</groupId>
<artifactId>javaee-api</artifactId>
<version>7.0</version>
<scope>provided</scope>
</dependency>
<dependency>
<groupId>javax.servlet</groupId>
<artifactId>jsp-api</artifactId>
<version>2.0</version>
<scope>provided</scope>
</dependency>
<dependency>
<groupId>javax.servlet</groupId>
<artifactId>javax.servlet-api</artifactId>
<version>3.1.0</version>
<scope>provided</scope>
</dependency>
</dependencies>
<build>
<finalName>tomcat</finalName>
<plugins>
<plugin>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-compiler-plugin</artifactId>
<version>3.1</version>
<configuration>
<source>1.7</source>
<target>1.7</target>
</configuration>
</plugin>
</plugins>
</build>
</project>
package com.lsy.websocket.demo;
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;
/**
* @ServerEndpoint 定义当前类为websocket服务器端口,参数是访问的路径
*/
@ServerEndpoint("/websocket")
public class websocketTomcat {
// 记录当前在线人数
private static int onlineCount = 0;
// CopyOnWriteArraySet是安全的set集合,用来存放客户端的每个websocket对象
private static CopyOnWriteArraySet<websocketTomcat> set = new CopyOnWriteArraySet<websocketTomcat>();
// 与客户端的连接会话
private Session session;
// 返回的信息
private String message;
// 当前用户的线程名字
private String username;
/**
* 连接建立成功调用的方法
* @param session 可选的参数。session为与某个客户端的连接会话,需要通过它来给客户端发送数据
*/
@OnOpen
public void onOpen(Session session) throws IOException {
// 初始化session
this.session = session;
// 将当前session添加到set集合中
set.add(this);
// 在线人数加1
onlineCount++;
// 修改当前线程的名字
Thread.currentThread().setName("用户" + onlineCount);
// 获取当前线程的名字
username = Thread.currentThread().getName();
// 编写返回的信息
message = username + "登录成功,当前在线人数为:" + onlineCount + "人";
// 群发消息
for (websocketTomcat websocketTomcat : set) {
websocketTomcat.session.getBasicRemote().sendText(message);
}
}
/**
* 连接关闭调用的方法
*/
@OnClose
public void onClose() throws IOException {
// 将当前session从set集合中移除
set.remove(this);
// 在线人数减1
onlineCount--;
// 编写返回的信息
message = username + "下线成功,当前在线人数为:" + onlineCount + "人";
// 判断set集合中是否还有websocket对象
if (set.size() > 0) {
// 群发消息
for (websocketTomcat websocketTomcat : set) {
websocketTomcat.session.getBasicRemote().sendText(message);
}
}
}
/**
* 连接关闭调用的方法
*/
@OnError
public void onError(Throwable error) {
error.printStackTrace();
}
/**
* 收到客户端消息后调用的方法
* @param message 客户端发送过来的消息
* @param session 可选的参数
*/
@OnMessage
public void onMessage(String message) throws IOException {
// 群发消息
for (websocketTomcat websocketTomcat : set) {
websocketTomcat.session.getBasicRemote().sendText(username + "说:" + message);
}
}
}
4.测试
点击进入在线测试工具:websocket在线测试工具
输入websocket测试地址:ws://localhost:8080/websocket+tomcat/websocket
5.结果展示
分别打开该地址两次
第一个人登录:
第二个登录的时候:
然后就可以在对话框进行聊天了