/**
* The contents of this file are subject to the license and copyright
* detailed in the LICENSE and NOTICE files at the root of the source
* tree and available online at
*
* http://www.dspace.org/license/
*/
package org.dspace.storage.rdbms;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import org.apache.log4j.Logger;
import org.dspace.browse.BrowseException;
import org.dspace.browse.IndexBrowse;
import org.dspace.core.ConfigurationManager;
/**
* Command-line executed class for initializing the DSpace database. This should
* be invoked with a single argument, the filename of the database schema file.
*
* @author Robert Tansley
* @version $Revision$
*/
public class InitializeDatabase
{
/** log4j category */
private static Logger log = Logger.getLogger(InitializeDatabase.class);
public static void main(String[] argv)
{
// Usage checks
if (argv.length != 1)
{
log.warn("Schema file not specified");
System.exit(1);
}
ConfigurationManager.loadConfig(null);
log.info("Initializing Database");
try
{
if("clean-database.sql".equals(argv[0]))
{
try
{
IndexBrowse browse = new IndexBrowse();
browse.setDelete(true);
browse.setExecute(true);
browse.clearDatabase();
}
catch (BrowseException e)
{
log.error(e.getMessage(),e);
throw new IllegalStateException(e.getMessage(),e);
}
DatabaseManager.loadSql(getScript(argv[0]));
}
else
{
DatabaseManager.loadSql(getScript(argv[0]));
try
{
IndexBrowse browse = new IndexBrowse();
browse.setRebuild(true);
browse.setExecute(true);
browse.initBrowse();
}
catch (BrowseException e)
{
log.error(e.getMessage(),e);
throw new IllegalStateException(e.getMessage(),e);
}
}
System.exit(0);
}
catch (Exception e)
{
log.fatal("Caught exception:", e);
System.exit(1);
}
}
/**
* Attempt to get the named script, with the following rules:
* etc/<db.name>/<name>
* etc/<name>
* <name>
*/
private static FileReader getScript(String name) throws FileNotFoundException, IOException
{
String dbName = ConfigurationManager.getProperty("db.name");
File myFile = null;
if (dbName != null)
{
myFile = new File("etc/" + dbName + "/" + name);
if (myFile.exists())
{
return new FileReader(myFile.getCanonicalPath());
}
}
myFile = new File("etc/" + name);
if (myFile.exists())
{
return new FileReader(myFile.getCanonicalPath());
}
return new FileReader(name);
}
}