HBASE-24658 Update PolicyBasedChaosMonkey to handle uncaught exceptions

Running `ServerKillingChaosMonkey` via `RESTApiClusterManager` for any
duration of time slowly leaks region servers. I see failures on the
RESTApi side go unreported on the ChaosMonkey side. It seems like
`RuntimeException`s are being thrown and lost.

`PolicyBasedChaosMonkey` uses a primitive means of thread management
anyway. Update to use a thread pool, thread groups, and an
uncaughtExceptionHandler.

Signed-off-by: Bharath Vissapragada <bharathv@apache.org>
Signed-off-by: Viraj Jasani <vjasani@apache.org>
This commit is contained in:
Nick Dimiduk 2020-06-29 16:35:26 -07:00 committed by Nick Dimiduk
parent a857c6d90c
commit 7ebc617026
1 changed files with 36 additions and 36 deletions

View File

@ -1,4 +1,4 @@
/**
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
@ -22,14 +22,18 @@ import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Properties;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import org.apache.commons.lang3.RandomUtils;
import org.apache.hadoop.hbase.IntegrationTestingUtility;
import org.apache.hadoop.hbase.chaos.policies.Policy;
import org.apache.hadoop.hbase.util.Pair;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.apache.hbase.thirdparty.com.google.common.util.concurrent.ThreadFactoryBuilder;
/**
* Chaos monkey that given multiple policies will run actions against the cluster.
@ -38,7 +42,6 @@ public class PolicyBasedChaosMonkey extends ChaosMonkey {
private static final Logger LOG = LoggerFactory.getLogger(PolicyBasedChaosMonkey.class);
private static final long ONE_SEC = 1000;
private static final long FIVE_SEC = 5 * ONE_SEC;
private static final long ONE_MIN = 60 * ONE_SEC;
public static final long TIMEOUT = ONE_MIN;
@ -46,6 +49,9 @@ public class PolicyBasedChaosMonkey extends ChaosMonkey {
final IntegrationTestingUtility util;
final Properties monkeyProps;
private final Policy[] policies;
private final ExecutorService monkeyThreadPool;
/**
* Construct a new ChaosMonkey
* @param util the HBaseIntegrationTestingUtility already configured
@ -60,19 +66,30 @@ public class PolicyBasedChaosMonkey extends ChaosMonkey {
}
public PolicyBasedChaosMonkey(Properties monkeyProps, IntegrationTestingUtility util,
Policy... policies) {
this.monkeyProps = monkeyProps;
this.util = util;
this.policies = policies;
Collection<Policy> policies) {
this(monkeyProps, util, policies.toArray(new Policy[0]));
}
public PolicyBasedChaosMonkey(Properties monkeyProps, IntegrationTestingUtility util,
Collection<Policy> policies) {
Policy... policies) {
this.monkeyProps = monkeyProps;
this.util = util;
this.policies = policies.toArray(new Policy[policies.size()]);
this.util = Objects.requireNonNull(util);
this.policies = Objects.requireNonNull(policies);
if (policies.length == 0) {
throw new IllegalArgumentException("policies may not be empty");
}
this.monkeyThreadPool = buildMonkeyThreadPool(policies.length);
}
private static ExecutorService buildMonkeyThreadPool(final int size) {
return Executors.newFixedThreadPool(size, new ThreadFactoryBuilder()
.setDaemon(false)
.setNameFormat("ChaosMonkey-%d")
.setUncaughtExceptionHandler((t, e) -> {
throw new RuntimeException(e);
})
.build());
}
/** Selects a random item from the given items */
public static <T> T selectRandomItem(T[] items) {
@ -114,27 +131,20 @@ public class PolicyBasedChaosMonkey extends ChaosMonkey {
return originalItems.subList(startIndex, startIndex + selectedNumber);
}
private Policy[] policies;
private Thread[] monkeyThreads;
@Override
public void start() throws Exception {
monkeyThreads = new Thread[policies.length];
Policy.PolicyContext context = new Policy.PolicyContext(monkeyProps, this.util);
for (int i=0; i<policies.length; i++) {
policies[i].init(context);
Thread monkeyThread = new Thread(policies[i], "ChaosMonkey");
monkeyThread.start();
monkeyThreads[i] = monkeyThread;
final Policy.PolicyContext context = new Policy.PolicyContext(monkeyProps, util);
for (final Policy policy : policies) {
policy.init(context);
monkeyThreadPool.execute(policy);
}
}
@Override
public void stop(String why) {
if (policies == null) {
return;
}
// stop accepting new work (shouldn't be any with a fixed-size pool)
monkeyThreadPool.shutdown();
// notify all executing policies that it's time to halt.
for (Policy policy : policies) {
policy.stop(why);
}
@ -142,22 +152,12 @@ public class PolicyBasedChaosMonkey extends ChaosMonkey {
@Override
public boolean isStopped() {
return policies[0].isStopped();
return monkeyThreadPool.isTerminated();
}
/**
* Wait for ChaosMonkey to stop.
* @throws InterruptedException
*/
@Override
public void waitForStop() throws InterruptedException {
if (monkeyThreads == null) {
return;
}
for (Thread monkeyThread : monkeyThreads) {
// TODO: bound the wait time per policy
monkeyThread.join();
}
monkeyThreadPool.awaitTermination(1, TimeUnit.MINUTES);
}
@Override