qing
59c5fe1feb
- 新增 ConfsTask（ITask，与 HttpTask 同构）：ConfigStore + INI/JSON 取值 + get/list/reload/health 四路由，协议与 Python confs 逐字节兼容 - 新增 ConfsLogC 异步日志客户端（沿用 confs2 LogC，服务名 confs） - main.cpp 新增 --confs <port> 选项与 CONFS_* 环境变量；不传则行为不变 - Makefile 链接 -lus_msg，版本 0.0.4 → 0.1.0 - _run.sh 启用 --confs 8010；_set.sh 自建 configs → ../confs/configs 链接 - init.txt 移除 confs2 自启动行（由 core 内嵌承担） - 黑盒对比测试 16/16 通过（tests/blackbox/test_core_confs.py）
133 lines
3.7 KiB
C++
133 lines
3.7 KiB
C++
/*
|
||
* ConfsLogC.hpp — core 内嵌 confs 服务的异步日志客户端
|
||
*
|
||
* 基于 libmsg AsyncQueue，异步批量推送日志到 db:8005 的 Logs 表。
|
||
* - 内存队列 + 后台线程攒批（满 50 条或 2 秒 flush）
|
||
* - fail-safe：db 不可用时静默丢弃
|
||
* - API: info() / warning() / error()
|
||
*
|
||
* 改编自 confs2/inc/LogC.hpp，仅改 namespace（qing::confs），
|
||
* 服务名保持 "confs" 与 Python/standalone confs2 日志连续。
|
||
*/
|
||
#pragma once
|
||
|
||
#include <us/Msg.hpp>
|
||
#include <us/AsyncQueue.hpp>
|
||
#include <boost/json.hpp>
|
||
#include <iostream>
|
||
#include <string>
|
||
#include <vector>
|
||
#include <chrono>
|
||
#include <cstdlib>
|
||
#include <algorithm>
|
||
#include <cctype>
|
||
|
||
namespace qing {
|
||
namespace confs {
|
||
|
||
struct ConfsLogEntry {
|
||
std::string ts;
|
||
std::string service;
|
||
std::string level;
|
||
std::string message;
|
||
std::string trace_id;
|
||
std::string extra;
|
||
};
|
||
|
||
class ConfsLogC {
|
||
public:
|
||
static ConfsLogC& instance() {
|
||
static ConfsLogC inst;
|
||
return inst;
|
||
}
|
||
|
||
void info(const std::string& svc, const std::string& msg) {
|
||
push(svc, "INFO", msg);
|
||
}
|
||
void warning(const std::string& svc, const std::string& msg) {
|
||
push(svc, "WARNING", msg);
|
||
}
|
||
void error(const std::string& svc, const std::string& msg) {
|
||
push(svc, "ERROR", msg);
|
||
}
|
||
|
||
private:
|
||
ConfsLogC() {
|
||
const char* host = std::getenv("DB_HOST");
|
||
const char* port = std::getenv("DB_PORT");
|
||
db_host_ = host ? host : "127.0.0.1";
|
||
db_port_ = port ? std::atoi(port) : 8005;
|
||
|
||
queue_ = std::make_unique<pmc::net::AsyncQueue<ConfsLogEntry>>(
|
||
[this](std::vector<ConfsLogEntry>&& batch) { send_batch(std::move(batch)); },
|
||
pmc::net::AsyncQueueConfig{.batch_size = 50, .flush_interval = 2.0, .max_queue_size = 10000},
|
||
[](const std::exception& e, std::vector<ConfsLogEntry>&&) {
|
||
std::cerr << "[core/confs-log] send failed: " << e.what() << std::endl;
|
||
});
|
||
queue_->start();
|
||
}
|
||
|
||
~ConfsLogC() { if (queue_) queue_->shutdown(); }
|
||
|
||
// P2-3 修复：过滤控制字符，防止日志注入
|
||
static std::string sanitize_for_log(std::string s) {
|
||
s.erase(std::remove_if(s.begin(), s.end(), [](unsigned char c) {
|
||
return c < 0x20 && c != '\t';
|
||
}), s.end());
|
||
return s;
|
||
}
|
||
|
||
void push(const std::string& svc, const std::string& level, const std::string& msg) {
|
||
auto now = std::chrono::system_clock::now();
|
||
auto t = std::chrono::system_clock::to_time_t(now);
|
||
// 系统本地时间 + 时区偏移
|
||
std::tm* lt = std::localtime(&t);
|
||
char buf[40];
|
||
std::strftime(buf, sizeof(buf), "%Y-%m-%dT%H:%M:%S", lt);
|
||
char tz_raw[8];
|
||
std::strftime(tz_raw, sizeof(tz_raw), "%z", lt);
|
||
std::string tz = tz_raw;
|
||
if (tz.length() >= 5)
|
||
tz = tz.substr(0, 3) + ":" + tz.substr(3);
|
||
|
||
ConfsLogEntry e;
|
||
e.ts = std::string(buf) + tz;
|
||
e.service = sanitize_for_log(svc);
|
||
e.level = level;
|
||
e.message = sanitize_for_log(msg);
|
||
queue_->push(std::move(e));
|
||
}
|
||
|
||
void send_batch(std::vector<ConfsLogEntry>&& batch) {
|
||
try {
|
||
pmc::net::MsgClient client(db_host_, db_port_, "confs-log");
|
||
for (auto& e : batch) {
|
||
boost::json::object vals;
|
||
vals["ts"] = e.ts;
|
||
vals["service"] = e.service;
|
||
vals["level"] = e.level;
|
||
vals["message"] = e.message;
|
||
vals["trace_id"] = e.trace_id.empty() ? nullptr : boost::json::value(e.trace_id);
|
||
vals["extra"] = e.extra.empty() ? nullptr : boost::json::value(e.extra);
|
||
|
||
boost::json::object req;
|
||
req["_route"] = "execute";
|
||
req["action"] = "insert";
|
||
req["table"] = "Logs";
|
||
req["values"] = vals;
|
||
|
||
client.call(boost::json::serialize(req), 1.0);
|
||
}
|
||
} catch (const std::exception& e) {
|
||
std::cerr << "[core/confs-log] batch drop: " << e.what() << std::endl;
|
||
}
|
||
}
|
||
|
||
std::string db_host_;
|
||
int db_port_;
|
||
std::unique_ptr<pmc::net::AsyncQueue<ConfsLogEntry>> queue_;
|
||
};
|
||
|
||
} // namespace confs
|
||
} // namespace qing
|