introduce DelegatingBuildProcess, avoid exception from listener on LocateNuGetConfigB...
[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   public final void start() throws RunBuildException {\r
40     super.start();\r
41     if (isInterrupted()) return;\r
42 \r
43     myReference.set(myAction.startImpl());\r
44   }\r
45 \r
46   @Override\r
47   protected final void interruptImpl() {\r
48     super.interruptImpl();\r
49     BuildProcess process = myReference.get();\r
50     if (process != null) process.interrupt();\r
51   }\r
52 \r
53   @Override\r
54   protected final BuildFinishedStatus waitForImpl() throws RunBuildException {\r
55     BuildProcess process = myReference.get();\r
56     if (isInterrupted()) return BuildFinishedStatus.INTERRUPTED;\r
57 \r
58     try {\r
59       if (process != null) {\r
60         process.start();\r
61         return process.waitFor();\r
62       }\r
63     } finally {\r
64       myAction.finishedImpl();\r
65     }\r
66 \r
67     if (isInterrupted()) return BuildFinishedStatus.INTERRUPTED;\r
68     return BuildFinishedStatus.FINISHED_SUCCESS;\r
69   }\r
70 \r
71   public static interface Action {\r
72     @NotNull\r
73     BuildProcess startImpl() throws RunBuildException;\r
74     void finishedImpl();\r
75   }\r
76 }\r