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