150行C代码的comet服务器

本文主要是介绍150行C代码的comet服务器，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Comet 技术就是常见的 Web 服务器"推"技术, 用于向网页实时地推送数据. 最常见的 Comet 技术应用在网页聊天, 当然还可以应用于很多的方面, 如微博更新, 热点新闻推送, 股票即时行情等等, 甚至是网页游戏!

Comet 技术如此重要, 但市面上并没有真正流行通用的 Comet 服务器和解决方案, 比较知名的互联网公司大多是自己开发, 或者基于开源服务器进行二次开发, 例如基于 Jetty(一个开源 Java Web 容器), 而 Facebook 的聊天系统的 Comet 服务器是基于 Mochiweb(一个开源的 Erlang Web 服务器).

当然还有比较知名的以 nginx 模块形式出现的 nginx-push-stream, 但根据实际使用经验, 这个模块无法稳定支撑 10 万个并发连接, 更别谈百万同时在线了. 这也是这个模块为什么没有被普遍大规模应用的原因.

既然大家都开发自己的 Comet 服务器, 那必然有其中的道理, 说是核心技术倒说不上, 不过是为了便于扩展, 能很好地和现有系统整合, 易于运维和管理而已. 那么, 要开发一个 Comet 服务器到底有多难呢? 其实, 一个最简单的 Comet 服务器只需要 150 行 C 语言代码!

先说一下 Comet 技术, 从浏览器支持考虑, long-polling 技术显然是最佳的选择, 又从跨域方面考虑, 那必然是 script tag long-polling 技术获胜. 这也是 Facebook 的选择. 所以, 最简单的 Comet 服务器只支持 Script tag long-polling 即可.

Long-polling 技术要求浏览器的每一个网页和服务器保持一个 HTTP 请求连接(TCP 连接), 服务器收到这样的连接后, 会立即返回 HTTP 首部, 接着通过 chunk 传输编码, 源源不断地将一个个消息发送给浏览器.

一个完整的 chunk 编码的 HTTP 响应如下:

HTTP/1.1 200 OK
Date: Fri, 31 Dec 1999 23:59:59 GMT
Content-Type: text/plain
Transfer-Encoding: chunked1a; ignore-stuff-here
abcdefghijklmnopqrstuvwxyz
10
1234567890abcdef
0
[blank line here]

只要服务器不返回只有"0"的那一行以及紧接着的空白行, 那么就可以保持向网页推数据.

最简单的 Comet 服务器使用了 libevent 框架, 你可以在这里得到它的代码: https://github.com/ideawu/icomet. 欢迎对 Comet 了解的前端工程师贡献 JavaScript 相关的代码!

使用方式:

订阅: curl -v "http://127.0.0.1:8000/sub?id=12"
推送: curl -v "http://127.0.0.1:8000/pub?id=12&content=hi"

这个 Comet 服务器的最大并发数并没有进行测试, 但 last.fm 的 CTO 对一个同样是基于 libevent 的类似程序进行测试, 100 万连接只需要 2GB 内存.

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <signal.h>
#include <fcntl.h>
#include <unistd.h>
#include <event.h>
#include <evhttp.h>
#include <event2/event.h>
#include <event2/http.h>
#include <event2/buffer.h>
#include <event2/util.h>
#include <event2/keyvalq_struct.h>
#include <netinet/in.h>
#include <arpa/inet.h>#define MAX_CHANNELS    1000struct Channel{int id;struct evhttp_request *req;
};struct Channel channels[MAX_CHANNELS];void init(){int i;for(i=0; i<MAX_CHANNELS; i++){channels[i].id = i;channels[i].req = NULL;}
}// called when user disconnects
void cleanup(struct evhttp_connection *evcon, void *arg){struct Channel *sub = (struct Channel *)arg;printf("disconnected uid %d\n", sub->id);sub->req = NULL;
}void sub_handler(struct evhttp_request *req, void *arg)
{struct evkeyvalq params;const char *uri = evhttp_request_get_uri(req);evhttp_parse_query(uri, &params);struct evbuffer *buf;int uid = -1;struct evkeyval *kv;for(kv = params.tqh_first; kv; kv = kv->next.tqe_next){if(strcmp(kv->key, "id") == 0){uid = atoi(kv->value);}}if(uid < 0 || uid >= MAX_CHANNELS){buf = evbuffer_new();evhttp_send_reply_start(req, HTTP_NOTFOUND, "Not Found");evbuffer_free(buf);return;}printf("sub: %d\n", uid);struct Channel *sub = &channels[uid];sub->req = req;buf = evbuffer_new();evhttp_send_reply_start(req, HTTP_OK, "OK");evhttp_add_header(req->output_headers, "Content-Type", "text/html; charset=utf-8");evbuffer_add_printf(buf, "{type: \"welcome\", id: \"%d\", content: \"hello world!\"}\n", uid);evhttp_send_reply_chunk(req, buf);evbuffer_free(buf);evhttp_connection_set_closecb(req->evcon, cleanup, &channels[uid]);
}void pub_handler(struct evhttp_request *req, void *arg){struct evkeyvalq params;const char *uri = evhttp_request_get_uri(req);evhttp_parse_query(uri, &params);struct evbuffer *buf;int uid = -1;const char *content = "";struct evkeyval *kv;for(kv = params.tqh_first; kv; kv = kv->next.tqe_next){if(strcmp(kv->key, "id") == 0){uid = atoi(kv->value);}else if(strcmp(kv->key, "content") == 0){content = kv->value;}}struct Channel *sub = NULL;if(uid < 0 || uid >= MAX_CHANNELS){sub = NULL;}else{sub = &channels[uid];}if(sub && sub->req){printf("pub: %d content: %s\n", uid, content);// push to browserbuf = evbuffer_new();evbuffer_add_printf(buf, "{type: \"data\", id: \"%d\", content: \"%s\"}\n", uid, content);evhttp_send_reply_chunk(sub->req, buf);evbuffer_free(buf);// response to publisherbuf = evbuffer_new();evhttp_add_header(req->output_headers, "Content-Type", "text/html; charset=utf-8");evbuffer_add_printf(buf, "ok\n");evhttp_send_reply(req, 200, "OK", buf);evbuffer_free(buf);}else{buf = evbuffer_new();evbuffer_add_printf(buf, "id: %d not connected\n", uid);evhttp_send_reply(req, 404, "Not Found", buf);evbuffer_free(buf);}
}int main(int argc, char **argv){signal(SIGPIPE, SIG_IGN);struct event_base *base;struct evhttp *http;struct evhttp_bound_socket *handle;unsigned short port = 8000;init();base = event_base_new();http = evhttp_new(base);evhttp_set_cb(http, "/sub", sub_handler, NULL);evhttp_set_cb(http, "/pub", pub_handler, NULL);//evhttp_set_gencb(http, request_handler, NULL);handle = evhttp_bind_socket_with_handle(http, "0.0.0.0", port);printf("server listen at 127.0.0.1:%d\n", port);event_base_dispatch(base);return 0;
}