diff --git a/spring-session-jdbc/src/main/java/org/springframework/session/jdbc/aot/hint/SessionJdbcSecurityHints.java b/spring-session-jdbc/src/main/java/org/springframework/session/jdbc/aot/hint/SessionJdbcSecurityHints.java new file mode 100644 index 00000000..b2806d73 --- /dev/null +++ b/spring-session-jdbc/src/main/java/org/springframework/session/jdbc/aot/hint/SessionJdbcSecurityHints.java @@ -0,0 +1,38 @@ +/* + * Copyright 2014-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.session.jdbc.aot.hint; + +import org.springframework.aot.hint.MemberCategory; +import org.springframework.aot.hint.RuntimeHints; +import org.springframework.aot.hint.RuntimeHintsRegistrar; +import org.springframework.aot.hint.TypeReference; + +/** + * A {@link RuntimeHintsRegistrar} for JDBC Session hints. + * + * @author Marcus Da Coregio + */ +public class SessionJdbcSecurityHints implements RuntimeHintsRegistrar { + + @Override + public void registerHints(RuntimeHints hints, ClassLoader classLoader) { + hints.reflection().registerType(TypeReference.of("javax.sql.DataSource"), + (hint) -> hint.withMembers(MemberCategory.INVOKE_DECLARED_METHODS)); + hints.resources().registerPattern("org/springframework/session/jdbc/*.sql"); + } + +} diff --git a/spring-session-jdbc/src/main/resources/META-INF/spring/aot.factories b/spring-session-jdbc/src/main/resources/META-INF/spring/aot.factories new file mode 100644 index 00000000..8097ed4e --- /dev/null +++ b/spring-session-jdbc/src/main/resources/META-INF/spring/aot.factories @@ -0,0 +1,2 @@ +org.springframework.aot.hint.RuntimeHintsRegistrar=\ +org.springframework.session.jdbc.aot.hint.SessionJdbcSecurityHints diff --git a/spring-session-jdbc/src/test/java/org/springframework/session/jdbc/aot/hint/SessionJdbcSecurityHintsTests.java b/spring-session-jdbc/src/test/java/org/springframework/session/jdbc/aot/hint/SessionJdbcSecurityHintsTests.java new file mode 100644 index 00000000..b2f480a6 --- /dev/null +++ b/spring-session-jdbc/src/test/java/org/springframework/session/jdbc/aot/hint/SessionJdbcSecurityHintsTests.java @@ -0,0 +1,62 @@ +/* + * Copyright 2014-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.session.jdbc.aot.hint; + +import org.junit.jupiter.api.Test; + +import org.springframework.aot.hint.RuntimeHints; +import org.springframework.aot.hint.RuntimeHintsPredicates; +import org.springframework.aot.hint.RuntimeHintsRegistrar; +import org.springframework.aot.hint.TypeReference; +import org.springframework.core.io.support.SpringFactoriesLoader; + +import static org.assertj.core.api.Assertions.assertThat; + +/** + * Tests for {@link SessionJdbcSecurityHints} + * + * @author Marcus Da Coregio + */ +class SessionJdbcSecurityHintsTests { + + private final RuntimeHints hints = new RuntimeHints(); + + private final SessionJdbcSecurityHints sessionJdbcSecurityHints = new SessionJdbcSecurityHints(); + + @Test + void aotFactoriesContainsRegistrar() { + boolean match = SpringFactoriesLoader.forResourceLocation("META-INF/spring/aot.factories") + .load(RuntimeHintsRegistrar.class).stream() + .anyMatch((registrar) -> registrar instanceof SessionJdbcSecurityHints); + assertThat(match).isTrue(); + } + + @Test + void jdbcSchemasHasHints() { + this.sessionJdbcSecurityHints.registerHints(this.hints, getClass().getClassLoader()); + assertThat(RuntimeHintsPredicates.resource().forResource("org/springframework/session/jdbc/schema.sql")) + .accepts(this.hints); + } + + @Test + void dataSourceHasHints() { + this.sessionJdbcSecurityHints.registerHints(this.hints, getClass().getClassLoader()); + assertThat(RuntimeHintsPredicates.reflection().onType(TypeReference.of("javax.sql.DataSource"))) + .accepts(this.hints); + } + +}