libevent 线程池的设计

Thread pool

背景

在我所做过的一个基于libevent项目中, 我所使用的线程模型是 one event_base per thread + thread pool 模型。每个线程最多有一个event_base, 每一个 TCP 连接必须由某一个event_base 管理,所有这个线程的 IO 都会转移到这个 event_base 上面来处理。换句话来说,一个 file descriptor 只能由一个线程来读写。这样,我们就很方便地把不同的 TCP 连接放到不同的线程里面去。

一个节点支持多线程,它有两种模式

  • 单线程, accept 与 TCP 的消息处理在同一个线程做 IO
  • 多线程, accept 有一个专门的线程接受连接,然后创建一个新的 Thread-Pool, 新的连接会按round-robin的方式分配。

具体实现

在其他的线程中,只要拿到一个 libevent_thread_t 的对象,往里面的 base 添加事件就好了。

typedef struct libevent_thread_s
{
    pthread_t thread_id;    //  pid of this thread
    struct event_base * base;   // libevent handle this thread uses
    
    // this pipe is used to stop the event_base from its own thread
    struct event * notify_event;
    int notify_receive_fd;
    int notify_send_fd;
    int no;                 // the thread number, which is needed for test
} libevent_thread_t;

创建新线程调用的函数(一直循环):

static void* worker_libevent(void *arg)
{
    libevent_thread_t* me = (libevent_thread_t *)arg;
    event_base_dispatch(me->base);
    return NULL;
}

虽然其他的线程可以往这个线程添加事件,但是当一个 libevent_thread_t 对象释放的时候, 其他的线程不能 break 这个线程的 event_base,只能是这个线程自己 break 自己的 event_base,所以在实现上我们可以首先要打开一个管道

libevent_thread_t* libevent_thread_new(){
    int fds[2];
    if(pipe(fds)){
        printf("Can't create notify pipe");
        return NULL;
    }
    pr_libevent_thread_t* t = g_new0(pr_libevent_thread_t, 1);
    t->notify_receive_fd = fds[0];
    t->notify_send_fd = fds[1];
    setup_thread(t);
    return t;
}

为这个管道一端的 fd 在event_base上面添加监听事件,有数据可读的时候就 break event_base 就可以退出了

static void thread_libevent_process(int fd, short which, void * arg)
{
    libevent_thread_t* me = (libevent_thread_t *)arg;
    char buf[1];
    read(fd, buf, 1);
    event_base_loopbreak(me->base);
}

static int setup_thread(libevent_thread_t* me)
{
    me->base = event_base_new();
    if(!me->base)
    {
        printf( "Can't allocate event base\n");
        return -1;
    }
    me->notify_event = event_new(me->base, me->notify_receive_fd, EV_READ | EV_PERSIST,
                                 thread_libevent_process, me);
    if(event_add(me->notify_event, NULL) == -1)
    {
        printf("Can't monitor libevent notify pipe\n");
        return -1;
    }
    return 0;
}

所以当我们想要退出时,向 fd 发送数据

void libevent_thread_wakeup(libevent_thread_t* me)
{
    char one[1];
    one[0] = 1;
    write(me->notify_send_fd, one, sizeof(one));
}

至于线程池,就是一个 libevent_thread_t 的数组,通过next来指定函数返回的线程,其中 n 为线程池容量大小, next 自增到 n 时变成 0

libevent_thread_t* event_pool_get_next_thread(libevent_thread_pool_t*);

typedef struct libevent_thread_pool_s
{
    int next;
    struct spinlock lock;
    int n;
    pr_libevent_thread_t** slot;
} libevent_thread_pool_t;

注意事项

在多线程环境中,libevent 的 event_base 的 loopbreak必须由他自己的线程来实现,所以其他线程只能是通过管道来通知。
在一个线程往另一个线程的 event_base 添加事件的时候,也就是在多个线程对于一个 event_base 操作的时候,event_base 需要对它自己的数据结构加锁,所以在使用线程池或者写多线程的程序的时候,开始时需要调用, 文档

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

推荐阅读更多精彩内容