mirror of
https://github.com/MaskRay/ccls.git
synced 2024-11-28 02:21:57 +00:00
stdin: synthesize an "exit" NotificationMessage in abnormal termination
This commit is contained in:
parent
401f057027
commit
cff00a8711
@ -503,13 +503,14 @@ void LaunchStdin() {
|
||||
set_thread_name("stdin");
|
||||
std::string str;
|
||||
const std::string_view kContentLength("Content-Length: ");
|
||||
bool received_exit = false;
|
||||
while (true) {
|
||||
int len = 0;
|
||||
str.clear();
|
||||
while (true) {
|
||||
int c = getchar();
|
||||
if (c == EOF)
|
||||
return;
|
||||
goto quit;
|
||||
if (c == '\n') {
|
||||
if (str.empty())
|
||||
break;
|
||||
@ -525,7 +526,7 @@ void LaunchStdin() {
|
||||
for (int i = 0; i < len; ++i) {
|
||||
int c = getchar();
|
||||
if (c == EOF)
|
||||
return;
|
||||
goto quit;
|
||||
str[i] = c;
|
||||
}
|
||||
|
||||
@ -549,16 +550,28 @@ void LaunchStdin() {
|
||||
LOG_V(2) << "receive NotificationMessage " << method;
|
||||
if (method.empty())
|
||||
continue;
|
||||
bool should_exit = method == "exit";
|
||||
received_exit = method == "exit";
|
||||
// g_config is not available before "initialize". Use 0 in that case.
|
||||
on_request->PushBack(
|
||||
{id, std::move(method), std::move(message), std::move(document),
|
||||
chrono::steady_clock::now() +
|
||||
chrono::milliseconds(g_config ? g_config->request.timeout : 0)});
|
||||
|
||||
if (should_exit)
|
||||
if (received_exit)
|
||||
break;
|
||||
}
|
||||
|
||||
quit:
|
||||
if (!received_exit) {
|
||||
const std::string_view str("{\"jsonrpc\":\"2.0\",\"method\":\"exit\"}");
|
||||
auto message = std::make_unique<char[]>(str.size());
|
||||
std::copy(str.begin(), str.end(), message.get());
|
||||
auto document = std::make_unique<rapidjson::Document>();
|
||||
document->Parse(message.get(), str.size());
|
||||
on_request->PushBack({RequestId(), std::string("exit"),
|
||||
std::move(message), std::move(document),
|
||||
chrono::steady_clock::now()});
|
||||
}
|
||||
ThreadLeave();
|
||||
}).detach();
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user