From 9c42586bc406015be0145576fd6cb3586686b4ca Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Mon, 7 Jun 2021 23:24:43 +0800
Subject: import(life): ...

---
 works/life/computer-network-experiment/IO.cpp | 80 +++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 works/life/computer-network-experiment/IO.cpp

(limited to 'works/life/computer-network-experiment/IO.cpp')

diff --git a/works/life/computer-network-experiment/IO.cpp b/works/life/computer-network-experiment/IO.cpp
new file mode 100644
index 0000000..0c20537
--- /dev/null
+++ b/works/life/computer-network-experiment/IO.cpp
@@ -0,0 +1,80 @@
+#include "IO.h"
+
+#include <folly/CancellationToken.h>
+
+#include <mutex>
+#include <ostream>
+#include <thread>
+#include <type_traits>
+
+folly::MPMCQueue<Output> output_queue(100);
+
+folly::CancellationSource cancellation_source;
+
+std::thread io_thread;
+
+void PrintOutput(const Output &output) {
+  std::basic_ostream<Char> *stream;
+
+  switch (output.type) {
+  case OutputType::Error:
+    stream = &error_stream;
+    break;
+  default:
+    stream = &output_stream;
+    break;
+  }
+
+  switch (output.color) {
+  case OutputColor::Normal:
+    (*stream) << output.message;
+    break;
+  case OutputColor::Green:
+    (*stream) << CRUT("\x1b[32m") << output.message << CRUT("\x1b[39m")
+              << std::flush;
+    break;
+  case OutputColor::Red:
+    (*stream) << CRUT("\x1b[31m") << output.message << CRUT("\x1b[39m")
+              << std::flush;
+    break;
+  case OutputColor::Yellow:
+    (*stream) << CRUT("\x1b[33m") << output.message << CRUT("\x1b[39m")
+              << std::flush;
+    break;
+  }
+}
+
+String ReadInputLine() {
+  String line;
+  std::getline(input_stream, line);
+  return line;
+}
+
+void IOThread() {
+  while (true) {
+    if (cancellation_source.isCancellationRequested()) {
+      while (true) {
+        Output output;
+        if (output_queue.readIfNotEmpty(output)) {
+          PrintOutput(output);
+        } else {
+          return;
+        }
+      }
+    }
+
+    Output output;
+    while (output_queue.readIfNotEmpty(output))
+      PrintOutput(output);
+
+    PrintOutput({CRUT("> ")});
+    OnInputLine(ReadInputLine());
+  }
+}
+
+void SignalAndWaitForOutputThreadStop() {
+  cancellation_source.requestCancellation();
+  io_thread.join();
+}
+
+void StartIOThread() { io_thread = std::thread(IOThread); }
-- 
cgit v1.2.3