一文学会在Netty中使用framecodec

简介

netty为我们提供了很多http2的封装,让我们可以轻松的搭建出一个支持http2的服务器。其中唯一需要我们自定义的就是http2 handler。

在之前的文章中,我们介绍了自定义http2handler继承自Http2ConnectionHandler并且实现Http2FrameListener。这种实现方式是netty目前比较推荐的实现方式,今天给大家介绍的一种实现方式是netty中准备替换继承Http2ConnectionHandler的实现方式,但是这种实现方式并不成熟,还在不断的完善中。

今天给大家介绍一下这种实现方式。

Http2FrameCodec

这种实现方式的核心类是Http2FrameCodec。事实上Http2FrameCodec也是继承自Http2ConnectionHandler。

它的主要作用是将HTTP/2中的frames和Http2Frame对象进行映射。Http2Frame是netty中对应所有http2 frame的封装,这样就可以在后续的handler中专注于处理Http2Frame对象即可,从而摆脱了http2协议的各种细节,可以减少使用者的工作量。

对于每个进入的HTTP/2 frame,Http2FrameCodec都会创建一个Http2Frame对象,并且将其传递给channelRead方法,用于对该对象进行处理。

通过调用write方法,可以对outbound的 Http2Frame 转换成为http2 frame的格式。

Http2Frame、Http2FrameStream和Http2StreamFrame

netty中有三个非常类似的类,他们是Http2Frame、Http2FrameStream和Http2StreamFrame。

我们知道netty中一个tcp连接可以建立多个stream,Http2FrameStream就是和stream对应的类,这个类中包含了stream的id和stream当前的状态。

一个stream中又包含了多个消息,每个消息都是由多个frame组成的,所以Http2Frame是和这些frame对应的netty类。

Http2StreamFrame本身也是一个frame,事实上它继承自Http2Frame。

为什么会有这个类呢?因为对应frame本身来说,一般情况下它是和一个特定的stream相关联的,Http2StreamFrame表示这种关联关系,可以通过它的set stream方法来指定其关联的stream。如果想要该frame应用到整个连接而不是特定的某个stream,如果是关联到整个连接,那么stream()方法的返回就是null。

Http2FrameCodec的构造

虽然Http2FrameCodec有构造函数,但是netty推荐使用Http2FrameCodecBuilder来构造它:

Http2FrameCodecBuilder.forServer().build();
复制代码

可以看到Http2FrameCodecBuilder有一个forServer还有一个forClient方法。他们一个是使用在服务器端,一个是使用在客户端。

主要是通过里面的server属性来进行区分。

Stream的生命周期

frame codec将会向有效的stream发送和写入frames。之前讲过了 Http2StreamFrame 是和一个Http2FrameStream对象相关联的。

对于一个有效的stream来说,如果任意一方发送一个RST_STREAM frame,那么该stream就会被关闭。

或者发送方或者接收方任意一方发送的frame中带有END_STREAM标记,该stream也会被关闭。

流控制

Http2FrameCodec提供了对流的自动控制,但是我们仍然需要做一些操作,来对window进行更新。

具体而言,当我们在接收到Http2DataFrame消息的时候,对消息进行处理之后,需要增大window的大小,表示该data已经被处理了,可以有更多的空间去容纳新的数据。

也就是说需要向ctx中写入一个Http2WindowUpdateFrame,在这个Http2WindowUpdateFrame中需要传入处理的data的大小和对应stream的id,下面是一个处理data frame的例子:

    /**
     * 处理data frame消息
     */
    private static void onDataRead(ChannelHandlerContext ctx, Http2DataFrame data){
        Http2FrameStream stream = data.stream();
        if (data.isEndStream()) {
            sendResponse(ctx, stream, data.content());
        } else {
            // 不是end stream不发送,但是需要释放引用
            data.release();
        }
        // 处理完data,需要更新window frame,增加处理过的Data大小
        ctx.write(new DefaultHttp2WindowUpdateFrame(data.initialFlowControlledBytes()).stream(stream));
    }
复制代码

上的例子中,我们向DefaultHttp2WindowUpdateFrame传入了对应的stream id,如果stream id为0,则表示处理的是整个connection,而不是单独的某个stream。

除了window update frame之外,对于某个特定stream的初始window还可以发送一个 Http2SettingsFrame,通过设置Http2Settings.initialWindowSize() 达到初始化的目的。

接收消息

对于每个HTTP/2 stream来说,其中包含的frame可以分为 Http2HeadersFrame和Http2DataFrame,而Http2HeadersFrame必定是第一个接收到的frame,并且这个headerFrame还关联了对应的stream对象:Http2FrameStream。

所以我们在处理的时候可以针对这两种不同的frame分别进行处理:

    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        if (msg instanceof Http2HeadersFrame) {
            onHeadersRead(ctx, (Http2HeadersFrame) msg);
        } else if (msg instanceof Http2DataFrame) {
            onDataRead(ctx, (Http2DataFrame) msg);
        } else {
            super.channelRead(ctx, msg);
        }
    }
复制代码

自定义handler

如果使用Http2FrameCodec,我们只需要在pipline中添加Http2FrameCodec之后,再添加自定义的handler即可:

ctx.pipeline().addLast(Http2FrameCodecBuilder.forServer().build(), new CustHttp2Handler());
复制代码

因为Http2FrameCodec已经对http2中的frame进行了转换,所以我们在CustHttp2Handler中只需要处理自定义逻辑即可。

netty推荐自定义的handler继承自Http2ChannelDuplexHandler,因为它比普通的ChannelDuplexHandler多了一个创建newStream()和遍历所有有效stream的 forEachActiveStream(Http2FrameStreamVisitor)方法。

总结

本文讲解了Http2FrameCodec的原理,和与其搭配的handler实现中要注意的事项。


链接:https://juejin.cn/post/7033198544936927239
 

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务