package com.websocket;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
public class WebsocketServer {
public static void main(String[] args) throws InterruptedException {
EventLoopGroup base = new NioEventLoopGroup();
EventLoopGroup work = new NioEventLoopGroup();
new ServerBootstrap().group(base, work).channel(NioServerSocketChannel.class).option(ChannelOption.SO_BACKLOG, 128).childOption(ChannelOption.SO_KEEPALIVE, true)
.childHandler(new ChannelInitializer<SocketChannel>() {
@Override
protected void initChannel(SocketChannel ch) throws Exception {
ch.pipeline().addLast(new HttpServerCodec()).addLast(new HttpObjectAggregator(2048))
.addLast(new WebSocketServerProtocolHandler("/hello")).addLast(new Handler());
}
}).bind(7000).sync().channel().closeFuture().sync();
}
public static class Handler extends SimpleChannelInboundHandler<TextWebSocketFrame> {
@Override
protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
System.out.println("客户端:" + ctx.channel().id().asLongText() + " 消息: " + msg.text());
TextWebSocketFrame textWebSocketFrame = new TextWebSocketFrame("服务器时间: " + LocalDateTime.now().format(DateTimeFormatter.ISO_LOCAL_DATE_TIME) + " 消息:" + msg.text());
ctx.writeAndFlush(textWebSocketFrame);
}
@Override
public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
System.out.println("客户端已链接:" + ctx.channel().id().asLongText());
}
@Override
public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
System.out.println("客户端已断开:" + ctx.channel().id().asLongText());
}
}
}
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<script>
let webSocket = new WebSocket("ws://localhost:7000/hello")
webSocket.onmessage = function (event) {
document.getElementById('acceptText').value += event.data + "\n";
}
webSocket.onopen = function () {
document.getElementById('acceptText').value += "服务器连接成功" + "\n";
}
webSocket.onclose = function () {
document.getElementById('acceptText').value += "服务器已断开连接" + "\n";
}
function sendMethod(message) {
debugger
if (!webSocket) {
return;
}
if (webSocket.readyState == WebSocket.OPEN) {
webSocket.send(message);
}
}
</script>
<body>
<form onsubmit="return false">
<textarea id="sendText" style="height: 300px;width: 200px"></textarea>
<button id="sendBut" onclick="sendMethod(this)">发送消息</button>
<textarea id="acceptText" style="height: 300px;width: 200px"></textarea>
</form>
</body>
</html>
build.gradle
dependencies {
// https://mvnrepository.com/artifact/io.netty/netty-all
implementation group: 'io.netty', name: 'netty-all', version: '4.1.75.Final'
}