AOT Optimizations

.github/workflows/aot.yml

@@ -0,0 +1,40 @@
+name: Run AOT Tests
+on:
+  push:
+    branches:
+      - master
+      - '[1-9]+.[0-9]+.x'
+  pull_request:
+    branches:
+      - master
+      - '[1-9]+.[0-9]+.x'
+jobs:
+  build:
+    if: github.repository != 'micronaut-projects/micronaut-project-template'
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        java: ['17']
+    steps:
+       # https://github.com/actions/virtual-environments/issues/709
+      - name: Free disk space
+        run: |
+         sudo rm -rf "/usr/local/share/boost"
+         sudo rm -rf "$AGENT_TOOLSDIRECTORY"
+         sudo apt-get clean
+         df -h
+      - uses: actions/checkout@v3
+      - uses: actions/cache@v3
+        with:
+          path: ~/.gradle/caches
+          key: ${{ runner.os }}-gradle-${{ hashFiles('**/*.gradle') }}
+          restore-keys: |
+            ${{ runner.os }}-gradle-
+      - name: Run AOT Tests
+        run: ./test-suite-aot.sh
+        env:
+           TESTCONTAINERS_RYUK_DISABLED: true
+           GRADLE_ENTERPRISE_ACCESS_KEY: ${{ secrets.GRADLE_ENTERPRISE_ACCESS_KEY }}
+           GRADLE_ENTERPRISE_CACHE_USERNAME: ${{ secrets.GRADLE_ENTERPRISE_CACHE_USERNAME }}
+           GRADLE_ENTERPRISE_CACHE_PASSWORD: ${{ secrets.GRADLE_ENTERPRISE_CACHE_PASSWORD }}
+           PREDICTIVE_TEST_SELECTION: "${{'false'}}"

buildSrc/src/main/groovy/io.micronaut.build.internal.aot-module.gradle

@@ -0,0 +1,4 @@
+plugins {
+    id "io.micronaut.build.internal.security-base"
+    id "io.micronaut.build.internal.module"
+}

gradle.properties

@@ -7,6 +7,8 @@ projectUrl=https://micronaut.io
 githubSlug=micronaut-projects/micronaut-security
 developers=James Kleeh,Sergio del Amo
 
+micronautAotVersion=1.1.1
+micronautGradlePluginVersion=3.6.3
 githubCoreBranch=3.8.x
 
 bomProperty=micronautSecurityVersion

security-aot/build.gradle.kts

@@ -0,0 +1,25 @@
+plugins {
+    id("io.micronaut.build.internal.aot-module")
+}
+micronautBuild {
+    aot {
+        version.set("1.1.1")
+    }
+}
+micronautBuild {
+    binaryCompatibility {
+        enabled.set(false)
+    }
+}
+dependencies {
+    compileOnly(platform(mn.micronaut.bom))
+    implementation(projects.securityOauth2)
+    implementation(projects.securityJwt)
+
+    testImplementation(mn.spock) {
+        exclude("org.codehaus.groovy", "groovy-all")
+    }
+    testImplementation(platform(mn.micronaut.bom))
+    testImplementation(mn.micronaut.http.server.netty)
+    testImplementation(projects.securityOauth2)
+}

security-aot/src/main/java/io/micronaut/security/aot/AOTContextUtils.java

