55 lines
1.1 KiB
C++
55 lines
1.1 KiB
C++
#include "stream.h"
|
|
|
|
#include "index.h"
|
|
|
|
namespace firesse {
|
|
|
|
Stream::Stream(firecgi::Request* request, Index* index)
|
|
: request_(request), index_(index) {
|
|
index_->Add(this);
|
|
}
|
|
|
|
Stream::~Stream() {
|
|
std::lock_guard l(mu_);
|
|
index_->Remove(this);
|
|
}
|
|
|
|
void Stream::OnClose(const std::function<void()>& callback) {
|
|
on_close_ = callback;
|
|
}
|
|
|
|
bool Stream::WriteEvent(const std::string_view& data, uint64_t id,
|
|
const std::string& type) {
|
|
std::lock_guard l(mu_);
|
|
|
|
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();
|
|
});
|
|
}
|
|
|
|
bool Stream::WriteRaw(const std::string_view& data) {
|
|
std::lock_guard l(mu_);
|
|
|
|
request_->WriteBody(data);
|
|
return request_->Flush();
|
|
}
|
|
|
|
bool Stream::End() { return request_->End(); }
|
|
|
|
void Stream::Close() {
|
|
if (on_close_) {
|
|
on_close_();
|
|
}
|
|
}
|
|
|
|
} // namespace firesse
|