org.wamblee.system.spring.component.HibernateComponent Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2005-2010 the original author or authors.
 * 
 * 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.wamblee.system.spring.component;

import org.hibernate.SessionFactory;

import org.hibernate.dialect.DerbyDialect;
import org.hibernate.dialect.MySQLInnoDBDialect;

import org.springframework.orm.hibernate3.HibernateTemplate;

import org.springframework.transaction.PlatformTransactionManager;

import org.wamblee.persistence.hibernate.HibernateMappingFiles;

import org.wamblee.system.components.ORMappingConfig;
import org.wamblee.system.components.ORMappingConfig.DatabaseType;
import org.wamblee.system.core.DefaultProvidedInterface;
import org.wamblee.system.core.DefaultRequiredInterface;
import org.wamblee.system.core.ProvidedInterface;
import org.wamblee.system.core.RequiredInterface;
import org.wamblee.system.core.Scope;
import org.wamblee.system.spring.SpringComponent;

import java.io.IOException;

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

import javax.sql.DataSource;

/**
 * 
 * @author $author$
 * @version $Revision$
 */
public class HibernateComponent extends SpringComponent {
    private static final String HIBERNATE_DIALECT_PROP = "hibernate.dialect";

    private static final String HIBERNATE_SCHEMAUPDATE_PROP = "hibernate.schemaupdate";

    private static final String HIBERNATE_PROPS_KEY = "hibernateProperties";

    private static final String HIBERNATE_SPRING_CONFIG = "spring/org.wamblee.system.spring.component.hibernate.xml";

    private final RequiredInterface config = new DefaultRequiredInterface(
        "config", ORMappingConfig.class);

    /**
     * Creates a new HibernateComponent object.
     * 
     * 
     */
    public HibernateComponent(String aName) throws IOException {
        super(aName, new String[] { HIBERNATE_SPRING_CONFIG },
            createProvided(), createRequired());

        Properties props = new Properties();
        addProperties(HIBERNATE_PROPS_KEY, props);

        addRequiredInterface(config);
    }

    @Override
    protected Scope doStart(Scope aExternalScope) {
        ORMappingConfig orMappingConfig = aExternalScope.getInterfaceImplementation(
            config.getProvider(), ORMappingConfig.class);
        setProperty(HIBERNATE_SCHEMAUPDATE_PROP, "" + orMappingConfig.isSchemaUpdate());

        DatabaseType db = orMappingConfig.getType();
        String dialect = db.handleCases(new DatabaseType.Switch() {
            @Override
            public String handleMySqlInnoDb() {
                return MySQLInnoDBDialect.class.getName();
            }

            @Override
            public String handleDerby() {
                return DerbyDialect.class.getName();
            }
        });

        getHibernateProperties().put(HIBERNATE_DIALECT_PROP, dialect);

        return super.doStart(aExternalScope);
    }

    private Properties getHibernateProperties() {
        return getProperties(HIBERNATE_PROPS_KEY);
    }

    private static Map createRequired() {
        Map required = new HashMap();
        required.put(new DefaultRequiredInterface("datasource",
            DataSource.class), "dataSource");
        required.put(new DefaultRequiredInterface("mappingFiles",
            HibernateMappingFiles.class), "hibernateMappingFiles");

        return required;
    }

    private static Map createProvided() {
        Map provided = new HashMap();

        provided.put("transactionManager", new DefaultProvidedInterface(
            "transactionMgr", PlatformTransactionManager.class));
        provided.put("sessionFactory", new DefaultProvidedInterface(
            "sessionFactory", SessionFactory.class));
        provided.put("org.springframework.orm.hibernate3.HibernateTemplate",
            new DefaultProvidedInterface("hibernateTemplate",
                HibernateTemplate.class));

        return provided;
    }
}