Twemproxy源码分析(二)进程&事件模型

23 篇文章 0 订阅
5 篇文章 0 订阅

概述:

上一节我们了解了Twemproxy主程序的启动过程,但没有深入细节。作为一个proxy服务,我们应该深入了解它的进程和事件模型。本章我们从nc_run()开始深入了解。

进程模型:

在看代码之前,我们得先大体了解Twemproxy使用的是单进程、多进程还是多线程的模型。通过grep代码,查找fork()和pthread_create()函数的调用,发现Twemproxy使用的是单进程单线程来处理请求,只是另外起了一个线程来处理统计数据。

这么看就简单多了,既然是单进程单线程,我们就不用关心进程管理进程间通信上面的问题了,可以把重点放在事件驱动模型上来。

事件模型:

了解libevent的同学可能都知道,利用libevent做事件驱动模型编程,基本步骤是:

  1. 调用event_init()创建event_base
  2. 创建event,调用event_set()设置文件描述符、文件描述符上需要监听的事件以及事件的回调函数
  3. 调用event_add()将创建的event加入event_base
  4. 调用event_dispatch()进入事件循环

Twemproxy没有使用libevent,而是自己封装了一个简单的事件驱动模型,步骤其实都差不多,相关代码在event目录下。

我们可以先看一下event/nc_event.h:

文件开头定义了几个宏,然后定义了两个回调函数指针,分别是event的回调函数event_cb_t和状态统计的回调函数event_stats_cb_t。紧接下来是一个条件宏定义:

可以看到,Twemproxy针对不同的平台定义了不同的event_base结构(bsd下是kqueue,linux下是epoll,evport是什么我也没见过了),因为我们用的是Linux,所以只关心epoll下的代码就行了:

epoll下的event_base定义很简单:一个epoll句柄ep,一个指向epoll_event数组的指针*event,一个event计数nevent,一个callback函数。

那么需要实现哪些函数呢?在event/nc_event.h的最底下有声明:

创建:

我们先看看如何创建event_base,下面是event/nc_epoll.c当中的event_base_create()函数定义:

创建event_base的过程,就是初始化前面定义的event_base结构体的内容:

  1. 调用epoll_create()创建epoll句柄
  2. 申请nevent * sizeof(struct epoll_event)大小的空间,存放epoll_event指针数组
  3. 申请sizeof(struct event_base),存放event_base
  4. 对event_base中的ep、event、nevent、cb进行赋值。

调用event_base_create的地点在nc_core.c当中:

其中EVENT_SIZE刚才nc_event.h的代码里有定义(1024),而core_core函数则在nc_core.c当中定义。调用event_base_create()之后,我们便得到了一个可以容纳1024个event的event_base。

添加event:

创建完event_base之后,便要向event_base中添加event,需要用到event_add_*函数,我们先来看看event_add_conn()函数:

conn的结构非常复杂,我们这里不去细看,因为这个函数只用到了conn->sd、conn->send_active和conn->recv_active,其中sd是要注册的文件描述符,而send_active和read_active只是布尔型的标记位,用来标记当前连接是否开启了读/写。

初始化event的时候,加入了EPOLLIN和EPOLLOUT事件,并指定了边缘触发EPOLLET模式,即当文件描述符的可读/可写状态切换的时候返回,成功后设置conn的send_active和read_active为1。

event_add_in()和event_add_out()其实是修改event的,具体分别将其修改为只读或者读写,同时修改send_active或者read_active。

那么什么时候添加event呢,一般是在创建连接之后,我们检索一下event_add_conn的调用位置:

其中nc_server.c中的调用是在Twemproxy创建了与后端Memcached Server的链接之后,而nc_proxy.c中的调用则分别是在创建listen端口之后,以及accept客户端的链接之后,这符合我们的预期。接下来我们分别来看看这3处。

监听端口:

针对每一个server pool,Twemproxy会创建一个proxy,每个proxy监听一个指定的端口,监听后即需要像event_base添加事件,相关代码在nc_proxy.c的proxy_listen()函数中。

代码已经省略掉了不相关的部分,可以看出,proxy_listen()函数先调用event_add_conn添加一个事件,然后将其修改为只读(等待客户端连接)。

等待连接:

accept的代码在nc_proxy.c的proxy_accept()函数中:

同样省略了不相关的部分,可以看出在accept到客户端的文件描述符后,也是通过event_add_conn添加事件,不过这时没有调用event_del_out(),因为我们是需要向客户端写回数据的。

连接Memcached:

连接后端Memcached的代码在nc_server.c当中的server_connect()函数:

初始化连接之后,调用event_add_conn()来添加事件,然后连接Memcached,因为连接需要读写,所以没有调用event_del_out()。

从上面的分析看,添加事件的过程基本上是一致的。

事件循环:

完成了event_base的创建和event的添加之后,就进入事件循环,前一节我们讲到时间循环是在core_loop()中做的:

core_loop()调用了event_wait()函数做事件循环,event_wait()的代码如下:

epoll_wait()返回当前I/O就绪的句柄数nsd,同时将epoll_event放入*event指向的数组中,随后就可以根据nsd和*event去依次访问每一个句柄,调用event_base的cb函数进行处理。

事件触发:

前面我们分析event_base_create()的时候了解到,event_base_create()这个函数仅在nc_core.c当中调用,也就是说全局只有一个event_base,而这个event_base上所有的事件触发的回调函数event_base->cb都是指向调用event_base_create()函数时所设置的core_core()。那么问题来了,如何区分不同的连接(后端Memcached连接句柄,客户端连接句柄以及proxy的监听端口句柄)呢?

我们需要具体看一下core_core()这个函数的代码,这个函数非常短:

可以看出,针对不同的事件:EVENT_READ、EVENT_WRITE,core_core()函数只是分别去调用了core_recv()和core_send()函数,而EVENT_ERR事件则报错。

那么core_recv()和core_send()函数做了什么呢?

可以看出他们只是分别调用了conn->recv()和conn->send(),我们看一下conn这个结构体相关部分的定义:

省略掉了无关的部分,可以看到conn包含了若干handler的函数指针,这其中就包含recv和send。那么这些字段是在创建conn的时候初始化的。针对不同的conn,为他们初始化不同的handler即可。

那么,首先我们看看listen的过程,调用proxy_listen()函数的位置是在nc_proxy.c的proxy_each_init()当中:

在调用proxy_listen()之前,调用了conn_get_proxy()来创建并初始化conn:

通过这个函数定义,我们看到为conn初始化了recv、close、ref和unref这4个handler,其中recv的handler是proxy_recv,那么也就是说,当监听句柄由不可读变为可读时,将会触发proxy_recv这个回调函数:

果不其然,这个函数调用了proxy_accept():

代码已经滤掉了无关部分,proxy_accept在accept到客户端的句柄后,通过调用conn_get()创建并初始化conn,在conn_get()中对conn做了如下初始化:

而在创建Memcached连接时,是这么初始化的:

这些handler的作用,这里暂时不细究,只需要理解事件触发回调函数的过程就可以了。

小结:

通过上面的分析,我们知道了Twemproxy使用了单进程模型,并且自己封装了事件模型,这与redis是类似的。这也就意味着,Twemproxy不适合做CPU密集的操作,仅作为proxy来用是OK的,若要对Twemproxy的功能进行扩展,要特别注意这一点,以免影响Twemproxy的吞吐量。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值