122 lines
3.9 KiB
Java
Executable File
122 lines
3.9 KiB
Java
Executable File
/*
|
|
* Copyright (C) 2001-2004 Red Hat Inc. All Rights Reserved.
|
|
*
|
|
* This library 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 2.1 of
|
|
* the License, or (at your option) any later version.
|
|
*
|
|
* This library 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 this library; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*
|
|
*/
|
|
package com.arsdigita.installer;
|
|
|
|
import com.arsdigita.db.DbHelper;
|
|
import com.arsdigita.util.UncheckedWrapperException;
|
|
import java.io.FileNotFoundException;
|
|
import java.io.FileReader;
|
|
import java.io.Reader;
|
|
import java.sql.Connection;
|
|
import java.sql.DriverManager;
|
|
import java.sql.SQLException;
|
|
import org.apache.log4j.BasicConfigurator;
|
|
import org.apache.log4j.Logger;
|
|
|
|
/**
|
|
*
|
|
* @version $Id: LoadSQLPlusScript.java 287 2005-02-22 00:29:02Z sskracic $
|
|
*/
|
|
public class LoadSQLPlusScript {
|
|
|
|
|
|
private static final Logger s_log =
|
|
Logger.getLogger(LoadSQLPlusScript.class);
|
|
|
|
private Connection m_con;
|
|
|
|
public static void main (String args[]) {
|
|
BasicConfigurator.configure();
|
|
|
|
if (args.length != 4) {
|
|
s_log.error("Usage: LoadSQLPlusScript " +
|
|
"<JDBC_URL> <username> <password> <script_filename>");
|
|
System.exit(1);
|
|
}
|
|
|
|
String jdbcUrl = args[0];
|
|
String dbUsername = args[1];
|
|
String dbPassword = args[2];
|
|
String scriptFilename = args[3];
|
|
|
|
LoadSQLPlusScript loader = new LoadSQLPlusScript();
|
|
loader.setConnection (jdbcUrl, dbUsername, dbPassword);
|
|
loader.loadSQLPlusScript(scriptFilename);
|
|
}
|
|
|
|
public void setConnection (Connection connection) {
|
|
m_con = connection;
|
|
}
|
|
|
|
public void setConnection (String jdbcUrl, String dbUsername,
|
|
String dbPassword) {
|
|
try {
|
|
int db = DbHelper.getDatabaseFromURL(jdbcUrl);
|
|
|
|
switch (db) {
|
|
case DbHelper.DB_POSTGRES:
|
|
Class.forName("org.postgresql.Driver");
|
|
break;
|
|
case DbHelper.DB_ORACLE:
|
|
Class.forName("oracle.jdbc.driver.OracleDriver");
|
|
break;
|
|
default:
|
|
throw new IllegalArgumentException("unsupported database");
|
|
}
|
|
|
|
s_log.warn("Using database " + DbHelper.getDatabaseName(db));
|
|
m_con = DriverManager.getConnection(jdbcUrl, dbUsername,
|
|
dbPassword);
|
|
} catch (SQLException e) {
|
|
throw new UncheckedWrapperException(e);
|
|
} catch (ClassNotFoundException e) {
|
|
throw new UncheckedWrapperException(e);
|
|
}
|
|
}
|
|
|
|
public void loadSQLPlusScript (String scriptFilename) {
|
|
if (scriptFilename != null) {
|
|
loadScript(scriptFilename.replace('\\', '/'));
|
|
}
|
|
}
|
|
|
|
protected void loadScript(String scriptFilename) {
|
|
if (s_log.isInfoEnabled()) {
|
|
s_log.info("Loading: '" + scriptFilename + "'");
|
|
}
|
|
SQLLoader loader = new SQLLoader(m_con) {
|
|
protected Reader open(String name) {
|
|
try {
|
|
return new FileReader(name);
|
|
} catch (FileNotFoundException e) {
|
|
s_log.warn("File not found: " + name);
|
|
return null;
|
|
}
|
|
}
|
|
};
|
|
loader.load(scriptFilename);
|
|
try {
|
|
m_con.commit();
|
|
} catch (SQLException e) {
|
|
throw new UncheckedWrapperException(e);
|
|
}
|
|
}
|
|
|
|
}
|