websocket入门代码

实时通信

• Ajax轮训
  • 异步的每隔一段时间向服务器发送一次请求
• Long pull
  • 类似Ajax请求,是阻塞的方式,请求服务器端后会一直等待服务器响应
• websocket
  • 基于HTTP协议的持久化协议
  • WebSocket 是 HTML5 开始提供的一种在单个 TCP 连接上进行全双工通讯的协议。
  • WebSocket 使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据。在 WebSocket API 中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输。

服务端实现代码

public class WSServer {

    public static void main(String[] args) throws InterruptedException {
        EventLoopGroup mainGroup = new NioEventLoopGroup();
        EventLoopGroup subGroup = new NioEventLoopGroup();

        try {
            ServerBootstrap server = new ServerBootstrap();
            server.group(mainGroup,subGroup)
                    .channel(NioServerSocketChannel.class)
                    .childHandler(new WSServerInitializer());
            ChannelFuture channelFuture = server.bind(8080).sync();
            channelFuture.channel().closeFuture().sync();
        } finally {
            mainGroup.shutdownGracefully();
            subGroup.shutdownGracefully();
        }
    }
}

public class WSServerInitializer extends ChannelInitializer {


    @Override
    protected void initChannel(SocketChannel ch) throws Exception {
        ChannelPipeline pipeline = ch.pipeline();
        // websocket基于http协议,所以要有http编解码器
        pipeline.addLast(new HttpServerCodec());
        // 对写大数据流的支持
        pipeline.addLast(new ChunkedWriteHandler());
        // 对httpMessage进行聚合, 聚合成FullHttpRequest或fullHttpResponse
        pipeline.addLast(new HttpObjectAggregator(1024*64));
        // ----------------- 以上用于支持HTTP协议------------------
        // websocket处理的协议,用于指定给客户端连接访问的路由, : /ws
        // 本handler会帮你处理一些繁重的复杂的事
        // 会帮你处理握手动作: handshaking(close,ping,pong) ping+pong=心跳
        // 对于websocket来说,都是以frames进行传输的,不同的数据类型对应的frames也不同
        pipeline.addLast(new WebSocketServerProtocolHandler("/ws"));
        // 自定义的handler
        pipeline.addLast(new ChatHandler());
    }
}

/**
 * @author: webin
 * @date: 2020/4/4 15:57
 * @description:    处理消息的handler
 *  TextWebSocketFrame: 在netty中,是用于为websocket专门处理文本的对象,frame是消息的载体
 * @version: 0.0.1
 */
public class ChatHandler extends SimpleChannelInboundHandler {

    /** 用于置入和管理所有客户端的channel,把所有的channel都保存到组里面去 */
    private static ChannelGroup clients = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    @Override
    protected void channelRead0(ChannelHandlerContext channelHandlerContext, TextWebSocketFrame msg) throws Exception {
        // 获取客户端传输过来的消息
        String text = msg.text();
        System.out.println("接收到的输出: " + text);

        for (Channel channel : clients) {
            channel.writeAndFlush(
                    new TextWebSocketFrame("[服务器接收到消息:]"+ LocalDateTime.now()+" 接收到消息,消息内容为: "+text)
            );
        }
        // 和上面作用一样
//        clients.writeAndFlush(
//                new TextWebSocketFrame("[服务器接收到消息:]"+ LocalDateTime.now()+" 接收到消息,消息内容为: "+text)
//        );

    }

    /**
     *  当客户端连接服务端会后
     *  获取客户端的channel,并且放到channelGroup中进行管理
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {

        clients.add(ctx.channel());
    }

    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
       // 当触发handlerRemoved,ChannelGroup会自动移除对应客户端的Channel
        clients.remove(ctx.channel());
        System.out.println("客户端断开,channel对应的长id:" + ctx.channel().id().asLongText());
        System.out.println("客户端断开,channel对应的短id:" + ctx.channel().id().asShortText());
    }
}

客户端实现代码

    
    


    
发送消息:
    
    
    
接收消息: