diff options
4 files changed, 190 insertions, 11 deletions
diff --git a/sonar-runner-dist/src/main/java/org/sonar/runner/Main.java b/sonar-runner-dist/src/main/java/org/sonar/runner/Main.java index d991dc2..ea472d1 100644 --- a/sonar-runner-dist/src/main/java/org/sonar/runner/Main.java +++ b/sonar-runner-dist/src/main/java/org/sonar/runner/Main.java @@ -43,15 +43,15 @@ import org.sonar.runner.impl.Logs; */ public class Main { - private final Exit exit; + private final Shutdown shutdown; private final Cli cli; private final Conf conf; private final RunnerFactory runnerFactory; private Runner<?> runner; private BufferedReader inputReader; - Main(Exit exit, Cli cli, Conf conf, RunnerFactory runnerFactory) { - this.exit = exit; + Main(Shutdown shutdown, Cli cli, Conf conf, RunnerFactory runnerFactory) { + this.shutdown = shutdown; this.cli = cli; this.conf = conf; this.runnerFactory = runnerFactory; @@ -59,9 +59,10 @@ public class Main { public static void main(String[] args) { Exit exit = new Exit(); + Shutdown shutdown = new Shutdown(exit); Cli cli = new Cli(exit).parse(args); cli.verify(); - Main main = new Main(exit, cli, new Conf(cli), new RunnerFactory()); + Main main = new Main(shutdown, cli, new Conf(cli), new RunnerFactory()); main.execute(); } @@ -84,17 +85,17 @@ public class Main { } catch (Exception e) { displayExecutionResult(stats, "FAILURE"); showError("Error during Sonar runner execution", e, cli.isDisplayStackTrace()); - exit.exit(Exit.ERROR); + shutdown.exit(Exit.ERROR); } runner.stop(); - exit.exit(Exit.SUCCESS); + shutdown.exit(Exit.SUCCESS); } private void init(Properties p) throws IOException { SystemInfo.print(); if (cli.isDisplayVersionOnly()) { - exit.exit(Exit.SUCCESS); + shutdown.exit(Exit.SUCCESS); } if (cli.isDisplayStackTrace()) { @@ -113,9 +114,18 @@ public class Main { if (inputReader == null) { inputReader = new BufferedReader(new InputStreamReader(System.in, StandardCharsets.UTF_8)); } + + shutdown.signalReady(true); + if(shutdown.shouldExit()) { + //exit before displaying message + return false; + } + Logs.info("<Press enter to restart analysis>"); - - return inputReader.readLine() != null; + String line = inputReader.readLine(); + shutdown.signalReady(false); + + return line != null; } private static void displayExecutionResult(Stats stats, String resultMsg) { diff --git a/sonar-runner-dist/src/main/java/org/sonar/runner/Shutdown.java b/sonar-runner-dist/src/main/java/org/sonar/runner/Shutdown.java new file mode 100644 index 0000000..70ec898 --- /dev/null +++ b/sonar-runner-dist/src/main/java/org/sonar/runner/Shutdown.java @@ -0,0 +1,92 @@ +/* + * SonarQube Runner - Distribution + * Copyright (C) 2011 SonarSource + * dev@sonar.codehaus.org + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 3 of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02 + */ +package org.sonar.runner; + +class Shutdown { + static final int SUCCESS = 0; + static final int ERROR = 1; + private static final long DEFAULT_MAX_WAIT = 10_000; + + private long maxWait; + ShutdownHook hook = new ShutdownHook(); + private boolean isReady = false; + private boolean exiting = false; + private Object lock = new Object(); + private Exit exit; + private Thread t; + + Shutdown(Exit exit) { + this(exit, DEFAULT_MAX_WAIT); + } + + Shutdown(Exit exit, long maxWait) { + this.maxWait = maxWait; + this.exit = exit; + Runtime.getRuntime().addShutdownHook(hook); + } + + void exit(int status) { + synchronized (lock) { + signalReady(true); + } + exit.exit(status); + } + + void signalReady(boolean ready) { + synchronized (lock) { + System.out.println("READY: " + ready); + this.isReady = ready; + lock.notifyAll(); + } + } + + boolean shouldExit() { + synchronized (lock) { + return exiting; + } + } + + class ShutdownHook extends Thread { + private ShutdownHook() { + this.setName("shutdown-hook"); + } + + @Override + public void run() { + long startTime = System.currentTimeMillis(); + synchronized (lock) { + exiting = true; + + while (!isReady) { + long waitTime = startTime + maxWait - System.currentTimeMillis(); + if (waitTime <= 0) { + break; + } + + try { + lock.wait(waitTime); + } catch (InterruptedException e) { + // continue + } + } + } + } + } +} diff --git a/sonar-runner-dist/src/test/java/org/sonar/runner/MainTest.java b/sonar-runner-dist/src/test/java/org/sonar/runner/MainTest.java index 773632d..7b7e019 100644 --- a/sonar-runner-dist/src/test/java/org/sonar/runner/MainTest.java +++ b/sonar-runner-dist/src/test/java/org/sonar/runner/MainTest.java @@ -30,7 +30,6 @@ import org.sonar.runner.api.Runner; import java.io.ByteArrayInputStream; import java.io.IOException; import java.io.InputStream; -import java.io.UnsupportedEncodingException; import java.nio.charset.StandardCharsets; import java.util.Properties; @@ -44,7 +43,7 @@ import static org.mockito.Mockito.when; public class MainTest { @Mock - private Exit exit; + private Shutdown exit; @Mock private Cli cli; @Mock diff --git a/sonar-runner-dist/src/test/java/org/sonar/runner/ShutdownTest.java b/sonar-runner-dist/src/test/java/org/sonar/runner/ShutdownTest.java new file mode 100644 index 0000000..9de7f4c --- /dev/null +++ b/sonar-runner-dist/src/test/java/org/sonar/runner/ShutdownTest.java @@ -0,0 +1,78 @@ +/* + * SonarQube Runner - Distribution + * Copyright (C) 2011 SonarSource + * dev@sonar.codehaus.org + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 3 of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02 + */ +package org.sonar.runner; + +import static org.mockito.Mockito.verify; +import static org.fest.assertions.Assertions.assertThat; +import org.mockito.MockitoAnnotations; +import org.mockito.Mock; +import org.junit.Test; +import org.junit.Before; + +public class ShutdownTest { + @Mock + private Exit exit; + private Shutdown shutdown; + + @Before + public void setUp() { + MockitoAnnotations.initMocks(this); + shutdown = new Shutdown(exit); + } + + @Test + public void testShutdown() { + shutdown.exit(3); + verify(exit).exit(3); + } + + @Test(timeout = 60_000) + public void testWaitReady() throws InterruptedException { + shutdown = new Shutdown(exit, 100_000); + shutdown.signalReady(false); + assertThat(shutdown.shouldExit()).isFalse(); + + Thread t = new HookCaller(); + t.start(); + Thread.sleep(1000); + + assertThat(t.isAlive()).isTrue(); + assertThat(shutdown.shouldExit()).isTrue(); + + shutdown.signalReady(true); + t.join(); + } + + @Test(timeout = 60_000) + public void testTimeout() throws InterruptedException { + shutdown = new Shutdown(exit, 0); + + Thread t = new HookCaller(); + t.start(); + t.join(); + } + + private class HookCaller extends Thread { + @Override + public void run() { + shutdown.hook.run(); + } + } +} |