firesse/stream.cc

#include "stream.h"

namespace firesse {

Stream::Stream(firecgi::Request* request)
		: request_(request) {
	request->OnClose([this]() {
		if (on_close_) {
			on_close_();
		}
	});
}

void Stream::OnClose(const std::function<void()>& callback) {
	on_close_ = callback;
}

bool Stream::WriteEvent(const std::string& data, uint64_t id, const std::string& type) {
	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::End() {
	return request_->End();
}

} // namespace firesse