qing
1d44e0dcb8
- Add libmsg module: pymsg protocol implementation (4-byte BE length prefix + UTF-8) - Add AsyncQueue<T>: generic async batch queue with backpressure (aligned with Python AsyncRpcQueue) - Add MsgServer: select-based multi-client TCP server with optional thread pool - Add RpcClient/RpcServer: JSON RPC convention layer with automatic route injection - Refactor gateway/LogC.hpp to use AsyncQueue for async batch log delivery - Update gateway/DbC.hpp to add call_msg() using pymsg TCP protocol - Remove duplicated gateway/inc/Msg.hpp (now provided by libmsg) - Update gateway/Makefile and _inst.sh build order Fixes included: - pmsg_send size check uses kMaxMsgSize (64MiB) instead of 4GiB - MsgClient._running uses std::atomic<bool> for thread safety - Deadline-based recv timeout to prevent Slowloris (header+payload shared deadline)
58 lines
2.1 KiB
C++
58 lines
2.1 KiB
C++
#pragma once
|
||
#include <us/HttpClient.hpp>
|
||
#include <us/Msg.hpp>
|
||
#include <boost/json.hpp>
|
||
namespace qing {
|
||
namespace json = boost::json;
|
||
class DbC {
|
||
public:
|
||
/* 本机 db 轻量查询的默认超时（秒）；慢操作由调用方显式放宽。
|
||
* smtp/captcha 等其他下游的超时常量见 main.cpp。 */
|
||
static constexpr double DEFAULT_TIMEOUT = 5.0;
|
||
|
||
/// 初始化 db 服务地址（启动时从配置设置，默认 127.0.0.1:8005）
|
||
static void init(const std::string& host, int port) {
|
||
_host = host;
|
||
_port = port;
|
||
}
|
||
|
||
// 【保留】旧版 HTTP 接口：通过 HttpClient POST 到 dbv2 HTTP /execute 端点。
|
||
// 保留用于兼容性参考，新代码优先使用 call_msg()（pymsg TCP 协议，更可靠）。
|
||
static json::value call_http(json::value& params) {
|
||
|
||
auto client = pmc::net::HttpClient();
|
||
client.set_header("Content-Type: application/json");
|
||
std::string payload = json::serialize(params);
|
||
std::string res = "";
|
||
std::string url = "http://" + _host + ":" + std::to_string(_port) + "/execute";
|
||
client.post_stream_sync(url, payload, [&res](const auto& ret) {
|
||
res += ret;
|
||
});
|
||
return json::parse(res);
|
||
}
|
||
|
||
// 新版 pymsg 接口：通过 MsgClient（TCP 长度前缀帧协议）调用 dbv2。
|
||
// 自动添加 _route = "execute"，返回 dbv2 pymsg 完整 JSON 响应
|
||
// {"success":..., "data":[...]}。
|
||
// 响应格式与 call_http 兼容，调用方可直接替换。
|
||
// db 是强依赖，启用 retry_on_refused：db 未启动时阻塞等待，避免启动顺序问题。
|
||
static json::value call_msg(const json::value& params,
|
||
double timeout = DEFAULT_TIMEOUT) {
|
||
json::value jv_route = params;
|
||
jv_route.as_object()["_route"] = "execute";
|
||
pmc::net::MsgClient client(_host, _port, "gateway");
|
||
std::string resp = client.call(
|
||
json::serialize(jv_route), timeout,
|
||
/*retry_on_refused=*/true);
|
||
return json::parse(resp);
|
||
}
|
||
|
||
static const std::string& host() { return _host; }
|
||
static int port() { return _port; }
|
||
|
||
private:
|
||
static inline std::string _host = "127.0.0.1";
|
||
static inline int _port = 8005;
|
||
};
|
||
}
|