-
Notifications
You must be signed in to change notification settings - Fork 99
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Showing
35 changed files
with
1,958 additions
and
292 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
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
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
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
116 changes: 116 additions & 0 deletions
116
src/main/java/com/zhongan/devpilot/agents/AgentsRunner.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,116 @@ | ||
| package com.zhongan.devpilot.agents; | ||
|
|
||
| import com.intellij.openapi.diagnostic.Logger; | ||
| import com.zhongan.devpilot.util.ProcessUtils; | ||
|
|
||
| import java.io.File; | ||
| import java.io.FileWriter; | ||
| import java.io.IOException; | ||
| import java.net.ServerSocket; | ||
| import java.util.ArrayList; | ||
| import java.util.List; | ||
| import java.util.concurrent.Executors; | ||
| import java.util.concurrent.ScheduledExecutorService; | ||
| import java.util.concurrent.TimeUnit; | ||
|
|
||
| import org.jetbrains.annotations.NotNull; | ||
|
|
||
| public class AgentsRunner { | ||
| private static final Logger LOG = Logger.getInstance(AgentsRunner.class); | ||
|
|
||
| public static final AgentsRunner INSTANCE = new AgentsRunner(); | ||
|
|
||
| private final ScheduledExecutorService executorService = Executors.newSingleThreadScheduledExecutor(); | ||
|
|
||
| public synchronized boolean run(boolean force) throws Exception { | ||
| File homeDir = BinaryManager.INSTANCE.getHomeDir(); | ||
| if (homeDir == null) { | ||
| LOG.warn("Home dir is null, skip running DevPilot-Agents."); | ||
| return false; | ||
| } | ||
| BinaryManager.AgentCheckResult checkRes = BinaryManager.INSTANCE.checkIfAgentRunning(homeDir); | ||
| if (!force && checkRes.isRunning()) { | ||
| LOG.info("Skip running DevPilot-Agents for already running."); | ||
| return true; | ||
| } | ||
| boolean processRes = BinaryManager.INSTANCE.postProcessBeforeRunning(homeDir); | ||
| if (!processRes) { | ||
| LOG.info("Skip running DevPilot-Agents for failure of init binary."); | ||
| return false; | ||
| } | ||
| int port = getAvailablePort(); | ||
| List<String> commands = createCommand(BinaryManager.INSTANCE.getBinaryPath(homeDir), port); | ||
| ProcessBuilder builder = new ProcessBuilder(commands); | ||
| builder.directory(homeDir); | ||
| Process process = builder.start(); | ||
|
|
||
| boolean aliveFlag = process.isAlive(); | ||
| if (aliveFlag) { | ||
| writeInfoFile(homeDir, ProcessUtils.findDevPilotAgentPidList(), port); | ||
| BinaryManager.INSTANCE.setCurrentPort(port); | ||
| } | ||
| if (aliveFlag && checkRes.isRunning()) { | ||
| executorService.schedule(() -> BinaryManager.INSTANCE.killOldProcess(checkRes.getPid()), 30, TimeUnit.SECONDS); | ||
| } | ||
| return aliveFlag; | ||
| } | ||
|
|
||
| public synchronized boolean run() throws Exception { | ||
| return run(false); | ||
| } | ||
|
|
||
| private int getAvailablePort() { | ||
| try (ServerSocket socket = new ServerSocket(0)) { | ||
| return socket.getLocalPort(); | ||
| } catch (IOException e) { | ||
| LOG.warn("No available port", e); | ||
| throw new RuntimeException("No available port", e); | ||
| } | ||
| } | ||
|
|
||
| public void writeInfoFile(File homeDir, List<Long> pids, int port) { | ||
| if (homeDir != null) { | ||
| File infoFile = new File(homeDir, BinaryManager.INSTANCE.getIdeInfoPath()); | ||
| try (FileWriter writer = new FileWriter(infoFile)) { | ||
| writer.write(port + System.lineSeparator()); | ||
| for (Long pid : pids) { | ||
| writer.write(pid + System.lineSeparator()); | ||
| } | ||
| LOG.info(String.format("Write info file to %s with port %s success.", homeDir.getName(), port)); | ||
| } catch (IOException e) { | ||
| LOG.warn(String.format("Failed to write info file: %s.", homeDir.getName()), e); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| private List<String> createCommand(@NotNull String binaryPath, int port) { | ||
| List<String> commands = new ArrayList<>(); | ||
| if (ProcessUtils.isWindowsPlatform()) { | ||
| String command = "cmd"; | ||
| File cmdFile = new File(System.getenv("WINDIR") + "\\system32\\cmd.exe"); | ||
| if (!cmdFile.exists()) { | ||
| cmdFile = new File("c:\\Windows\\system32\\cmd.exe"); | ||
| } | ||
|
|
||
| if (cmdFile.exists()) { | ||
| command = "\"" + cmdFile.getAbsolutePath() + "\""; | ||
| } | ||
|
|
||
| commands.add(command); | ||
| commands.add("/c"); | ||
| } | ||
|
|
||
| if (ProcessUtils.isWindowsPlatform()) { | ||
| binaryPath = binaryPath.replace("(", "^(").replace(")", "^)").replace("&", "^&").replace(">", "^>").replace("<", "^<").replace("|", "^|"); | ||
| binaryPath = "\"" + binaryPath + "\""; | ||
| } | ||
|
|
||
| commands.add(binaryPath); | ||
| commands.add("--port"); | ||
| commands.add(String.valueOf(port)); | ||
|
|
||
| LOG.info("Starting DevPilot-Agents with command: " + commands); | ||
| return commands; | ||
| } | ||
|
|
||
| } |
Oops, something went wrong.