blob: 5d3fe12ffa4138f9f54d49432fbd2879d223c8ef (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
#include "IO.h"
#include <folly/CancellationToken.h>
#include <mutex>
#include <ostream>
#include <thread>
#include <type_traits>
folly::MPMCQueue<Output> output_queue(100);
namespace {
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); }
|
