osgi.jee.samples.jpa.aries.hibernate.derby.internal.services.AriesHibernateDerbyPersistenceService.java Source code

Introduction

Here is the source code for osgi.jee.samples.jpa.aries.hibernate.derby.internal.services.AriesHibernateDerbyPersistenceService.java

Source

/**
 * OSGi/JEE Sample.
 * 
 * Copyright (C) 2014 Goulwen Le Fur
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License.
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package osgi.jee.samples.jpa.aries.hibernate.derby.internal.services;

import java.io.InputStream;
import java.sql.Connection;
import java.sql.SQLException;

import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;

import org.apache.derby.tools.ij;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.engine.spi.SessionFactoryImplementor;
import org.hibernate.service.jdbc.connections.spi.ConnectionProvider;

import osgi.jee.samples.jpa.api.services.persistence.PersistenceService;
import osgi.jee.samples.jpa.model.Constants;

/**
 * @author <a href="mailto:goulwen.lefur@gmail.com">Goulwen Le Fur</a>.
 *
 */
public class AriesHibernateDerbyPersistenceService implements PersistenceService {

    private EntityManagerFactory entityManagerFactory;

    /**
     * @param entityManagerFactory the entityManagerFactory to set
     */
    public void setEntityManagerFactory(EntityManagerFactory entityManagerFactory) {
        this.entityManagerFactory = entityManagerFactory;
    }

    /**
     * {@inheritDoc}
     * @see osgi.jee.samples.jpa.api.services.persistence.PersistenceService#createEntityManager()
     */
    public EntityManager createEntityManager() {
        return entityManagerFactory.createEntityManager();
    }

    /**
     * {@inheritDoc}
     * @see osgi.jee.samples.jpa.api.services.persistence.PersistenceService#extractConnection(javax.persistence.EntityManager)
     */
    public Connection extractConnection(EntityManager entityManager) {
        Session session = entityManager.unwrap(Session.class);
        SessionFactory sessionFactory = session.getSessionFactory();
        if (sessionFactory instanceof SessionFactoryImplementor) {
            @SuppressWarnings("deprecation")
            ConnectionProvider connectionProvider = ((SessionFactoryImplementor) sessionFactory)
                    .getConnectionProvider();
            try {
                return connectionProvider.getConnection();
            } catch (SQLException e) {
                return null;
            }
        }
        return null;
    }

    /**
     * {@inheritDoc}
     * @see osgi.jee.samples.jpa.api.services.persistence.PersistenceService#initSchema(java.sql.Connection, java.io.InputStream)
     */
    public void initSchema(Connection connection, InputStream schemaResource) throws Exception {
        ij.runScript(connection, schemaResource, Constants.UTF_8_ENCODING, System.out, Constants.UTF_8_ENCODING);
    }

}