mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-12-03 20:37:50 +08:00
151 lines
4.1 KiB
C++
151 lines
4.1 KiB
C++
/*
|
||
* Copyright(C) 2021 hikyuu.org
|
||
*
|
||
* Create on: 2021-02-28
|
||
* Author: fasiondog
|
||
*/
|
||
|
||
#pragma once
|
||
|
||
#include <string_view>
|
||
#include <vector>
|
||
#include <functional>
|
||
|
||
#include "HttpValidError.h"
|
||
#include "../common/json.hpp"
|
||
#include "../common/log.h"
|
||
|
||
using json = nlohmann::json; // 不保持插入排序
|
||
using ordered_json = nlohmann::ordered_json; // 保持插入排序
|
||
|
||
namespace hku {
|
||
|
||
// 仅内部使用
|
||
#define NNG_CHECK(rv) \
|
||
{ \
|
||
if (rv != 0) { \
|
||
HKU_THROW("[NNG_ERROR] {}", nng_strerror(rv)); \
|
||
} \
|
||
}
|
||
|
||
// 仅内部使用
|
||
#define NNG_CHECK_M(rv, msg) \
|
||
{ \
|
||
if (rv != 0) { \
|
||
HKU_THROW("[HTTP_ERROR] {} err: {}", msg, nng_strerror(rv)); \
|
||
} \
|
||
}
|
||
|
||
class HttpHandle {
|
||
CLASS_LOGGER(HttpHandle)
|
||
|
||
public:
|
||
HttpHandle() = delete;
|
||
HttpHandle(nng_aio *aio);
|
||
|
||
/** 前处理 */
|
||
virtual void before_run() {}
|
||
|
||
/** 响应处理 */
|
||
virtual void run() = 0;
|
||
|
||
void addFilter(std::function<void(HttpHandle *)> filter) {
|
||
m_filters.push_back(filter);
|
||
}
|
||
|
||
/**
|
||
* 获取请求头部信息
|
||
* @param name 头部信息名称
|
||
* @return 如果获取不到将返回 NULL
|
||
*/
|
||
const char *getReqHeader(const char *name) {
|
||
return nng_http_req_get_header(m_nng_req, name);
|
||
}
|
||
|
||
std::string getReqHeader(const std::string &name) {
|
||
const char *head = nng_http_req_get_header(m_nng_req, name.c_str());
|
||
return head ? std::string(head) : std::string();
|
||
}
|
||
|
||
/**
|
||
* 获取请求数据
|
||
* @param[out] data 请求中的数据起始地址，无数据时返回 NULL
|
||
* @param[out] len 请求中的数据长度
|
||
* @note 请求中无数据时, len返回的长度可能不为0
|
||
*/
|
||
void getReqData(void **data, size_t *len) {
|
||
nng_http_req_get_data(m_nng_req, data, len);
|
||
}
|
||
|
||
std::string getReqData();
|
||
|
||
json getReqJson() {
|
||
void *data;
|
||
size_t len;
|
||
nng_http_req_get_data(m_nng_req, &data, &len);
|
||
HTTP_VALID_CHECK(data, INVALID_JSON_REQUEST, "Req data is empty!");
|
||
json result;
|
||
try {
|
||
result = json::parse((const char *)data);
|
||
} catch (json::exception &e) {
|
||
throw HttpValidError(INVALID_JSON_REQUEST, e.what());
|
||
}
|
||
return result;
|
||
}
|
||
|
||
/**
|
||
* 请求的 ulr 中是否包含 query 参数
|
||
*/
|
||
bool haveQueryParams();
|
||
|
||
typedef std::unordered_map<std::string, std::string> QueryParams;
|
||
|
||
/**
|
||
* 获取 query 参数
|
||
* @param query_params [out] 输出 query 参数
|
||
* @return true | false 获取或解析失败
|
||
*/
|
||
bool getQueryParams(QueryParams &query_params);
|
||
|
||
void setResStatus(uint16_t status) {
|
||
NNG_CHECK(nng_http_res_set_status(m_nng_res, status));
|
||
}
|
||
|
||
void setResHeader(const char *key, const char *val) {
|
||
NNG_CHECK(nng_http_res_set_header(m_nng_res, key, val));
|
||
}
|
||
|
||
void setResData(const char *data) {
|
||
NNG_CHECK(nng_http_res_copy_data(m_nng_res, data, strlen(data)));
|
||
}
|
||
|
||
void setResData(const std::string &content) {
|
||
NNG_CHECK(nng_http_res_copy_data(m_nng_res, content.data(), content.size()));
|
||
}
|
||
|
||
void setResData(const json &data) {
|
||
std::string x = data.dump();
|
||
setResData(x);
|
||
}
|
||
|
||
void setResData(const ordered_json &data) {
|
||
std::string x = data.dump();
|
||
setResData(x);
|
||
}
|
||
|
||
void operator()();
|
||
|
||
private:
|
||
// error未捕获的信息，统一返回500页面
|
||
void unknown_error(const std::string &errmsg);
|
||
|
||
protected:
|
||
nng_aio *m_http_aio{nullptr};
|
||
nng_http_res *m_nng_res{nullptr};
|
||
nng_http_req *m_nng_req{nullptr};
|
||
nng_http_conn *m_nng_conn{nullptr};
|
||
std::vector<std::function<void(HttpHandle *)>> m_filters;
|
||
};
|
||
|
||
} // namespace hku
|