使用mio开发web framework - base

最近,笔者要用rust实现一个高性能网络服务,首先就需要选择一个好的异步网络库,在c++里面我们有太多选择,libev,libevent,libuv,甚至笔者自己也写过一个libtnet,不过在rust里面,mio几乎就是唯一的方案了。

mio实现很简单,就是在系统相关函数epoll,kqueue上面封装了自己的相关逻辑,抽象出好用的接口供外面使用。如果大家熟悉python tornado,就可以发现,mio就非常类似于tornado里面的IOLoop。

mio的使用很简单,我们只需要创建一个EventLoop,然后实现自己的Handler,就可以运行起来了。

mio的Handler是一个Trait,定义如下:

pub trait Handler: Sized {
    type Timeout;
    type Message;
    
    // 注册的token有相关事件的时候调用,譬如我们注册了一个socket,
    // socket可读或者可写的时候就会调用ready。
    fn ready(&mut self, event_loop: &mut EventLoop<Self>, token: Token, events: EventSet) {
    }

    // 有通知的时候调用。
    fn notify(&mut self, event_loop: &mut EventLoop<Self>, msg: Self::Message) {
    }

    // 注册的timer到期了调用
    fn timeout(&mut self, event_loop: &mut EventLoop<Self>, timeout: Self::Timeout) {
    }

    // 收到interupted signal的时候调用。
    fn interrupted(&mut self, event_loop: &mut EventLoop<Self>) {
    }

    // 每轮末尾的时候调用。
    fn tick(&mut self, event_loop: &mut EventLoop<Self>) {
    }
}

一个简单的例子:

use mio::{EventLoop, Handler};
struct BaseHandler;

impl Handler for BaseHandler {
    type Timeout = ();
    type Message = ();
}

let mut event_loop = EventLoop::<BaseHandler>::new().unwrap();
event_loop.shutdown();

在上面的例子中,我们实现了一个简单的BaseHandler,因为Handler这个Trait本身有默认的函数实现,所以BaseHandler没有干任何事情。

我们通过EventLoop::new创建一个event loop,但是没有干任何事情就shutdown了。

Timeout的例子:

struct TimeoutHandler;

impl Handler for TimeoutHandler {
    type Timeout = u32;
    type Message = ();

    fn timeout(&mut self, event_loop: &mut EventLoop<Self>, _: Self::Timeout) {
        event_loop.shutdown();
    }
}

let mut event_loop = EventLoop::new().unwrap();
event_loop.timeout_ms(100, 0).unwrap();
event_loop.run(&mut TimeoutHandler).unwrap();

上面的TimeoutHandler我们实现了timeout函数,里面直接shutdown这个event loop了,然后在run之前,我们通过timeout_ms这个函数,注册了一个timer,100ms之后过期,调用对应的timeout函数。

Notify例子:

struct NotifyHandler;

impl Handler for NotifyHandler {
    type Timeout = ();
    type Message = u32;

    fn notify(&mut self, event_loop: &mut EventLoop<Self>, _: Self::Message) {
        event_loop.shutdown();
    }
}

let mut event_loop = EventLoop::new().unwrap();
let sender = event_loop.channel();
let child = thread::spawn(move || {
   sender.send(0).unwrap();
});
event_loop.run(&mut NotifyHandler).unwrap();
child.join().unwrap();

上面的NotifyHandler实现了notify接口,在run之前,我们通过channel函数生成了一个sender,mio的EventLoop是一个单线程的loop,如果要在其他线程跟EventLoop进行交互,只能通过channel生成的sender。

上面的例子,我们生成了sender并且将其move到另一个新的线程中,然后在新的线程里面send了一个message,event loop会在自己的事件循环里面检测到这个事件,触发notify操作。

Tick的例子:

struct TickHandler;

impl Handler for TickHandler {
    type Timeout = ();
    type Message = ();

    fn tick(&mut self, event_loop: &mut EventLoop<Self>) {
        if !event_loop.is_running() {
            // Handle quit here.
            return;
        }
    }

    fn notify(&mut self, event_loop: &mut EventLoop<Self>, _: Self::Message) {
        event_loop.shutdown();
    }
}

let mut event_loop = EventLoop::new().unwrap();
let sender = event_loop.channel();
sender.send(0).unwrap();
event_loop.run(&mut TickHandler).unwrap();

EventLoop在每轮处理的最后,会调用tick这个回调函数,有时候我们会通过notify来告诉EventLoop结束,但是如果我们在notify这个message里面处理close,没准这个message后面还有一些message需要处理,没准这时候因为之前就已经在close里面释放相关资源了导致panic,所以笔者更倾向的一种做法就是close在tick里面处理,因为tick是每轮最后一个调用的函数,我们可以保证之后不会再处理任何的事件了。

TODO:

这里并没有说明ready的用法,因为这个跟socket关系比较大,后续笔者开始实现TcpServer的时候在详细说明。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,064评论 5 466
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,606评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,011评论 0 328
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,550评论 1 269
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,465评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 47,919评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,428评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,075评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,208评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,185评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,191评论 1 328
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,914评论 3 316
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,482评论 3 302
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,585评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,825评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,194评论 2 344
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,703评论 2 339

推荐阅读更多精彩内容