firesse/stream.cc

#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