actix websocket消息处理流程源码剖析

简介

Actix Web | A powerful, pragmatic, and extremely fast web framework for Rust. 是一个rust web框架。本文以官方的例子（websocket 聊天室examples/websockets/chat at master · actix/examples · GitHub）为切入口，通过阅读源码剖析其消息处理的流程。

1、服务端向客户端发送

链接建立时，actix首先将http的请求路由到这个方法里。然后ws:start开启websocket的链接处理

ws:start里完成了websocket协议的握手，并开始一个actor

在WebsocketContext::create里调用了create_with_addr：创建了一个Mailbox并把Mailbox的的发送端给了WebsocketContext。另外创建了一个空的VecDeque给WebsocketContext。

WebsocketContext的addr实际上就是Mailbox的发送端。

Mailbox里实际上是一个channel，默认大小是16

 ws:start返回的是一个HttpResponse，它的body实际上是一个stream，是WebsocketContextFut。WebsocketContextFut实现了Stream trait的poll_next方法，也就是说可以被tokio await。

 

 WebsocketContextFut 中的poll_next实际上就是从messages队列中不断取出消息并编码成字节。上层也就是说通过await WebsocketContextFut来从中取出字节并发送给客户端。

最后回到用户接口层面，WsChatSession里调用ctx.text();这样的接口，实际上就是把消息放到了message队列中。放进去之后，上面await的时候会从中取出来。 

 2、接受客户端的消息

ws:start的时候，回传进来一个stream，这个stream最终放到了WebsocketContext里。WebsocketContext又会放在ContextFut里，ContextFut又会放在WebsocketContextFut里

前面说过WebsocketContextFut会被await 调用poll_next方法。而他的poll_next中实际上调用了ContextFut的poll方法

ContextFut 的poll方法里实际上做了很多事情。包括对刚刚说的stream进行poll_next从而取出客户端发来的消息。 初次之外还做了许多事情，比如actor自己创建的future、stream等，都是在这里poll的。

 

 总结：

其实把上面的套路弄清楚之后，actix这整套逻辑就很容易看懂了，实际上就是各种poll来poll去。