Examples with DefaultEnhancementContext org.hibernate.bytecode.enhance.spi.DefaultEnhancementContext used on opensource projects

Example 1 with DefaultEnhancementContext

use of org.hibernate.bytecode.enhance.spi.DefaultEnhancementContext in project hibernate-orm by hibernate.

the class MavenEnhancePlugin method execute.

public void execute() throws MojoExecutionException, MojoFailureException {
    if (!shouldApply()) {
        getLog().warn("Skipping Hibernate bytecode enhancement plugin execution since no feature is enabled");
        return;
    }
    if (!dir.startsWith(base)) {
        throw new MojoExecutionException("The enhancement directory 'dir' (" + dir + ") is no subdirectory of 'base' (" + base + ")");
    }
    // Perform a depth first search for sourceSet
    File root = new File(this.dir);
    if (!root.exists()) {
        getLog().info("Skipping Hibernate enhancement plugin execution since there is no classes dir " + dir);
        return;
    }
    walkDir(root);
    if (sourceSet.isEmpty()) {
        getLog().info("Skipping Hibernate enhancement plugin execution since there are no classes to enhance on " + dir);
        return;
    }
    getLog().info("Starting Hibernate enhancement for classes on " + dir);
    final ClassLoader classLoader = toClassLoader(Collections.singletonList(new File(base)));
    EnhancementContext enhancementContext = new DefaultEnhancementContext() {

        @Override
        public ClassLoader getLoadingClassLoader() {
            return classLoader;
        }

        @Override
        public boolean doBiDirectionalAssociationManagement(UnloadedField field) {
            return enableAssociationManagement;
        }

        @Override
        public boolean doDirtyCheckingInline(UnloadedClass classDescriptor) {
            return enableDirtyTracking;
        }

        @Override
        public boolean hasLazyLoadableAttributes(UnloadedClass classDescriptor) {
            return enableLazyInitialization;
        }

        @Override
        public boolean isLazyLoadable(UnloadedField field) {
            return enableLazyInitialization;
        }

        @Override
        public boolean doExtendedEnhancement(UnloadedClass classDescriptor) {
            return enableExtendedEnhancement;
        }
    };
    if (enableExtendedEnhancement) {
        getLog().warn("Extended enhancement is enabled. Classes other than entities may be modified. You should consider access the entities using getter/setter methods and disable this property. Use at your own risk.");
    }
    final Enhancer enhancer = Environment.getBytecodeProvider().getEnhancer(enhancementContext);
    for (File file : sourceSet) {
        final byte[] enhancedBytecode = doEnhancement(file, enhancer);
        if (enhancedBytecode == null) {
            continue;
        }
        writeOutEnhancedClass(enhancedBytecode, file);
        getLog().info("Successfully enhanced class [" + file + "]");
    }
}

Example 2 with DefaultEnhancementContext

use of org.hibernate.bytecode.enhance.spi.DefaultEnhancementContext in project hibernate-orm by hibernate.

the class HibernatePlugin method applyEnhancement.

private void applyEnhancement(final Project project, final HibernateExtension hibernateExtension) {
    if (!hibernateExtension.enhance.shouldApply()) {
        project.getLogger().warn("Skipping Hibernate bytecode enhancement since no feature is enabled");
        return;
    }
    for (final SourceSet sourceSet : hibernateExtension.getSourceSets()) {
        project.getLogger().debug("Applying Hibernate enhancement action to SourceSet.{}", sourceSet.getName());
        final Task compileTask = project.getTasks().findByName(sourceSet.getCompileJavaTaskName());
        compileTask.doLast(new Action<Task>() {

            @Override
            public void execute(Task task) {
                project.getLogger().debug("Starting Hibernate enhancement on SourceSet.{}", sourceSet.getName());
                final ClassLoader classLoader = toClassLoader(sourceSet.getRuntimeClasspath());
                EnhancementContext enhancementContext = new DefaultEnhancementContext() {

                    @Override
                    public ClassLoader getLoadingClassLoader() {
                        return classLoader;
                    }

                    @Override
                    public boolean doBiDirectionalAssociationManagement(UnloadedField field) {
                        return hibernateExtension.enhance.getEnableAssociationManagement();
                    }

                    @Override
                    public boolean doDirtyCheckingInline(UnloadedClass classDescriptor) {
                        return hibernateExtension.enhance.getEnableDirtyTracking();
                    }

                    @Override
                    public boolean hasLazyLoadableAttributes(UnloadedClass classDescriptor) {
                        return hibernateExtension.enhance.getEnableLazyInitialization();
                    }

                    @Override
                    public boolean isLazyLoadable(UnloadedField field) {
                        return hibernateExtension.enhance.getEnableLazyInitialization();
                    }

                    @Override
                    public boolean doExtendedEnhancement(UnloadedClass classDescriptor) {
                        return hibernateExtension.enhance.getEnableExtendedEnhancement();
                    }
                };
                if (hibernateExtension.enhance.getEnableExtendedEnhancement()) {
                    logger.warn("Extended enhancement is enabled. Classes other than entities may be modified. You should consider access the entities using getter/setter methods and disable this property. Use at your own risk.");
                }
                final Enhancer enhancer = Environment.getBytecodeProvider().getEnhancer(enhancementContext);
                final FileTree fileTree = project.fileTree(sourceSet.getOutput().getClassesDir());
                for (File file : fileTree) {
                    if (!file.getName().endsWith(".class")) {
                        continue;
                    }
                    final byte[] enhancedBytecode = doEnhancement(sourceSet.getOutput().getClassesDir(), file, enhancer);
                    if (enhancedBytecode != null) {
                        writeOutEnhancedClass(enhancedBytecode, file);
                        logger.info("Successfully enhanced class [" + file + "]");
                    } else {
                        logger.info("Skipping class [" + file.getAbsolutePath() + "], not an entity nor embeddable");
                    }
                }
            }
        });
    }
}