近来在项目中需要实现一个http service的功能。虽然可以通过封装socket自己实现http的发送和解析。但考虑到目前网络上还是有大量的http的解析源码,自己再从头实现一番稍显麻烦。因此在网络上搜刮了一番,看到很多实现http的代码, 无一不体现了其轻量级的特点。然而,虽然轻量级,但从代码的量上来说,为了一个http service的功能,引入这许多的代码,也着实不是十分情愿呢。后在同事的推荐下选择了mongoose的代码。
mongoose的代码着实轻量,先看看它的特点:
1. 在整个的实现是使用编写
2. 整个代码也只有一个mongoose.c和mongoose.h两个文件, 从引入第三方的考虑上也着实不多。
3. 实现的功能还是非常多的,从使用的层面上来说功能还是比较全面。只不过不知道是否是为了第三方使用的方便还是怎么地,它的代码只用了两个源文件罢了。诸多的功能也大以宏的开始与结束来区分。
4. 示例非常齐全,所有的功能都有单独的示例
然而,不管它实现多少功能,对于我来说只需要三个:
1. 有http的解析等
2. 文件少,使用方便,不需要因为使用一个简单的功能引入大量代码,而且引入的代码着实没有用到。
3. 有完整的示例
当我们拿到一个第三方库或者第三方源码的时候,第一件事情就是看看代码的示例,并且自己动手谢谢的代码,完成自己想要的功能。于是,我花了一点时间自己写了一个测试的代码,最后发现测试的时候并不通。虽然它的代码示例很全面,然而对于我们来说,或许在它的代码中,有些函数我们不需要,而有些函数却不再这个示例中使用,因此需要自己测试。耗费了一些时间以后,我个人做了一下的简单封装,算是简单的实现一个http service的功能,其中使用到了一点11的特性。可丢砖头,也可交流。
- // File: basic_http.h
- // Description: ---
- // Notes: ---
- // Author: Haust <wyy123_2008@qq.com>
- // Revision: 2015-11-19 by Haust
- #pragma once
- #include "mongoose.h"
- #include <map>
- #include <string>
- #include <functional>
- class BasicHttp {
- public:
- using handler = std::function<void(std::string, std::string)>;
- public:
- virtual ~BasicHttp(){};
- void Init(uint32_t port);
- bool Start();
- bool Close();
- bool RegisterHandler(std::string uri, handler f);
- void UnRegisterHandler(std::string uri);
- void Loop(int milli);
- void SendReply(std::string uri, std::string reply);
- void SendError(std::string uri, int errcode, std::string reply);
- protected:
- using handler_map = std::map<std::string, handler>;
- using connection_map = std::multimap<std::string, mg_connection*>;
- private:
- static void EvHandler(struct mg_connection* nc, int ev, void* ev_data);
- static void HandleRequst(struct mg_connection* nc, int ev, void *ev_data);
- public:
- static handler_map _handlers;
- static connection_map _connections;
- char _port[11];
- struct mg_mgr _mgr;
- };
- // File: basic_http.cpp
- // Description: ---
- // Notes: ---
- // Author: Haust <wyy123_2008@qq.com>
- // Revision: 2015-11-19 by Haust
- #include "basic_http.h"
- BasicHttp::handler_map BasicHttp::_handlers;
- BasicHttp::connection_map BasicHttp::_connections;
- void BasicHttp::Init(uint32_t port){
- memset(_port, 0, sizeof(_port));
- snprintf(_port, sizeof(_port), "%u", port);
- }
- bool BasicHttp::Start(){
- mg_mgr_init(&_mgr, NULL);
- auto nc = mg_bind(&_mgr, _port, EvHandler);
- if(nullptr == nc)
- return false;
- mg_set_protocol_http_websocket(nc);
- return true;
- }
- bool BasicHttp::Close(){
- mg_mgr_free(&_mgr);
- return true;
- }
- bool BasicHttp::RegisterHandler(std::string uri, handler f){
- auto it = _handlers.find(uri);
- if(_handlers.end() != it)
- return false;
- return _handlers.emplace(uri, f).second;
- }
- void BasicHttp::UnRegisterHandler(std::string uri){
- auto it = _handlers.find(uri);
- if(_handlers.end() != it)
- _handlers.erase(it);
- }
- void BasicHttp::Loop(int milli){
- mg_mgr_poll(&_mgr, milli);
- }
- void BasicHttp::EvHandler(struct mg_connection* nc, int ev, void* ev_data){
- switch(ev){
- case MG_EV_HTTP_REQUEST:
- HandleRequst(nc, ev, ev_data);
- break;
- default:
- break;
- }
- }
- void BasicHttp::HandleRequst(struct mg_connection *nc, int ev, void* ev_data){
- http_message* hm = (http_message*)ev_data;
- std::string uri(hm->uri.p, hm->uri.len);
- auto it = _handlers.find(uri);
- if(_handlers.end() == it)
- return;
- _connections.emplace(uri, nc);
- it->second(std::string(hm->query_string.p, hm->query_string.len),
- std::string(hm->body.p, hm->body.len));
- }
- void BasicHttp::SendReply(std::string uri, std::string reply){
- auto range = _connections.equal_range(uri);
- if(range.first == range.second)
- return;
- auto it = range.first;
- mg_printf(it->second, "HTTP/1.1 200 OK\r\niConnection: close\r\nContent-Type: text/html\r\nContent-Length: %u\r\n\r\n%s\r\n",
- (uint32_t)reply.length(), reply.c_str());
- it->second->flags |= MG_F_SEND_AND_CLOSE;
- _connections.erase(it);
- }
- void BasicHttp::SendError(std::string uri, int errcode, std::string reply){
- auto range = _connections.equal_range(uri);
- if(range.first == range.second)
- return;
- auto it = range.first;
- mg_printf(it->second, "HTTP/1.1 %d %s\r\n", errcode, reply.c_str());
- it->second->flags |= MG_F_SEND_AND_CLOSE;
- _connections.erase(it);
- }
- #include "mongoose.c"