com.bstek.dorado.data.resource.DefaultModelResourceBundleManager.java Source code

Introduction

Here is the source code for com.bstek.dorado.data.resource.DefaultModelResourceBundleManager.java

Source

/*
 * This file is part of Dorado 7.x (http://dorado7.bsdn.org).
 * 
 * Copyright (c) 2002-2012 BSTEK Corp. All rights reserved.
 * 
 * This file is dual-licensed under the AGPLv3 (http://www.gnu.org/licenses/agpl-3.0.html) 
 * and BSDN commercial (http://www.bsdn.org/licenses) licenses.
 * 
 * If you are unsure which license is appropriate for your use, please contact the sales department
 * at http://www.bstek.com/contact.
 */

package com.bstek.dorado.data.resource;

import java.io.IOException;
import java.io.InputStream;
import java.util.Locale;
import java.util.Properties;

import net.sf.ehcache.Ehcache;
import net.sf.ehcache.Element;

import org.apache.commons.collections.keyvalue.MultiKey;
import org.apache.commons.lang.StringUtils;

import com.bstek.dorado.config.definition.Definition;
import com.bstek.dorado.core.io.Resource;
import com.bstek.dorado.core.resource.ResourceBundle;
import com.bstek.dorado.util.PathUtils;

/**
 * @author Benny Bao (mailto:benny.bao@bstek.com)
 * @since 2012-5-5
 */
public class DefaultModelResourceBundleManager implements ModelResourceBundleManager {
    private static final String RESOURCE_FILE_SUFFIX = ".properties";

    private Ehcache cache;

    public void setCache(Ehcache cache) {
        this.cache = cache;
    }

    protected Resource findResource(Resource modelResource, Locale locale) throws IOException {
        if (modelResource != null) {
            Resource resource;
            String path = modelResource.getPath();
            if (StringUtils.isEmpty(path)) {
                return null;
            }

            int i = path.lastIndexOf(PathUtils.PATH_DELIM);
            if (i >= 0) {
                path = path.substring(i + 1);
            } else {
                i = path.lastIndexOf(':');
                if (i >= 0) {
                    path = path.substring(i + 1);
                }
            }
            i = path.indexOf('.');
            if (i >= 0) {
                path = path.substring(0, i);
            }

            if (locale != null) {
                String localeSuffix = '.' + locale.toString();
                try {
                    resource = modelResource.createRelative(path + localeSuffix + RESOURCE_FILE_SUFFIX);
                    if (resource != null && resource.exists()) {
                        return resource;
                    }
                } catch (Exception e) {
                    // JBOSS 5.1snowdrop?VFS???
                }
            }

            try {
                resource = modelResource.createRelative(path + RESOURCE_FILE_SUFFIX);
                if (resource != null && resource.exists()) {
                    return resource;
                }
            } catch (Exception e) {
                // JBOSS 5.1snowdrop?VFS???
            }
        }
        return null;
    }

    protected ResourceBundle doGetBundle(Resource modelResource, Locale locale) throws Exception {
        Resource resource = findResource(modelResource, locale);
        if (resource != null) {
            InputStream in = resource.getInputStream();
            try {
                Properties properties = new Properties();
                properties.load(in);
                return new ModelResourceBundle(properties);
            } finally {
                in.close();
            }
        }
        return null;
    }

    public ResourceBundle getBundle(Resource resource, Locale locale) throws Exception {
        if (resource == null) {
            return null;
        }
        String path = resource.getPath();
        if (StringUtils.isEmpty(path)) {
            return null;
        }

        Object cacheKey = new MultiKey(path, locale);
        synchronized (cache) {
            Element element = cache.get(cacheKey);
            if (element == null) {
                ResourceBundle bundle = doGetBundle(resource, locale);
                element = new Element(cacheKey, bundle);
                cache.put(element);
            }
            return (ResourceBundle) element.getObjectValue();
        }
    }

    public ResourceBundle getBundle(Definition definition, Locale locale) throws Exception {
        return getBundle(definition.getResource(), locale);
    }

}