org.kuali.db.SQLGenerator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jdbc-support Show documentation
Various utility methods for JDBC
There is a newer version: 1.1.3
package org.kuali.db;

import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.Properties;

import org.apache.commons.io.IOUtils;
import org.springframework.core.io.DefaultResourceLoader;
import org.springframework.core.io.Resource;
import org.springframework.util.PropertyPlaceholderHelper;

/**
 * 
 *
 */
public class SQLGenerator {
	PropertyPlaceholderHelper helper = new PropertyPlaceholderHelper("${", "}");
	JDBCUtils jdbcUtils = new JDBCUtils();
	DefaultResourceLoader loader = new DefaultResourceLoader();
	String encoding;
	Properties properties;
	String url;
	DatabaseCommand command;
	String prefix = "classpath:org/kuali/db/";

	public SQLGenerator() {
		this(null, null, null);
	}

	public SQLGenerator(Properties properties, String url, DatabaseCommand command) {
		super();
		this.properties = properties;
		this.url = url;
		this.command = command;
	}

	protected String getSQLLocation() {
		return getLocation("/" + getPackageFriendlyName(command.toString()) + ".sql");
	}

	public String getSQL() throws IOException {
		String location = getSQLLocation();
		Resource resource = loader.getResource(location);
		String sql = IOUtils.toString(resource.getInputStream(), getEncoding());
		Properties customProperties = getCustomProperties();
		if (properties != null) {
			customProperties.putAll(properties);
		}
		sql = helper.replacePlaceholders(sql, customProperties);
		return sql;
	}

	protected String getPackageFriendlyName(String s) {
		return s.toLowerCase().replace("_", "");
	}

	protected String getLocation(String suffix) {
		JDBCConfiguration jdbcConfiguration = jdbcUtils.getDatabaseConfiguration(url);
		DatabaseType type = jdbcConfiguration.getType();
		String packageFriendlyName = getPackageFriendlyName(type.toString());
		String location = prefix + packageFriendlyName + suffix;
		return location;
	}

	protected String getCustomPropertiesLocation() {
		return getLocation("/custom.properties");
	}

	protected Properties getCustomProperties() throws IOException {
		String location = getCustomPropertiesLocation();
		Reader input = null;
		try {
			Resource resource = loader.getResource(location);
			input = new InputStreamReader(resource.getInputStream(), getEncoding());
			Properties p = new Properties();
			p.load(input);
			return p;
		} finally {
			IOUtils.closeQuietly(input);
		}
	}

	public String getEncoding() {
		return encoding;
	}

	public void setEncoding(String encoding) {
		this.encoding = encoding;
	}

	public Properties getProperties() {
		return properties;
	}

	public void setProperties(Properties properties) {
		this.properties = properties;
	}

	public String getUrl() {
		return url;
	}

	public void setUrl(String url) {
		this.url = url;
	}

	public DatabaseCommand getCommand() {
		return command;
	}

	public void setCommand(DatabaseCommand command) {
		this.command = command;
	}

	public PropertyPlaceholderHelper getHelper() {
		return helper;
	}

	public void setHelper(PropertyPlaceholderHelper helper) {
		this.helper = helper;
	}

	public JDBCUtils getJdbcUtils() {
		return jdbcUtils;
	}

	public void setJdbcUtils(JDBCUtils jdbcUtils) {
		this.jdbcUtils = jdbcUtils;
	}

	public DefaultResourceLoader getLoader() {
		return loader;
	}

	public void setLoader(DefaultResourceLoader loader) {
		this.loader = loader;
	}

	public String getPrefix() {
		return prefix;
	}

	public void setPrefix(String prefix) {
		this.prefix = prefix;
	}

}