/*
    * Licensed under the GPL License. You may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *   https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
    *
    * THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
    * WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
    * PURPOSE.
   */
  package psiprobe.controllers.sql;
  
  import java.util.List;
  import java.util.Map;
  
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpServletResponse;
  import javax.servlet.http.HttpSession;
  
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import org.springframework.beans.factory.annotation.Value;
  import org.springframework.stereotype.Controller;
  import org.springframework.web.bind.ServletRequestUtils;
  import org.springframework.web.bind.annotation.RequestMapping;
  import org.springframework.web.servlet.ModelAndView;
  
  import psiprobe.PostParameterizableViewController;
  import psiprobe.model.sql.DataSourceTestInfo;
  
  /**
   * Displays a result set cached in an attribute of HttpSession object to support result set
   * pagination feature without re-executing a query that created the result set.
   */
  @Controller
  public class CachedRecordSetController extends PostParameterizableViewController {
  
    /** The Constant logger. */
    private static final Logger logger = LoggerFactory.getLogger(CachedRecordSetController.class);
  
    @RequestMapping(path = "/sql/cachedRecordset.ajax")
    @Override
    public ModelAndView handleRequest(HttpServletRequest request, HttpServletResponse response)
        throws Exception {
      return super.handleRequest(request, response);
    }
  
    @Override
    protected ModelAndView handleRequestInternal(HttpServletRequest request,
        HttpServletResponse response) throws Exception {
  
      int rowsPerPage = ServletRequestUtils.getIntParameter(request, "rowsPerPage", 0);
      List<Map<String, String>> results = null;
      int rowsAffected = 0;
      HttpSession sess = request.getSession(false);
  
      if (sess == null) {
        request.setAttribute("errorMessage", getMessageSourceAccessor()
            .getMessage("probe.src.dataSourceTest.cachedResultSet.failure"));
        logger.error("Cannot retrieve a cached result set. Http session is NULL.");
      } else {
        DataSourceTestInfo sessData =
            (DataSourceTestInfo) sess.getAttribute(DataSourceTestInfo.DS_TEST_SESS_ATTR);
  
        if (sessData == null) {
          request.setAttribute("errorMessage", getMessageSourceAccessor()
              .getMessage("probe.src.dataSourceTest.cachedResultSet.failure"));
          logger.error("Cannot retrieve a cached result set. {} session attribute is NULL.",
              DataSourceTestInfo.DS_TEST_SESS_ATTR);
        } else {
          synchronized (sess) {
            sessData.setRowsPerPage(rowsPerPage);
          }
  
          results = sessData.getResults();
  
          if (results == null) {
            request.setAttribute("errorMessage", getMessageSourceAccessor()
                .getMessage("probe.src.dataSourceTest.cachedResultSet.failure"));
            logger.error("Cached results set is NULL.");
          } else {
            rowsAffected = results.size();
          }
        }
      }
  
      ModelAndView mv = new ModelAndView(getViewName(), "results", results);
      mv.addObject("rowsAffected", String.valueOf(rowsAffected));
      mv.addObject("rowsPerPage", String.valueOf(rowsPerPage));
  
      return mv;
    }
  
    @Value("ajax/sql/recordset")
    @Override
    public void setViewName(String viewName) {
      super.setViewName(viewName);
    }
  
 }