core: minor cleanup of NameResolverProvider

* Make the list of providers an immutable List
* Make obvious that the list is statically initialized
* Add documentation for when methods were added.
* Use RuntimeException, rather than IllegalStateException.
This commit is contained in:
Carl Mastrangelo 2018-08-20 12:59:55 -07:00 committed by GitHub
parent c318b4e9d6
commit 3f05a6e331
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 65 additions and 43 deletions

View File

@ -17,11 +17,13 @@
package io.grpc;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.base.Preconditions;
import java.net.URI;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.Collections;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.annotation.Nullable;
/**
* Provider of name resolvers for name agnostic consumption.
@ -32,40 +34,42 @@ import java.util.List;
*/
@ExperimentalApi("https://github.com/grpc/grpc-java/issues/4159")
public abstract class NameResolverProvider extends NameResolver.Factory {
private static final Logger logger = Logger.getLogger(NameResolverProvider.class.getName());
/**
* The port number used in case the target or the underlying naming system doesn't provide a
* port number.
*
* @since 1.0.0
*/
@SuppressWarnings("unused") // Avoids outside callers accidentally depending on the super class.
public static final Attributes.Key<Integer> PARAMS_DEFAULT_PORT =
NameResolver.Factory.PARAMS_DEFAULT_PORT;
@VisibleForTesting
static final Iterable<Class<?>> HARDCODED_CLASSES = new HardcodedClasses();
static final Iterable<Class<?>> HARDCODED_CLASSES = getHardCodedClasses();
private static final List<NameResolverProvider> providers = ServiceProviders.loadAll(
NameResolverProvider.class,
HARDCODED_CLASSES,
NameResolverProvider.class.getClassLoader(),
new ServiceProviders.PriorityAccessor<NameResolverProvider>() {
@Override
public boolean isAvailable(NameResolverProvider provider) {
return provider.isAvailable();
}
@Override
public int getPriority(NameResolverProvider provider) {
return provider.priority();
}
});
new NameResolverPriorityAccessor());
private static final NameResolver.Factory factory = new NameResolverFactory(providers);
/**
* Returns non-{@code null} ClassLoader-wide providers, in preference order.
*
* @since 1.0.0
*/
public static List<NameResolverProvider> providers() {
return providers;
}
/**
* @since 1.0.0
*/
public static NameResolver.Factory asFactory() {
return factory;
}
@ -78,6 +82,8 @@ public abstract class NameResolverProvider extends NameResolver.Factory {
/**
* Whether this provider is available for use, taking the current environment into consideration.
* If {@code false}, no other methods are safe to be called.
*
* @since 1.0.0
*/
protected abstract boolean isAvailable();
@ -86,17 +92,20 @@ public abstract class NameResolverProvider extends NameResolver.Factory {
* consideration. 5 should be considered the default, and then tweaked based on environment
* detection. A priority of 0 does not imply that the provider wouldn't work; just that it should
* be last in line.
*
* @since 1.0.0
*/
protected abstract int priority();
private static class NameResolverFactory extends NameResolver.Factory {
private static final class NameResolverFactory extends NameResolver.Factory {
private final List<NameResolverProvider> providers;
public NameResolverFactory(List<NameResolverProvider> providers) {
this.providers = providers;
NameResolverFactory(List<NameResolverProvider> providers) {
this.providers = Collections.unmodifiableList(new ArrayList<NameResolverProvider>(providers));
}
@Override
@Nullable
public NameResolver newNameResolver(URI targetUri, Attributes params) {
checkForProviders();
for (NameResolverProvider provider : providers) {
@ -115,26 +124,41 @@ public abstract class NameResolverProvider extends NameResolver.Factory {
}
private void checkForProviders() {
Preconditions.checkState(!providers.isEmpty(),
"No NameResolverProviders found via ServiceLoader, including for DNS. "
+ "This is probably due to a broken build. If using ProGuard, check your configuration");
if (providers.isEmpty()) {
String msg = "No NameResolverProviders found via ServiceLoader, including for DNS. "
+ "This is probably due to a broken build. If using ProGuard, check your configuration";
throw new RuntimeException(msg);
}
}
}
@VisibleForTesting
static final class HardcodedClasses implements Iterable<Class<?>> {
static final List<Class<?>> getHardCodedClasses() {
// Class.forName(String) is used to remove the need for ProGuard configuration. Note that
// ProGuard does not detect usages of Class.forName(String, boolean, ClassLoader):
// https://sourceforge.net/p/proguard/bugs/418/
try {
return Collections.<Class<?>>singletonList(
Class.forName("io.grpc.internal.DnsNameResolverProvider"));
} catch (ClassNotFoundException e) {
logger.log(Level.FINE, "Unable to find DNS NameResolver", e);
}
return Collections.emptyList();
}
private static final class NameResolverPriorityAccessor
implements ServiceProviders.PriorityAccessor<NameResolverProvider> {
NameResolverPriorityAccessor() {}
@Override
public Iterator<Class<?>> iterator() {
List<Class<?>> list = new ArrayList<Class<?>>();
// Class.forName(String) is used to remove the need for ProGuard configuration. Note that
// ProGuard does not detect usages of Class.forName(String, boolean, ClassLoader):
// https://sourceforge.net/p/proguard/bugs/418/
try {
list.add(Class.forName("io.grpc.internal.DnsNameResolverProvider"));
} catch (ClassNotFoundException e) {
// ignore
}
return list.iterator();
public boolean isAvailable(NameResolverProvider provider) {
return provider.isAvailable();
}
@Override
public int getPriority(NameResolverProvider provider) {
return provider.priority();
}
}
}

View File

@ -16,14 +16,13 @@
package io.grpc;
import static com.google.common.truth.Truth.assertThat;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import com.google.common.collect.ImmutableSet;
import io.grpc.NameResolverProvider.HardcodedClasses;
import io.grpc.internal.DnsNameResolverProvider;
import java.net.URI;
import java.util.Collections;
@ -47,7 +46,7 @@ public class NameResolverProviderTest {
try {
factory.getDefaultScheme();
fail("Expected exception");
} catch (IllegalStateException ex) {
} catch (RuntimeException ex) {
assertTrue(ex.toString(), ex.getMessage().contains("No NameResolverProviders found"));
}
}
@ -73,7 +72,7 @@ public class NameResolverProviderTest {
try {
factory.newNameResolver(uri, attributes);
fail("Expected exception");
} catch (IllegalStateException ex) {
} catch (RuntimeException ex) {
assertTrue(ex.toString(), ex.getMessage().contains("No NameResolverProviders found"));
}
}
@ -87,11 +86,10 @@ public class NameResolverProviderTest {
}
@Test
public void getClassesViaHardcoded_triesToLoadClasses() throws Exception {
ServiceProvidersTestUtil.testHardcodedClasses(
HardcodedClassesCallable.class.getName(),
getClass().getClassLoader(),
ImmutableSet.of("io.grpc.internal.DnsNameResolverProvider"));
public void getClassesViaHardcoded_classesPresent() throws Exception {
List<Class<?>> classes = NameResolverProvider.getHardCodedClasses();
assertThat(classes).hasSize(1);
assertThat(classes.get(0).getName()).isEqualTo("io.grpc.internal.DnsNameResolverProvider");
}
@Test
@ -119,8 +117,8 @@ public class NameResolverProviderTest {
public static final class HardcodedClassesCallable implements Callable<Iterator<Class<?>>> {
@Override
public Iterator<Class<?>> call() throws Exception {
return new HardcodedClasses().iterator();
public Iterator<Class<?>> call() {
return NameResolverProvider.getHardCodedClasses().iterator();
}
}