View Javadoc
1   package org.apache.maven.surefire.junitcore.pc;
2   
3   /*
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *     http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing,
15   * software distributed under the License is distributed on an
16   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17   * KIND, either express or implied.  See the License for the
18   * specific language governing permissions and limitations
19   * under the License.
20   */
21  
22  import java.util.concurrent.CancellationException;
23  import java.util.concurrent.ConcurrentLinkedQueue;
24  import java.util.concurrent.ExecutionException;
25  import java.util.concurrent.ExecutorService;
26  import java.util.concurrent.Future;
27  
28  /**
29   * Parallel strategy for shared thread pool in private package.
30   *
31   * @author Tibor Digana (tibor17)
32   * @see AbstractThreadPoolStrategy
33   * @since 2.16
34   */
35  final class SharedThreadPoolStrategy
36      extends AbstractThreadPoolStrategy
37  {
38      SharedThreadPoolStrategy( ExecutorService threadPool )
39      {
40          super( threadPool, new ConcurrentLinkedQueue<Future<?>>() );
41      }
42  
43      @Override
44      public boolean hasSharedThreadPool()
45      {
46          return true;
47      }
48  
49      @Override
50      public boolean finished()
51          throws InterruptedException
52      {
53          boolean wasRunningAll = disable();
54          for ( Future<?> futureResult : getFutureResults() )
55          {
56              try
57              {
58                  futureResult.get();
59              }
60              catch ( InterruptedException e )
61              {
62                  // after called external ExecutorService#shutdownNow()
63                  wasRunningAll = false;
64              }
65              catch ( ExecutionException e )
66              {
67                  // JUnit core throws exception.
68                  if ( e.getCause() != null )
69                  {
70                      logQuietly( e.getCause() );
71                  }
72              }
73              catch ( CancellationException e )
74              {
75                  /**
76                   * Cancelled by {@link Future#cancel(boolean)} in {@link stop()} and {@link stopNow()}.
77                   */
78              }
79          }
80          return wasRunningAll;
81      }
82  
83      @Override
84      protected boolean stop()
85      {
86          return stop( false );
87      }
88  
89      @Override
90      protected boolean stopNow()
91      {
92          return stop( true );
93      }
94  
95      private boolean stop( boolean interrupt )
96      {
97          final boolean wasRunning = disable();
98          for ( Future<?> futureResult : getFutureResults() )
99          {
100             futureResult.cancel( interrupt );
101         }
102         return wasRunning;
103     }
104 }