-
Notifications
You must be signed in to change notification settings - Fork 2.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Speedup Hibernate ORM's enhancement of large models
- Loading branch information
Showing
3 changed files
with
161 additions
and
30 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
132 changes: 132 additions & 0 deletions
132
...nt/src/main/java/io/quarkus/hibernate/orm/deployment/integration/DeferredClassLoader.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,132 @@ | ||
| package io.quarkus.hibernate.orm.deployment.integration; | ||
|
|
||
| import java.io.IOException; | ||
| import java.io.InputStream; | ||
| import java.net.URL; | ||
| import java.util.Enumeration; | ||
| import java.util.stream.Stream; | ||
|
|
||
| /** | ||
| * This is meant to be passed exclusively to the ByteBuddy based entity Enhancer of Hibernate ORM. | ||
| * | ||
| * High couple warning: we rely on the knowledge that ByteBuddy will exclusively invoke method | ||
| * {@link ClassLoader#getResourceAsStream(String)} on the passed classloader; to verify that | ||
| * this assumption doesn't break in the future we throw exceptions on all other operations. | ||
| * | ||
| * Why do we do this? | ||
| * We need the Enhancer to source such resources from the thread context classloader which is set during the | ||
| * enhancement process, but we can't capture this yet as it hasn't been created at the point in which we | ||
| * need to pass a ClassLoader to the Hibernate ORM context. | ||
| * On the other hand, deferring the creation of the ORM enhancer implies needing to create a new enhancer | ||
| * instance for each processed entity, which very significantly slows down the process. | ||
| */ | ||
| final class DeferredClassLoader extends ClassLoader { | ||
|
|
||
| public static final ClassLoader INSTANCE = new DeferredClassLoader(); | ||
|
|
||
| protected DeferredClassLoader() { | ||
| super(); | ||
| } | ||
|
|
||
| @Override | ||
| public String getName() { | ||
| return "Defferred Enhancement Classloader"; | ||
| } | ||
|
|
||
| @Override | ||
| public Class<?> loadClass(String name) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| protected Class<?> loadClass(String name, boolean resolve) throws ClassNotFoundException { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| protected Object getClassLoadingLock(String className) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| protected Class<?> findClass(String name) throws ClassNotFoundException { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| protected Class<?> findClass(String moduleName, String name) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| protected URL findResource(String moduleName, String name) throws IOException { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| public URL getResource(String name) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| public Enumeration<URL> getResources(String name) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| public Stream<URL> resources(String name) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| protected URL findResource(String name) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| protected Enumeration<URL> findResources(String name) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| /** | ||
| * The only implemented method: delegate to the currently set context classloader. | ||
| */ | ||
| @Override | ||
| public InputStream getResourceAsStream(String name) { | ||
| return Thread.currentThread().getContextClassLoader().getResourceAsStream(name); | ||
| } | ||
|
|
||
| @Override | ||
| protected Package definePackage( | ||
| String name, | ||
| String specTitle, | ||
| String specVersion, | ||
| String specVendor, | ||
| String implTitle, | ||
| String implVersion, | ||
| String implVendor, | ||
| URL sealBase) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| protected Package getPackage(String name) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| protected Package[] getPackages() { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| @Override | ||
| protected String findLibrary(String libname) { | ||
| throw dontUse(); | ||
| } | ||
|
|
||
| private UnsupportedOperationException dontUse() { | ||
| return new UnsupportedOperationException( | ||
| "This classloader only supports the #getResourceAsStream() operation and should not be used for anything else"); | ||
| } | ||
|
|
||
| } |
21 changes: 21 additions & 0 deletions
21
.../main/java/io/quarkus/hibernate/orm/deployment/integration/QuarkusEnhancementContext.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,21 @@ | ||
| package io.quarkus.hibernate.orm.deployment.integration; | ||
|
|
||
| import org.hibernate.bytecode.enhance.spi.DefaultEnhancementContext; | ||
| import org.hibernate.bytecode.enhance.spi.UnloadedField; | ||
|
|
||
| public final class QuarkusEnhancementContext extends DefaultEnhancementContext { | ||
|
|
||
| @Override | ||
| public boolean doBiDirectionalAssociationManagement(final UnloadedField field) { | ||
| //Don't enable automatic association management as it's often too surprising. | ||
| //Also, there's several cases in which its semantics are of unspecified, | ||
| //such as what should happen when dealing with ordered collections. | ||
| return false; | ||
| } | ||
|
|
||
| @Override | ||
| public ClassLoader getLoadingClassLoader() { | ||
| return DeferredClassLoader.INSTANCE; | ||
| } | ||
|
|
||
| } |