mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-12-05 05:17:47 +08:00
82 lines
2.0 KiB
C++
82 lines
2.0 KiB
C++
/*
|
|
* Copyright(C) 2021 hikyuu.org
|
|
*
|
|
* Create on: 2021-02-28
|
|
* Author: fasiondog
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <string>
|
|
#include <unordered_set>
|
|
#include <hikyuu/utilities/thread/MQStealThreadPool.h>
|
|
#include <hikyuu/utilities/thread/FuncWrapper.h>
|
|
#include "HttpHandle.h"
|
|
|
|
namespace hku {
|
|
|
|
class HttpServer {
|
|
CLASS_LOGGER(HttpServer)
|
|
|
|
public:
|
|
HttpServer(const char *host, uint16_t port);
|
|
virtual ~HttpServer();
|
|
|
|
static void start();
|
|
static void stop();
|
|
|
|
/**
|
|
* 设置 handle 无法捕获的错误返回信息，如 404
|
|
* @param http_status http状态码
|
|
* @param body 返回消息
|
|
*/
|
|
static void set_error_msg(int16_t http_status, const std::string &body);
|
|
|
|
template <typename Handle>
|
|
void GET(const char *path) {
|
|
regHandle("GET", path, [](nng_aio *aio) { ms_tg.submit(Handle(aio)); });
|
|
}
|
|
|
|
template <typename Handle>
|
|
void POST(const char *path) {
|
|
regHandle("POST", path, [](nng_aio *aio) { ms_tg.submit(Handle(aio)); });
|
|
}
|
|
|
|
template <typename Handle>
|
|
void PUT(const char *path) {
|
|
regHandle("PUT", path, [](nng_aio *aio) { ms_tg.submit(Handle(aio)); });
|
|
}
|
|
|
|
template <typename Handle>
|
|
void DEL(const char *path) {
|
|
regHandle("DELETE", path, [](nng_aio *aio) { ms_tg.submit(Handle(aio)); });
|
|
}
|
|
|
|
template <typename Handle>
|
|
void PATCH(const char *path) {
|
|
regHandle("PATCH", path, [](nng_aio *aio) { ms_tg.submit(Handle(aio)); });
|
|
}
|
|
|
|
template <typename Handle>
|
|
void regHandle(const char *method, const char *path) {
|
|
regHandle(method, path, [](nng_aio *aio) { ms_tg.submit(Handle(aio)); });
|
|
}
|
|
|
|
private:
|
|
void regHandle(const char *method, const char *path, void (*rest_handle)(nng_aio *));
|
|
|
|
private:
|
|
std::string m_root_url;
|
|
std::string m_host;
|
|
uint16_t m_port;
|
|
|
|
private:
|
|
static void http_exit();
|
|
static void signal_handler(int signal);
|
|
|
|
private:
|
|
static nng_http_server *ms_server;
|
|
static MQStealThreadPool ms_tg;
|
|
};
|
|
|
|
} // namespace hku
|