From 0229cc45a03191880cd74b871bb419da836ae87a Mon Sep 17 00:00:00 2001
From: Stephane Nicoll <snicoll@vmware.com>
Date: Fri, 12 Aug 2022 16:34:13 +0200
Subject: [PATCH] Add support for Async invocation in a native image

This commit adds necessary hints so that Async-annotated method can be
discovered in a native image.

Closes gh-28710
---
 .../scheduling/annotation/Async.java          |  3 ++
 .../annotation/SchedulingRuntimeHints.java    | 37 +++++++++++++++++++
 .../resources/META-INF/spring/aot.factories   |  5 ++-
 3 files changed, 44 insertions(+), 1 deletion(-)
 create mode 100644 spring-context/src/main/java/org/springframework/scheduling/annotation/SchedulingRuntimeHints.java

diff --git a/spring-context/src/main/java/org/springframework/scheduling/annotation/Async.java b/spring-context/src/main/java/org/springframework/scheduling/annotation/Async.java
index 1eba40b325..5d3c290aa3 100644
--- a/spring-context/src/main/java/org/springframework/scheduling/annotation/Async.java
+++ b/spring-context/src/main/java/org/springframework/scheduling/annotation/Async.java
@@ -22,6 +22,8 @@ import java.lang.annotation.Retention;
 import java.lang.annotation.RetentionPolicy;
 import java.lang.annotation.Target;
 
+import org.springframework.aot.hint.annotation.Reflective;
+
 /**
  * Annotation that marks a method as a candidate for <i>asynchronous</i> execution.
  *
@@ -54,6 +56,7 @@ import java.lang.annotation.Target;
 @Target({ElementType.TYPE, ElementType.METHOD})
 @Retention(RetentionPolicy.RUNTIME)
 @Documented
+@Reflective
 public @interface Async {
 
 	/**
diff --git a/spring-context/src/main/java/org/springframework/scheduling/annotation/SchedulingRuntimeHints.java b/spring-context/src/main/java/org/springframework/scheduling/annotation/SchedulingRuntimeHints.java
new file mode 100644
index 0000000000..7603ea9ebf
--- /dev/null
+++ b/spring-context/src/main/java/org/springframework/scheduling/annotation/SchedulingRuntimeHints.java
@@ -0,0 +1,37 @@
+/*
+ * Copyright 2002-2022 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.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.scheduling.annotation;
+
+import org.springframework.aot.hint.RuntimeHints;
+import org.springframework.aot.hint.RuntimeHintsRegistrar;
+import org.springframework.aot.hint.support.RuntimeHintsUtils;
+import org.springframework.lang.Nullable;
+
+/**
+ * {@link RuntimeHintsRegistrar} for scheduling annotations.
+ *
+ * @author Stephane Nicoll
+ * @since 6.0
+ */
+class SchedulingRuntimeHints implements RuntimeHintsRegistrar {
+
+	@Override
+	public void registerHints(RuntimeHints hints, @Nullable ClassLoader classLoader) {
+		RuntimeHintsUtils.registerAnnotation(hints, Async.class);
+	}
+
+}
diff --git a/spring-context/src/main/resources/META-INF/spring/aot.factories b/spring-context/src/main/resources/META-INF/spring/aot.factories
index 67939a421d..ff2960e07d 100644
--- a/spring-context/src/main/resources/META-INF/spring/aot.factories
+++ b/spring-context/src/main/resources/META-INF/spring/aot.factories
@@ -1,2 +1,5 @@
+org.springframework.aot.hint.RuntimeHintsRegistrar= \
+org.springframework.scheduling.annotation.SchedulingRuntimeHints
+
 org.springframework.beans.factory.aot.BeanRegistrationAotProcessor= \
-org.springframework.context.aot.ReflectiveProcessorBeanRegistrationAotProcessor
\ No newline at end of file
+org.springframework.context.aot.ReflectiveProcessorBeanRegistrationAotProcessor