datarouter-prov/src/main/java/org/onap/dmaap/datarouter/provisioning/PublishServlet.java

   1 /*******************************************************************************\r
   2  * ============LICENSE_START==================================================\r
   3  * * org.onap.dmaap\r
   4  * * ===========================================================================\r
   5  * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
   6  * * ===========================================================================\r
   7  * * Licensed under the Apache License, Version 2.0 (the "License");\r
   8  * * you may not use this file except in compliance with the License.\r
   9  * * You may obtain a copy of the License at\r
  10  * *\r
  11  *  *      http://www.apache.org/licenses/LICENSE-2.0\r
  12  * *\r
  13  *  * Unless required by applicable law or agreed to in writing, software\r
  14  * * distributed under the License is distributed on an "AS IS" BASIS,\r
  15  * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  16  * * See the License for the specific language governing permissions and\r
  17  * * limitations under the License.\r
  18  * * ============LICENSE_END====================================================\r
  19  * *\r
  20  * * ECOMP is a trademark and service mark of AT&T Intellectual Property.\r
  21  * *\r
  22  ******************************************************************************/\r
  23 \r
  24 \r
  25 package org.onap.dmaap.datarouter.provisioning;\r
  26 \r
  27 import com.att.eelf.configuration.EELFLogger;\r
  28 import com.att.eelf.configuration.EELFManager;\r
  29 import java.io.IOException;\r
  30 import java.util.ArrayList;\r
  31 import java.util.Collection;\r
  32 import java.util.List;\r
  33 import javax.servlet.ServletConfig;\r
  34 import javax.servlet.ServletException;\r
  35 import javax.servlet.http.HttpServletRequest;\r
  36 import javax.servlet.http.HttpServletResponse;\r
  37 import org.json.JSONArray;\r
  38 import org.json.JSONObject;\r
  39 import org.json.JSONTokener;\r
  40 import org.onap.dmaap.datarouter.provisioning.beans.EventLogRecord;\r
  41 import org.onap.dmaap.datarouter.provisioning.beans.Feed;\r
  42 import org.onap.dmaap.datarouter.provisioning.beans.IngressRoute;\r
  43 import org.onap.dmaap.datarouter.provisioning.eelf.EelfMsgs;\r
  44 import org.onap.dmaap.datarouter.provisioning.utils.DB;\r
  45 \r
  46 /**\r
  47  * This servlet handles redirects for the &lt;publishURL&gt; on the provisioning server, which is generated by the\r
  48  * provisioning server to handle a particular subscriptions to a feed. See the <b>File Publishing and Delivery API</b>\r
  49  * document for details on how these methods should be invoked.\r
  50  *\r
  51  * @author Robert Eby\r
  52  * @version $Id: PublishServlet.java,v 1.8 2014/03/12 19:45:41 eby Exp $\r
  53  */\r
  54 @SuppressWarnings("serial")\r
  55 public class PublishServlet extends BaseServlet {\r
  56 \r
  57     private int next_node;\r
  58     private String provstring;\r
  59     private List<IngressRoute> irt;\r
  60     //Adding EELF Logger Rally:US664892\r
  61     private static EELFLogger eelflogger = EELFManager.getInstance()\r
  62         .getLogger("org.onap.dmaap.datarouter.provisioning.PublishServlet");\r
  63     private static final Object lock = new Object();\r
  64 \r
  65 \r
  66     @Override\r
  67     public void init(ServletConfig config) throws ServletException {\r
  68         super.init(config);\r
  69         next_node = 0;\r
  70         provstring = "";\r
  71         irt = new ArrayList<IngressRoute>();\r
  72 \r
  73     }\r
  74 \r
  75     @Override\r
  76     public void doDelete(HttpServletRequest req, HttpServletResponse resp) throws IOException {\r
  77         setIpAndFqdnForEelf("doDelete");\r
  78         eelflogger.info(EelfMsgs.MESSAGE_WITH_BEHALF_AND_FEEDID, req.getHeader(BEHALF_HEADER), getIdFromPath(req) + "");\r
  79         redirect(req, resp);\r
  80     }\r
  81 \r
  82     @Override\r
  83     public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {\r
  84         setIpAndFqdnForEelf("doGet");\r
  85         eelflogger.info(EelfMsgs.MESSAGE_WITH_BEHALF_AND_FEEDID, req.getHeader(BEHALF_HEADER), getIdFromPath(req) + "");\r
  86         redirect(req, resp);\r
  87     }\r
  88 \r
  89     @Override\r
  90     public void doPut(HttpServletRequest req, HttpServletResponse resp) throws IOException {\r
  91         setIpAndFqdnForEelf("doPut");\r
  92         eelflogger.info(EelfMsgs.MESSAGE_WITH_BEHALF_AND_FEEDID, req.getHeader(BEHALF_HEADER), getIdFromPath(req) + "");\r
  93         redirect(req, resp);\r
  94     }\r
  95 \r
  96     @Override\r
  97     public void doPost(HttpServletRequest req, HttpServletResponse resp) throws IOException {\r
  98         setIpAndFqdnForEelf("doPost");\r
  99         eelflogger.info(EelfMsgs.MESSAGE_WITH_BEHALF, req.getHeader(BEHALF_HEADER));\r
 100         redirect(req, resp);\r
 101 \r
 102     }\r
 103 \r
 104     private void redirect(HttpServletRequest req, HttpServletResponse resp) {\r
 105         try {\r
 106             String[] nodes = getNodes();\r
 107             if (nodes == null || nodes.length == 0) {\r
 108                 resp.sendError(HttpServletResponse.SC_SERVICE_UNAVAILABLE,\r
 109                     "There are no nodes defined in the DR network.");\r
 110             } else {\r
 111                 EventLogRecord elr = new EventLogRecord(req);\r
 112                 int feedid = checkPath(req);\r
 113                 if (feedid < 0) {\r
 114                     String message = (feedid == -1)\r
 115                         ? "Invalid request - Missing or bad feed number."\r
 116                         : "Invalid request - Missing file ID.";\r
 117                     elr.setMessage(message);\r
 118                     elr.setResult(HttpServletResponse.SC_NOT_FOUND);\r
 119                     eventlogger.info(elr);\r
 120 \r
 121                     resp.sendError(HttpServletResponse.SC_NOT_FOUND, message);\r
 122                 } else {\r
 123                     // Generate new URL\r
 124                     String nextnode = getRedirectNode(feedid, req);\r
 125                     nextnode = nextnode + ":" + DB.getHttpsPort();\r
 126                     String newurl = "https://" + nextnode + "/publish" + req.getPathInfo();\r
 127                     String qs = req.getQueryString();\r
 128                     if (qs != null) {\r
 129                         newurl += "?" + qs;\r
 130                     }\r
 131 \r
 132                     // Log redirect in event log\r
 133                     String message = "Redirected to: " + newurl;\r
 134                     elr.setMessage(message);\r
 135                     elr.setResult(HttpServletResponse.SC_MOVED_PERMANENTLY);\r
 136                     eventlogger.info(elr);\r
 137 \r
 138                     resp.setStatus(HttpServletResponse.SC_MOVED_PERMANENTLY);\r
 139                     resp.setHeader("Location", newurl);\r
 140                 }\r
 141             }\r
 142         } catch (IOException ioe) {\r
 143             intlogger.error("IOException" + ioe.getMessage());\r
 144 \r
 145         }\r
 146     }\r
 147 \r
 148     private String getRedirectNode(int feedid, HttpServletRequest req) {\r
 149         // Check to see if the IRT needs to be updated\r
 150         Poker p = Poker.getPoker();\r
 151         String s = p.getProvisioningString();\r
 152         synchronized (lock) {\r
 153             if (irt == null || (s.length() != provstring.length()) || !s.equals(provstring)) {\r
 154                 // Provisioning string has changed -- update the IRT\r
 155                 provstring = s;\r
 156                 JSONObject jo = new JSONObject(new JSONTokener(provstring));\r
 157                 JSONArray ja = jo.getJSONArray("ingress");\r
 158                 List<IngressRoute> newlist = new ArrayList<IngressRoute>();\r
 159                 for (int i = 0; i < ja.length(); i++) {\r
 160                     IngressRoute iroute = new IngressRoute(ja.getJSONObject(i));\r
 161                     newlist.add(iroute);\r
 162                 }\r
 163                 irt = newlist;\r
 164             }\r
 165         }\r
 166 \r
 167         // Look in IRT for next node\r
 168         for (IngressRoute route : irt) {\r
 169             if (route.matches(feedid, req)) {\r
 170                 // pick a node at random from the list\r
 171                 Collection<String> nodes = route.getNodes();\r
 172                 String[] arr = nodes.toArray(new String[0]);\r
 173                 long id = System.currentTimeMillis() % arr.length;\r
 174                 String node = arr[(int) id];\r
 175                 intlogger.info("Redirecting to " + node + " because of route " + route);\r
 176                 return node;\r
 177             }\r
 178         }\r
 179 \r
 180         // No IRT rule matches, do round robin of all active nodes\r
 181         String[] nodes = getNodes();\r
 182         if (next_node >= nodes.length)    // The list of nodes may have grown/shrunk\r
 183         {\r
 184             next_node = 0;\r
 185         }\r
 186         return nodes[next_node++];\r
 187     }\r
 188 \r
 189     private int checkPath(HttpServletRequest req) {\r
 190         String path = req.getPathInfo();\r
 191         if (path == null || path.length() < 2) {\r
 192             return -1;\r
 193         }\r
 194         path = path.substring(1);\r
 195         int ix = path.indexOf('/');\r
 196         if (ix < 0 || ix == path.length() - 1) {\r
 197             return -2;\r
 198         }\r
 199         try {\r
 200             int feedid = Integer.parseInt(path.substring(0, ix));\r
 201             if (!Feed.isFeedValid(feedid)) {\r
 202                 return -1;\r
 203             }\r
 204             return feedid;\r
 205         } catch (NumberFormatException e) {\r
 206             return -1;\r
 207         }\r
 208     }\r
 209 }\r