Track streams

stream.cc

@@ -3,19 +3,18 @@
 namespace firesse {
 
 Stream::Stream(firecgi::Request* request)
-		: request_(request) {
-	request->OnClose([this]() {
-		if (on_close_) {
-			on_close_();
-		}
-	});
-}
+		: request_(request) {}
 
 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) {
+	{
+		std::lock_guard l(mu_);
+		last_message_time_ = std::chrono::steady_clock::now();
+	}
+
 	return request_->InTransaction<bool>([=]() {
 		if (id) {
 			request_->WriteBody("id: ", std::to_string(id), "\n");
@@ -32,4 +31,19 @@ bool Stream::End() {
 	return request_->End();
 }
 
+std::chrono::steady_clock::time_point Stream::LastMessageTime() {
+	std::lock_guard l(mu_);
+	return last_message_time_;
+}
+
+void Stream::Close() {
+	if (on_close_) {
+		on_close_();
+	}
+}
+
+bool IsFresherStream::operator() (Stream* a, Stream* b) const {
+	return a->LastMessageTime() > b->LastMessageTime();
+}
+
 } // namespace firesse