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