Merge branch '1.4.x' into 1.5.x

Author: wilkinsona

spring-boot-actuator/src/main/java/org/springframework/boot/actuate/autoconfigure/ManagementContextConfiguration.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2015 the original author or authors.
+ * Copyright 2012-2017 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -23,15 +23,21 @@
 import java.lang.annotation.Target;
 
 import org.springframework.context.annotation.Configuration;
+import org.springframework.core.Ordered;
+import org.springframework.core.annotation.Order;
 
 /**
  * Specialized {@link Configuration @Configuration} class that defines configuration
  * specific for the management context. Configurations should be registered in
  * {@code /META-INF/spring.factories} under the
  * {@code org.springframework.boot.actuate.autoconfigure.ManagementContextConfiguration}
  * key.
+ * <p>
+ * {@code ManagementContextConfiguration} classes can be ordered using {@link Order}.
+ * Ordering by implementing {@link Ordered} is not supported and will have no effect.
  *
  * @author Phillip Webb
+ * @author Andy Wilkinson
  * @since 1.3.0
  */
 @Target(ElementType.TYPE)

spring-boot-actuator/src/main/java/org/springframework/boot/actuate/autoconfigure/ManagementContextConfigurationsImportSelector.java

@@ -16,19 +16,20 @@
 
 package org.springframework.boot.actuate.autoconfigure;
 
+import java.io.IOException;
 import java.util.ArrayList;
-import java.util.LinkedHashSet;
 import java.util.List;
-import java.util.Set;
+import java.util.Map;
 
 import org.springframework.beans.factory.BeanClassLoaderAware;
 import org.springframework.context.annotation.DeferredImportSelector;
+import org.springframework.core.OrderComparator;
 import org.springframework.core.Ordered;
-import org.springframework.core.annotation.AnnotationAwareOrderComparator;
 import org.springframework.core.annotation.Order;
 import org.springframework.core.io.support.SpringFactoriesLoader;
 import org.springframework.core.type.AnnotationMetadata;
-import org.springframework.util.ClassUtils;
+import org.springframework.core.type.classreading.MetadataReader;
+import org.springframework.core.type.classreading.SimpleMetadataReaderFactory;
 
 /**
  * Selects configuration classes for the management context configuration. Entries are
@@ -38,6 +39,7 @@
  *
  * @author Dave Syer
  * @author Phillip Webb
+ * @author Andy Wilkinson
  * @see ManagementContextConfiguration
  */
 @Order(Ordered.LOWEST_PRECEDENCE)
@@ -48,32 +50,81 @@ class ManagementContextConfigurationsImportSelector
 
 	@Override
 	public String[] selectImports(AnnotationMetadata metadata) {
-		// Find all possible auto configuration classes, filtering duplicates
-		List<String> names = loadFactoryNames();
-		Set<Class<?>> classes = new LinkedHashSet<Class<?>>();
-		for (String factoryName : names) {
-			classes.add(ClassUtils.resolveClassName(factoryName, this.classLoader));
+		// Find all management context configuration classes, filtering duplicates
+		List<ManagementConfiguration> configurations = getConfigurations();
+		OrderComparator.sort(configurations);
+		List<String> names = new ArrayList<String>();
+		for (ManagementConfiguration configuration : configurations) {
+			names.add(configuration.getClassName());
 		}
-		return getSortedClassNames(new ArrayList<Class<?>>(classes));
+		return names.toArray(new String[names.size()]);
 	}
 
-	protected List<String> loadFactoryNames() {
-		return SpringFactoriesLoader
-				.loadFactoryNames(ManagementContextConfiguration.class, this.classLoader);
+	private List<ManagementConfiguration> getConfigurations() {
+		SimpleMetadataReaderFactory readerFactory = new SimpleMetadataReaderFactory(
+				this.classLoader);
+		List<ManagementConfiguration> configurations = new ArrayList<ManagementConfiguration>();
+		for (String className : loadFactoryNames()) {
+			getConfiguration(readerFactory, configurations, className);
+		}
+		return configurations;
 	}
 
-	private String[] getSortedClassNames(List<Class<?>> classes) {
-		AnnotationAwareOrderComparator.sort(classes);
-		List<String> names = new ArrayList<String>();
-		for (Class<?> sourceClass : classes) {
-			names.add(sourceClass.getName());
+	private void getConfiguration(SimpleMetadataReaderFactory readerFactory,
+			List<ManagementConfiguration> configurations, String className) {
+		try {
+			MetadataReader metadataReader = readerFactory.getMetadataReader(className);
+			configurations.add(new ManagementConfiguration(metadataReader));
 		}
-		return names.toArray(new String[names.size()]);
+		catch (IOException ex) {
+			throw new RuntimeException(
+					"Failed to read annotation metadata for '" + className + "'", ex);
+		}
+	}
+
+	protected List<String> loadFactoryNames() {
+		return SpringFactoriesLoader
+				.loadFactoryNames(ManagementContextConfiguration.class, this.classLoader);
 	}
 
 	@Override
 	public void setBeanClassLoader(ClassLoader classLoader) {
 		this.classLoader = classLoader;
 	}
 
+	/**
+	 * A management configuration class which can be sorted according to {@code @Order}.
+	 */
+	private static final class ManagementConfiguration implements Ordered {
+
+		private final String className;
+
+		private final int order;
+
+		ManagementConfiguration(MetadataReader metadataReader) {
+			AnnotationMetadata annotationMetadata = metadataReader
+					.getAnnotationMetadata();
+			this.order = readOrder(annotationMetadata);
+			this.className = metadataReader.getClassMetadata().getClassName();
+		}
+
+		private int readOrder(AnnotationMetadata annotationMetadata) {
+			Map<String, Object> attributes = annotationMetadata
+					.getAnnotationAttributes(Order.class.getName());
+			Integer order = (attributes == null ? null
+					: (Integer) attributes.get("value"));
+			return (order == null ? Ordered.LOWEST_PRECEDENCE : order);
+		}
+
+		public String getClassName() {
+			return this.className;
+		}
+
+		@Override
+		public int getOrder() {
+			return this.order;
+		}
+
+	}
+
 }

spring-boot-actuator/src/test/java/org/springframework/boot/actuate/autoconfigure/ManagementContextConfigurationsImportSelectorTests.java

@@ -29,6 +29,7 @@
  * Tests for {@link ManagementContextConfigurationsImportSelector}.
  *
  * @author Phillip Webb
+ * @author Andy Wilkinson
  */
 public class ManagementContextConfigurationsImportSelectorTests {
 
@@ -37,15 +38,16 @@ public void selectImportsShouldOrderResult() throws Exception {
 		String[] imports = new TestManagementContextConfigurationsImportSelector()
 				.selectImports(null);
 		assertThat(imports).containsExactly(A.class.getName(), B.class.getName(),
-				C.class.getName());
+				C.class.getName(), D.class.getName());
 	}
 
 	private static class TestManagementContextConfigurationsImportSelector
 			extends ManagementContextConfigurationsImportSelector {
 
 		@Override
 		protected List<String> loadFactoryNames() {
-			return Arrays.asList(C.class.getName(), A.class.getName(), B.class.getName());
+			return Arrays.asList(C.class.getName(), A.class.getName(), D.class.getName(),
+					B.class.getName());
 		}
 
 	}
@@ -65,4 +67,8 @@ private static class C {
 
 	}
 
+	static class D {
+
+	}
+
 }