nuget-agent/src/jetbrains/buildServer/nuget/agent/util/DelegatingBuildProcess.java

   1 /*\r
   2  * Copyright 2000-2011 JetBrains s.r.o.\r
   3  *\r
   4  * Licensed under the Apache License, Version 2.0 (the "License");\r
   5  * you may not use this file except in compliance with the License.\r
   6  * You may obtain a copy of the License at\r
   7  *\r
   8  * http://www.apache.org/licenses/LICENSE-2.0\r
   9  *\r
  10  * Unless required by applicable law or agreed to in writing, software\r
  11  * distributed under the License is distributed on an "AS IS" BASIS,\r
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  13  * See the License for the specific language governing permissions and\r
  14  * limitations under the License.\r
  15  */\r
  16 \r
  17 package jetbrains.buildServer.nuget.agent.util;\r
  18 \r
  19 import jetbrains.buildServer.RunBuildException;\r
  20 import jetbrains.buildServer.agent.BuildFinishedStatus;\r
  21 import jetbrains.buildServer.agent.BuildProcess;\r
  22 import org.jetbrains.annotations.NotNull;\r
  23 \r
  24 import java.util.concurrent.atomic.AtomicReference;\r
  25 \r
  26 /**\r
  27  * Created by Eugene Petrenko (eugene.petrenko@gmail.com)\r
  28  * Date: 07.07.11 18:41\r
  29  */\r
  30 public class DelegatingBuildProcess extends BuildProcessBase {\r
  31   private final AtomicReference<BuildProcess> myReference = new AtomicReference<BuildProcess>();\r
  32   private final Action myAction;\r
  33 \r
  34   public DelegatingBuildProcess(@NotNull final Action action) {\r
  35     myAction = action;\r
  36   }\r
  37 \r
  38   @Override\r
  39   protected final void interruptImpl() {\r
  40     super.interruptImpl();\r
  41     BuildProcess process = myReference.get();\r
  42     if (process != null) process.interrupt();\r
  43   }\r
  44 \r
  45   @NotNull\r
  46   @Override\r
  47   protected final BuildFinishedStatus waitForImpl() throws RunBuildException {\r
  48     try {\r
  49       BuildProcess process = myAction.startImpl();\r
  50       myReference.set(process);\r
  51 \r
  52       if (isInterrupted()) return BuildFinishedStatus.INTERRUPTED;\r
  53       process.start();\r
  54       return process.waitFor();\r
  55     } finally {\r
  56       myReference.set(null);\r
  57       myAction.finishedImpl();\r
  58     }\r
  59   }\r
  60 \r
  61   public static interface Action {\r
  62     @NotNull\r
  63     BuildProcess startImpl() throws RunBuildException;\r
  64     void finishedImpl();\r
  65   }\r
  66 }\r