HBASE-14960 Fallback to using default RPCControllerFactory if class cannot be loaded

This commit is contained in:
Enis Soztutar 2015-12-10 19:11:57 -08:00
parent abb2e95f66
commit cff664c5e2
2 changed files with 40 additions and 3 deletions

View File

@ -19,6 +19,8 @@ package org.apache.hadoop.hbase.ipc;
import java.util.List; import java.util.List;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.CellScannable; import org.apache.hadoop.hbase.CellScannable;
import org.apache.hadoop.hbase.CellScanner; import org.apache.hadoop.hbase.CellScanner;
@ -30,7 +32,13 @@ import org.apache.hadoop.hbase.util.ReflectionUtils;
*/ */
@InterfaceAudience.Private @InterfaceAudience.Private
public class RpcControllerFactory { public class RpcControllerFactory {
private static final Log LOG = LogFactory.getLog(RpcControllerFactory.class);
/**
* Custom RPC Controller factory allows frameworks to change the RPC controller. If the configured
* controller cannot be found in the classpath or loaded, we fall back to the default RPC
* controller factory.
*/
public static final String CUSTOM_CONTROLLER_CONF_KEY = "hbase.rpc.controllerfactory.class"; public static final String CUSTOM_CONTROLLER_CONF_KEY = "hbase.rpc.controllerfactory.class";
protected final Configuration conf; protected final Configuration conf;
@ -55,7 +63,21 @@ public class RpcControllerFactory {
String rpcControllerFactoryClazz = String rpcControllerFactoryClazz =
configuration.get(CUSTOM_CONTROLLER_CONF_KEY, configuration.get(CUSTOM_CONTROLLER_CONF_KEY,
RpcControllerFactory.class.getName()); RpcControllerFactory.class.getName());
return ReflectionUtils.instantiateWithCustomCtor(rpcControllerFactoryClazz, try {
new Class[] { Configuration.class }, new Object[] { configuration }); return ReflectionUtils.instantiateWithCustomCtor(rpcControllerFactoryClazz,
new Class[] { Configuration.class }, new Object[] { configuration });
} catch (UnsupportedOperationException | NoClassDefFoundError ex) {
// HBASE-14960: In case the RPCController is in a non-HBase jar (Phoenix), but the application
// is a pure HBase application, we want to fallback to the default one.
String msg = "Cannot load configured \"" + CUSTOM_CONTROLLER_CONF_KEY + "\" ("
+ rpcControllerFactoryClazz + ") from hbase-site.xml, falling back to use "
+ "default RpcControllerFactory";
if (LOG.isDebugEnabled()) {
LOG.warn(msg, ex); // if DEBUG enabled, we want the exception, but still log in WARN level
} else {
LOG.warn(msg);
}
return new RpcControllerFactory(configuration);
}
} }
} }

View File

@ -19,6 +19,7 @@ package org.apache.hadoop.hbase.client;
import static org.apache.hadoop.hbase.HBaseTestingUtility.fam1; import static org.apache.hadoop.hbase.HBaseTestingUtility.fam1;
import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import java.io.IOException; import java.io.IOException;
import java.util.List; import java.util.List;
@ -54,14 +55,17 @@ public class TestRpcControllerFactory {
super(conf); super(conf);
} }
@Override
public PayloadCarryingRpcController newController() { public PayloadCarryingRpcController newController() {
return new CountingRpcController(super.newController()); return new CountingRpcController(super.newController());
} }
@Override
public PayloadCarryingRpcController newController(final CellScanner cellScanner) { public PayloadCarryingRpcController newController(final CellScanner cellScanner) {
return new CountingRpcController(super.newController(cellScanner)); return new CountingRpcController(super.newController(cellScanner));
} }
@Override
public PayloadCarryingRpcController newController(final List<CellScannable> cellIterables) { public PayloadCarryingRpcController newController(final List<CellScannable> cellIterables) {
return new CountingRpcController(super.newController(cellIterables)); return new CountingRpcController(super.newController(cellIterables));
} }
@ -103,7 +107,7 @@ public class TestRpcControllerFactory {
Configuration conf = UTIL.getConfiguration(); Configuration conf = UTIL.getConfiguration();
conf.set(CoprocessorHost.REGION_COPROCESSOR_CONF_KEY, conf.set(CoprocessorHost.REGION_COPROCESSOR_CONF_KEY,
ProtobufCoprocessorService.class.getName()); ProtobufCoprocessorService.class.getName());
UTIL.startMiniCluster(); UTIL.startMiniCluster();
} }
@ -202,4 +206,15 @@ public class TestRpcControllerFactory {
assertEquals(0, CountingRpcController.INT_PRIORITY.get()); assertEquals(0, CountingRpcController.INT_PRIORITY.get());
return counter + 1; return counter + 1;
} }
@Test
public void testFallbackToDefaultRpcControllerFactory() {
Configuration conf = new Configuration(UTIL.getConfiguration());
conf.set(RpcControllerFactory.CUSTOM_CONTROLLER_CONF_KEY, "foo.bar.Baz");
// Should not fail
RpcControllerFactory factory = RpcControllerFactory.instantiate(conf);
assertNotNull(factory);
assertEquals(factory.getClass(), RpcControllerFactory.class);
}
} }