Netty学习笔记(六) 简单的聊天室功能之WebSocket客户端开发实例

时间:2023-12-24 23:21:31

在之前的Netty相关学习笔记中,学习了如何去实现聊天室的服务段,这里我们来实现聊天室的客户端,聊天室的客户端使用的是Html5和WebSocket实现,下面我们继续学习.

创建客户端

接着第五个笔记说,第五个笔记实现了简单的静态资源服务起,那么我们利用这个静态资源服务起为我们提供页面,创建一个socket.html页面,在这个页面中我们实现Socket连接,连接到我们的Netty搭建的聊天服务器上,因此我们需要创建一个聊天页面和Socket连接,这里我们假定Socket连接地址为 http://localhost:8080/socket 我们首先实现页面的开发和Socket的连接。

客户端代码

<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8">
<title>WebSocket Chat</title>
</head>
<body>
<script type="text/javascript">
var socket;
if (!window.WebSocket) {
window.WebSocket = window.MozWebSocket;
}
if (window.WebSocket) {
socket = new WebSocket("ws://localhost:8080/socket");
socket.onmessage = function (event) {
var ta = document.getElementById('chatContent');
ta.value = ta.value + '\n' + event.data
};
socket.onopen = function (event) {
var ta = document.getElementById('chatContent');
ta.value = "连接开启!";
};
socket.onclose = function (event) {
var ta = document.getElementById('chatContent');
ta.value = ta.value + "连接被关闭";
};
} else {
alert("浏览器不支持 WebSocket!");
} function send(message) {
if (!window.WebSocket) {
return;
}
if (socket.readyState == WebSocket.OPEN) {
socket.send(message);
} else {
alert("连接没有开启.");
}
}
</script>
<form onsubmit="return false;">
<h1>基于Netty构建的聊天室</h1>
<textarea id="chatContent" style="width: 100%; height: 400px;"/>
<hr>
<input type="text" name="message" style="width: 300px">
<input type="button" value="发送消息" onclick="send(this.form.message.value)">
<input type="button" onclick="javascript:document.getElementById('chatContent').value=''" value="清空记录">
</form>
</body>
</html>

页面效果

启动完成后,输入http://localhost:8080/socket.html 应当可以看到如下的界面,这些方法和第五个笔记的没有什么新得东西,就是新增了一个HTML页面而已.

[Netty学习笔记(六) 简单的聊天室功能之WebSocket客户端开发实例

可以看到连接被关闭的字样,这是因为我们的服务端还没有完成,我们将服务端继续改进即可。

完善服务端

在第四个笔记的时候,我们只实现了一个简单的服务端,使用telnet进行测试,我们可以将起拿过来修改一下,为我们所用..

修改内容分析

修改传递类型

第四篇文章中我们实现了基于String类型的数据,这里我们修改为TextWebSocketFrame 当然相应的返回类型和读取也需要对应修改(这里为了演示,移除了时间,不影响整体逻辑,只是界面效果).x修改后的代码如下:

package com.zhoutao123.simpleChat.html;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.util.concurrent.GlobalEventExecutor; import static com.zhoutao123.simpleChat.utils.DatetimeUtils.getNowDatetime; public class TextWebSocketServiceAdapter extends SimpleChannelInboundHandler<TextWebSocketFrame> { // 创建ChannelGroup 用于保存连接的Channel
public static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE); // 当有新的Channel增加的时候
@Override
public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
// 获取当前的Channel
Channel channel = ctx.channel();
// 向其他Channel发送上线消息
channelGroup.writeAndFlush(
new TextWebSocketFrame(String.format("[服务器]\t用户:%s 加入聊天室!\n", channel.remoteAddress())));
// 添加Channel到Group里面
channelGroup.add(channel);
// 向新用户发送欢迎信息
channel.writeAndFlush(
new TextWebSocketFrame(String.format("你好,%s欢迎来到Netty聊天室\n", channel.remoteAddress())));
} @Override
public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
// 用户退出后向全部Channel发送下线消息
channelGroup.writeAndFlush(
new TextWebSocketFrame(
String.format("[服务器]\t用户:%s 离开聊天室!\n", ctx.channel().remoteAddress())));
// 移除
channelGroup.remove(ctx.channel());
} @Override
protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
// 服务器接收到新的消息后之后执行
// 获取当前的Channel
Channel currentChannel = ctx.channel();
// 遍历
for (Channel channel : channelGroup) {
String sendMessage = "";
// 如果是当前的用户发送You的信息,不是则发送带有发送人的信息
if (channel == currentChannel) {
sendMessage = String.format("[You]\t%s\n", msg.text());
} else {
sendMessage = String.format("[%s]\t %s\n", currentChannel.remoteAddress(), msg.text());
}
channel.writeAndFlush(new TextWebSocketFrame(sendMessage));
}
} @Override
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
// 发送异常的时候通知移除
Channel channel = ctx.channel();
channelGroup.writeAndFlush(
new TextWebSocketFrame(String.format("[服务器]\t 用户 %s 出现异常掉线!\n", channel.remoteAddress())));
ctx.close();
}
}

新增解码器和编码

既然接收和发送的数据类型改变了,那么我们也要相应的修改解码器和编码器,添加以下代码即可.

package com.zhoutao123.simpleChat.html;

import com.waylau.netty.demo.websocketchat.TextWebSocketFrameHandler;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.handler.stream.ChunkedWriteHandler; /**
* 服务端 ChannelInitializer
*
* @author waylau.com
* @date 2015-3-13
*/
public class HttpServerInitializer extends ChannelInitializer<SocketChannel> { @Override
public void initChannel(SocketChannel ch) throws Exception {
ChannelPipeline pipeline = ch.pipeline();
//将请求和应答消息编码或解码为HTTP消息
pipeline.addLast(new HttpServerCodec());
//将HTTP消息的多个部分组合成一条完整的HTTP消息
pipeline.addLast(new HttpObjectAggregator(64 * 1024));
pipeline.addLast(new ChunkedWriteHandler());
pipeline.addLast(new HttpServerHandleAdapter());
// 添加以下代码
pipeline.addLast(new WebSocketServerProtocolHandler("/socket"));
pipeline.addLast(new TextWebSocketServiceAdapter());
}
}

测试效果

启动服务器,打开聊天界面测试:

Netty学习笔记(六) 简单的聊天室功能之WebSocket客户端开发实例

可以看到实现了基本的简单的聊天效果,至于压力测试还没有测试,不知道能承载多少个用户,至此我们从丢弃服务到实现在线群聊的实现,大致对Netty有了基本的了解,后期我们将对Netty实现源码级别的学习和了解.