Open Source Repository

Home /hibernate/hibernate-3.2.4.ga | Repository Home



org/hibernate/util/ConfigHelper.java
// $Id: ConfigHelper.java 9972 2006-05-31 16:59:05Z [email protected] $
package org.hibernate.util;

import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Properties;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.hibernate.HibernateException;
import org.hibernate.cfg.Environment;

/**
 * A simple class to centralize logic needed to locate config files on the system.
 *
 @author Steve
 */
public final class ConfigHelper {
  private static final Log log = LogFactory.getLog(ConfigHelper.class);

  /** Try to locate a local URL representing the incoming path.  The first attempt
   * assumes that the incoming path is an actual URL string (file://, etc).  If this
   * does not work, then the next attempts try to locate this UURL as a java system
   * resource.
   *
   @param path The path representing the config location.
   @return An appropriate URL or null.
   */
  public static final URL locateConfig(final String path) {
    try {
      return new URL(path);
    }
    catch(MalformedURLException e) {
      return findAsResource(path);
    }
  }

  /** 
   * Try to locate a local URL representing the incoming path.  
   * This method <b>only</b> attempts to locate this URL as a 
   * java system resource.
   *
   @param path The path representing the config location.
   @return An appropriate URL or null.
   */
  public static final URL findAsResource(final String path) {
    URL url = null;

    // First, try to locate this resource through the current
    // context classloader.
    ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
    if (contextClassLoader!=null) {
      url = contextClassLoader.getResource(path);
    }
    if (url != null)
      return url;

    // Next, try to locate this resource through this class's classloader
    url = ConfigHelper.class.getClassLoader().getResource(path);
    if (url != null)
      return url;

    // Next, try to locate this resource through the system classloader
    url = ClassLoader.getSystemClassLoader().getResource(path);

    // Anywhere else we should look?
    return url;
  }

  /** Open an InputStream to the URL represented by the incoming path.  First makes a call
   * to {@link #locateConfig(java.lang.String)} in order to find an appropriate URL.
   {@link java.net.URL#openStream()} is then called to obtain the stream.
   *
   @param path The path representing the config location.
   @return An input stream to the requested config resource.
   @throws HibernateException Unable to open stream to that resource.
   */
  public static final InputStream getConfigStream(final String paththrows HibernateException {
    final URL url = ConfigHelper.locateConfig(path);

    if (url == null) {
      String msg = "Unable to locate config file: " + path;
      log.fatal(msg);
      throw new HibernateException(msg);
    }

    try {
      return url.openStream();
        }
    catch(IOException e) {
          throw new HibernateException("Unable to open config file: " + path, e);
        }
  }

  /** Open an Reader to the URL represented by the incoming path.  First makes a call
   * to {@link #locateConfig(java.lang.String)} in order to find an appropriate URL.
   {@link java.net.URL#openStream()} is then called to obtain a stream, which is then
   * wrapped in a Reader.
   *
   @param path The path representing the config location.
   @return An input stream to the requested config resource.
   @throws HibernateException Unable to open reader to that resource.
   */
  public static final Reader getConfigStreamReader(final String paththrows HibernateException {
    return new InputStreamReadergetConfigStream(path) );
  }

  /** Loads a properties instance based on the data at the incoming config location.
   *
   @param path The path representing the config location.
   @return The loaded properties instance.
   @throws HibernateException Unable to load properties from that resource.
   */
  public static final Properties getConfigProperties(String paththrows HibernateException {
    try {
      Properties properties = new Properties();
      properties.loadgetConfigStream(path) );
      return properties;
    }
    catch(IOException e) {
      throw new HibernateException("Unable to load properties from specified config file: " + path, e);
    }
  }
  
  private ConfigHelper() {}

  public static InputStream getResourceAsStream(String resource) {
    String stripped = resource.startsWith("/"?
        resource.substring(1: resource;

    InputStream stream = null;
    ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
    if (classLoader!=null) {
      stream = classLoader.getResourceAsStreamstripped );
    }
    if stream == null ) {
      stream = Environment.class.getResourceAsStreamresource );
    }
    if stream == null ) {
      stream = Environment.class.getClassLoader().getResourceAsStreamstripped );
    }
    if stream == null ) {
      throw new HibernateExceptionresource + " not found" );
    }
    return stream;
  }


  public static InputStream getUserResourceAsStream(String resource) {
    boolean hasLeadingSlash = resource.startsWith"/" );
    String stripped = hasLeadingSlash ? resource.substring(1: resource;

    InputStream stream = null;

    ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
    if classLoader != null ) {
      stream = classLoader.getResourceAsStreamresource );
      if stream == null && hasLeadingSlash ) {
        stream = classLoader.getResourceAsStreamstripped );
      }
    }

    if stream == null ) {
      stream = Environment.class.getClassLoader().getResourceAsStreamresource );
    }
    if stream == null && hasLeadingSlash ) {
      stream = Environment.class.getClassLoader().getResourceAsStreamstripped );
    }

    if stream == null ) {
      throw new HibernateExceptionresource + " not found" );
    }

    return stream;
  }

}