/*
***************************************************************************************
* Copyright (C) 2006 EsperTech, Inc. All rights reserved. *
* http://www.espertech.com/esper *
* http://www.espertech.com *
* ---------------------------------------------------------------------------------- *
* The software in this package is published under the terms of the GPL license *
* a copy of which has been included with this distribution in the license.txt file. *
***************************************************************************************
*/
package com.espertech.esperio.db.config;
import com.espertech.esper.client.ConfigurationDBRef;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.Document;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import java.io.*;
import java.net.URL;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class ConfigurationDBAdapter {
private static Logger log = LoggerFactory.getLogger(ConfigurationDBAdapter.class);
private Map<String, ConfigurationDBRef> jdbcConnections;
private List<DMLQuery> dmlQueries;
private List<UpsertQuery> upsertQueries;
private Map<String, Executor> executors;
public ConfigurationDBAdapter() {
jdbcConnections = new HashMap<String, ConfigurationDBRef>();
dmlQueries = new ArrayList<DMLQuery>();
upsertQueries = new ArrayList<UpsertQuery>();
upsertQueries = new ArrayList<UpsertQuery>();
executors = new HashMap<String, Executor>();
}
public Map<String, ConfigurationDBRef> getJdbcConnections() {
return jdbcConnections;
}
public void setJdbcConnections(Map<String, ConfigurationDBRef> jdbcConnections) {
this.jdbcConnections = jdbcConnections;
}
public List<DMLQuery> getDmlQueries() {
return dmlQueries;
}
public void setDmlQueries(List<DMLQuery> dmlQueries) {
this.dmlQueries = dmlQueries;
}
public List<UpsertQuery> getUpsertQueries() {
return upsertQueries;
}
public void setUpsertQueries(List<UpsertQuery> upsertQueries) {
this.upsertQueries = upsertQueries;
}
public Map<String, Executor> getExecutors() {
return executors;
}
public void setExecutors(Map<String, Executor> executors) {
this.executors = executors;
}
/**
* Use the configuration specified in an application
* resource named <tt>esper.cfg.xml</tt>.
*
* @return Configuration initialized from the resource
* @throws RuntimeException thrown to indicate error reading configuration
*/
public ConfigurationDBAdapter configure() throws RuntimeException {
configure('/' + "esperio.db.cfg.xml");
return this;
}
/**
* Use the ConfigurationDBAdapter specified in the given application
* resource. The format of the resource is defined in
* <tt>esper-configuration-2.0.xsd</tt>.\
* <p>
* The resource is found via <tt>getConfigurationInputStream(resource)</tt>.
* That method can be overridden to implement an arbitrary lookup strategy.
* </p>
* See <tt>getResourceAsStream</tt> for information on how the resource name is resolved.
*
* @param resource if the file name of the resource
* @return ConfigurationDBAdapter initialized from the resource
* @throws RuntimeException thrown to indicate error reading configuration
*/
public ConfigurationDBAdapter configure(String resource) throws RuntimeException {
if (log.isInfoEnabled()) {
log.info("Configuring from resource: " + resource);
}
InputStream stream = getConfigurationInputStream(resource);
ConfigurationDBAdapterParser.doConfigure(this, stream, resource);
return this;
}
/**
* Get the ConfigurationDBAdapter file as an <tt>InputStream</tt>. Might be overridden
* by subclasses to allow the ConfigurationDBAdapter to be located by some arbitrary
* mechanism.
* <p>
* See <tt>getResourceAsStream</tt> for information on how the resource name is resolved.
*
* @param resource is the resource name
* @return input stream for resource
* @throws RuntimeException thrown to indicate error reading configuration
*/
protected static InputStream getConfigurationInputStream(String resource) throws RuntimeException {
return getResourceAsStream(resource);
}
/**
* Use the ConfigurationDBAdapter specified by the given XML String.
* The format of the document obtained from the URL is defined in
* <tt>esper-configuration-2.0.xsd</tt>.
*
* @param xml XML string
* @return A ConfigurationDBAdapter configured via the file
* @throws RuntimeException is thrown when the URL could not be access
*/
public ConfigurationDBAdapter configureFromString(String xml) throws RuntimeException {
if (log.isInfoEnabled()) {
log.info("Configuring from string");
}
try {
InputSource source = new InputSource(new StringReader(xml));
DocumentBuilderFactory builderFactory = DocumentBuilderFactory.newInstance();
builderFactory.setNamespaceAware(true);
Document doc = builderFactory.newDocumentBuilder().parse(source);
ConfigurationDBAdapterParser.doConfigure(this, doc);
return this;
} catch (IOException ioe) {
throw new RuntimeException("could not configure from String: " + ioe.getMessage(), ioe);
} catch (SAXException e) {
throw new RuntimeException("could not configure from String: " + e.getMessage(), e);
} catch (ParserConfigurationException e) {
throw new RuntimeException("could not configure from String: " + e.getMessage(), e);
}
}
/**
* Use the ConfigurationDBAdapter specified by the given URL.
* The format of the document obtained from the URL is defined in
* <tt>esper-configuration-2.0.xsd</tt>.
*
* @param url URL from which you wish to load the configuration
* @return A ConfigurationDBAdapter configured via the file
* @throws RuntimeException is thrown when the URL could not be access
*/
public ConfigurationDBAdapter configure(URL url) throws RuntimeException {
if (log.isInfoEnabled()) {
log.info("Configuring from url: " + url.toString());
}
try {
ConfigurationDBAdapterParser.doConfigure(this, url.openStream(), url.toString());
return this;
} catch (IOException ioe) {
throw new RuntimeException("could not configure from URL: " + url, ioe);
}
}
/**
* Returns an input stream from an application resource in the classpath.
* <p>
* The method first removes the '/' character from the resource name if
* the first character is '/'.
* <p>
* The lookup order is as follows:
* <p>
* If a thread context class loader exists, use <tt>Thread.currentThread().getResourceAsStream</tt>
* to obtain an InputStream.
* <p>
* If no input stream was returned, use the <tt>Configuration.class.getResourceAsStream</tt>.
* to obtain an InputStream.
* <p>
* If no input stream was returned, use the <tt>Configuration.class.getClassLoader().getResourceAsStream</tt>.
* to obtain an InputStream.
* <p>
* If no input stream was returned, throw an Exception.
*
* @param resource to get input stream for
* @return input stream for resource
*/
protected 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.getResourceAsStream(stripped);
}
if (stream == null) {
stream = ConfigurationDBAdapter.class.getResourceAsStream(resource);
}
if (stream == null) {
stream = ConfigurationDBAdapter.class.getClassLoader().getResourceAsStream(stripped);
}
if (stream == null) {
throw new RuntimeException(resource + " not found");
}
return stream;
}
/**
* Use the ConfigurationDBAdapter specified in the given application
* file. The format of the file is defined in
* <tt>esper-configuration-2.0.xsd</tt>.
*
* @param configFile <tt>File</tt> from which you wish to load the configuration
* @return A ConfigurationDBAdapter configured via the file
* @throws RuntimeException when the file could not be found
*/
public ConfigurationDBAdapter configure(File configFile) throws RuntimeException {
if (log.isDebugEnabled()) {
log.debug("configuring from file: " + configFile.getName());
}
try {
ConfigurationDBAdapterParser.doConfigure(this, new FileInputStream(configFile), configFile.toString());
} catch (FileNotFoundException fnfe) {
throw new RuntimeException("could not find file: " + configFile, fnfe);
}
return this;
}
}