firesse/server.cc

#include "server.h"

namespace firesse {

Server::Server(int port, const std::function<void(Stream*)>& callback)
		: callback_(callback),
		  keep_alive_(std::chrono::seconds(15), &index_),
		  firecgi_server_(port,
		                  [this](firecgi::Request* request) { OnRequest(request); },
						  1) {}

void Server::Serve() {
	keep_alive_.Start();
	firecgi_server_.Serve();
	keep_alive_.Stop();
}

void Server::Shutdown() {
	firecgi_server_.Shutdown();
}

void Server::RegisterSignalHandlers() {
	firecgi_server_.RegisterSignalHandlers();
}

void Server::OnRequest(firecgi::Request* request) {
	request->WriteHeader("Content-Type", "text/event-stream; charset=utf-8");
	request->WriteHeader("Cache-Control", "no-cache");
	request->WriteHeader("X-Accel-Buffering", "no");
	request->WriteBody("");
	auto stream = new Stream(request, &index_);

	{
		request->OnClose([stream]() {
			stream->Close();
			delete stream;
		});
	}

	callback_(stream);
}

} // namespace firesse