org.unitils.orm.jpa.util.JpaEntityManagerFactoryLoader.java Source code

Introduction

Here is the source code for org.unitils.orm.jpa.util.JpaEntityManagerFactoryLoader.java

Source

/*
 * Copyright 2008,  Unitils.org
 *
 * 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
 *
 *     http://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.unitils.orm.jpa.util;

import org.apache.commons.lang.StringUtils;
import org.springframework.instrument.classloading.LoadTimeWeaver;
import org.springframework.orm.jpa.AbstractEntityManagerFactoryBean;
import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
import org.unitils.core.Unitils;
import org.unitils.core.UnitilsException;
import org.unitils.database.DatabaseModule;
import org.unitils.orm.common.util.ConfiguredOrmPersistenceUnit;
import org.unitils.orm.common.util.OrmPersistenceUnitLoader;
import org.unitils.orm.jpa.JpaModule;
import org.unitils.util.ReflectionUtils;

import javax.persistence.EntityManagerFactory;
import javax.sql.DataSource;

import java.lang.reflect.InvocationTargetException;

/**
 * Loads an <code>EntityManagerFactory</code> given a {@link JpaConfig} object
 *
 * @author Filip Neven
 * @author Tim Ducheyne
 */
public class JpaEntityManagerFactoryLoader
        implements OrmPersistenceUnitLoader<EntityManagerFactory, Object, JpaConfig> {

    protected String databaseName;

    /**
     * @param databaseName
     */
    public JpaEntityManagerFactoryLoader(String databaseName) {
        this.databaseName = databaseName;
    }

    public ConfiguredOrmPersistenceUnit<EntityManagerFactory, Object> getConfiguredOrmPersistenceUnit(
            Object testObject, JpaConfig entityManagerConfig) {
        AbstractEntityManagerFactoryBean factoryBean = createEntityManagerFactoryBean(testObject,
                entityManagerConfig);
        EntityManagerFactory entityManagerFactory = factoryBean.getObject();
        Object providerSpecificConfigurationObject = getJpaProviderSupport()
                .getProviderSpecificConfigurationObject(factoryBean.getPersistenceProvider());
        return new ConfiguredOrmPersistenceUnit<EntityManagerFactory, Object>(entityManagerFactory,
                providerSpecificConfigurationObject);
    }

    /**
     * @param testObject The test instance, not null
     * @param jpaConfig  The configuration parameters for the <code>EntityManagerFactory</code>
     * @return A completely configured <code>AbstractEntityManagerFactoryBean</code>
     */
    protected AbstractEntityManagerFactoryBean createEntityManagerFactoryBean(Object testObject,
            JpaConfig jpaConfig) {
        LocalContainerEntityManagerFactoryBean factoryBean = new LocalContainerEntityManagerFactoryBean();
        factoryBean.setDataSource(getDataSource());
        factoryBean.setJpaVendorAdapter(getJpaProviderSupport().getSpringJpaVendorAdaptor());
        String persistenceXmlFile = jpaConfig.getConfigFiles().iterator().next();
        if (!StringUtils.isEmpty(persistenceXmlFile)) {
            factoryBean.setPersistenceXmlLocation(persistenceXmlFile);
        }
        factoryBean.setPersistenceUnitName(jpaConfig.getPersistenceUnitName());
        LoadTimeWeaver loadTimeWeaver = getJpaProviderSupport().getLoadTimeWeaver();
        if (loadTimeWeaver != null) {
            factoryBean.setLoadTimeWeaver(loadTimeWeaver);
        }
        if (jpaConfig.getConfigMethod() != null) {
            try {
                ReflectionUtils.invokeMethod(testObject, jpaConfig.getConfigMethod(), factoryBean);
            } catch (InvocationTargetException e) {
                throw new UnitilsException("Error while invoking custom config method", e.getCause());
            }
        }
        factoryBean.afterPropertiesSet();
        return factoryBean;
    }

    protected DataSource getDataSource() {
        return getDatabaseModule().getWrapper(databaseName).getDataSource();
    }

    protected JpaProviderSupport getJpaProviderSupport() {
        return getJpaModule().getJpaProviderSupport();
    }

    protected DatabaseModule getDatabaseModule() {
        return Unitils.getInstance().getModulesRepository().getModuleOfType(DatabaseModule.class);
    }

    protected JpaModule getJpaModule() {
        return Unitils.getInstance().getModulesRepository().getModuleOfType(JpaModule.class);
    }
}