webSocket协议初体验

最新推荐文章于 2023-06-30 13:40:25 发布

我是一个菜鸟程序猿

最新推荐文章于 2023-06-30 13:40:25 发布

阅读量230

点赞数

分类专栏： WebSocket 文章标签： java websocket协议

本文链接：https://blog.youkuaiyun.com/weixin_43075298/article/details/89212745

版权

WebSocket 专栏收录该内容

0 篇文章

订阅专栏

WebSocket协议提供双向通信，允许服务器主动推送信息。相比于HTTP，WebSocket弥补了不能主动推送的缺陷，常用于实时交互场景。本文介绍了WebSocket的基本概念、使用原因，并提供了简单的客户端和服务器端示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

WebSocket协议体验

1.什么是WebSocket？

和http协议差不多，只不过WebSocket协议的服务器可以主动向客户端推送信息，客户端也可以主动向服务器发送信息，是真正的双向平等对话，属于服务器推送技术的一种。建立在 TCP 协议之上，服务器端的实现比较容易。协议标识符是ws（如果加密，则为wss），服务器网址就是 URL。

2.为什么要使用WebSocket？

因为HTTP 协议有一个缺陷：通信只能由客户端发起

3.使用例子：

首先导入jar包：
在这里插入图片描述
客户端：

<%@ page language="java" contentType="text/html; charset=ISO-8859-1"
	pageEncoding="ISO-8859-1"%>
<!DOCTYPE html>
<html>
<head>
<meta charset="ISO-8859-1">
<title>Insert title here</title>
</head>
<body>

	<h1>Welcome</h1>
	<br />
	<input id="text" type="text" />
	<button "send()">发送消息</button>
	<hr />
	<button "closeWebSocket()">关闭WebSocket连接</button>
	<hr />
	<div id="message"></div>

	<script type="text/javascript">
		var websocket = null;
		//判断当前浏览器是否支持WebSocket
		if ('WebSocket' in window) {
			websocket = new WebSocket("ws://localhost:8081/TestWebSocket/websocket");
		} else {
			alert('当前浏览器 Not support websocket')
		}

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

		//连接成功建立的回调方法
		websocket.onopen = function() {
			setMessageInnerHTML("WebSocket连接成功");
		}

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

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

		//监听窗口关闭事件，当窗口关闭时，主动去关闭websocket连接，防止连接还没断开就关闭窗口，server端会抛异常。
		window.onbeforeunload = function() {
			closeWebSocket();
		}

		//将消息显示在网页上
		function setMessageInnerHTML(innerHTML) {
			document.getElementById('message').innerHTML += innerHTML + '<br/>';
		}

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

		//发送消息
		function send() {
			var message = document.getElementById('text').value;
			websocket.send(message);
		}
	</script>

</body>
</html>

服务器：


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")
public class WebSocketTest {
	//静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
	private static int onlineCount = 0;
	//concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。
	//若要实现服务端与单一客户端通信的话，可以使用Map来存放，其中Key可以为用户标识
	private static CopyOnWriteArraySet<WebSocketTest> webSocketSet = new CopyOnWriteArraySet<WebSocketTest>();
	//与某个客户端的连接会话，需要通过它来给客户端发送数据
	private Session session;
	public Session getSession(){
		return this.session;
	}
	/**
	 * 连接建立成功调用的方法
	 * @param session  可选的参数。session为与某个客户端的连接会话，需要通过它来给客户端发送数据
	 */
	@OnOpen
	public void onOpen(Session session){
		this.session = session;
		webSocketSet.add(this);         //加入set中
		addOnlineCount();      //在线数加1
		System.out.println("有新连接加入！当前在线人数为" + getOnlineCount());
	}
	/**
	 * 连接关闭调用的方法
	 */
	@OnClose
	public void onClose(){
		webSocketSet.remove(this);       //从set中删除
		subOnlineCount();       //在线数减1
		System.out.println("有一连接关闭！当前在线人数为" + getOnlineCount());
	}
	/**
	 * 收到客户端消息后调用的方法
	 * @param message 客户端发送过来的消息
	 * @param session 可选的参数
	 */
	@OnMessage
	public void onMessage(String message, Session session) {
		//System.out.println("来自客户端的消息:" + message);
		//群发消息
		for(WebSocketTest item: webSocketSet){
			try {
				item.sendMessage(message);
			} catch (IOException e) {
				e.printStackTrace();
				continue;
			}
		}
	}
	/**
	 * 发生错误时调用
	 * @param session
	 * @param error
	 */
	@OnError
	public void onError(Session session, Throwable error){
		System.out.println("发生错误");
		error.printStackTrace();
	}
	/**
	 * 这个方法与上面几个方法不一样。没有用注解，是根据自己需要添加的方法。
	 * @param message
	 * @throws IOException
	 */
	public void sendMessage(String message) throws IOException{
		this.session.getBasicRemote().sendText(message);
		//this.session.getAsyncRemote().sendText(message);
	}
	public static synchronized int getOnlineCount() {
		return onlineCount;
	}
	
	public static synchronized void addOnlineCount() {
		WebSocketTest.onlineCount++;
	}
	
	public static synchronized void subOnlineCount() {
		WebSocketTest.onlineCount--;
	}

}

4.详细请看：

WebSocket教程