cpp/firecgi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Simplify API with recursive mutex

Browse Source
This commit is contained in:
flamingcow
2019-05-09 23:22:45 -07:00
parent 10bdc8e775
commit 8bd0813161
2 changed files with 17 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
request.cc
View File

@@ -54,7 +54,7 @@ const std::string_view& Request::GetBody() {
}
void Request::WriteHeader(const std::string_view& name, const std::string_view& value) {
std::lock_guard<std::mutex> l(output_mu_);
std::lock_guard<std::recursive_mutex> l(output_mu_);
CHECK(!body_written_);
CHECK(out_buf_.Write(name));
@@ -64,12 +64,17 @@ void Request::WriteHeader(const std::string_view& name, const std::string_view&
}
void Request::WriteBody(const std::string_view& body) {
std::lock_guard<std::mutex> l(output_mu_);
WriteBodyLocked(body);
std::lock_guard<std::recursive_mutex> l(output_mu_);
if (!body_written_) {
CHECK(out_buf_.Write("\n"));
body_written_ = true;
}
// TODO: make this able to span multiple packets
CHECK(out_buf_.Write(body));
}
bool Request::Flush() {
std::lock_guard<std::mutex> l(output_mu_);
std::lock_guard<std::recursive_mutex> l(output_mu_);
std::vector<iovec> vecs;
@@ -87,9 +92,9 @@ bool Request::Flush() {
}
bool Request::End() {
std::lock_guard<std::mutex> l(output_mu_);
std::lock_guard<std::recursive_mutex> l(output_mu_);
WriteBodyLocked("");
WriteBody("");
std::vector<iovec> vecs;
@@ -120,15 +125,4 @@ Header Request::OutputHeader() {
return Header(6, request_id_, out_buf_.ReadMaxLen());
}
void Request::WriteBodyLocked(const std::string_view& body) {
CHECK(!output_mu_.try_lock());
if (!body_written_) {
CHECK(out_buf_.Write("\n"));
body_written_ = true;
}
// TODO: make this able to span multiple packets
CHECK(out_buf_.Write(body));
}
} // namespace firecgi