Transport Pipeline-Bound 扩展

remote_module

Transport Pipeline 参考 Netty ChannelPipeline,提供 Inbound 和 Outbound 接口,支持对消息或 I/O 事件扩展。基于 In/OutboundHandler 可以扩展实现 TLS、限流、透传信息处理等。如下图所示,各个 BoundHandler 会串行依次执行。

trans_pipeline

接口定义

// OutboundHandler is used to process write event.
type OutboundHandler interface {
  Write(ctx context.Context, conn net.Conn, send Message) (context.Context, error)
}

// InboundHandler is used to process read event.
type InboundHandler interface {
  OnActive(ctx context.Context, conn net.Conn) (context.Context, error)

  OnInactive(ctx context.Context, conn net.Conn) context.Context

  OnRead(ctx context.Context, conn net.Conn) (context.Context, error)

  OnMessage(ctx context.Context, args, result Message) (context.Context, error)
}

默认扩展

  • 服务端限流 Handler

    Kitex 支持连接级别和请求级别限流,限流是为了保障服务的可用性,当达到阈值应当及时限流,放到 Transport 层可以达到及时限流的目的,实现见 limiter_inbound.go。

    • 连接级别限流 OnActive(), OnInactive()
    • 请求级别限流 OnRead()
  • 元信息透传 Handler

    元信息透传是基于传输协议透传一些 RPC 额外的信息给下游,同时读取传输协议中上游透传的信息,实现见 transmeta_bound.go。

    • 写入透传信息 Write()
    • 读取透传信息 OnMessage()

    为更明确的为使用者元信息透传的扩展能力,Kitex 单独定义了信息透传的处理接口 MetaHandler,这里会执行 MetaHandler 进行透传信息的处理。

    // MetaHandler reads or writes metadata through certain protocol.
    type MetaHandler interface {
      WriteMeta(ctx context.Context, msg Message) (context.Context, error)
      ReadMeta(ctx context.Context, msg Message) (context.Context, error)
    }
    

指定自定义的 BoundHandler

  • 服务端

    option: WithBoundHandler

    svr := xxxservice.NewServer(handler, server.WithBoundHandler(yourBoundHandler))
    
  • 调用端

    option: WithBoundHandler

    cli, err := xxxservice.NewClient(targetService, client.WithBoundHandler(yourBoundHandler))
    

最后修改 July 30, 2023 : fix type error (#735) (50c91bf)