267 lines
9.6 KiB
C
267 lines
9.6 KiB
C
|
#ifndef HV_HTTP_SERVICE_H_
|
||
|
|
#define HV_HTTP_SERVICE_H_
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
#include <map>
|
||
|
|
#include <unordered_map>
|
||
|
|
#include <list>
|
||
|
|
#include <memory>
|
||
|
|
#include <functional>
|
||
|
|
|
||
|
|
#include "hexport.h"
|
||
|
|
#include "HttpMessage.h"
|
||
|
|
#include "HttpResponseWriter.h"
|
||
|
|
#include "HttpContext.h"
|
||
|
|
|
||
|
|
#define DEFAULT_BASE_URL "/api/v1"
|
||
|
|
#define DEFAULT_DOCUMENT_ROOT "/var/www/html"
|
||
|
|
#define DEFAULT_HOME_PAGE "index.html"
|
||
|
|
#define DEFAULT_ERROR_PAGE "error.html"
|
||
|
|
#define DEFAULT_INDEXOF_DIR "/downloads/"
|
||
|
|
#define DEFAULT_KEEPALIVE_TIMEOUT 75000 // ms
|
||
|
|
|
||
|
|
// for FileCache
|
||
|
|
#define MAX_FILE_CACHE_SIZE (1 << 22) // 4M
|
||
|
|
#define DEFAULT_FILE_CACHE_STAT_INTERVAL 10 // s
|
||
|
|
#define DEFAULT_FILE_CACHE_EXPIRED_TIME 60 // s
|
||
|
|
|
||
|
|
/*
|
||
|
|
* @param[in] req: parsed structured http request
|
||
|
|
* @param[out] resp: structured http response
|
||
|
|
* @return 0: handle unfinished
|
||
|
|
* http_status_code: handle done
|
||
|
|
*/
|
||
|
|
#define HTTP_STATUS_UNFINISHED 0
|
||
|
|
// NOTE: http_sync_handler run on IO thread
|
||
|
|
typedef std::function<int(HttpRequest* req, HttpResponse* resp)> http_sync_handler;
|
||
|
|
// NOTE: http_async_handler run on hv::async threadpool
|
||
|
|
typedef std::function<void(const HttpRequestPtr& req, const HttpResponseWriterPtr& writer)> http_async_handler;
|
||
|
|
// NOTE: http_ctx_handler run on IO thread, you can easily post HttpContextPtr to your consumer thread for processing.
|
||
|
|
typedef std::function<int(const HttpContextPtr& ctx)> http_ctx_handler;
|
||
|
|
|
||
|
|
struct http_handler {
|
||
|
|
http_sync_handler sync_handler;
|
||
|
|
http_async_handler async_handler;
|
||
|
|
http_ctx_handler ctx_handler;
|
||
|
|
|
||
|
|
http_handler() {}
|
||
|
|
http_handler(http_sync_handler fn) : sync_handler(std::move(fn)) {}
|
||
|
|
http_handler(http_async_handler fn) : async_handler(std::move(fn)) {}
|
||
|
|
http_handler(http_ctx_handler fn) : ctx_handler(std::move(fn)) {}
|
||
|
|
http_handler(const http_handler& rhs)
|
||
|
|
: sync_handler(std::move(rhs.sync_handler))
|
||
|
|
, async_handler(std::move(rhs.async_handler))
|
||
|
|
, ctx_handler(std::move(rhs.ctx_handler))
|
||
|
|
{}
|
||
|
|
|
||
|
|
const http_handler& operator=(http_sync_handler fn) {
|
||
|
|
sync_handler = std::move(fn);
|
||
|
|
return *this;
|
||
|
|
}
|
||
|
|
const http_handler& operator=(http_async_handler fn) {
|
||
|
|
async_handler = std::move(fn);
|
||
|
|
return *this;
|
||
|
|
}
|
||
|
|
const http_handler& operator=(http_ctx_handler fn) {
|
||
|
|
ctx_handler = std::move(fn);
|
||
|
|
return *this;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool isNull() {
|
||
|
|
return sync_handler == NULL &&
|
||
|
|
async_handler == NULL &&
|
||
|
|
ctx_handler == NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
operator bool() {
|
||
|
|
return !isNull();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
struct http_method_handler {
|
||
|
|
http_method method;
|
||
|
|
http_handler handler;
|
||
|
|
|
||
|
|
http_method_handler() {}
|
||
|
|
http_method_handler(http_method m, const http_handler& h) : method(m), handler(h) {}
|
||
|
|
};
|
||
|
|
|
||
|
|
// method => http_method_handler
|
||
|
|
typedef std::list<http_method_handler> http_method_handlers;
|
||
|
|
// path => http_method_handlers
|
||
|
|
typedef std::unordered_map<std::string, std::shared_ptr<http_method_handlers>> http_api_handlers;
|
||
|
|
|
||
|
|
namespace hv {
|
||
|
|
|
||
|
|
struct HV_EXPORT HttpService {
|
||
|
|
// preprocessor -> processor -> postprocessor
|
||
|
|
http_handler preprocessor;
|
||
|
|
// processor: api_handlers -> staticHandler -> errorHandler
|
||
|
|
http_handler processor;
|
||
|
|
http_handler postprocessor;
|
||
|
|
|
||
|
|
// api service (that is http.APIServer)
|
||
|
|
std::string base_url;
|
||
|
|
http_api_handlers api_handlers;
|
||
|
|
|
||
|
|
// file service (that is http.FileServer)
|
||
|
|
http_handler staticHandler;
|
||
|
|
http_handler largeFileHandler;
|
||
|
|
std::string document_root;
|
||
|
|
std::string home_page;
|
||
|
|
std::string error_page;
|
||
|
|
// indexof service (that is http.DirectoryServer)
|
||
|
|
std::string index_of;
|
||
|
|
http_handler errorHandler;
|
||
|
|
|
||
|
|
// options
|
||
|
|
int keepalive_timeout;
|
||
|
|
int max_file_cache_size; // cache small file
|
||
|
|
int file_cache_stat_interval; // stat file is modified
|
||
|
|
int file_cache_expired_time; // remove expired file cache
|
||
|
|
/*
|
||
|
|
* @test limit_rate
|
||
|
|
* @build make examples
|
||
|
|
* @server bin/httpd -c etc/httpd.conf -s restart -d
|
||
|
|
* @client bin/wget http://127.0.0.1:8080/downloads/test.zip
|
||
|
|
*/
|
||
|
|
int limit_rate; // limit send rate, unit: KB/s
|
||
|
|
|
||
|
|
HttpService() {
|
||
|
|
// base_url = DEFAULT_BASE_URL;
|
||
|
|
|
||
|
|
document_root = DEFAULT_DOCUMENT_ROOT;
|
||
|
|
home_page = DEFAULT_HOME_PAGE;
|
||
|
|
// error_page = DEFAULT_ERROR_PAGE;
|
||
|
|
// index_of = DEFAULT_INDEXOF_DIR;
|
||
|
|
|
||
|
|
keepalive_timeout = DEFAULT_KEEPALIVE_TIMEOUT;
|
||
|
|
max_file_cache_size = MAX_FILE_CACHE_SIZE;
|
||
|
|
file_cache_stat_interval = DEFAULT_FILE_CACHE_STAT_INTERVAL;
|
||
|
|
file_cache_expired_time = DEFAULT_FILE_CACHE_EXPIRED_TIME;
|
||
|
|
limit_rate = -1; // unlimited
|
||
|
|
}
|
||
|
|
|
||
|
|
void AddApi(const char* path, http_method method, const http_handler& handler);
|
||
|
|
// @retval 0 OK, else HTTP_STATUS_NOT_FOUND, HTTP_STATUS_METHOD_NOT_ALLOWED
|
||
|
|
int GetApi(const char* url, http_method method, http_handler** handler);
|
||
|
|
// RESTful API /:field/ => req->query_params["field"]
|
||
|
|
int GetApi(HttpRequest* req, http_handler** handler);
|
||
|
|
|
||
|
|
hv::StringList Paths() {
|
||
|
|
hv::StringList paths;
|
||
|
|
for (auto& pair : api_handlers) {
|
||
|
|
paths.emplace_back(pair.first);
|
||
|
|
}
|
||
|
|
return paths;
|
||
|
|
}
|
||
|
|
|
||
|
|
// github.com/gin-gonic/gin
|
||
|
|
void Handle(const char* httpMethod, const char* relativePath, http_sync_handler handlerFunc) {
|
||
|
|
AddApi(relativePath, http_method_enum(httpMethod), http_handler(handlerFunc));
|
||
|
|
}
|
||
|
|
void Handle(const char* httpMethod, const char* relativePath, http_async_handler handlerFunc) {
|
||
|
|
AddApi(relativePath, http_method_enum(httpMethod), http_handler(handlerFunc));
|
||
|
|
}
|
||
|
|
void Handle(const char* httpMethod, const char* relativePath, http_ctx_handler handlerFunc) {
|
||
|
|
AddApi(relativePath, http_method_enum(httpMethod), http_handler(handlerFunc));
|
||
|
|
}
|
||
|
|
|
||
|
|
// HEAD
|
||
|
|
void HEAD(const char* relativePath, http_sync_handler handlerFunc) {
|
||
|
|
Handle("HEAD", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void HEAD(const char* relativePath, http_async_handler handlerFunc) {
|
||
|
|
Handle("HEAD", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void HEAD(const char* relativePath, http_ctx_handler handlerFunc) {
|
||
|
|
Handle("HEAD", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
|
||
|
|
// GET
|
||
|
|
void GET(const char* relativePath, http_sync_handler handlerFunc) {
|
||
|
|
Handle("GET", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void GET(const char* relativePath, http_async_handler handlerFunc) {
|
||
|
|
Handle("GET", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void GET(const char* relativePath, http_ctx_handler handlerFunc) {
|
||
|
|
Handle("GET", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
|
||
|
|
// POST
|
||
|
|
void POST(const char* relativePath, http_sync_handler handlerFunc) {
|
||
|
|
Handle("POST", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void POST(const char* relativePath, http_async_handler handlerFunc) {
|
||
|
|
Handle("POST", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void POST(const char* relativePath, http_ctx_handler handlerFunc) {
|
||
|
|
Handle("POST", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
|
||
|
|
// PUT
|
||
|
|
void PUT(const char* relativePath, http_sync_handler handlerFunc) {
|
||
|
|
Handle("PUT", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void PUT(const char* relativePath, http_async_handler handlerFunc) {
|
||
|
|
Handle("PUT", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void PUT(const char* relativePath, http_ctx_handler handlerFunc) {
|
||
|
|
Handle("PUT", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
|
||
|
|
// DELETE
|
||
|
|
// NOTE: Windows <winnt.h> #define DELETE as a macro, we have to replace DELETE with Delete.
|
||
|
|
void Delete(const char* relativePath, http_sync_handler handlerFunc) {
|
||
|
|
Handle("DELETE", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void Delete(const char* relativePath, http_async_handler handlerFunc) {
|
||
|
|
Handle("DELETE", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void Delete(const char* relativePath, http_ctx_handler handlerFunc) {
|
||
|
|
Handle("DELETE", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
|
||
|
|
// PATCH
|
||
|
|
void PATCH(const char* relativePath, http_sync_handler handlerFunc) {
|
||
|
|
Handle("PATCH", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void PATCH(const char* relativePath, http_async_handler handlerFunc) {
|
||
|
|
Handle("PATCH", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void PATCH(const char* relativePath, http_ctx_handler handlerFunc) {
|
||
|
|
Handle("PATCH", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Any
|
||
|
|
void Any(const char* relativePath, http_sync_handler handlerFunc) {
|
||
|
|
Handle("HEAD", relativePath, handlerFunc);
|
||
|
|
Handle("GET", relativePath, handlerFunc);
|
||
|
|
Handle("POST", relativePath, handlerFunc);
|
||
|
|
Handle("PUT", relativePath, handlerFunc);
|
||
|
|
Handle("DELETE", relativePath, handlerFunc);
|
||
|
|
Handle("PATCH", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void Any(const char* relativePath, http_async_handler handlerFunc) {
|
||
|
|
Handle("HEAD", relativePath, handlerFunc);
|
||
|
|
Handle("GET", relativePath, handlerFunc);
|
||
|
|
Handle("POST", relativePath, handlerFunc);
|
||
|
|
Handle("PUT", relativePath, handlerFunc);
|
||
|
|
Handle("DELETE", relativePath, handlerFunc);
|
||
|
|
Handle("PATCH", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
void Any(const char* relativePath, http_ctx_handler handlerFunc) {
|
||
|
|
Handle("HEAD", relativePath, handlerFunc);
|
||
|
|
Handle("GET", relativePath, handlerFunc);
|
||
|
|
Handle("POST", relativePath, handlerFunc);
|
||
|
|
Handle("PUT", relativePath, handlerFunc);
|
||
|
|
Handle("DELETE", relativePath, handlerFunc);
|
||
|
|
Handle("PATCH", relativePath, handlerFunc);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif // HV_HTTP_SERVICE_H_
|