/** * Copyright 2012 Comcast Corporation * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.comcast.cns.controller; import java.io.IOException; import java.io.PrintWriter; import java.util.Map; import javax.servlet.ServletException; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.apache.log4j.Logger; import org.json.JSONObject; import com.amazonaws.services.sns.model.GetTopicAttributesRequest; import com.amazonaws.services.sns.model.GetTopicAttributesResult; import com.amazonaws.services.sns.model.SetTopicAttributesRequest; import com.comcast.cmb.common.controller.AdminServletBase; import com.comcast.cmb.common.controller.CMBControllerServlet; import com.comcast.cns.model.CNSRetryPolicy; import com.comcast.cns.model.CNSThrottlePolicy; import com.comcast.cns.model.CNSTopicDeliveryPolicy; import com.comcast.cns.model.CNSRetryPolicy.CnsBackoffFunction; /** * Admin page for editing topic delivery policy * @author tina, bwolf, aseem * */ public class CNSEditTopicDeliveryPolicyPage extends AdminServletBase { private static final long serialVersionUID = 1L; private static Logger logger = Logger.getLogger(CNSEditTopicDeliveryPolicyPage.class); @Override public void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { if (redirectUnauthenticatedUser(request, response)) { return; } CMBControllerServlet.valueAccumulator.initializeAllCounters(); response.setContentType("text/html"); PrintWriter out = response.getWriter(); String topicArn = request.getParameter("topicArn"); String userId = request.getParameter("userId"); Map<?, ?> params = request.getParameterMap(); connect(request); out.println("<html>"); simpleHeader(request, out, "View/Edit Topic Delivery Policy"); if (params.containsKey("Update")) { String numRetries = request.getParameter("numRetries"); String retriesNoDelay = request.getParameter("retriesNoDelay"); String minDelay = request.getParameter("minDelay"); String minDelayRetries = request.getParameter("minDelayRetries"); String maxDelay = request.getParameter("maxDelay"); String maxDelayRetries = request.getParameter("maxDelayRetries"); String maxReceiveRate = request.getParameter("maxReceiveRate"); String backoffFunc = request.getParameter("backoffFunc"); String ignoreOverride = request.getParameter("ignoreOverride"); CNSTopicDeliveryPolicy deliveryPolicy = new CNSTopicDeliveryPolicy(); CNSRetryPolicy defaultHealthyRetryPolicy = new CNSRetryPolicy(); if (maxDelay.trim().length() > 0) { defaultHealthyRetryPolicy.setMaxDelayTarget(Integer.parseInt(maxDelay)); } if (minDelay.trim().length() > 0) { defaultHealthyRetryPolicy.setMinDelayTarget(Integer.parseInt(minDelay)); } if (maxDelayRetries.trim().length() > 0) { defaultHealthyRetryPolicy.setNumMaxDelayRetries(Integer.parseInt(maxDelayRetries)); } if (minDelayRetries.trim().length() > 0) { defaultHealthyRetryPolicy.setNumMinDelayRetries(Integer.parseInt(minDelayRetries)); } if (retriesNoDelay.trim().length() > 0) { defaultHealthyRetryPolicy.setNumNoDelayRetries(Integer.parseInt(retriesNoDelay)); } if (numRetries.trim().length() > 0) { defaultHealthyRetryPolicy.setNumRetries(Integer.parseInt(numRetries)); } defaultHealthyRetryPolicy.setBackOffFunction(CnsBackoffFunction.valueOf(backoffFunc)); deliveryPolicy.setDefaultHealthyRetryPolicy(defaultHealthyRetryPolicy); deliveryPolicy.setDisableSubscriptionOverrides(ignoreOverride != null ? true: false); CNSThrottlePolicy defaultThrottle = new CNSThrottlePolicy(); if (maxReceiveRate.trim().length() > 0) { defaultThrottle.setMaxReceivesPerSecond(Integer.parseInt(maxReceiveRate)); } deliveryPolicy.setDefaultThrottlePolicy(defaultThrottle ); try { SetTopicAttributesRequest setTopicAttributesRequest = new SetTopicAttributesRequest(topicArn, "DeliveryPolicy", deliveryPolicy.toString()); sns.setTopicAttributes(setTopicAttributesRequest); logger.debug("event=set_delivery_policy topic_arn=" + topicArn + " userId= " + userId); } catch (Exception ex) { logger.error("event=set_delivery_policy user_id= " + userId, ex); throw new ServletException(ex); } out.println("<body onload='javascript:window.opener.location.reload();window.close();'>"); } else { int numRetries=0, retriesNoDelay = 0, minDelay = 0, minDelayRetries = 0, maxDelay = 0, maxDelayRetries = 0, maxReceiveRate = 0; String retryBackoff = "linear"; boolean ignoreOverride = false; if (topicArn != null) { Map<String, String> attributes = null; CNSTopicDeliveryPolicy deliveryPolicy = null; try { GetTopicAttributesRequest getTopicAttributesRequest = new GetTopicAttributesRequest(topicArn); GetTopicAttributesResult getTopicAttributesResult = sns.getTopicAttributes(getTopicAttributesRequest); attributes = getTopicAttributesResult.getAttributes(); deliveryPolicy = new CNSTopicDeliveryPolicy(new JSONObject(attributes.get("DeliveryPolicy"))); } catch (Exception ex) { logger.error("event=failed_to_get_attributes arn=" + topicArn, ex); throw new ServletException(ex); } if (deliveryPolicy != null) { CNSRetryPolicy healPol = deliveryPolicy.getDefaultHealthyRetryPolicy(); if (healPol != null) { numRetries= healPol.getNumRetries(); retriesNoDelay = healPol.getNumNoDelayRetries(); minDelay = healPol.getMinDelayTarget(); minDelayRetries = healPol.getNumMinDelayRetries(); maxDelay = healPol.getMaxDelayTarget(); maxDelayRetries = healPol.getNumMaxDelayRetries(); retryBackoff = healPol.getBackOffFunction().toString(); } CNSThrottlePolicy throttlePol = deliveryPolicy.getDefaultThrottlePolicy(); if (throttlePol != null) { if (throttlePol.getMaxReceivesPerSecond() != null) { maxReceiveRate = throttlePol.getMaxReceivesPerSecond().intValue(); } } ignoreOverride = deliveryPolicy.isDisableSubscriptionOverrides(); } } out.println("<body>"); out.println("<h1>View/Edit Topic Delivery Policy</h1>"); out.println("<form action=\"/webui/cnsuser/editdeliverypolicy?topicArn="+topicArn+"\" method=POST>"); out.println("<input type='hidden' name='userId' value='"+ userId +"'>"); out.println("<table>"); out.println("<tr><td colspan=2><b><font color='orange'>Delivery Policy</font></b></td></tr>"); out.println("<tr><td colspan=2><b>Apply these delivery policies for the topic:</b></td></tr>"); out.println("<tr><td>Number of retries:</td><td><input type='text' name='numRetries' size='50' value='" + numRetries + "'></td></tr>"); out.println("<tr><td> </td><td><I><font color='grey'>Between 0 - 100</font></I></td></tr>"); out.println("<tr><td>Retries with no delay:</td><td><input type='text' name='retriesNoDelay' size='50' value='" + retriesNoDelay + "'></td></tr>"); out.println("<tr><td> </td><td><I><font color='grey'>Between (0 - number of retries)</font></I></td></tr>"); out.println("<tr><td>Minimum delay:</td><td><input type='text' name='minDelay' size='50' value='" + minDelay + "'></td></tr>"); out.println("<tr><td> </td><td><I><font color='grey'>In seconds.Between 0 - maximum delay</font></I></td></tr>"); out.println("<tr><td>Minimum delay retries:</td><td><input type='text' name='minDelayRetries' size='50' value='" + minDelayRetries + "'></td></tr>"); out.println("<tr><td> </td><td><I><font color='grey'>Between (0 - number of retries)</font></I></td></tr>"); out.println("<tr><td>Maximum delay:</td><td><input type='text' name='maxDelay' size='50' value='" + maxDelay + "'></td></tr>"); out.println("<tr><td> </td><td><I><font color='grey'>In seconds. Between minimum delay - 3600</font></I></td></tr>"); out.println("<tr><td>Maximum delay retries:</td><td><input type='text' name='maxDelayRetries' size='50' value='" + maxDelayRetries + "'></td></tr>"); out.println("<tr><td> </td><td><I><font color='grey'>Between (0 - number of retries)</font></I></td></tr>"); out.println("<tr><td>Maximum receive rate:</td><td><input type='text' name='maxReceiveRate' size='50' value='" + maxReceiveRate + "'></td></tr>"); out.println("<tr><td> </td><td><I><font color='grey'>Receives per second. >= 1</font></I></td></tr>"); out.println("<tr><td> </td><td> </td></tr>"); if (retryBackoff.equals("linear")) { out.println("<tr><td>Retry backoff function:</td><td><select name='backoffFunc'><option value='linear' selected>Linear</option><option value='arithmetic'>Arithmetic</option><option value='geometric'>Geometric</option><option value='exponential'>Exponential</option></select></td></tr>"); } else if (retryBackoff.equals("arithmetic")) { out.println("<tr><td>Retry backoff function:</td><td><select name='backoffFunc'><option value='linear'>Linear</option><option value='arithmetic' selected>Arithmetic</option><option value='geometric'>Geometric</option><option value='exponential'>Exponential</option></select></td></tr>"); } else if (retryBackoff.equals("geometric")) { out.println("<tr><td>Retry backoff function:</td><td><select name='backoffFunc'><option value='linear'>Linear</option><option value='arithmetic'>Arithmetic</option><option value='geometric' selected>Geometric</option><option value='exponential'>Exponential</option></select></td></tr>"); } else if (retryBackoff.equals("exponential")) { out.println("<tr><td>Retry backoff function:</td><td><select name='backoffFunc'><option value='linear'>Linear</option><option value='arithmetic'>Arithmetic</option><option value='geometric'>Geometric</option><option value='exponential' selected>Exponential</option></select></td></tr>"); } if (ignoreOverride) { out.println("<tr><td>Ignore subscription override:</td><td><input type='checkbox' name='ignoreOverride' checked></td></tr>"); } else { out.println("<tr><td>Ignore subscription override:</td><td><input type='checkbox' name='ignoreOverride'></td></tr>"); } out.println("<tr><td colspan=2><hr/></td></tr>"); out.println("<tr><td colspan=2 align=right><input type='button' onclick='window.close()' value='Cancel'><input type='submit' name='Update' value='Update'></td></tr></table></form>"); } out.println("</body></html>"); CMBControllerServlet.valueAccumulator.deleteAllCounters(); } @Override public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { doGet(request, response); } }