+++ /dev/null
-/*******************************************************************************\r
- * ============LICENSE_START==================================================\r
- * * org.onap.dmaap\r
- * * ===========================================================================\r
- * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
- * * ===========================================================================\r
- * * Licensed under the Apache License, Version 2.0 (the "License");\r
- * * you may not use this file except in compliance with the License.\r
- * * You may obtain a copy of the License at\r
- * *\r
- * * http://www.apache.org/licenses/LICENSE-2.0\r
- * *\r
- * * Unless required by applicable law or agreed to in writing, software\r
- * * distributed under the License is distributed on an "AS IS" BASIS,\r
- * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
- * * See the License for the specific language governing permissions and\r
- * * limitations under the License.\r
- * * ============LICENSE_END====================================================\r
- * *\r
- * * ECOMP is a trademark and service mark of AT&T Intellectual Property.\r
- * *\r
- ******************************************************************************/\r
-\r
-package org.onap.dmaap.datarouter.provisioning;\r
-\r
-import com.att.eelf.configuration.EELFLogger;\r
-import com.att.eelf.configuration.EELFManager;\r
-import java.io.IOException;\r
-import java.net.HttpURLConnection;\r
-import java.net.InetAddress;\r
-import java.net.MalformedURLException;\r
-import java.net.URL;\r
-import java.net.UnknownHostException;\r
-import java.util.Arrays;\r
-import java.util.HashSet;\r
-import java.util.Map;\r
-import java.util.Set;\r
-import java.util.Timer;\r
-import java.util.TimerTask;\r
-import java.util.TreeSet;\r
-import org.json.JSONException;\r
-import org.json.JSONObject;\r
-import org.json.JSONTokener;\r
-\r
-import org.onap.dmaap.datarouter.provisioning.beans.EgressRoute;\r
-import org.onap.dmaap.datarouter.provisioning.beans.Feed;\r
-import org.onap.dmaap.datarouter.provisioning.beans.Group;\r
-import org.onap.dmaap.datarouter.provisioning.beans.IngressRoute;\r
-import org.onap.dmaap.datarouter.provisioning.beans.NetworkRoute;\r
-import org.onap.dmaap.datarouter.provisioning.beans.Parameters;\r
-import org.onap.dmaap.datarouter.provisioning.beans.Subscription;\r
-import org.onap.dmaap.datarouter.provisioning.utils.DB;\r
-\r
-/**\r
- * This class handles the two timers (described in R1 Design Notes), and takes care of issuing the GET to each node of\r
- * the URL to "poke".\r
- *\r
- * @author Robert Eby\r
- * @version $Id: Poker.java,v 1.11 2014/01/08 16:13:47 eby Exp $\r
- */\r
-\r
-public class Poker extends TimerTask {\r
-\r
- /**\r
- * Template used to generate the URL to issue the GET against.\r
- */\r
- private static final String POKE_URL_TEMPLATE = "http://%s/internal/fetchProv";\r
-\r
- private static final Object lock = new Object();\r
- private static final String CARRIAGE_RETURN = "\n],\n";\r
-\r
- /**\r
- * This is a singleton -- there is only one Poker object in the server.\r
- */\r
- private static Poker poker;\r
- private long timer1;\r
- private long timer2;\r
- private String thisPod; // DNS name of this machine\r
- private EELFLogger logger;\r
- private String provString;\r
-\r
-\r
- private Poker() {\r
- timer1 = timer2 = 0;\r
- logger = EELFManager.getInstance().getLogger("InternalLog");\r
- try {\r
- thisPod = InetAddress.getLocalHost().getHostName();\r
- } catch (UnknownHostException e) {\r
- thisPod = "*UNKNOWN_POD*"; // not a major problem\r
- logger.info("UnknownHostException: Setting thisPod to \"*UNKNOWN_POD*\"", e);\r
- }\r
- provString = buildProvisioningString();\r
- Timer rolex = new Timer();\r
- rolex.scheduleAtFixedRate(this, 0L, 1000L); // Run once a second to check the timers\r
- }\r
-\r
- /**\r
- * Get the singleton Poker object.\r
- *\r
- * @return the Poker\r
- */\r
- public static synchronized Poker getPoker() {\r
- if (poker == null) {\r
- poker = new Poker();\r
- }\r
- return poker;\r
- }\r
-\r
- /**\r
- * This method sets the two timers described in the design notes.\r
- *\r
- * @param t1 the first timer controls how long to wait after a provisioning request before poking each node This\r
- * timer can be reset if it has not "gone off".\r
- * @param t2 the second timer set the outer bound on how long to wait. It cannot be reset.\r
- */\r
- public void setTimers(long t1, long t2) {\r
- synchronized (lock) {\r
- if (timer1 == 0 || t1 > timer1) {\r
- timer1 = t1;\r
- }\r
- if (timer2 == 0) {\r
- timer2 = t2;\r
- }\r
- }\r
- if (logger.isDebugEnabled()) {\r
- logger.debug("Poker timers set to " + timer1 + " and " + timer2);\r
- }\r
-\r
-\r
- }\r
-\r
- /**\r
- * Return the last provisioning string built.\r
- *\r
- * @return the last provisioning string built.\r
- */\r
- public String getProvisioningString() {\r
- return provString;\r
- }\r
-\r
- /**\r
- * The method to run at the predefined interval (once per second). This method checks to see if either of the two\r
- * timers has expired, and if so, will rebuild the provisioning string, and poke all the nodes and other PODs. The\r
- * timers are then reset to 0.\r
- */\r
- @Override\r
- public void run() {\r
- try {\r
- if (timer1 > 0) {\r
- long now = System.currentTimeMillis();\r
- boolean fire = false;\r
- synchronized (lock) {\r
- if (now > timer1 || now > timer2) {\r
- timer1 = timer2 = 0;\r
- fire = true;\r
- }\r
- }\r
- if (fire) {\r
- pokeNodes();\r
- }\r
- }\r
- } catch (Exception e) {\r
- logger.warn("PROV0020: Caught exception in Poker: " + e);\r
- }\r
- }\r
-\r
- private void pokeNodes() {\r
- // Rebuild the prov string\r
- provString = buildProvisioningString();\r
- // Only the active POD should poke nodes, etc.\r
- boolean active = SynchronizerTask.getSynchronizer().isActive();\r
- if (active) {\r
- // Poke all the DR nodes\r
- for (String n : BaseServlet.getNodes()) {\r
- pokeNode(n);\r
- }\r
- // Poke the pod that is not us\r
- for (String n : BaseServlet.getPods()) {\r
- if (n.length() > 0 && !n.equals(thisPod)) {\r
- pokeNode(n);\r
- }\r
- }\r
- }\r
- }\r
-\r
- private void pokeNode(final String nodename) {\r
- logger.debug("PROV0012 Poking node " + nodename + " ...");\r
- String nodeUrl = String.format(POKE_URL_TEMPLATE, nodename + ":" + DB.getHttpPort());\r
- Runnable runn = () -> {\r
- try {\r
- URL url = new URL(nodeUrl);\r
- HttpURLConnection conn = (HttpURLConnection) url.openConnection();\r
- conn.setConnectTimeout(60000); //Fixes for Itrack DATARTR-3, poke timeout\r
- conn.connect();\r
- conn.getContentLength(); // Force the GET through\r
- conn.disconnect();\r
- } catch (MalformedURLException e) {\r
- logger.warn(\r
- "PROV0013 MalformedURLException Error poking node at " + nodeUrl + " : " + e\r
- .getMessage(), e);\r
- } catch (IOException e) {\r
- logger.warn("PROV0013 IOException Error poking node at " + nodeUrl + " : " + e\r
- .getMessage(), e);\r
- }\r
- };\r
- runn.run();\r
- }\r
-\r
- private String buildProvisioningString() {\r
- StringBuilder sb = new StringBuilder("{\n");\r
-\r
- // Append Feeds to the string\r
- String pfx = "\n";\r
- sb.append("\"feeds\": [");\r
- for (Feed f : Feed.getAllFeeds()) {\r
- sb.append(pfx);\r
- sb.append(f.asJSONObject().toString());\r
- pfx = ",\n";\r
- }\r
- sb.append(CARRIAGE_RETURN);\r
-\r
- //Append groups to the string - Rally:US708115 - 1610\r
- pfx = "\n";\r
- sb.append("\"groups\": [");\r
- for (Group s : Group.getAllgroups()) {\r
- sb.append(pfx);\r
- sb.append(s.asJSONObject().toString());\r
- pfx = ",\n";\r
- }\r
- sb.append(CARRIAGE_RETURN);\r
-\r
- // Append Subscriptions to the string\r
- pfx = "\n";\r
- sb.append("\"subscriptions\": [");\r
- for (Subscription s : Subscription.getAllSubscriptions()) {\r
- sb.append(pfx);\r
- if (s != null) {\r
- sb.append(s.asJSONObject().toString());\r
- }\r
- pfx = ",\n";\r
- }\r
- sb.append(CARRIAGE_RETURN);\r
-\r
- // Append Parameters to the string\r
- pfx = "\n";\r
- sb.append("\"parameters\": {");\r
- Map<String, String> props = Parameters.getParameters();\r
- Set<String> ivals = new HashSet<>();\r
- String intv = props.get("_INT_VALUES");\r
- if (intv != null) {\r
- ivals.addAll(Arrays.asList(intv.split("\\|")));\r
- }\r
- for (String key : new TreeSet<String>(props.keySet())) {\r
- String val = props.get(key);\r
- sb.append(pfx);\r
- sb.append(" \"").append(key).append("\": ");\r
- if (ivals.contains(key)) {\r
- // integer value\r
- sb.append(val);\r
- } else if (key.endsWith("S")) {\r
- // Split and append array of strings\r
- String[] pp = val.split("\\|");\r
- String p2 = "";\r
- sb.append("[");\r
- for (String t : pp) {\r
- sb.append(p2).append("\"").append(quote(t)).append("\"");\r
- p2 = ",";\r
- }\r
- sb.append("]");\r
- } else {\r
- sb.append("\"").append(quote(val)).append("\"");\r
- }\r
- pfx = ",\n";\r
- }\r
- sb.append("\n},\n");\r
-\r
- // Append Routes to the string\r
- pfx = "\n";\r
- sb.append("\"ingress\": [");\r
- for (IngressRoute in : IngressRoute.getAllIngressRoutes()) {\r
- sb.append(pfx);\r
- sb.append(in.asJSONObject().toString());\r
- pfx = ",\n";\r
- }\r
- sb.append(CARRIAGE_RETURN);\r
-\r
- pfx = "\n";\r
- sb.append("\"egress\": {");\r
- for (EgressRoute eg : EgressRoute.getAllEgressRoutes()) {\r
- sb.append(pfx);\r
- String str = eg.asJSONObject().toString();\r
- str = str.substring(1, str.length() - 1);\r
- sb.append(str);\r
- pfx = ",\n";\r
- }\r
- sb.append("\n},\n");\r
-\r
- pfx = "\n";\r
- sb.append("\"routing\": [");\r
- for (NetworkRoute ne : NetworkRoute.getAllNetworkRoutes()) {\r
- sb.append(pfx);\r
- sb.append(ne.asJSONObject().toString());\r
- pfx = ",\n";\r
- }\r
- sb.append("\n]");\r
- sb.append("\n}");\r
-\r
- // Convert to string and verify it is valid JSON\r
- String tempProvString = sb.toString();\r
- try {\r
- new JSONObject(new JSONTokener(tempProvString));\r
- } catch (JSONException e) {\r
- logger.warn("PROV0016: Possible invalid prov string: " + e);\r
- }\r
- return tempProvString;\r
- }\r
-\r
- private String quote(String str) {\r
- StringBuilder sb = new StringBuilder();\r
- for (char ch : str.toCharArray()) {\r
- if (ch == '\\' || ch == '"') {\r
- sb.append('\\');\r
- }\r
- sb.append(ch);\r
- }\r
- return sb.toString();\r
- }\r
-}\r