-
Notifications
You must be signed in to change notification settings - Fork 206
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Shutdown LS eventually if parent shell process is gone
- Loading branch information
Showing
4 changed files
with
191 additions
and
48 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
141 changes: 141 additions & 0 deletions
141
...java/org/springframework/ide/vscode/commons/languageserver/util/ParentProcessWatcher.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,141 @@ | ||
| /******************************************************************************* | ||
| * Copyright (c) 2022 VMware, Inc. | ||
| * All rights reserved. This program and the accompanying materials | ||
| * are made available under the terms of the Eclipse Public License v1.0 | ||
| * which accompanies this distribution, and is available at | ||
| * https://www.eclipse.org/legal/epl-v10.html | ||
| * | ||
| * Contributors: | ||
| * VMware, Inc. - initial API and implementation | ||
| *******************************************************************************/ | ||
| package org.springframework.ide.vscode.commons.languageserver.util; | ||
|
|
||
| import java.io.IOException; | ||
| import java.util.concurrent.Executors; | ||
| import java.util.concurrent.ScheduledExecutorService; | ||
| import java.util.concurrent.ScheduledFuture; | ||
| import java.util.concurrent.TimeUnit; | ||
| import java.util.function.Function; | ||
|
|
||
| import org.eclipse.lsp4j.jsonrpc.MessageConsumer; | ||
| import org.slf4j.Logger; | ||
| import org.slf4j.LoggerFactory; | ||
|
|
||
| import com.google.common.io.Closeables; | ||
|
|
||
| /** | ||
| * Watches the parent process PID and invokes exit if it is no longer available. | ||
| * This implementation waits for periods of inactivity to start querying the PIDs. | ||
| * Copied from JDT LS: | ||
| * https://github.com/eclipse/eclipse.jdt.ls/blob/64b15c5a9e5b11f62ceb5163ceb6930d5dea7129/org.eclipse.jdt.ls.core/src/org/eclipse/jdt/ls/core/internal/ParentProcessWatcher.java | ||
| */ | ||
| public final class ParentProcessWatcher implements Runnable, Function<MessageConsumer, MessageConsumer> { | ||
|
|
||
| private static Logger logger = LoggerFactory.getLogger(ParentProcessWatcher.class); | ||
|
|
||
| private static final long INACTIVITY_DELAY = 30_000; | ||
| private static final boolean isJava1x = System.getProperty("java.version").startsWith("1."); | ||
| private static final int POLL_DELAY_SECS = 10; | ||
| private volatile long lastActivityTime; | ||
| private final SimpleLanguageServer server; | ||
| private ScheduledFuture<?> task; | ||
| private ScheduledExecutorService service; | ||
|
|
||
| public ParentProcessWatcher(SimpleLanguageServer server ) { | ||
| this.server = server; | ||
| service = Executors.newScheduledThreadPool(1); | ||
| task = service.scheduleWithFixedDelay(this, POLL_DELAY_SECS, POLL_DELAY_SECS, TimeUnit.SECONDS); | ||
| } | ||
|
|
||
| public void run() { | ||
| if (!parentProcessStillRunning()) { | ||
| logger.info("Parent process stopped running, forcing server exit"); | ||
| task.cancel(true); | ||
| server.exit(); | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * Checks whether the parent process is still running. | ||
| * If not, then we assume it has crashed, and we have to terminate the Java Language Server. | ||
| * | ||
| * @return true if the parent process is still running | ||
| */ | ||
| private boolean parentProcessStillRunning() { | ||
| // Wait until parent process id is available | ||
| final Integer pid = server.getParentProcessId(); | ||
| if (pid == null || lastActivityTime > (System.currentTimeMillis() - INACTIVITY_DELAY)) { | ||
| return true; | ||
| } | ||
| String command; | ||
| if (isWindows()) { | ||
| command = "cmd /c \"tasklist /FI \"PID eq " + pid + "\" | findstr " + pid + "\""; | ||
| } else { | ||
| command = "kill -0 " + pid; | ||
| } | ||
| Process process = null; | ||
| boolean finished = false; | ||
| try { | ||
| process = Runtime.getRuntime().exec(command); | ||
| finished = process.waitFor(POLL_DELAY_SECS, TimeUnit.SECONDS); | ||
| if (!finished) { | ||
| process.destroy(); | ||
| finished = process.waitFor(POLL_DELAY_SECS, TimeUnit.SECONDS); // wait for the process to stop | ||
| } | ||
| if (isWindows() && finished && process.exitValue() > 1) { | ||
| // the tasklist command should return 0 (parent process exists) or 1 (parent process doesn't exist) | ||
| logger.info("The tasklist command: '{}' returns {}", command, process.exitValue()); | ||
| return true; | ||
| } | ||
| return !finished || process.exitValue() == 0; | ||
| } catch (IOException | InterruptedException e) { | ||
| logger.error("", e); | ||
| return true; | ||
| } finally { | ||
| if (process != null) { | ||
| if (!finished) { | ||
| process.destroyForcibly(); | ||
| } | ||
| // Terminating or destroying the Process doesn't close the process handle on Windows. | ||
| // It is only closed when the Process object is garbage collected (in its finalize() method). | ||
| // On Windows, when the Java LS is idle, we need to explicitly request a GC, | ||
| // to prevent an accumulation of zombie processes, as finalize() will be called. | ||
| if (isWindows()) { | ||
| // Java >= 9 doesn't close the handle when the process is garbage collected | ||
| // We need to close the opened streams | ||
| if (!isJava1x) { | ||
| Closeables.closeQuietly(process.getInputStream()); | ||
| Closeables.closeQuietly(process.getErrorStream()); | ||
| try { | ||
| Closeables.close(process.getOutputStream(), false); | ||
| } catch (IOException e) { | ||
| } | ||
| } | ||
| System.gc(); | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public MessageConsumer apply(final MessageConsumer consumer) { | ||
| //inject our own consumer to refresh the timestamp | ||
| return message -> { | ||
| lastActivityTime = System.currentTimeMillis(); | ||
| try { | ||
| consumer.consume(message); | ||
| } catch (UnsupportedOperationException e) { | ||
| //log a warning and ignore. We are getting some messages from vsCode the server doesn't know about | ||
| logger.warn("Unsupported message was ignored!", e); | ||
| } | ||
| }; | ||
| } | ||
|
|
||
| private static boolean isWindows() { | ||
| String os = System.getProperty("os.name"); | ||
| if (os != null) { | ||
| return os.toLowerCase().indexOf("win") >= 0; | ||
| } | ||
| return false; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters