org.beangle.orm.hibernate.tool.ConfigBuilder.java Source code

Introduction

Here is the source code for org.beangle.orm.hibernate.tool.ConfigBuilder.java

Source

/*
 * Beangle, Agile Development Scaffold and Toolkit
 *
 * Copyright (c) 2005-2014, Beangle Software.
 *
 * Beangle 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 3 of the License, or
 * (at your option) any later version.
 *
 * Beangle 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.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Beangle.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.beangle.orm.hibernate.tool;

import java.io.InputStream;
import java.util.Enumeration;
import java.util.Properties;

import org.beangle.commons.entity.orm.AbstractPersistModule;
import org.beangle.commons.entity.orm.EntityPersistConfig;
import org.beangle.commons.io.IOs;
import org.beangle.commons.lang.ClassLoaders;
import org.beangle.orm.hibernate.DefaultTableNamingStrategy;
import org.beangle.orm.hibernate.RailsNamingStrategy;
import org.hibernate.cfg.Configuration;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;

public class ConfigBuilder {

    /**
     * build configration
     * 
     * @throws Exception
     */
    @SuppressWarnings("unchecked")
    public static Configuration build(Configuration cfg) throws Exception {
        PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver(
                DdlGenerator.class.getClassLoader());

        // config naming strategy
        DefaultTableNamingStrategy tableNamingStrategy = new DefaultTableNamingStrategy();
        for (Resource resource : resolver.getResources("classpath*:META-INF/beangle/table.properties"))
            tableNamingStrategy.addConfig(resource.getURL());
        RailsNamingStrategy namingStrategy = new RailsNamingStrategy();
        namingStrategy.setTableNamingStrategy(tableNamingStrategy);
        cfg.setNamingStrategy(namingStrategy);

        for (Resource resource : resolver.getResources("classpath*:META-INF/hibernate.cfg.xml"))
            cfg.configure(resource.getURL());

        for (Resource resource : resolver.getResources("classpath*:META-INF/beangle/persist*.properties")) {
            InputStream is = resource.getURL().openStream();
            Properties props = new Properties();
            if (null != is)
                props.load(is);

            Object module = props.remove("module");
            if (null == module)
                continue;

            Class<? extends AbstractPersistModule> moduleClass = (Class<? extends AbstractPersistModule>) ClassLoaders
                    .loadClass(module.toString());
            addPersistInfo(cfg, moduleClass.newInstance().getConfig());
            Enumeration<String> enumer = (Enumeration<String>) props.propertyNames();
            while (enumer.hasMoreElements()) {
                String propertyName = enumer.nextElement();
                cfg.setProperty(propertyName, props.getProperty(propertyName));
            }
            IOs.close(is);
        }
        cfg.buildMappings();
        return cfg;
    }

    private static void addPersistInfo(Configuration cfg, EntityPersistConfig epconfig) {
        for (EntityPersistConfig.EntityDefinition definition : epconfig.getEntites()) {
            cfg.addAnnotatedClass(definition.getClazz());
            if (null != definition.getCacheUsage()) {
                String region = (null == definition.getCacheRegion()) ? definition.getEntityName()
                        : definition.getCacheRegion();
                cfg.setCacheConcurrencyStrategy(definition.getEntityName(), definition.getCacheUsage(), region,
                        true);
            }
        }
        for (EntityPersistConfig.CollectionDefinition definition : epconfig.getCollections()) {
            if (null == definition.getCacheUsage())
                continue;
            String role = epconfig.getEntity(definition.getClazz()).getEntityName() + "."
                    + definition.getProperty();
            String region = (null == definition.getCacheRegion()) ? role : definition.getCacheRegion();
            cfg.setCollectionCacheConcurrencyStrategy(role, definition.getCacheUsage(), region);
        }
    }

}