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

Google format

Browse Source
This commit is contained in:
Ian Gulliver
2019-05-18 12:18:26 -07:00
parent 07c5e5661d
commit 83dfa81138
11 changed files with 397 additions and 250 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
stream.cc
View File

@@ -5,52 +5,50 @@
namespace firesse {
Stream::Stream(firecgi::Request* request, Index* index)
: request_(request),
index_(index) {
index_->Add(this);
: request_(request), index_(index) {
index_->Add(this);
}
Stream::~Stream() {
std::lock_guard l(mu_);
index_->Remove(this);
std::lock_guard l(mu_);
index_->Remove(this);
}
void Stream::OnClose(const std::function<void()>& callback) {
on_close_ = callback;
on_close_ = callback;
}
bool Stream::WriteEvent(const std::string_view& data, uint64_t id, const std::string& type) {
std::lock_guard l(mu_);
bool Stream::WriteEvent(const std::string_view& data, uint64_t id,
const std::string& type) {
std::lock_guard l(mu_);
index_->Freshen(this);
index_->Freshen(this);
return request_->InTransaction<bool>([=]() {
if (id) {
request_->WriteBody("id: ", std::to_string(id), "\n");
}
if (!type.empty()) {
request_->WriteBody("event: ", type, "\n");
}
request_->WriteBody("data: ", data, "\n\n");
return request_->Flush();
});
return request_->InTransaction<bool>([=]() {
if (id) {
request_->WriteBody("id: ", std::to_string(id), "\n");
}
if (!type.empty()) {
request_->WriteBody("event: ", type, "\n");
}
request_->WriteBody("data: ", data, "\n\n");
return request_->Flush();
});
}
bool Stream::WriteRaw(const std::string_view& data) {
std::lock_guard l(mu_);
std::lock_guard l(mu_);
request_->WriteBody(data);
return request_->Flush();
request_->WriteBody(data);
return request_->Flush();
}
bool Stream::End() {
return request_->End();
}
bool Stream::End() { return request_->End(); }
void Stream::Close() {
if (on_close_) {
on_close_();
}
if (on_close_) {
on_close_();
}
}
} // namespace firesse
} // namespace firesse