基于Tomcate、java、websocket 简单在线聊天

基于Tomcate、java、websocket 简单在线聊天

web.xml配置

跟所有servlet 在web.xml进行配置

<servlet>
		<servlet-name>socketHome</servlet-name>
		<servlet-class>SocketServer</servlet-class>
	</servlet>
	<servlet-mapping>
		<servlet-name>socketHome</servlet-name>
		<url-pattern>/ws</url-pattern>
	</servlet-mapping>

后台代码
SocketServer.java，ws://协议的请求就必须实现WebSocketServlet这个类

import java.util.concurrent.atomic.AtomicInteger;

import javax.servlet.http.HttpServletRequest;

import org.apache.catalina.websocket.StreamInbound;
import org.apache.catalina.websocket.WebSocketServlet;

/**
 * 
 * @author wangbg 通讯入口
 */
@SuppressWarnings("deprecation")
public class SocketServer extends WebSocketServlet {

	/**
	 * 
	 */
	private static final long serialVersionUID = 1L;

	/**
	 * 生成/获取用户
	 * 
	 * @param reuqest
	 * @return
	 */
	private String getUser(HttpServletRequest reuqest) {
		Object objectUser = reuqest.getSession().getAttribute("user");
		if (objectUser == null) {
			objectUser = "游客" + reuqest.getSession().getId();
		}
		return objectUser.toString();
	}

	// 实现createWebSocketInbound，在这里初始化自定义的WebSocket连接对象
	@Override
	protected StreamInbound createWebSocketInbound(String arg0,
			HttpServletRequest reuqest) {
		// TODO Auto-generated method stub
		return new MessageInbound(getUser(reuqest));
	}
}

MessageInbound.java,WebSocket连接对象类
代码中的主要实现了onOpen、onTextMessage方法，分别处理用户上线、发送消息,还有onClose方法 不过我没有实现。

import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;

import net.sf.json.JSONObject;

import org.apache.catalina.websocket.WsOutbound;


/**
 * 
 * @author wangbg 通訊交互
 */
@SuppressWarnings("deprecation")
public class MessageInbound extends
		org.apache.catalina.websocket.MessageInbound {

	private final String user;

	public MessageInbound(String user) {
		// TODO Auto-generated constructor stub
		this.user = user;
	}

	public String getUser() {
		return this.user;
	}

	/**
	 * 用户连接触发事件 1 向连接池添加当前用户 2 向在线用户发送当前用户上线消息 3 向当前用户发送在线用户列表
	 */
	@Override
	protected void onOpen(WsOutbound outbound) {
		// TODO Auto-generated method stub
		super.onOpen(outbound);
		// 向连接池添加当前用户
		InboundPool.addInbound(this);
		JSONObject result = new JSONObject();
		// 向在线用户发送当前用户上线消息
		InboundPool.sendMessage(this.user + "上线");
		// 向当前用户发送在线用户列表
		result = new JSONObject();
		result.element("onlineUser", InboundPool.getOnlineUser());
		InboundPool.sendMessageToUser(this.user, result.toString());
	}

	@Override
	protected void onBinaryMessage(ByteBuffer arg0) throws IOException {
		// TODO Auto-generated method stub

	}

	/**
	 * 客户端发送消息时触发
	 */
	@Override
	protected void onTextMessage(CharBuffer message) throws IOException {
		// TODO Auto-generated method stub
		InboundPool.sendMessage(this.user + ":" + message.toString());
	}

}

InboundPool.java,连接对象池，管理在线用户的连接

import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;

import net.sf.json.JSONObject;

import org.apache.catalina.websocket.WsOutbound;

/**
 * 
 * @author wangbg 通訊交互
 */
@SuppressWarnings("deprecation")
public class MessageInbound extends
		org.apache.catalina.websocket.MessageInbound {

	private final String user;

	public MessageInbound(String user) {
		// TODO Auto-generated constructor stub
		this.user = user;
	}

	public String getUser() {
		return this.user;
	}

	/**
	 * 用户连接触发事件 1 向连接池添加当前用户 2 向在线用户发送当前用户上线消息 3 向当前用户发送在线用户列表
	 */
	@Override
	protected void onOpen(WsOutbound outbound) {
		// TODO Auto-generated method stub
		super.onOpen(outbound);
		// 向连接池添加当前用户
		InboundPool.addInbound(this);
		JSONObject result = new JSONObject();
		// 向在线用户发送当前用户上线消息
		InboundPool.sendMessage(this.user + "上线");
		// 向当前用户发送在线用户列表
		result = new JSONObject();
		result.element("onlineUser", InboundPool.getOnlineUser());
		InboundPool.sendMessageToUser(this.user, result.toString());
	}

	@Override
	protected void onBinaryMessage(ByteBuffer arg0) throws IOException {
		// TODO Auto-generated method stub

	}

	/**
	 * 客户端发送消息时触发
	 */
	@Override
	protected void onTextMessage(CharBuffer message) throws IOException {
		// TODO Auto-generated method stub
		InboundPool.sendMessage(this.user + ":" + message.toString());
	}

}

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>
<body>
	<input type="text" id="message" size="60" />
	<input type="button" id="send" value="发送" />
	<br>
	<div id="panel"></div>
</body>
<script type="text/javascript">
	var panel = document.getElementById('panel');  
	var message = document.getElementById('message');  
	var send = document.getElementById('send') 
	// 控制台输出对象  
	var console = {
		log : function(text) {
			panel.innerHTML += text + "<br>";
		}
	};
	// WebSocket
	var socketBox = {
		socket : null, // WebSocket连接对象  
		host : '', // WebSocket连接 url  
		connect : function() { // 连接服务器  
			window.WebSocket = window.WebSocket || window.MozWebSocket;
			if (!window.WebSocket) { // 检测浏览器支持  
				console.log('Error: 请升级到ie10+浏览器 .');
				return;
			}
			// 创建连接并注册响应函数  
			this.socket = new WebSocket(this.host);
			this.socket.onopen = function() {
				console.log("连接服务器.");
			};
			this.socket.onmessage = function(res) {
				if (res.type == "message") {
					console.log(res.data);
				}
			};
			this.socket.onclose = function() {
				console.log("服务器断开 .");
				socketBox.socket.onclose();
				socketBox.socket = null; // 清理  
			};
		},
		send : function(message) { // 发送消息方法  
			if (this.socket) {
				this.socket.send(message);
				return true;
			}
			console.log('请先连接服务器 !!!');
			return false;
		}
	};
	// 初始化WebSocket连接 url  
	socketBox.host = (window.location.protocol == 'http:') ? 'ws://' : 'wss://';
	socketBox.host += window.location.host + '/websocket/ws';
	// 初始化按钮点击事件函数  
	send.onclick = function() {
		var ms = message.value;
		if (!ms)
			return;
		if (!socketBox.send(ms))
			return;
		message.value = '';
	};
	if (!socketBox.socket)
		socketBox.connect();
</script>
</body>
</html>