qing
81ba942b8e
- Add libmsg: C++20 header-only pymsg protocol implementation - Msg.hpp: pmsg_send/pmsg_recv (blocking + deadline modes) + MsgClient - MsgServer.hpp: multi-client TCP server with select event loop + thread pool - AsyncQueue<T>: generic async batch processing queue - RpcClient/RpcServer: JSON RPC convention layer (requires boost::json) - Msg_Basic.hpp/cpp: exception hierarchy, constants, version - Fix thread safety: MsgClient _sockfd/_running use std::atomic with exchange() - Fix deadline recv: client call() uses shared deadline for header+payload - Fix graceful shutdown: thread pool drains queued tasks before exit - Fix pmsg_send size check: enforce 64MiB limit (was 4GiB no-op check) - Align AsyncQueue batch semantics with Python: batch_size is threshold - Refactor gateway/LogC.hpp to use libmsg AsyncQueue<T> - Update gateway Makefile to link -lus_msg, DbC.hpp to use <us/Msg.hpp> - Add libmsg to _inst.sh build order
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;
|
||
};
|
||
}
|