more infrastructure:
[teamcity/dotNetPackagesSupport.git] / nuget-agent / src / jetbrains / buildServer / nuget / agent / util / impl / CompositeBuildProcessImpl.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.impl;\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 jetbrains.buildServer.nuget.agent.util.CompositeBuildProcess;\r
23 import org.jetbrains.annotations.NotNull;\r
24 \r
25 import java.util.concurrent.BlockingQueue;\r
26 import java.util.concurrent.LinkedBlockingQueue;\r
27 import java.util.concurrent.atomic.AtomicBoolean;\r
28 import java.util.concurrent.atomic.AtomicReference;\r
29 \r
30 /**\r
31  * Created by Eugene Petrenko (eugene.petrenko@gmail.com)\r
32  * Date: 07.07.11 14:04\r
33  */\r
34 public class CompositeBuildProcessImpl implements CompositeBuildProcess {\r
35   private final BlockingQueue<BuildProcess> myProcessList = new LinkedBlockingQueue<BuildProcess>();\r
36   private final AtomicBoolean myIsInterrupted = new AtomicBoolean();\r
37   private final AtomicBoolean myIsFinished = new AtomicBoolean();\r
38   private final AtomicReference<BuildProcess> myCurrentProcess = new AtomicReference<BuildProcess>();\r
39 \r
40   public void pushBuildProcess(@NotNull final BuildProcess process) {\r
41     myProcessList.add(process);\r
42   }\r
43 \r
44   public void start() throws RunBuildException {\r
45   }\r
46 \r
47   public boolean isInterrupted() {\r
48     return myIsInterrupted.get();\r
49   }\r
50 \r
51   public boolean isFinished() {\r
52     return myIsFinished.get();\r
53   }\r
54 \r
55   public void interrupt() {\r
56     myIsInterrupted.set(true);\r
57     BuildProcess process = myCurrentProcess.get();\r
58     if (process != null) {\r
59       process.interrupt();\r
60     }\r
61   }\r
62 \r
63   @NotNull\r
64   public BuildFinishedStatus waitFor() throws RunBuildException {\r
65     for(BuildProcess proc = myProcessList.poll(); proc != null; proc = myProcessList.poll()) {\r
66       myCurrentProcess.set(proc);\r
67       try {\r
68         proc.start();\r
69         BuildFinishedStatus status = proc.waitFor();\r
70         if (status == BuildFinishedStatus.FINISHED_FAILED ) return BuildFinishedStatus.FINISHED_FAILED;\r
71       } finally {\r
72         myCurrentProcess.set(null);\r
73       }\r
74       if (isInterrupted()) return BuildFinishedStatus.INTERRUPTED;\r
75     }\r
76     myIsFinished.set(true);\r
77     if (isInterrupted()) return BuildFinishedStatus.INTERRUPTED;\r
78 \r
79     return BuildFinishedStatus.FINISHED_SUCCESS;\r
80   }\r
81 }\r