/*
Copyright (C) 2005-2012, by the President and Fellows of Harvard College.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
Dataverse Network - A web application to share, preserve and analyze research data.
Developed at the Institute for Quantitative Social Science, Harvard University.
Version 3.0.
*/
/*
* VDCIngestBean.java
*
* Created on October 2, 2007, 3:15 PM
*/
package edu.harvard.iq.dvn.core.web.servlet;
import edu.harvard.iq.dvn.core.admin.NetworkRoleServiceLocal;
import edu.harvard.iq.dvn.core.admin.VDCUser;
import edu.harvard.iq.dvn.ingest.dsb.DSBIngestMessage;
import edu.harvard.iq.dvn.core.study.Study;
import edu.harvard.iq.dvn.core.study.StudyFile;
import edu.harvard.iq.dvn.core.study.StudyFileEditBean;
import edu.harvard.iq.dvn.core.study.StudyFileServiceLocal;
import edu.harvard.iq.dvn.core.study.StudyServiceLocal;
import edu.harvard.iq.dvn.core.study.TabularDataFile;
import edu.harvard.iq.dvn.core.web.common.VDCSessionBean;
import java.io.*;
import java.util.ArrayList;
import java.util.List;
import javax.annotation.Resource;
import javax.ejb.EJB;
import javax.inject.Inject;
import javax.jms.Queue;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.QueueConnection;
import javax.jms.QueueConnectionFactory;
import javax.jms.QueueSender;
import javax.jms.QueueSession;
import javax.servlet.*;
import javax.servlet.http.*;
/**
*
* @author Gustavo
* @version
*/
public class VDCIngestServlet extends HttpServlet {
@EJB StudyServiceLocal studyService;
@EJB StudyFileServiceLocal studyFileService;
@Resource(mappedName="jms/DSBIngest") Queue queue;
@Resource(mappedName="jms/DSBQueueConnectionFactory") QueueConnectionFactory factory;
@Inject VDCSessionBean vdcSession;
private boolean isNetworkAdmin(HttpServletRequest req) {
VDCUser user = null;
if ( vdcSession.getLoginBean() != null ) {
user= vdcSession.getLoginBean().getUser();
if (user.getNetworkRole()!=null && user.getNetworkRole().getName().equals(NetworkRoleServiceLocal.ADMIN) ) {
return true;
}
}
return false;
}
/** Handles the HTTP <code>GET</code> method.
* @param request servlet request
* @param response servlet response
*/
protected void doGet(HttpServletRequest req, HttpServletResponse res)
throws ServletException, IOException {
res.setContentType("text/html;charset=UTF-8");
PrintWriter out = res.getWriter();
if (isNetworkAdmin(req)) {
Long studyId = null;
try {
studyId = new Long( req.getParameter("studyId") );
Study study = studyService.getStudy(studyId);
String selectString = "";
for (StudyFile file : study.getStudyFiles()) {
selectString += "<option value=" + file.getId()+ ">" + file.getFileName();
}
beginPage(out);
out.println("<h3>Ingest</h3>");
out.println("<form method=POST>");
out.println("To make a file subsettable, please select the data file and the control card file below:<br/>");
out.print("<input type=hidden name=\"studyId\" value=" + studyId + ">");
out.println("<br/><br/><table>");
out.print("<tr><td>Data file</td>");
out.println("<td><select name=\"dataFileId\">");
out.println(selectString);
out.print("</select></td></tr>");
out.print("<td>Control Card file: </td>");
out.println("<td><select name=\"controlCardFileId\">");
out.println(selectString);
out.print("</select></td></tr>");
out.print("<td>Email: </td>");
out.println("<td><input name=\"emailAddress\" value=\"" + vdcSession.getLoginBean().getUser().getEmail() + "\" size=30 >");
out.print("</td></tr>");
out.println("</table><br/><br/>");
out.print("<input name=ingest value=\"Make subsettable\" type=submit />");
out.print("<hr>");
out.println("</form>");
endPage(out);
} catch (NumberFormatException nfe) {
displayMessage(out, "Action failed.", "The study id must be of type Long.");
} catch (Exception e) {
e.printStackTrace();
displayMessage(out, "Action failed.", "An unknown error occurred trying to get study; id = " + studyId);
}
} else {
displayMessage(out, "You are not authorized for this action.");
}
}
/** Handles the HTTP <code>POST</code> method.
* @param request servlet request
* @param response servlet response
*/
protected void doPost(HttpServletRequest req, HttpServletResponse res)
throws ServletException, IOException {
res.setContentType("text/html;charset=UTF-8");
PrintWriter out = res.getWriter();
if (isNetworkAdmin(req) ) {
Long studyId = new Long( req.getParameter("studyId") );
Long dataFileId = new Long( req.getParameter("dataFileId") );
Long controlCardFileId = new Long( req.getParameter("controlCardFileId") );
StudyFileEditBean fileBean = new StudyFileEditBean( studyFileService.getStudyFile(dataFileId) );
if (fileBean.getStudyFile() instanceof TabularDataFile) {
((TabularDataFile) fileBean.getStudyFile()).getDataTable(); // this is to instantiate lazy relationship, before serializing
}
fileBean.setTempSystemFileLocation( fileBean.getStudyFile().getFileSystemLocation() );
fileBean.setControlCardSystemFileLocation( studyFileService.getStudyFile(controlCardFileId).getFileSystemLocation() );
List tempList = new ArrayList();
tempList.add(fileBean);
QueueConnection conn = null;
QueueSession session = null;
QueueSender sender = null;
try {
conn = factory.createQueueConnection();
session = conn.createQueueSession(false,0);
sender = session.createSender(queue);
DSBIngestMessage ingestMessage = new DSBIngestMessage();
ingestMessage.setFileBeans(tempList);
ingestMessage.setIngestEmail(req.getParameter("emailAddress"));
ingestMessage.setIngestUserId(vdcSession.getLoginBean().getUser().getId());
ingestMessage.setStudyId(studyId);
Message message = session.createObjectMessage(ingestMessage);
sender.send(message);
} catch (JMSException ex) {
ex.printStackTrace();
} finally {
try {
if (sender != null) {sender.close();}
if (session != null) {session.close();}
if (conn != null) {conn.close();}
} catch (JMSException ex) {
ex.printStackTrace();
}
}
displayMessage(out,"Request in process. You will receive an e-mail when it is complete.",
"(for study id = " + studyId +
"; data file id = " + dataFileId +
"; control card file id = " +controlCardFileId + ")");
} else {
displayMessage(out, "You are not authorized for this action.");
}
}
private void beginPage(PrintWriter out) {
out.println("<html>");
out.println("<head>");
out.println("<title>Ingest Servlet</title>");
out.println("</head>");
out.println("<body><center>");
}
private void endPage(PrintWriter out) {
out.println("</center></body>");
out.println("</html>");
out.close();
}
private void displayMessage(PrintWriter out, String title) {
displayMessage(out, title, null);
}
private void displayMessage(PrintWriter out, String title, String message) {
beginPage(out);
out.println("<h3>" + title + "</h3>");
if (message != null && !message.trim().equals("") ) {
out.println(message);
}
endPage(out);
}
/**
* Processes requests for both HTTP <code>GET</code> and <code>POST</code> methods.
* @param request servlet request
* @param response servlet response
*/
protected void processRequest(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
response.setContentType("text/html;charset=UTF-8");
PrintWriter out = response.getWriter();
/* TODO output your page here
out.println("<html>");
out.println("<head>");
out.println("<title>Servlet VDCIngestBean</title>");
out.println("</head>");
out.println("<body>");
out.println("<h1>Servlet VDCIngestBean at " + request.getContextPath () + "</h1>");
out.println("</body>");
out.println("</html>");
*/
out.close();
}
/**
* Returns a short description of the servlet.
*/
public String getServletInfo() {
return "Short description";
}
// </editor-fold>
}