首页 > Web开发 > 详细

skynet源代码学习 - 从全局队列中弹出/压入一个消息队列过程

时间:2014-09-06 17:24:23      阅读:257      评论:0      收藏:0      [点我收藏+]


学习云风的skynet源代码,简单记录下。

void 
skynet_globalmq_push(struct message_queue * queue) {
	struct global_queue *q= Q;

	uint32_t tail = GP(__sync_fetch_and_add(&q->tail,1));

	// only one thread can set the slot (change q->queue[tail] from NULL to queue)
	if (!__sync_bool_compare_and_swap(&q->queue[tail], NULL, queue)) {
		// The queue may full seldom, save queue in list
		// 假设swap失败说明queue[] 满了,达到了64K个队列,出现的几率非常小
		// 假设这种话,就把其保存在Q的list中
		assert(queue->next == NULL);
		struct message_queue * last;
		do {
			last = q->list;
			queue->next = last;
		} while(!__sync_bool_compare_and_swap(&q->list, last, queue));

		return;
	}
}

// 结构体global_queue中的 head, tail 字段分别控制着Q的取,存过程
// GP呢能够看做是一个hash的过程,以此来确定其维护的queues的index
struct message_queue * 
skynet_globalmq_pop() {
	struct global_queue *q = Q;
	uint32_t head =  q->head;

	if (head == q->tail) {
		// The queue is empty.
		return NULL;
	}

	uint32_t head_ptr = GP(head);

	struct message_queue * list = q->list;
	// 假设list非空,说明Q->queue以前满过,就把他们转移回queue[]中,由于那里速度更快
	if (list) {
		// If q->list is not empty, try to load it back to the queue
		struct message_queue *newhead = list->next;
		if (__sync_bool_compare_and_swap(&q->list, list, newhead)) {
			// try load list only once, if success , push it back to the queue.
			list->next = NULL;
			skynet_globalmq_push(list);
		}
	}

	// 从头取一个消息队列
	struct message_queue * mq = q->queue[head_ptr];
	if (mq == NULL) {
		// globalmq push not complete
		return NULL;
	}
	// 取走一个消息后自然要将index往后移动一个位置,而且刚那个position置为空
	if (!__sync_bool_compare_and_swap(&q->head, head, head+1)) {
		return NULL;
	}
	// only one thread can get the slot (change q->queue[head_ptr] to NULL)
	if (!__sync_bool_compare_and_swap(&q->queue[head_ptr], mq, NULL)) {
		return NULL;
	}

	return mq;
}


skynet源代码学习 - 从全局队列中弹出/压入一个消息队列过程

原文:http://www.cnblogs.com/hrhguanli/p/3959579.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!