fz-mirrors/MaaAssistantArknights
0
0
Fork 0
mirror of https://github.com/MaaAssistantArknights/MaaAssistantArknights.git synced 2026-07-18 10:10:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

perf: 围绕 AsstHttp 的各种优化

Browse Source
This commit is contained in:
zzyyyl
2022-09-05 23:44:48 +08:00
parent d2e22059c9
commit 42ff5eaddf
5 changed files with 58 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
src/MeoAssistant/AsstHttp.hpp
View File

@@ -10,23 +10,25 @@
namespace asst::http
{
class Response : public std::string
class Response
{
private:
std::string m_raw_data;
std::string m_last_error;
unsigned m_status_code = 0;
std::string_view m_protocol_version;
std::string_view m_status_code_info;
std::string_view m_data;
std::string_view m_body;
std::unordered_map<std::string, std::string_view> m_headers;
static const inline std::unordered_set<std::string_view> _accepted_protocol_version = { "HTTP/1.1",
"HTTP/2.0" };
bool _analyze_status_line(std::string_view status_line)
{
size_t _word_count = 0;
for (const auto& word : utils::string_split(status_line, " ")) {
++_word_count;
if (_word_count == 1) {
static const std::unordered_set<std::string_view> accepted_protocol_version = { "HTTP/1.1" };
if (!accepted_protocol_version.contains(word)) {
if (!_accepted_protocol_version.contains(word)) {
m_last_error = "unsupported protocol version`" + std::string(word) + "`";
return false;
}
@@ -58,7 +60,7 @@ namespace asst::http
{
size_t _colon_pos = status_line.find(':');
if (_colon_pos == status_line.npos) {
m_last_error = "connot decode header `" + std::string(status_line) + "`";
m_last_error = "cannot decode header `" + std::string(status_line) + "`";
return false;
}
m_headers[utils::view_cast<std::string>(status_line.substr(0, _colon_pos) | views::transform([](char c) {
@@ -71,10 +73,11 @@ namespace asst::http
public:
template <typename... Args>
requires std::is_constructible_v<std::string, Args...>
Response(Args&&... args) : std::string(std::forward<Args>(args)...)
Response(Args&&... args) : m_raw_data(std::forward<Args>(args)...)
{
std::string_view this_view = *this;
std::string_view this_view = m_raw_data;
bool _is_status_line = true;
bool _is_data_line = false;
for (const auto& line : utils::string_split(this_view, "\r\n")) {
// status
if (_is_status_line) {
@@ -84,23 +87,37 @@ namespace asst::http
_is_status_line = false;
}
// headers
else if (!line.empty()) {
if (!_analyze_headers_line(line)) {
return;
else if (!_is_data_line) {
if (!line.empty()) {
if (!_analyze_headers_line(line)) {
return;
}
}
else {
_is_data_line = true; // 当前行是空行，则下一行开始都是 data 段
}
}
// data
else {
m_data = std::string_view(line.begin(), this_view.end());
m_body = std::string_view(line.begin(), this_view.end());
return;
}
}
}
Response() = default;
~Response() = default;
Response(Response&&) = default;
Response& operator=(Response&&) = default;
Response(const Response&) = delete;
Response& operator=(const Response&) = delete;
unsigned status_code() const { return m_status_code; }
std::string_view protocol_version() const { return m_protocol_version; }
std::string_view status_code_info() const { return m_status_code_info; }
std::string_view data() const { return m_data; }
std::string_view body() const { return m_body; }
std::optional<std::string_view> find_header(const std::string& key) const
{
if (auto iter = m_headers.find(key); iter != m_headers.cend()) {
@@ -111,7 +128,8 @@ namespace asst::http
}
}
const auto& headers() const { return m_headers; }
const std::string& get_last_error() const noexcept { return m_last_error; }
const std::string& get_last_error() const { return m_last_error; }
operator std::string() const { return m_raw_data; }
bool success() const { return m_status_code == 200; }
bool status_2xx() const { return m_status_code >= 200 && m_status_code < 300; }