package org.kuali.ole.web;
   
   import org.kuali.ole.docstore.metrics.reindex.ReIndexingStatus;
   import org.kuali.ole.docstore.process.RebuildIndexesHandler;
   import org.kuali.ole.logger.DocStoreLogger;
   
   import javax.servlet.ServletException;
   import javax.servlet.http.HttpServlet;
   import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpServletResponse;
  import java.io.IOException;
  import java.io.PrintWriter;
  
  /**
   * Created by IntelliJ IDEA.
   * User: pvsubrah
   * Date: 11/12/11
   * Time: 3:41 PM
   * To change this template use File | Settings | File Templates.
   */
  public class RebuildIndexServlet extends HttpServlet {
  
      private static final long serialVersionUID = 1L;
      DocStoreLogger            docStoreLogger   = new DocStoreLogger(getClass().getName());
  
      @Override
      protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
          doPost(req, resp);
      }
  
      @Override
      protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
          try {
  
              String docCategory = req.getParameter("docCategory");
              String docType = req.getParameter("docType");
              String docFormat = req.getParameter("docFormat");
              String action = req.getParameter("action");
              RebuildIndexesHandler rebuildIndexesHandler = RebuildIndexesHandler.getInstance();
  
              if (action.equalsIgnoreCase("start")) {
                  String result = rebuildIndexesHandler.startProcess(docCategory, docType, docFormat);
                  outputMessage(resp, result);
              }
              else if (action.equalsIgnoreCase("stop")) {
                  String result = rebuildIndexesHandler.stopProcess();
                  outputMessage(resp, result);
              }
              else if (action.equalsIgnoreCase("status")) {
                  ReIndexingStatus reIndexingStatus = ReIndexingStatus.getInstance();
                  String result = reIndexingStatus.getJsonString();
                  outputMessage(resp, result);
              }
              else if (action.equalsIgnoreCase("Clear Status")) {
                  ReIndexingStatus reIndexingStatus = ReIndexingStatus.getInstance();
                  reIndexingStatus.reset();
              }
              else {
                  String result = "Invalid action :" + action ;
                  outputMessage(resp, result);
              }
          } catch (Exception e) {
              docStoreLogger.log("Error during rebuilding of the indexes from documentstore", e);
          }
      }
  
      private void outputMessage(HttpServletResponse resp, String s) throws IOException {
          PrintWriter out = resp.getWriter();
          out.println(s);
          out.flush();
          out.close();
      }
  }