root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Polish ContextAotProcessor

This commit is contained in:
Sam Brannen 2022-10-10 18:58:17 +02:00
parent cad7444afa
commit 7ed42ca4ca
1 changed files with 26 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
spring-context/src/main/java/org/springframework/context/aot/ContextAotProcessor.java
View File

@ -42,30 +42,32 @@ import org.springframework.util.CollectionUtils;
* @author Stephane Nicoll * @author Stephane Nicoll
* @author Andy Wilkinson * @author Andy Wilkinson
* @author Phillip Webb * @author Phillip Webb
* @author Sam Brannen
* @since 6.0 * @since 6.0
* @see org.springframework.test.context.aot.TestAotProcessor * @see org.springframework.test.context.aot.TestAotProcessor
*/ */
public abstract class ContextAotProcessor extends AbstractAotProcessor { public abstract class ContextAotProcessor extends AbstractAotProcessor {
private final Class<?> application; private final Class<?> applicationClass;
/** /**
* Create a new processor for the specified application entry point and * Create a new processor for the specified application entry point and
* common settings. * common settings.
* @param application the application entry point * @param applicationClass the application entry point (class with a {@code main()} method)
* @param settings the settings to apply * @param settings the settings to apply
*/ */
protected ContextAotProcessor(Class<?> application, Settings settings) { protected ContextAotProcessor(Class<?> applicationClass, Settings settings) {
super(settings); super(settings);
this.application = application; this.applicationClass = applicationClass;
} }
/** /**
* Get the the application entry point. * Get the the application entry point (typically a class with a {@code main()} method).
*/ */
protected Class<?> getApplication() { protected Class<?> getApplicationClass() {
return this.application; return this.applicationClass;
} }
@ -77,16 +79,16 @@ public abstract class ContextAotProcessor extends AbstractAotProcessor {
*/ */
public ClassName process() { public ClassName process() {
deleteExistingOutput(); deleteExistingOutput();
GenericApplicationContext applicationContext = prepareApplicationContext(getApplication()); GenericApplicationContext applicationContext = prepareApplicationContext(getApplicationClass());
return performAotProcessing(applicationContext); return performAotProcessing(applicationContext);
} }
/** /**
* Prepare the {@link GenericApplicationContext} for the specified * Prepare the {@link GenericApplicationContext} for the specified
* application to be used against an {@link ApplicationContextAotGenerator}. * application entry point to be used against an {@link ApplicationContextAotGenerator}.
* @return a non-refreshed {@link GenericApplicationContext} * @return a non-refreshed {@link GenericApplicationContext}
*/ */
protected abstract GenericApplicationContext prepareApplicationContext(Class<?> application); protected abstract GenericApplicationContext prepareApplicationContext(Class<?> applicationClass);
/** /**
* Perform ahead-of-time processing of the specified context. * Perform ahead-of-time processing of the specified context.
@ -104,31 +106,34 @@ public abstract class ContextAotProcessor extends AbstractAotProcessor {
registerEntryPointHint(generationContext, generatedInitializerClassName); registerEntryPointHint(generationContext, generatedInitializerClassName);
generationContext.writeGeneratedContent(); generationContext.writeGeneratedContent();
writeHints(generationContext.getRuntimeHints()); writeHints(generationContext.getRuntimeHints());
writeNativeImageProperties(getDefaultNativeImageArguments(getApplication().getName())); writeNativeImageProperties(getDefaultNativeImageArguments(getApplicationClass().getName()));
return generatedInitializerClassName; return generatedInitializerClassName;
} }
/** /**
* Callback to customize the {@link ClassNameGenerator}. By default, a * Callback to customize the {@link ClassNameGenerator}.
* standard {@link ClassNameGenerator} using the configured application * <p>By default, a standard {@link ClassNameGenerator} using the configured
* as the default target is used. * {@linkplain #getApplicationClass() application entry point} as the default
* target is used.
* @return the class name generator * @return the class name generator
*/ */
protected ClassNameGenerator createClassNameGenerator() { protected ClassNameGenerator createClassNameGenerator() {
return new ClassNameGenerator(ClassName.get(getApplication())); return new ClassNameGenerator(ClassName.get(getApplicationClass()));
} }
/** /**
* Return the native image arguments to use. By default, the main * Return the native image arguments to use.
* class to use, as well as standard application flags are added. * <p>By default, the main class to use, as well as standard application flags
* are added.
* <p>If the returned list is empty, no {@code native-image.properties} is * <p>If the returned list is empty, no {@code native-image.properties} is
* contributed. * contributed.
* @param application the application entry point * @param applicationClassName the fully qualified class name of the application
* entry point
* @return the native image options to contribute * @return the native image options to contribute
*/ */
protected List<String> getDefaultNativeImageArguments(String application) { protected List<String> getDefaultNativeImageArguments(String applicationClassName) {
List<String> args = new ArrayList<>(); List<String> args = new ArrayList<>();
args.add("-H:Class=" + application); args.add("-H:Class=" + applicationClassName);
args.add("--report-unsupported-elements-at-runtime"); args.add("--report-unsupported-elements-at-runtime");
args.add("--no-fallback"); args.add("--no-fallback");
args.add("--install-exit-handlers"); args.add("--install-exit-handlers");
@ -139,7 +144,7 @@ public abstract class ContextAotProcessor extends AbstractAotProcessor {
ClassName generatedInitializerClassName) { ClassName generatedInitializerClassName) {
TypeReference generatedType = TypeReference.of(generatedInitializerClassName.canonicalName()); TypeReference generatedType = TypeReference.of(generatedInitializerClassName.canonicalName());
TypeReference applicationType = TypeReference.of(getApplication()); TypeReference applicationType = TypeReference.of(getApplicationClass());
ReflectionHints reflection = generationContext.getRuntimeHints().reflection(); ReflectionHints reflection = generationContext.getRuntimeHints().reflection();
reflection.registerType(applicationType); reflection.registerType(applicationType);
reflection.registerType(generatedType, typeHint -> typeHint.onReachableType(applicationType) reflection.registerType(generatedType, typeHint -> typeHint.onReachableType(applicationType)