Java tutorial
/* * This file is part of Mockey, a tool for testing application * interactions over HTTP, with a focus on testing web services, * specifically web applications that consume XML, JSON, and HTML. * * Copyright (C) 2009-2010 Authors: * * chad.lafontaine (chad.lafontaine AT gmail DOT com) * neil.cronin (neil AT rackle DOT com) * lorin.kobashigawa (lkb AT kgawa DOT com) * rob.meyer (rob AT bigdis DOT com) * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License * as published by the Free Software Foundation; either version 2 * of the License, or (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. * */ package com.mockey.ui; import java.io.IOException; import java.io.PrintWriter; import java.util.ArrayList; import java.util.List; import javax.servlet.RequestDispatcher; import javax.servlet.ServletException; import javax.servlet.http.HttpServlet; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.json.JSONException; import org.json.JSONObject; import com.mockey.model.TwistInfo; import com.mockey.storage.IMockeyStorage; import com.mockey.storage.StorageRegistry; public class TwistInfoSetupServlet extends HttpServlet implements TwistInfoConfigurationAPI { /** * */ private static final long serialVersionUID = 2685147086312954142L; private static IMockeyStorage store = StorageRegistry.MockeyStorage; /** * Handles the following activities for <code>TwistInfo</code> * */ public void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException { List<TwistInfo> twistInfoList = store.getTwistInfoList(); String responseType = req.getParameter("response-type"); // If type is JSON, then respond with JSON // Otherwise, direct to JSP if (PARAMETER_KEY_RESPONSE_TYPE_VALUE_JSON.equalsIgnoreCase(responseType)) { // *********************** // BEGIN - JSON response // *********************** resp.setContentType("application/json"); PrintWriter out = resp.getWriter(); try { JSONObject jsonResponseObject = new JSONObject(); JSONObject jsonObject = null; // new JSONObject(); for (TwistInfo twistInfo : twistInfoList) { jsonObject = new JSONObject(); jsonObject.put("id", "" + twistInfo.getId()); jsonObject.put("name", twistInfo.getName()); for (PatternPair patternPair : twistInfo.getPatternPairList()) { JSONObject ppObj = new JSONObject(); ppObj.put("origination", patternPair.getOrigination()); ppObj.put("destination", patternPair.getDestination()); jsonObject.append("pattern-pair-list", ppObj); } } jsonResponseObject.put("result", jsonObject); out.println(jsonResponseObject.toString()); } catch (JSONException jsonException) { throw new ServletException(jsonException); } out.flush(); out.close(); return; // *********************** // END - JSON response // *********************** } else { req.setAttribute("twistInfoList", twistInfoList); req.setAttribute("twistInfoIdEnabled", store.getUniversalTwistInfoId()); RequestDispatcher dispatch = req.getRequestDispatcher("/twistinfo_setup.jsp"); dispatch.forward(req, resp); return; } } /** * Handles the following activities for <code>TwistInfo</code> * */ public void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException { // *********************** // 1. If type is JSON, then respond with JSON // 2. Otherwise, dispatch to JSP // 3. Create/delete/update TwistInfo // *********************** // *********************** // INITIALIZAION // *********************** Long twistInfoId = null; TwistInfo twistInfo = null; try { twistInfoId = new Long(req.getParameter(PARAMETER_KEY_TWIST_ID)); twistInfo = store.getTwistInfoById(twistInfoId); } catch (Exception e) { // Do nothing. If the value doesn't exist, // then we'll create a new TwistInfo } if (twistInfo == null) { twistInfo = new TwistInfo(); } // *********************** // DATA HANDLING // *********************** String name = req.getParameter(PARAMETER_KEY_TWIST_NAME); if (name == null || name.trim().length() == 0) { name = "TwistInfo (auto-generated)"; } String[] originationArguments = req.getParameterValues(PARAMETER_KEY_TWIST_ORIGINATION_LIST); String[] destinationArguments = req.getParameterValues(PARAMETER_KEY_TWIST_DESTINATION_LIST); // Remove any existing TwistInfo patterns. twistInfo.setPatternPairList(new ArrayList<PatternPair>()); if ((originationArguments != null && destinationArguments != null) && (originationArguments.length == destinationArguments.length)) { for (int i = 0; i < originationArguments.length; i++) { twistInfo.addPatternPair(new PatternPair(originationArguments[i], destinationArguments[i])); } } twistInfo.setName(name); twistInfo = store.saveOrUpdateTwistInfo(twistInfo); // *********************** // RESPONSE - in JSON or JSP // *********************** String responseType = req.getParameter(PARAMETER_KEY_RESPONSE_TYPE); if (PARAMETER_KEY_RESPONSE_TYPE_VALUE_JSON.equalsIgnoreCase(responseType)) { // BUILD JSON response resp.setContentType("application/json"); PrintWriter out = resp.getWriter(); try { JSONObject jsonResponseObject = new JSONObject(); JSONObject jsonObject = new JSONObject(); jsonObject.put("success", "Twist updated"); jsonObject.put("id", "" + twistInfo.getId()); jsonObject.put("name", twistInfo.getName()); jsonResponseObject.put("result", jsonObject); out.println(jsonResponseObject.toString()); } catch (JSONException jsonException) { throw new ServletException(jsonException); } out.flush(); out.close(); return; } else { List<TwistInfo> twistInfoList = store.getTwistInfoList(); req.setAttribute("twistInfoList", twistInfoList); req.setAttribute("twistInfoIdEnabled", store.getUniversalTwistInfoId()); RequestDispatcher dispatch = req.getRequestDispatcher("/twistinfo_setup.jsp"); dispatch.forward(req, resp); return; } } }