近来在项目中需要实现一个http service的功能。虽然可以通过封装socket自己实现http的发送和解析。但考虑到目前网络上还是有大量的http的解析源码,自己再从头实现一番稍显麻烦。因此在网络上搜刮了一番,看到很多实现http的代码, 无一不体现了其轻量级的特点。然而,虽然轻量级,但从代码的量上来说,为了一个http service的功能,引入这许多的代码,也着实不是十分情愿呢。后在同事的推荐下选择了mongoose的代码。
mongoose的代码着实轻量,先看看它的特点:
1. 在整个的实现是使用c语言编写
2. 整个代码也只有一个mongoose.c和mongoose.h两个文件, 从引入第三方的考虑上也着实不多。
3. 实现的功能还是非常多的,从使用的层面上来说功能还是比较全面。只不过不知道是否是为了第三方使用的方便还是怎么地,它的代码只用了两个源文件罢了。诸多的功能也大以宏的开始与结束来区分。
4. 示例非常齐全,所有的功能都有单独的示例
然而,不管它实现多少功能,对于我来说只需要三个:
1. 有http的解析等
2. 文件少,使用方便,不需要因为使用一个简单的功能引入大量代码,而且引入的代码着实没有用到。
3. 有完整的示例
当我们拿到一个第三方库或者第三方源码的时候,第一件事情就是看看代码的示例,并且自己动手谢谢测试的代码,完成自己想要的功能。于是,我花了一点时间自己写了一个测试的代码,最后发现测试的时候并不通。虽然它的代码示例很全面,然而对于我们来说,或许在它的代码中,有些函数我们不需要,而有些函数却不再这个示例中使用,因此需要自己测试。耗费了一些时间以后,我个人做了一下的简单封装,算是简单的实现一个http service的功能,其中使用到了一点c++11的特性。可丢砖头,也可交流。
// File: basic_http.h // Description: --- // Notes: --- // Author: Haust <[email protected]> // 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 <[email protected]> // 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"