litvinovg/vitro
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

NIHVIVO-1571 changes in SDBSetupController for rdb to sdb conversion.

Browse source
This commit is contained in:
ass92 2011-01-26 21:47:01 +00:00
parent be39ff7d5b
commit 65cf36c121
1 changed files with 57 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

74
webapp/src/edu/cornell/mannlib/vitro/webapp/controller/freemarker/SDBSetupController.java
View file

@ -1,12 +1,20 @@
/* $This file is distributed under the terms of the license in /doc/license.txt$ */ /* $This file is distributed under the terms of the license in /doc/license.txt$ */
package edu.cornell.mannlib.vitro.webapp.controller.freemarker; package edu.cornell.mannlib.vitro.webapp.controller.freemarker;
import java.sql.SQLException;
import java.util.HashMap; import java.util.HashMap;
import java.util.Map; import java.util.Map;
import org.apache.commons.dbcp.BasicDataSource;
import org.apache.commons.logging.Log; import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory; import org.apache.commons.logging.LogFactory;
import com.hp.hpl.jena.ontology.OntModel;
import com.hp.hpl.jena.ontology.OntModelSpec;
import com.hp.hpl.jena.rdf.model.ModelFactory;
import com.hp.hpl.jena.sdb.Store;
import com.hp.hpl.jena.sdb.StoreDesc;
import edu.cornell.mannlib.vedit.beans.LoginStatusBean; import edu.cornell.mannlib.vedit.beans.LoginStatusBean;
import edu.cornell.mannlib.vitro.webapp.controller.VitroRequest; import edu.cornell.mannlib.vitro.webapp.controller.VitroRequest;
import edu.cornell.mannlib.vitro.webapp.controller.freemarker.UrlBuilder.Route; import edu.cornell.mannlib.vitro.webapp.controller.freemarker.UrlBuilder.Route;
@ -15,12 +23,15 @@ import edu.cornell.mannlib.vitro.webapp.controller.freemarker.responsevalues.Red
import edu.cornell.mannlib.vitro.webapp.controller.freemarker.responsevalues.ResponseValues; import edu.cornell.mannlib.vitro.webapp.controller.freemarker.responsevalues.ResponseValues;
import edu.cornell.mannlib.vitro.webapp.controller.freemarker.responsevalues.TemplateResponseValues; import edu.cornell.mannlib.vitro.webapp.controller.freemarker.responsevalues.TemplateResponseValues;
import edu.cornell.mannlib.vitro.webapp.reasoner.SimpleReasoner; import edu.cornell.mannlib.vitro.webapp.reasoner.SimpleReasoner;
import edu.cornell.mannlib.vitro.webapp.servlet.setup.JenaDataSourceSetupSDB;
public class SDBSetupController extends FreemarkerHttpServlet { public class SDBSetupController extends FreemarkerHttpServlet {
private static final Log log = LogFactory.getLog( private static final Log log = LogFactory.getLog(
SimpleReasonerRecomputeController.class); SimpleReasonerRecomputeController.class);
private static final String SDB_SETUP_FTL = "sdbSetup.ftl";
protected ResponseValues processRequest(VitroRequest vreq) { protected ResponseValues processRequest(VitroRequest vreq) {
// Due to requiresLoginLevel(), we don't get here unless logged in as DBA // Due to requiresLoginLevel(), we don't get here unless logged in as DBA
if (!LoginStatusBean.getBean(vreq) if (!LoginStatusBean.getBean(vreq)
@ -31,41 +42,70 @@ public class SDBSetupController extends FreemarkerHttpServlet {
String messageStr = ""; String messageStr = "";
try { try {
JenaDataSourceSetupSDB jenaDataSourceSetupSDB = new JenaDataSourceSetupSDB();
if (false) { Boolean done = (Boolean)getServletContext().getAttribute("done");
String setupsignal = (String) vreq.getParameter("setupsignal");
} else { if (done!=null && done) {
if (false) { messageStr = "SDB is being setup";
} else{
} else { String sdbsetup = (String)getServletContext().getAttribute("sdbsetup");
new Thread(new SDBSetupRunner()).start(); if(sdbsetup == null || sdbsetup.equals("showButton") || setupsignal == null){
messageStr = "SDB setup started"; body.put("link", "show");
messageStr = null;
getServletContext().setAttribute("sdbsetup", "yes");
}
else if(setupsignal!=null && setupsignal.equals("setup")){
new Thread(new SDBSetupRunner(jenaDataSourceSetupSDB)).start();
messageStr = "SDB setup started";
getServletContext().setAttribute("sdbsetup", "showButton");
}
} }
}
} catch (Exception e) { } catch (Exception e) {
log.error("Error setting up SDB store", e); log.error("Error setting up SDB store", e);
body.put("errorMessage", body.put("errorMessage",
"Error setting up SDB store: " + "Error setting up SDB store: " +
e.getMessage()); e.getMessage());
return new ExceptionResponseValues( return new ExceptionResponseValues(
Template.ERROR_MESSAGE.toString(), body, e); SDB_SETUP_FTL, body, e);
} }
body.put("message", messageStr); body.put("message", messageStr);
return new TemplateResponseValues(Template.MESSAGE.toString(), body); return new TemplateResponseValues(SDB_SETUP_FTL, body);
} }
private class SDBSetupRunner implements Runnable { private class SDBSetupRunner implements Runnable {
private SimpleReasoner simpleReasoner; private JenaDataSourceSetupSDB jenaDataSourceSetupSDB;
final OntModelSpec MEM_ONT_MODEL_SPEC = OntModelSpec.OWL_MEM;
public SDBSetupRunner() { public SDBSetupRunner(JenaDataSourceSetupSDB jenaDataSourceSetupSDB) {
this.jenaDataSourceSetupSDB = jenaDataSourceSetupSDB;
} }
public void run() { public void run() {
Boolean done = true;
getServletContext().setAttribute("done",done);
StoreDesc storeDesc = jenaDataSourceSetupSDB.makeStoreDesc();
BasicDataSource bds = jenaDataSourceSetupSDB.makeDataSourceFromConfigurationProperties();
Store store = null;
try {
store = JenaDataSourceSetupSDB.connectStore(bds, storeDesc);
} catch (SQLException e) {
log.error("Error while getting the sdb store with given store description and basic data source", e);
}
OntModel memModel = (OntModel)getServletContext().getAttribute("jenaOntModel");
if (memModel == null) {
memModel = ModelFactory.createOntologyModel(MEM_ONT_MODEL_SPEC);
log.warn("WARNING: no database connected. Changes will disappear after context restart.");
}
OntModel inferenceModel = (OntModel)getServletContext().getAttribute("inferenceOntModel");
if(inferenceModel == null){
inferenceModel = ModelFactory.createOntologyModel(MEM_ONT_MODEL_SPEC);
}
if(store!=null)
jenaDataSourceSetupSDB.setupSDB(getServletContext(), store, memModel, inferenceModel);
done = false;
getServletContext().setAttribute("done",done);
} }
} }