Java WebSocket通讯

最新推荐文章于 2024-10-08 14:44:14 发布

hlt_寒

最新推荐文章于 2024-10-08 14:44:14 发布

阅读量577

点赞数 2

本文链接：https://blog.csdn.net/hlt_1176264794/article/details/88990002

版权

我项目的目录结构
在这里插入图片描述
ServerManager.java代码

package com.hlt.Communication;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;

/** 服务管理 */
public class ServerManager {
	// 声明一个存储链接的容器
	private static Collection<MyServer> servers = Collections
			.synchronizedCollection(new ArrayList<MyServer>());

	/** 向所有链接发送消息 */
	public static void broadCast(String msg) {
		// 遍历所有链接
		for (MyServer server : servers) {
			try {
				// 发送消息
				server.sendMessage(msg);
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}

	/** 获取当前链接总数 */
	public static int getTotal() {
		return servers.size();
	}

	/** 新链接 */
	public static void add(MyServer server) {
		servers.add(server);
		System.out.println("有新连接加入！ 当前总连接数是：" + servers.size());
	}

	/** 退出链接 */
	public static void remove(MyServer server) {
		servers.remove(server);
		System.out.println("有连接退出！ 当前总连接数是：" + servers.size());
	}

}

MyServer.java代码

package com.hlt.Communication;

import java.io.IOException;

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;

/**
 * 
 * 创建BitCoinServer类，用注解@ServerEndpoint("/ws/bitcoinServer")把它标记为一个WebSocket
 * Server ws/bitcoinServer 表示有通过这个地址访问该服务
 * OnOpen 表示有浏览器链接过来的时候被调用 
 * OnClose 表示浏览器发出关闭请求的时候被调用 
 * OnMessage 表示浏览器发消息的时候被调用
 * OnError 表示有错误发生，比如网络断开了等等
 * sendMessage 用于向浏览器回发消息
 * 其中OnOpen发生的时候，即有链接过来的时候，会把当前WebSocket
 * Server丢在ServerManager里管理起来，这样Tomcat才知道总共有哪些Server, 方便以后进行群发
 * 
 * @ServerEndpoint 注解是一个类层次的注解，它的功能主要是将目前的类定义成一个websocket服务器端,
 *                 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端
 */
@ServerEndpoint("/ws/Communication")
public class MyServer {

	// 与某个客户端的连接会话，需要通过它来给客户端发送数据

	private Session session;

	@OnOpen
	public void onOpen(Session session) {
		this.session = session;
		ServerManager.add(this);
	}

	public void sendMessage(String message) throws IOException {
		this.session.getBasicRemote().sendText(message);
	}

	@OnClose
	public void onClose() {
		ServerManager.remove(this);
		ServerManager.broadCast(String.format("{\"total\":"
				+ ServerManager.getTotal() + "}"));
	}

	@OnMessage
	public void onMessage(String message, Session session) {
		System.out.println("来自客户端的消息:" + message);
		int total = ServerManager.getTotal();
		message = String.format("{\"text\":\"" + message + "\",\"total\":"
				+ total + "}");
		ServerManager.broadCast(message);
	}

	@OnError
	public void onError(Session session, Throwable error) {
		System.out.println("发生错误");
		error.printStackTrace();
	}
}

communication.jsp代码

<%@ page language="java" pageEncoding="UTF-8"%>
<!DOCTYPE html>
<html>
<head>
<title>用WebSocket实时获知信息</title>
</head>
<body>
	<div>
		当前总共 <span id="total">0</span> 个人在线
	</div>
	<div>
		<p id="allInfo"></p>
	</div>
	<input type="text" id="textInfo">
	<button type="button" id="btnSend">发送</button>
</body>

<script type="text/javascript">
	var websocket = null;
	//判断当前浏览器是否支持WebSocket
	if ('WebSocket' in window) {
		//tomcat启动端口
		websocket = new WebSocket("ws://localhost:8080/WebSocketDemo/ws/Communication");
		
		//连接成功建立的回调方法
		websocket.onopen = function() {
			websocket.send("客户端链接成功");
		};

		//接收到消息的回调方法
		websocket.onmessage = function(event) {
			setMessageInnerHTML(event.data);
		};

		//连接发生错误的回调方法
		websocket.onerror = function() {
			alert("WebSocket连接发生错误");
		};

		//连接关闭的回调方法
		websocket.onclose = function() {
			alert("WebSocket连接关闭");
		};

		//监听窗口关闭事件，当窗口关闭时，主动去关闭websocket连接，防止连接还没断开就关闭窗口，server端会抛异常。
		window.onbeforeunload = function() {
			closeWebSocket();
		};
		
		document.getElementById('btnSend').onclick=function(){
			var textInfo = document.getElementById('textInfo').value;
			websocket.send(textInfo);
		};
	} else {
		alert('当前浏览器 Not support websocket');
	}

	//将消息显示在网页上
	function setMessageInnerHTML(innerHTML) {
		var bitcoin = eval("(" + innerHTML + ")");
		document.getElementById('total').innerHTML = bitcoin.total;
		if (bitcoin.text != undefined) {
			document.getElementById('allInfo').innerHTML += bitcoin.text + "<br>";
		}
	}

	//关闭WebSocket连接
	function closeWebSocket() {
		websocket.close();
	}
</script>
</html>