@@ -0,0 +1,105 @@
+/*
+ * Copyright 2017-2022 original 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 io.micronaut.security.aot;
+
+import io.micronaut.aot.core.AOTContext;
+import io.micronaut.context.Qualifier;
+import io.micronaut.core.annotation.Internal;
+import io.micronaut.core.annotation.NonNull;
+import io.micronaut.core.annotation.Nullable;
+
+import java.util.Collection;
+
+/**
+ * Utility to retrieve beans from the Application Context associated to the AOT Context.
+ * @author Sergio del Amo
+ * @since 3.9.0
+ */
+@Internal
+public final class AOTContextUtils {
+    private AOTContextUtils() {
+
+    }
+
+    /**
+     * Get all beans of the given type.
+     *
+     * @param beanType The bean type
+     * @param <T>      The bean type parameter
+     * @param aotContext The AOT Context
+     * @return The found beans
+     */
+    @NonNull
+    public static <T> Collection<T> getBeansOfType(@NonNull Class<T> beanType, @NonNull AOTContext aotContext) {
+        return  aotContext.getAnalyzer()
+            .getApplicationContext()
+            .getBeansOfType(beanType);
+    }
+
+    /**
+     * Obtains a Bean for the given type.
+     *
+     * @param beanType The bean type
+     * @param <T>      The bean type parameter
+     * @param aotContext The AOT Context
+     * @return An instanceof said bean
+     * @throws io.micronaut.context.exceptions.NonUniqueBeanException When multiple possible bean definitions exist
+     *                                                                for the given type
+     * @throws io.micronaut.context.exceptions.NoSuchBeanException If the bean doesn't exist
+     */
+    @NonNull
+    public static <T> T getBean(@NonNull Class<T> beanType, @NonNull AOTContext aotContext) {
+        return  aotContext.getAnalyzer()
+            .getApplicationContext()
+            .getBean(beanType);
+    }
+
+    /**
+     * Return whether the bean of the given type is contained within this context.
+     *
+     * @param beanType  The bean type
+     * @param qualifier The qualifier for the bean
+     * @param <T>       The concrete type
+     * @param aotContext The AOT Context
+     * @return True if it is
+     */
+    @NonNull
+    public static <T> boolean containsBean(@NonNull Class<T> beanType, @Nullable Qualifier<T> qualifier, @NonNull AOTContext aotContext) {
+        return  aotContext.getAnalyzer()
+            .getApplicationContext()
+            .containsBean(beanType, qualifier);
+    }
+
+    /**
+     * Obtains a Bean for the given type and qualifier.
+     *
+     * @param beanType  The bean type
+     * @param qualifier The qualifier
+     * @param <T>       The bean type parameter
+     * @param aotContext The AOT Context
+     * @return An instanceof said bean
+     * @throws io.micronaut.context.exceptions.NonUniqueBeanException When multiple possible bean definitions exist
+     *                                                                for the given type
+     * @see io.micronaut.inject.qualifiers.Qualifiers
+     */
+    @NonNull
+    public static <T> T getBean(@NonNull Class<T> beanType, @Nullable Qualifier<T> qualifier, @NonNull AOTContext aotContext) {
+        return  aotContext.getAnalyzer()
+            .getApplicationContext()
+            .getBean(beanType, qualifier);
+    }
+}
+

security-aot/src/main/java/io/micronaut/security/aot/GeneratedFile.java

@@ -0,0 +1,81 @@
+/*
+ * Copyright 2017-2021 original 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 io.micronaut.security.aot;
+
+import com.squareup.javapoet.JavaFile;
+import io.micronaut.core.annotation.Internal;
+import io.micronaut.core.annotation.NonNull;
+import io.micronaut.core.naming.Named;
+
+/**
+ * A generated file.
+ * @author Sergio del Amo
+ * @since 3.9.0
+ */
+@Internal
+public class GeneratedFile implements Named {
+
+    @NonNull
+    private final String name;
+
+    @NonNull
+    private final String simpleName;
+
+    @NonNull
+    private final JavaFile javaFile;
+
+    /**
+     *
+     * @param name The name qualifier
+     * @param simpleName The simple filename
+     * @param javaFile The java file
+     */
+    public GeneratedFile(@NonNull String name,
+                         @NonNull String simpleName,
+                         @NonNull JavaFile javaFile) {
+        this.name = name;
+        this.simpleName = simpleName;
+        this.javaFile = javaFile;
+    }
+
+    /**
+     *
+     * @return The name qualifier
+     */
+    @Override
+    @NonNull
+    public String getName() {
+        return name;
+    }
+
+    /**
+     *
+     * @return The simple filename
+     */
+    @NonNull
+    public String getSimpleName() {
+        return simpleName;
+    }
+
+    /**
+     *
+     * @return The java file
+     */
+    @NonNull
+    public JavaFile getJavaFile() {
+        return javaFile;
+    }
+}