edu.ucla.cs.scai.canali.webapp.QueryServlet.java Source code

Introduction

Here is the source code for edu.ucla.cs.scai.canali.webapp.QueryServlet.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package edu.ucla.cs.scai.canali.webapp;

import com.google.gson.Gson;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import com.google.gson.reflect.TypeToken;
import edu.ucla.cs.scai.canali.core.autocompleter.AutocompleteObject;
import edu.ucla.cs.scai.canali.core.query.QueryService;
import edu.ucla.cs.scai.canali.core.query.ResultWrapper;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import edu.ucla.cs.scai.canali.webapp.utils.WebAppUtils;

/**
 *
 * @author Giuseppe M. Mazzeo <mazzeo@cs.ucla.edu>
 */
public class QueryServlet extends HttpServlet {

    /**
     * Processes requests for both HTTP <code>GET</code> and <code>POST</code>
     * methods.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     *
     */
    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {

        WebAppUtils.printParameters(this.getServletName(), request);

        response.setContentType("application/json;charset=UTF-8");

        JsonParser jp = new JsonParser();
        JsonArray ja = jp.parse(request.getParameter("s")).getAsJsonArray();
        ArrayList<AutocompleteObject> s = new ArrayList<>();
        for (Iterator<JsonElement> it = ja.iterator(); it.hasNext();) {
            JsonObject jo = it.next().getAsJsonObject();
            String state = jo.get("s").getAsString();
            String labels = jo.get("l").getAsString();
            String type = jo.get("k").getAsString();
            String restrictedText = jo.get("r").getAsString();
            String freeText = jo.has("f") ? jo.get("f").getAsString() : null;
            Integer relatedTokenPosition = null;
            try {
                relatedTokenPosition = jo.get("c").getAsInt();
            } catch (Exception e) {
            }
            s.add(new AutocompleteObject(null, restrictedText, state, labels, type, null, relatedTokenPosition,
                    false, freeText));
        }
        String endpoint = request.getParameter("e");
        String limitS = request.getParameter("l");
        int limit = 100;
        try {
            limit = Integer.parseInt(limitS);
        } catch (Exception e) {
        }
        boolean disableSubclasses = false;
        try {
            disableSubclasses = request.getParameter("us").equals("false");
        } catch (Exception e) {
        }
        ResultWrapper res = null;
        try {
            res = new QueryService().answerQuery(s, endpoint, limit, disableSubclasses);
        } catch (Exception ex) {
            ex.printStackTrace();
            res = new ResultWrapper(null, ex.getMessage());
        }

        Gson gson = new Gson();
        try (PrintWriter out = response.getWriter()) {
            out.println(gson.toJson(res));
            /*
             if ($_GET["callback"]) {
             $json_response = $_GET["callback"] . "(" . $json_response . 
             }")";
             */
        }
    }

    // <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
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    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
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        processRequest(request, response);
    }

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

}