com.liferay.portal.dao.orm.hibernate.PortletSessionFactoryImpl.java Source code

Introduction

Here is the source code for com.liferay.portal.dao.orm.hibernate.PortletSessionFactoryImpl.java

Source

/**
 * Copyright (c) 2000-2012 Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library 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 Lesser General Public License for more
 * details.
 */

package com.liferay.portal.dao.orm.hibernate;

import com.liferay.portal.dao.shard.ShardDataSourceTargetSource;
import com.liferay.portal.kernel.dao.jdbc.CurrentConnectionUtil;
import com.liferay.portal.kernel.dao.orm.ORMException;
import com.liferay.portal.kernel.dao.orm.Session;
import com.liferay.portal.kernel.log.Log;
import com.liferay.portal.kernel.log.LogFactoryUtil;
import com.liferay.portal.kernel.util.InfrastructureUtil;
import com.liferay.portal.spring.hibernate.PortletHibernateConfiguration;
import com.liferay.portal.util.PropsValues;

import java.sql.Connection;

import java.util.HashMap;
import java.util.Map;

import javax.sql.DataSource;

import org.hibernate.SessionFactory;

/**
 * @author Shuyang Zhou
 * @author Alexander Chow
 */
public class PortletSessionFactoryImpl extends SessionFactoryImpl {

    public void afterPropertiesSet() {
        if (_dataSource == InfrastructureUtil.getDataSource()) {

            // Register only if the current session factory is using the portal
            // data source

            portletSessionFactories.add(this);
        }
    }

    @Override
    public void destroy() {
        portletSessionFactories.remove(this);
    }

    public DataSource getDataSource() {
        ShardDataSourceTargetSource shardDataSourceTargetSource = (ShardDataSourceTargetSource) InfrastructureUtil
                .getShardDataSourceTargetSource();

        if (shardDataSourceTargetSource != null) {
            return shardDataSourceTargetSource.getDataSource();
        } else {
            return _dataSource;
        }
    }

    @Override
    public Session openSession() throws ORMException {
        SessionFactory sessionFactory = getSessionFactory();

        org.hibernate.Session session = null;

        if (PropsValues.SPRING_HIBERNATE_SESSION_DELEGATED) {
            Connection connection = CurrentConnectionUtil.getConnection(getDataSource());

            if (connection == null) {
                session = sessionFactory.getCurrentSession();
            } else {
                session = sessionFactory.openSession(connection);
            }
        } else {
            session = sessionFactory.openSession();
        }

        if (_log.isDebugEnabled()) {
            org.hibernate.impl.SessionImpl sessionImpl = (org.hibernate.impl.SessionImpl) session;

            _log.debug("Session is using connection release mode " + sessionImpl.getConnectionReleaseMode());
        }

        return wrapSession(session);
    }

    public void setDataSource(DataSource dataSource) {
        _dataSource = dataSource;
    }

    protected SessionFactory getSessionFactory() {
        ShardDataSourceTargetSource shardDataSourceTargetSource = (ShardDataSourceTargetSource) InfrastructureUtil
                .getShardDataSourceTargetSource();

        if (shardDataSourceTargetSource == null) {
            return getSessionFactoryImplementor();
        }

        DataSource dataSource = shardDataSourceTargetSource.getDataSource();

        SessionFactory sessionFactory = _sessionFactories.get(dataSource);

        if (sessionFactory != null) {
            return sessionFactory;
        }

        PortletHibernateConfiguration portletHibernateConfiguration = new PortletHibernateConfiguration();

        portletHibernateConfiguration.setDataSource(dataSource);

        try {
            sessionFactory = portletHibernateConfiguration.buildSessionFactory();
        } catch (Exception e) {
            _log.error(e, e);

            return null;
        }

        _sessionFactories.put(dataSource, sessionFactory);

        return sessionFactory;
    }

    private static Log _log = LogFactoryUtil.getLog(PortletSessionFactoryImpl.class);

    private DataSource _dataSource;
    private Map<DataSource, SessionFactory> _sessionFactories = new HashMap<DataSource, SessionFactory>();

}