appserver/tests/v2-tests/appserv-tests/devtests/callflow/dbreader/servlet/dbReader.java

/*
 * Copyright (c) 2018 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

/*
 * dbReader.java
 *
 * Created on May 10, 2006, 10:31 AM
 */

package com.sun.enterprise.admin.monitor.callflow;

import java.io.*;
import java.net.*;
import java.sql.*;
import jakarta.annotation.Resource;

import jakarta.servlet.*;
import jakarta.servlet.http.*;
import javax.sql.DataSource;
import com.sun.enterprise.admin.monitor.callflow.Agent;
import com.sun.enterprise.Switch;

/**
 *
 * DB Reader Utility Servlet. Returns the Rows in each callflow table.
 * @author Harpreet Singh
 * @version
 */
public class dbReader extends HttpServlet {

    /** Processes requests for both HTTP <code>GET</code> and <code>POST</code> methods.
     * @param request servlet request
     * @param response servlet response
     */
    @Resource (name="jdbc/__CallFlowPool", mappedName="jdbc/__CallFlowPool")
    private DataSource callflowDS;
    private static final String __COUNT = " select count(*) from ";
    private static final String __SELECT = " select count(*) from  ";
    private static final String __WHERE = " where ";
    private static final String __DELETE =" delete from ";
    private static final String REQUEST_START_TBL__SERVER = "REQUEST_START_TBL__SERVER";
    private static final String RS = REQUEST_START_TBL__SERVER;

    private static final String REQUEST_END_TBL__SERVER = "REQUEST_END_TBL__SERVER";
    private static final String RE = REQUEST_END_TBL__SERVER;

    private static final String METHOD_START_TBL__SERVER = "METHOD_START_TBL__SERVER";
    private static final String MS = METHOD_START_TBL__SERVER;

    private static final String METHOD_END_TBL__SERVER ="METHOD_END_TBL__SERVER";
    private static final String ME = METHOD_END_TBL__SERVER;

    private static final String START_TIME_TBL__SERVER ="START_TIME_TBL__SERVER";
    private static final String ST = START_TIME_TBL__SERVER;

    private static final String END_TIME_TBL__SERVER   ="END_TIME_TBL__SERVER";
    private static final String ET = END_TIME_TBL__SERVER;

    private static final String AND = " AND ";
    private static final String RID = ".REQUEST_ID ";
    private static final String COMMA = ",";
    private static final String EQUAL = " = ";

   // name of the servlet whose db value is to be queried
    private static final String SERVLET_NAME = "";

    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
        response.setContentType("text/html;charset=UTF-8");
        PrintWriter out = response.getWriter();
        Connection con = null;
        con = getConnection();
        String cleandb = request.getParameter ("cleandb");
        if (cleandb != null){
            deleteRows (con);
        out.println ("Cleaned rows from all tables in db");
            return;
        }
        String servletName = request.getParameter ("servletName");
        if (servletName == null){
            out.println (" Specify servlet name whose data is to be looked up"+
                    " via a request parameter name ="+servletName);
        }
//        out.println (" Looking up information for "+ servletName);
//        out.println();
        String sql = __COUNT + MS +  COMMA + ME + COMMA + RS + COMMA + RE +
             __WHERE + MS +".MODULE_NAME = " + "'" + "/"+ servletName + "'" + AND +
                MS + RID + EQUAL + ME + RID + AND +
                MS + RID + EQUAL + RS + RID + AND +
                MS + RID + EQUAL + RE + RID;
//        out.println (" \nSQL : "+sql);
        int val = getCountCallFlowRows (con, sql);
        String sql2 = __COUNT + MS +  COMMA + ET +
             __WHERE + MS +".MODULE_NAME = " + "'" + "/" +servletName + "'" + AND +
                MS + RID + EQUAL + ET + RID;
        int et = getCountCallFlowRows (con, sql2);
        String sql3 = __COUNT + MS +  COMMA + ST +
             __WHERE + MS +".MODULE_NAME = " + "'" + "/"+ servletName + "'" + AND +
                MS + RID + EQUAL + ST + RID;

        int st = getCountCallFlowRows (con, sql3);
        out.println ("RS=RE=MS=ME="+ val +" ST="+st+" ET="+et);
//        out.println (" Cleaning up Database");
////        deleteRows (con);
        out.close();
        try {
            con.close();
        } catch (SQLException ex) {
            ex.printStackTrace();
        }
    }

    private Connection getConnection () throws ServletException{

        Connection con = null;
        try {
            con = callflowDS.getConnection();
        } catch (SQLException ex) {
           throw new ServletException (ex);
        }
        return con;
    }
    private String getRequestId (Connection con , String sql) throws ServletException {
        String result = null;
        try {
            Statement stmt = con.createStatement();
            stmt.executeQuery (sql);
            ResultSet set = stmt.getResultSet();
            set.next();
            result = set.getString(1);
        } catch (SQLException ex) {
            throw new ServletException (ex);
        }
        return result;
    }
    private int getCountCallFlowRows (Connection con, String sql) throws ServletException{
        int count = -1;
        try {
            Statement stmt = con.createStatement();
            stmt.executeQuery (sql);
            ResultSet set = stmt.getResultSet();
            set.next();
            count = set.getInt(1);
        } catch (SQLException ex) {
            throw new ServletException (ex);
        }
        return count;

    }
  private void deleteRows (Connection con) throws ServletException{
        try {
            Statement stmt = con.createStatement();
            stmt.execute (__DELETE + REQUEST_START_TBL__SERVER);
            stmt.execute (__DELETE + REQUEST_END_TBL__SERVER);

            stmt.execute (__DELETE + METHOD_START_TBL__SERVER);
            stmt.execute (__DELETE + METHOD_END_TBL__SERVER);

            stmt.execute (__DELETE + START_TIME_TBL__SERVER);
            stmt.execute (__DELETE + END_TIME_TBL__SERVER);

        } catch (SQLException ex) {
            throw new ServletException (ex);
        }
    }
    // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
    /** Handles the HTTP <code>GET</code> method.
     * @param request servlet request
     * @param response servlet response
     */
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
        processRequest(request, response);
    }

    /** Handles the HTTP <code>POST</code> method.
     * @param request servlet request
     * @param response servlet response
     */
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
        processRequest(request, response);
    }

    /** Returns a short description of the servlet.
     */
    public String getServletInfo() {
        return "Short description";
    }
    // </editor-fold>
}