当前位置:   article > 正文

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

actix websock

简介

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去。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/知新_RL/article/detail/889217
推荐阅读
相关标签
  

闽ICP备14008679号