memcache connect queue

在每个工作线程中，都有一个connect queue ,它其实是主线程和工作线程进行通信的中介，其里面的每个节点本质上就是对一个socket进行的封装，到主线程需要把一个任务分发给一个工作线程的时候，就会在对应的工作线程的connect queue中插入一个节点，下面我们简单下

 

一   节点结构

 

typedef struct conn_queue_item CQ_ITEM; struct conn_queue_item { int sfd; //socket文件描述符 enum conn_states init_state; //初始状态 int event_flags; int read_buffer_size; enum network_transport transport; //网络传输方式 CQ_ITEM *next; };

 

二 connect queue

typedef struct conn_queue CQ; struct conn_queue { CQ_ITEM *head; //头部指针 CQ_ITEM *tail; //尾部指针 pthread_mutex_t lock; //互斥锁 pthread_cond_t cond; //条件锁 };

 

三 基本的操作

 

 

1  初始化

/* * Initializes a connection queue. */ static void cq_init(CQ *cq) { pthread_mutex_init(&cq->lock, NULL); pthread_cond_init(&cq->cond, NULL); cq->head = NULL; cq->tail = NULL; }

 

2  把一个节点插入队列最后

static void cq_push(CQ *cq, CQ_ITEM *item) { item->next = NULL; pthread_mutex_lock(&cq->lock); if (NULL == cq->tail) cq->head = item; else cq->tail->next = item; cq->tail = item; pthread_cond_signal(&cq->cond); pthread_mutex_unlock(&cq->lock); }

 

3 从队列的最前面返回一个节点，并在队列中删除

static CQ_ITEM *cq_pop(CQ *cq) { CQ_ITEM *item; pthread_mutex_lock(&cq->lock); item = cq->head; if (NULL != item) { cq->head = item->next; if (NULL == cq->head) cq->tail = NULL; } pthread_mutex_unlock(&cq->lock); return item; }

 

4  新建一个节点

static CQ_ITEM *cqi_new(void) { CQ_ITEM *item = NULL; pthread_mutex_lock(&cqi_freelist_lock); //从free list中获取 if (cqi_freelist) { item = cqi_freelist; cqi_freelist = item->next; } pthread_mutex_unlock(&cqi_freelist_lock); if (NULL == item) { int i; //分配一系列的item，防止每次都分配 /* Allocate a bunch of items at once to reduce fragmentation */ item = malloc(sizeof(CQ_ITEM) * ITEMS_PER_ALLOC); if (NULL == item) return NULL; /* * Link together all the new items except the first one * (which we'll return to the caller) for placement on * the freelist. */ for (i = 2; i < ITEMS_PER_ALLOC; i++) item[i - 1].next = &item[i]; pthread_mutex_lock(&cqi_freelist_lock); item[ITEMS_PER_ALLOC - 1].next = cqi_freelist; cqi_freelist = &item[1]; pthread_mutex_unlock(&cqi_freelist_lock); } return item; }

 

从代码中我们可以看到，主线程是维护着一个空闲的链表，当需要节点的时候，首先从空闲链表中获取，如果空闲中也没有了，那么就进行新的分配，为了提高效率，采用批量分配的方法，除了当前返回的，其他的节点形成链表，挂载到空闲链表中。

 

 

5 释放一个节点（放入自由链表中）

static void cqi_free(CQ_ITEM *item) { pthread_mutex_lock(&cqi_freelist_lock); item->next =cqi_freelist; cqi_freelist = item; pthread_mutex_unlock(&cqi_freelist_lock); }

 

 

通过上面几个操作，主线程和工作线程通信就能实现了。