com.tvd12.ezydata.jpa.loader.EzyJpaEntityManagerFactoryLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ezydata-jpa Show documentation
There is a newer version: 1.2.9
package com.tvd12.ezydata.jpa.loader;

import com.tvd12.ezydata.database.util.EzyDatabasePropertiesKeeper;
import com.tvd12.ezyfox.reflect.EzyReflection;
import com.tvd12.ezyfox.reflect.EzyReflectionProxy;
import lombok.AllArgsConstructor;
import org.hibernate.jpa.HibernatePersistenceProvider;

import javax.persistence.*;
import javax.persistence.spi.ClassTransformer;
import javax.persistence.spi.PersistenceUnitInfo;
import javax.persistence.spi.PersistenceUnitTransactionType;
import javax.sql.DataSource;
import java.net.URL;
import java.util.*;

public class EzyJpaEntityManagerFactoryLoader
    extends EzyDatabasePropertiesKeeper {

    protected final Set entityPackages = new HashSet<>();
    protected final List managedClassNames = new ArrayList<>();
    protected final List mappingFileNames = new ArrayList<>();
    protected String jpaVersion = "2.2";
    protected DataSource dataSource;
    protected PersistenceUnitTransactionType transactionType = PersistenceUnitTransactionType.RESOURCE_LOCAL;


    public EzyJpaEntityManagerFactoryLoader jpaVersion(String jpaVersion) {
        this.jpaVersion = jpaVersion;
        return this;
    }

    public EzyJpaEntityManagerFactoryLoader dataSource(DataSource dataSource) {
        this.dataSource = dataSource;
        return this;
    }

    public EzyJpaEntityManagerFactoryLoader transactionType(PersistenceUnitTransactionType transactionType) {
        this.transactionType = transactionType;
        return this;
    }

    public EzyJpaEntityManagerFactoryLoader mappingFileName(String mappingFileName) {
        this.mappingFileNames.add(mappingFileName);
        return this;
    }

    public EzyJpaEntityManagerFactoryLoader mappingFileName(Iterable mappingFileNames) {
        for (String mappingFileName : mappingFileNames) {
            mappingFileName(mappingFileName);
        }
        return this;
    }

    public EzyJpaEntityManagerFactoryLoader entityClass(Class entityClass) {
        this.managedClassNames.add(entityClass.getName());
        return this;
    }

    public EzyJpaEntityManagerFactoryLoader entityClasses(Iterable> entityClasses) {
        for (Class entityClass : entityClasses) {
            entityClass(entityClass);
        }
        return this;
    }

    public EzyJpaEntityManagerFactoryLoader entityPackage(String entityPackage) {
        this.entityPackages.add(entityPackage);
        return this;
    }

    public EzyJpaEntityManagerFactoryLoader entityPackages(Iterable entityPackages) {
        for (String entityPackage : entityPackages) {
            entityPackage(entityPackage);
        }
        return this;
    }

    public EntityManagerFactory load(String persistenceUnitName) {
        scanEntityPackages();
        return doLoad(persistenceUnitName);
    }

    private EntityManagerFactory doLoad(String persistenceUnitName) {
        EntityManagerFactory entityManagerFactory = null;
        try {
            entityManagerFactory = loadByHibernate(persistenceUnitName);
        } catch (Throwable e) {
            logger.warn(
                "can't load EntityManagerFactory by hibernate " +
                    "(you can disable this warning by config log level to ERROR)",
                e
            );
        }
        if (entityManagerFactory == null) {
            entityManagerFactory = loadByDefaultJpa(persistenceUnitName);
        }
        return entityManagerFactory;
    }

    private EntityManagerFactory loadByHibernate(String persistenceUnitName) {
        PersistenceUnitInfo persistenceUnitInfo =
            new PersistenceUnitInfoImpl(persistenceUnitName);
        HibernatePersistenceProvider persistenceProvider =
            new HibernatePersistenceProvider();
        return persistenceProvider
            .createContainerEntityManagerFactory(
                persistenceUnitInfo,
                Collections.EMPTY_MAP
            );
    }

    private EntityManagerFactory loadByDefaultJpa(String persistenceUnitName) {
        return Persistence.createEntityManagerFactory(persistenceUnitName);
    }

    private void scanEntityPackages() {
        if (entityPackages.isEmpty()) {
            return;
        }
        EzyReflection reflection = new EzyReflectionProxy(entityPackages);
        entityClasses(reflection.getAnnotatedClasses(Entity.class));
    }

    @AllArgsConstructor
    private class PersistenceUnitInfoImpl implements PersistenceUnitInfo {

        private final String persistenceUnitName;

        @Override
        public String getPersistenceUnitName() {
            return persistenceUnitName;
        }

        @Override
        public String getPersistenceProviderClassName() {
            return HibernatePersistenceProvider.class.getName();
        }

        @Override
        public PersistenceUnitTransactionType getTransactionType() {
            return transactionType;
        }

        @Override
        public DataSource getJtaDataSource() {
            return dataSource;
        }

        @Override
        public DataSource getNonJtaDataSource() {
            return dataSource;
        }

        @Override
        public List getMappingFileNames() {
            return mappingFileNames;
        }

        @Override
        public List getJarFileUrls() {
            return Collections.emptyList();
        }

        @Override
        public URL getPersistenceUnitRootUrl() {
            return null;
        }

        @Override
        public List getManagedClassNames() {
            return managedClassNames;
        }

        @Override
        public boolean excludeUnlistedClasses() {
            return false;
        }

        @Override
        public SharedCacheMode getSharedCacheMode() {
            return SharedCacheMode.UNSPECIFIED;
        }

        @Override
        public ValidationMode getValidationMode() {
            return ValidationMode.AUTO;
        }

        public Properties getProperties() {
            return properties;
        }

        @Override
        public String getPersistenceXMLSchemaVersion() {
            return jpaVersion;
        }

        @Override
        public ClassLoader getClassLoader() {
            return Thread.currentThread().getContextClassLoader();
        }

        @Override
        public void addTransformer(ClassTransformer transformer) {}

        @Override
        public ClassLoader getNewTempClassLoader() {
            return null;
        }
    }
}