Browse Source
This commit adds a reflection hint on EmbeddedDatabaseFactory$EmbeddedDataSourceProxy#shutdown when EmbeddedDatabaseFactory is reachable in order to allow its usage as bean destroy method. Closes gh-29259pull/29282/head
3 changed files with 100 additions and 0 deletions
@ -0,0 +1,44 @@ |
|||||||
|
/* |
||||||
|
* 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.jdbc.datasource.embedded; |
||||||
|
|
||||||
|
import java.util.Collections; |
||||||
|
|
||||||
|
import org.springframework.aot.hint.ExecutableMode; |
||||||
|
import org.springframework.aot.hint.RuntimeHints; |
||||||
|
import org.springframework.aot.hint.RuntimeHintsRegistrar; |
||||||
|
|
||||||
|
/** |
||||||
|
* {@link RuntimeHintsRegistrar} implementation that registers reflection hints for |
||||||
|
* {@code EmbeddedDataSourceProxy#shutdown} in order to allow it to be used as a bean |
||||||
|
* destroy method. |
||||||
|
* |
||||||
|
* @author Sebastien Deleuze |
||||||
|
* @since 6.0 |
||||||
|
*/ |
||||||
|
class EmbeddedDatabaseFactoryRuntimeHints implements RuntimeHintsRegistrar { |
||||||
|
|
||||||
|
@Override |
||||||
|
public void registerHints(RuntimeHints hints, ClassLoader classLoader) { |
||||||
|
hints.reflection().registerTypeIfPresent(classLoader, |
||||||
|
"org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseFactory$EmbeddedDataSourceProxy", |
||||||
|
builder -> builder |
||||||
|
.onReachableType(EmbeddedDatabaseFactory.class) |
||||||
|
.withMethod("shutdown", Collections.emptyList(), ExecutableMode.INVOKE)); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,2 @@ |
|||||||
|
org.springframework.aot.hint.RuntimeHintsRegistrar=\ |
||||||
|
org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseFactoryRuntimeHints |
||||||
@ -0,0 +1,54 @@ |
|||||||
|
/* |
||||||
|
* 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.jdbc.datasource.embedded; |
||||||
|
|
||||||
|
import org.junit.jupiter.api.BeforeEach; |
||||||
|
import org.junit.jupiter.api.Test; |
||||||
|
|
||||||
|
import org.springframework.aot.hint.RuntimeHints; |
||||||
|
import org.springframework.aot.hint.RuntimeHintsRegistrar; |
||||||
|
import org.springframework.aot.hint.predicate.RuntimeHintsPredicates; |
||||||
|
import org.springframework.core.io.support.SpringFactoriesLoader; |
||||||
|
import org.springframework.util.ClassUtils; |
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||||
|
|
||||||
|
/** |
||||||
|
* Tests for {@link EmbeddedDatabaseFactoryRuntimeHints}. |
||||||
|
* |
||||||
|
* @author Sebastien Deleuze |
||||||
|
*/ |
||||||
|
public class EmbeddedDatabaseFactoryRuntimeHintsTests { |
||||||
|
|
||||||
|
private RuntimeHints hints; |
||||||
|
|
||||||
|
@BeforeEach |
||||||
|
void setup() { |
||||||
|
this.hints = new RuntimeHints(); |
||||||
|
SpringFactoriesLoader.forResourceLocation("META-INF/spring/aot.factories") |
||||||
|
.load(RuntimeHintsRegistrar.class).forEach(registrar -> registrar |
||||||
|
.registerHints(this.hints, ClassUtils.getDefaultClassLoader())); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
void embeddedDataSourceProxyTypeHasHint() throws ClassNotFoundException { |
||||||
|
assertThat(RuntimeHintsPredicates.reflection() |
||||||
|
.onMethod("org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseFactory$EmbeddedDataSourceProxy", "shutdown")) |
||||||
|
.accepts(this.hints); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue