欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

使用WebSocket进行通信的简单应用

程序员文章站 2022-07-08 07:59:20
...
一 应用
打开一个页面,将向服务器发送一段字符串,并通过onmessage事件监听器来接受服务器返回的消息。
 
二 代码
客户端代码
<!DOCTYPE html>
<html>
<head>
	<meta name="author" content="Yeeku.H.Lee(CrazyIt.org)" />
	<meta http-equiv="Content-Type" content="text/html; charset=UTF-8" />
	<title> 使用WebSocket通信 </title>
</head>
<body>
<script type="text/javascript">
	// 创建Web Socket对象
	var webSocket = new WebSocket("ws://127.0.0.1:30000");
	// 当WebSocket建立网络连接时激发该函数
	webSocket.onopen= function()
	{
		alert("已打开连接");
		// 发送消息
		webSocket.send("haha web socket");
	}
	// 为onmessage事件绑定监听器,接收消息
	webSocket.onmessage= function(event)
	{
		// 接收消息
		alert("收到的消息是:" + event.data);
	}
</script>
</body>
</html>
 
服务器代码
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.Charset;
import java.security.MessageDigest;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import sun.misc.BASE64Encoder;
public class SimpleServer
{
	public SimpleServer()throws Exception
	{
		// 创建ServerSocket,准备接受客户端连接
		ServerSocket ss = new ServerSocket(30000);
		// 接收到客户端连接
		Socket socket = ss.accept();
		// 得到Socket对应的输入流
		InputStream in = socket.getInputStream();
		// 得到Socket对应的输出流
		OutputStream out = socket.getOutputStream();
		byte[] buff = new byte[1024];
		int count = -1;
		String req = "";
		// 读取数据,此时建立与WebSocket的"握手"。
		count = in.read(buff);
		// 将读取的数据转化为字符串
		req = new String(buff , 0 , count);
		System.out.println("握手请求:" + req);
		// 获取WebSocket的key
		String secKey = getSecWebSocketKey(req);
		System.out.println("secKey = " + secKey);
		String response = "HTTP/1.1 101 Switching Protocols\r\nUpgrade: "
			+ "websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Accept: "
				+ getSecWebSocketAccept(secKey) + "\r\n\r\n";
		System.out.println("secAccept = " + getSecWebSocketAccept(secKey));
		out.write(response.getBytes());
		// 再次读取WebSocket发送过来的数据
		count = in.read(buff);
		System.out.println("接收的字节数:" + count);
		/*
			因为WebSocket发送过来的数据遵循了一定的协议格式,
			其中第3个~第6个字节是数据掩码。
			从第7个字节开始才是真正的有效数据。
			因此程序使用第3个~第6个字节对后面的数据进行了处理
		*/
		for (int i = 0 ; i < count - 6 ; i++ )
		{
			buff[i + 6] = (byte) (buff[i % 4 + 2] ^ buff[i + 6]);
		}
		// 显示读取得到的数据
		System.out.println("接收的内容:" +  new String(buff
			, 6 , count - 6 , "UTF-8"));
		// 发送数据时,第一个字节必须与读到的第一个字节相同
		byte[] pushHead = new byte[2];
		pushHead[0] = buff[0];
		String pushMsg = "收到,收到!欢迎加入WebSocket世界!";
		// 发送数据时,第二个字节记录发送数据的长度
		pushHead[1] = (byte) pushMsg.getBytes("UTF-8").length;
		// 发送前两个字节
		out.write(pushHead);
		// 发送有效数据
		out.write(pushMsg.getBytes("UTF-8"));
		// 关闭Socket
		socket.close();
		// 关闭ServerSocket
		ss.close();
	}
	// 获取WebSocket请求的SecKey
	private String getSecWebSocketKey(String req)
	{
		//构建正则表达式,获取Sec-WebSocket-Key: 后面的内容
		Pattern p = Pattern.compile("^(Sec-WebSocket-Key:).+",
				Pattern.CASE_INSENSITIVE | Pattern.MULTILINE);
		Matcher m = p.matcher(req);
		if (m.find())
		{
			// 提取Sec-WebSocket-Key
			String foundstring = m.group();
			return foundstring.split(":")[1].trim();
		}
		else
		{
			return null;
		}
	}
	// 根据WebSocket请求的SecKey计算SecAccept
	private String getSecWebSocketAccept(String key)
		throws Exception
	{
		String guid = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
		key += guid;
		MessageDigest md = MessageDigest.getInstance("SHA-1");
		md.update(key.getBytes("ISO-8859-1") , 0 , key.length());
		byte[] sha1Hash = md.digest();
		BASE64Encoder encoder = new BASE64Encoder();
		return encoder.encode(sha1Hash);
	}
	public static void main(String[] args)
		throws Exception
	{
		new SimpleServer();
	}
}
 
 
三 运行结果
使用WebSocket进行通信的简单应用
            
    
    博客分类: JavaScript WebSocket 
 
  • 使用WebSocket进行通信的简单应用
            
    
    博客分类: JavaScript WebSocket 
  • 大小: 29.1 KB
相关标签: WebSocket