a82171794ba807179135e214d63b24333db3a778
[teamcity/dotNetPackagesSupport.git] / 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   @Override\r
46   protected final BuildFinishedStatus waitForImpl() throws RunBuildException {\r
47     try {\r
48       BuildProcess process = myAction.startImpl();\r
49       myReference.set(process);\r
50 \r
51       if (isInterrupted()) return BuildFinishedStatus.INTERRUPTED;\r
52       process.start();\r
53       return process.waitFor();\r
54     } finally {\r
55       myReference.set(null);\r
56       myAction.finishedImpl();\r
57     }\r
58   }\r
59 \r
60   public static interface Action {\r
61     @NotNull\r
62     BuildProcess startImpl() throws RunBuildException;\r
63     void finishedImpl();\r
64   }\r
65 }\r