NIHVIVO-1307 new controller that supports uploading file on "Convert csv to rdf" tool.
This commit is contained in:
ass92
parent
17557c3ecb
commit
caa8bddf2b
1 changed files with 195 additions and 0 deletions
|
|
@ -0,0 +1,195 @@
|
||||||
|
package edu.cornell.mannlib.vitro.webapp.controller.jena;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.net.URL;
|
||||||
|
import java.util.Collection;
|
||||||
|
import java.util.Iterator;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
import javax.servlet.RequestDispatcher;
|
||||||
|
import javax.servlet.ServletException;
|
||||||
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
import javax.servlet.http.HttpServletResponse;
|
||||||
|
|
||||||
|
import org.apache.commons.fileupload.FileItem;
|
||||||
|
|
||||||
|
import com.hp.hpl.jena.ontology.OntModel;
|
||||||
|
import com.hp.hpl.jena.ontology.OntModelSpec;
|
||||||
|
import com.hp.hpl.jena.rdf.model.Model;
|
||||||
|
import com.hp.hpl.jena.rdf.model.ModelFactory;
|
||||||
|
import com.hp.hpl.jena.rdf.model.ModelMaker;
|
||||||
|
import com.hp.hpl.jena.shared.JenaException;
|
||||||
|
|
||||||
|
import edu.cornell.mannlib.vedit.beans.LoginStatusBean;
|
||||||
|
import edu.cornell.mannlib.vedit.controller.BaseEditController;
|
||||||
|
import edu.cornell.mannlib.vitro.webapp.beans.Portal;
|
||||||
|
import edu.cornell.mannlib.vitro.webapp.controller.Controllers;
|
||||||
|
import edu.cornell.mannlib.vitro.webapp.controller.VitroRequest;
|
||||||
|
import edu.cornell.mannlib.vitro.webapp.dao.jena.VitroJenaSpecialModelMaker;
|
||||||
|
import edu.cornell.mannlib.vitro.webapp.filestorage.uploadrequest.FileUploadServletRequest;
|
||||||
|
import edu.cornell.mannlib.vitro.webapp.utils.Csv2Rdf;
|
||||||
|
|
||||||
|
|
||||||
|
public class JenaCsv2RdfController extends BaseEditController{
|
||||||
|
private static final String CSV2RDF_JSP = "/jenaIngest/csv2rdf.jsp";
|
||||||
|
private static final String INGEST_MENU_JSP = "/jenaIngest/ingestMenu.jsp";
|
||||||
|
private static int maxFileSizeInBytes = 1024 * 1024 * 2000; //2000mb
|
||||||
|
|
||||||
|
public void doPost(HttpServletRequest rawRequest,
|
||||||
|
HttpServletResponse response) throws ServletException, IOException {
|
||||||
|
FileUploadServletRequest req = FileUploadServletRequest.parseRequest(rawRequest,
|
||||||
|
maxFileSizeInBytes);
|
||||||
|
if (req.hasFileUploadException()) {
|
||||||
|
forwardToFileUploadError(req.getFileUploadException().getLocalizedMessage(), req, response);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
VitroRequest request = new VitroRequest(req);
|
||||||
|
if (!checkLoginStatus(request,response) ){
|
||||||
|
try {
|
||||||
|
response.sendRedirect(getDefaultLandingPage(request));
|
||||||
|
} catch (IOException e) {
|
||||||
|
// TODO Auto-generated catch block
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
Map<String, List<FileItem>> fileStreams = req.getFiles();
|
||||||
|
FileItem fileStream = fileStreams.get("filePath").get(0);
|
||||||
|
String filePath = fileStreams.get("filePath").get(0).getName();
|
||||||
|
|
||||||
|
String actionStr = request.getParameter("action");
|
||||||
|
actionStr = (actionStr != null) ? actionStr : "";
|
||||||
|
|
||||||
|
if ("csv2rdf".equals(actionStr)) {
|
||||||
|
String csvUrl = request.getParameter("csvUrl");
|
||||||
|
if (!csvUrl.isEmpty() || !filePath.isEmpty()) {
|
||||||
|
doExecuteCsv2Rdf(request,fileStream,filePath);
|
||||||
|
request.setAttribute("title","IngestMenu");
|
||||||
|
request.setAttribute("bodyJsp", INGEST_MENU_JSP);
|
||||||
|
} else {
|
||||||
|
request.setAttribute("title","Convert CSV to RDF");
|
||||||
|
request.setAttribute("bodyJsp",CSV2RDF_JSP);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Portal portal = request.getPortal();
|
||||||
|
RequestDispatcher rd = request.getRequestDispatcher(Controllers.BASIC_JSP);
|
||||||
|
request.setAttribute("portalBean",portal);
|
||||||
|
request.setAttribute("css", "<link rel=\"stylesheet\" type=\"text/css\" href=\""+portal.getThemeDir()+"css/edit.css\"/>");
|
||||||
|
|
||||||
|
try {
|
||||||
|
rd.forward(request, response);
|
||||||
|
} catch (Exception e) {
|
||||||
|
System.out.println(this.getClass().getName()+" could not forward to view.");
|
||||||
|
System.out.println(e.getMessage());
|
||||||
|
System.out.println(e.getStackTrace());
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private void forwardToFileUploadError( String errrorMsg , HttpServletRequest req, HttpServletResponse response) throws ServletException{
|
||||||
|
req.setAttribute("errors", errrorMsg);
|
||||||
|
RequestDispatcher rd = req.getRequestDispatcher("/edit/fileUploadError.jsp");
|
||||||
|
try {
|
||||||
|
rd.forward(req, response);
|
||||||
|
} catch (IOException e1) {
|
||||||
|
throw new ServletException(e1);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void doExecuteCsv2Rdf(VitroRequest vreq,FileItem fileStream, String filePath) {
|
||||||
|
char[] quoteChars = {'"'};
|
||||||
|
String namespace = vreq.getParameter("namespace");
|
||||||
|
String tboxNamespace = vreq.getParameter("tboxNamespace");
|
||||||
|
String typeName = vreq.getParameter("typeName");
|
||||||
|
String csvUrl = vreq.getParameter("csvUrl");
|
||||||
|
Model destination = null;
|
||||||
|
String destinationModelNameStr = vreq.getParameter("destinationModelName");
|
||||||
|
if (destinationModelNameStr != null && destinationModelNameStr.length()>0) {
|
||||||
|
destination = getModel(destinationModelNameStr, vreq);
|
||||||
|
}
|
||||||
|
Model tboxDestination = null;
|
||||||
|
String tboxDestinationModelNameStr = vreq.getParameter("tboxDestinationModelName");
|
||||||
|
if (tboxDestinationModelNameStr != null && tboxDestinationModelNameStr.length()>0) {
|
||||||
|
tboxDestination = getModel(tboxDestinationModelNameStr, vreq);
|
||||||
|
}
|
||||||
|
|
||||||
|
char separatorChar = ',';
|
||||||
|
if ("tab".equalsIgnoreCase(vreq.getParameter("separatorChar"))) {
|
||||||
|
separatorChar = '\t';
|
||||||
|
}
|
||||||
|
|
||||||
|
Csv2Rdf c2r = new Csv2Rdf(separatorChar, quoteChars,namespace,tboxNamespace,typeName);
|
||||||
|
|
||||||
|
InputStream is = null;
|
||||||
|
|
||||||
|
try {
|
||||||
|
if(!csvUrl.isEmpty())
|
||||||
|
is = new URL(csvUrl).openStream();
|
||||||
|
else if(!filePath.isEmpty())
|
||||||
|
is = fileStream.getInputStream();
|
||||||
|
|
||||||
|
} catch (IOException e) {
|
||||||
|
System.out.println("IOException opening URL "+csvUrl);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
Model[] models = null;
|
||||||
|
|
||||||
|
try {
|
||||||
|
models = c2r.convertToRdf(is);
|
||||||
|
} catch (IOException e) {
|
||||||
|
System.out.println("IOException converting "+csvUrl+" to RDF");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (destination != null) {
|
||||||
|
destination.add(models[0]);
|
||||||
|
}
|
||||||
|
if (tboxDestination != null) {
|
||||||
|
tboxDestination.add(models[1]);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private Model getModel(String name, HttpServletRequest request) {
|
||||||
|
if ("vitro:jenaOntModel".equals(name)) {
|
||||||
|
Object sessionOntModel = request.getSession().getAttribute("jenaOntModel");
|
||||||
|
if (sessionOntModel != null && sessionOntModel instanceof OntModel) {
|
||||||
|
return (OntModel) sessionOntModel;
|
||||||
|
} else {
|
||||||
|
return (OntModel) getServletContext().getAttribute("jenaOntModel");
|
||||||
|
}
|
||||||
|
} else if ("vitro:baseOntModel".equals(name)) {
|
||||||
|
Object sessionOntModel = request.getSession().getAttribute("baseOntModel");
|
||||||
|
if (sessionOntModel != null && sessionOntModel instanceof OntModel) {
|
||||||
|
return (OntModel) sessionOntModel;
|
||||||
|
} else {
|
||||||
|
return (OntModel) getServletContext().getAttribute("baseOntModel");
|
||||||
|
}
|
||||||
|
} else if ("vitro:inferenceOntModel".equals(name)) {
|
||||||
|
Object sessionOntModel = request.getSession().getAttribute("inferenceOntModel");
|
||||||
|
if (sessionOntModel != null && sessionOntModel instanceof OntModel) {
|
||||||
|
return (OntModel) sessionOntModel;
|
||||||
|
} else {
|
||||||
|
return (OntModel) getServletContext().getAttribute("inferenceOntModel");
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
return getVitroJenaModelMaker(request).getModel(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private ModelMaker getVitroJenaModelMaker(HttpServletRequest request) {
|
||||||
|
ModelMaker myVjmm = (ModelMaker) request.getSession().getAttribute("vitroJenaModelMaker");
|
||||||
|
myVjmm = (myVjmm == null) ? (ModelMaker) getServletContext().getAttribute("vitroJenaModelMaker") : myVjmm;
|
||||||
|
return new VitroJenaSpecialModelMaker(myVjmm, request);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue