wording
[teamcity/dotNetPackagesSupport.git] / nuget-server / src / jetbrains / buildServer / nuget / server / trigger / ThreadedBuildTriggetPolicy.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.server.trigger;\r
18 \r
19 import jetbrains.buildServer.buildTriggers.BuildTriggerException;\r
20 import jetbrains.buildServer.buildTriggers.PolledBuildTrigger;\r
21 import jetbrains.buildServer.buildTriggers.PolledTriggerContext;\r
22 import org.jetbrains.annotations.NotNull;\r
23 \r
24 import java.util.concurrent.Callable;\r
25 import java.util.concurrent.ExecutionException;\r
26 import java.util.concurrent.ExecutorService;\r
27 import java.util.concurrent.Future;\r
28 \r
29 /**\r
30  * Created by Eugene Petrenko (eugene.petrenko@gmail.com)\r
31  * Date: 12.07.11 23:29\r
32  */\r
33 public class ThreadedBuildTriggetPolicy extends PolledBuildTrigger {\r
34   private final ExecutorService myExecutor;\r
35   private final TriggerUpdateChecker myUpdater;\r
36   private Future<BuildStartReason> myUpdateRequired;\r
37 \r
38   public ThreadedBuildTriggetPolicy(@NotNull final ExecutorService executor,\r
39                                     @NotNull final TriggerUpdateChecker updater) {\r
40     myExecutor = executor;\r
41     myUpdater = updater;\r
42   }\r
43 \r
44   @Override\r
45   public synchronized void triggerBuild(@NotNull PolledTriggerContext context) throws BuildTriggerException {\r
46     if (myUpdateRequired == null) {\r
47       myUpdateRequired = myExecutor.submit(createUpdateTask(context));\r
48     }\r
49 \r
50     if (!myUpdateRequired.isDone()) return;\r
51 \r
52     BuildStartReason result;\r
53     try {\r
54       result = myUpdateRequired.get();\r
55     } catch (InterruptedException e) {\r
56       return;\r
57     } catch (ExecutionException e) {\r
58       Throwable cause = e.getCause();\r
59       if (cause instanceof BuildTriggerException) throw (BuildTriggerException)cause;\r
60       throw new BuildTriggerException(cause.getMessage(), cause);\r
61     } finally {\r
62       myUpdateRequired = null;\r
63     }\r
64 \r
65     if (result != null) {\r
66       context.getBuildType().addToQueue(result.getReason());\r
67     }\r
68   }\r
69 \r
70   @NotNull\r
71   private Callable<BuildStartReason> createUpdateTask(@NotNull final PolledTriggerContext storage) {\r
72     return new Callable<BuildStartReason>() {\r
73       public BuildStartReason call() throws Exception {\r
74         return myUpdater.checkChanges(storage.getTriggerDescriptor(), storage.getCustomDataStorage());\r
75       }\r
76     };\r
77   }\r
78 }\r