aunix/ZLMediaKit
1
0
Fork 0
mirror of https://gitee.com/xia-chu/ZLMediaKit.git synced 2026-05-20 08:47:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

支持打印python http api调试日志

Browse Source
This commit is contained in:
xia-chu 2026-03-18 15:59:27 +08:00
parent fa7b0639d8
commit 752f705b36
2 changed files with 28 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
server/WebApi.h
View File

@ -56,6 +56,7 @@ typedef enum {
extern const std::string kSecret; extern const std::string kSecret;
extern const std::string kLegacyAuth; extern const std::string kLegacyAuth;
extern const std::string kApiDebug;
} // namespace API } // namespace API
class ApiRetException : public std::runtime_error { class ApiRetException : public std::runtime_error {

28
server/pyinvoker.cpp
View File

@ -13,6 +13,7 @@
#include "Util/File.h" #include "Util/File.h"
#include "Common/Parser.h" #include "Common/Parser.h"
#include "Http/HttpSession.h" #include "Http/HttpSession.h"
#include "WebApi.h"
using namespace toolkit; using namespace toolkit;
using namespace mediakit; using namespace mediakit;
@ -102,6 +103,30 @@ mINI to_native(const py::dict &opt) {
return ret; return ret;
} }
void python_api_debug(const Parser &parser, const std::string &body) {
GET_CONFIG(bool, api_debug, API::kApiDebug);
if (!api_debug) {
return;
}
ssize_t size = body.size();
LogContextCapture log(getLogger(), toolkit::LDebug, __FILE__, "python http api debug", __LINE__);
log << "\r\n# request:\r\n" << parser.method() << " " << parser.fullUrl() << "\r\n";
log << "# header:\r\n";
for (auto &pr : parser.getHeader()) {
log << pr.first << " : " << pr.second << "\r\n";
}
auto &content = parser.content();
log << "# content:\r\n" << (content.size() > 4 * 1024 ? content.substr(0, 4 * 1024) : content) << "\r\n";
if (size > 0 && size < 4 * 1024) {
log << "# response:\r\n" << body << "\r\n";
} else {
log << "# response size:" << size << "\r\n";
}
}
void handle_http_request(const py::object &check_route, const py::object &submit_coro, const Parser &parser, const HttpSession::HttpResponseInvoker &invoker, bool &consumed, toolkit::SockInfo &sender) { void handle_http_request(const py::object &check_route, const py::object &submit_coro, const Parser &parser, const HttpSession::HttpResponseInvoker &invoker, bool &consumed, toolkit::SockInfo &sender) {
py::gil_scoped_acquire guard; py::gil_scoped_acquire guard;
@ -150,7 +175,7 @@ void handle_http_request(const py::object &check_route, const py::object &submit
StrCaseMap resp_headers; StrCaseMap resp_headers;
std::string resp_body; std::string resp_body;
int status = 500; int status = 500;
auto send = py::cpp_function([invoker, status, resp_body, resp_headers](const py::dict &msg) mutable { auto send = py::cpp_function([parser, invoker, status, resp_body, resp_headers](const py::dict &msg) mutable {
auto type = msg["type"].cast<std::string>(); auto type = msg["type"].cast<std::string>();
if (type == "http.response.start") { if (type == "http.response.start") {
status = msg["status"].cast<int>(); status = msg["status"].cast<int>();
@ -166,6 +191,7 @@ void handle_http_request(const py::object &check_route, const py::object &submit
// 💥 只在 more_body=False 时回调 // 💥 只在 more_body=False 时回调
bool more = msg.contains("more_body") && msg["more_body"].cast<bool>(); bool more = msg.contains("more_body") && msg["more_body"].cast<bool>();
if (!more) { if (!more) {
python_api_debug(parser, resp_body);
invoker(status, resp_headers, resp_body); invoker(status, resp_headers, resp_body);
} }
} }