1 /*******************************************************************************
\r
2 * ============LICENSE_START==================================================
\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
11 * * http://www.apache.org/licenses/LICENSE-2.0
\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
20 * * ECOMP is a trademark and service mark of AT&T Intellectual Property.
\r
22 ******************************************************************************/
\r
24 package org.onap.dmaap.datarouter.provisioning;
\r
26 import java.io.IOException;
\r
27 import java.net.HttpURLConnection;
\r
28 import java.net.InetAddress;
\r
29 import java.net.MalformedURLException;
\r
30 import java.net.URL;
\r
31 import java.net.UnknownHostException;
\r
32 import java.util.Arrays;
\r
33 import java.util.HashSet;
\r
34 import java.util.Map;
\r
35 import java.util.Set;
\r
36 import java.util.Timer;
\r
37 import java.util.TimerTask;
\r
38 import java.util.TreeSet;
\r
39 import org.apache.log4j.Logger;
\r
40 import org.json.JSONException;
\r
41 import org.json.JSONObject;
\r
42 import org.json.JSONTokener;
\r
43 import org.onap.dmaap.datarouter.provisioning.beans.EgressRoute;
\r
44 import org.onap.dmaap.datarouter.provisioning.beans.Feed;
\r
45 import org.onap.dmaap.datarouter.provisioning.beans.Group;
\r
46 import org.onap.dmaap.datarouter.provisioning.beans.IngressRoute;
\r
47 import org.onap.dmaap.datarouter.provisioning.beans.NetworkRoute;
\r
48 import org.onap.dmaap.datarouter.provisioning.beans.Parameters;
\r
49 import org.onap.dmaap.datarouter.provisioning.beans.Subscription;
\r
50 import org.onap.dmaap.datarouter.provisioning.utils.DB;
\r
53 * This class handles the two timers (described in R1 Design Notes), and takes care of issuing the GET to each node of
\r
54 * the URL to "poke".
\r
56 * @author Robert Eby
\r
57 * @version $Id: Poker.java,v 1.11 2014/01/08 16:13:47 eby Exp $
\r
59 public class Poker extends TimerTask {
\r
62 * Template used to generate the URL to issue the GET against
\r
64 private static final String POKE_URL_TEMPLATE = "http://%s/internal/fetchProv";
\r
66 private static final Object lock = new Object();
\r
69 * This is a singleton -- there is only one Poker object in the server
\r
71 private static Poker poker;
\r
72 private long timer1;
\r
73 private long timer2;
\r
74 private String thisPod; // DNS name of this machine
\r
75 private Logger logger;
\r
76 private String provString;
\r
79 timer1 = timer2 = 0;
\r
80 Timer rolex = new Timer();
\r
81 logger = Logger.getLogger("org.onap.dmaap.datarouter.provisioning.internal");
\r
83 thisPod = InetAddress.getLocalHost().getHostName();
\r
84 } catch (UnknownHostException e) {
\r
85 thisPod = "*UNKNOWN*"; // not a major problem
\r
86 logger.info("UnknownHostException: Setting thisPod to \"*UNKNOWN*\"");
\r
88 provString = buildProvisioningString();
\r
90 rolex.scheduleAtFixedRate(this, 0L, 1000L); // Run once a second to check the timers
\r
94 * Get the singleton Poker object.
\r
98 public static synchronized Poker getPoker() {
\r
99 if (poker == null) {
\r
100 poker = new Poker();
\r
106 * This method sets the two timers described in the design notes.
\r
108 * @param t1 the first timer controls how long to wait after a provisioning request before poking each node This
\r
109 * timer can be reset if it has not "gone off".
\r
110 * @param t2 the second timer set the outer bound on how long to wait. It cannot be reset.
\r
112 public void setTimers(long t1, long t2) {
\r
113 synchronized (lock) {
\r
114 if (timer1 == 0 || t1 > timer1) {
\r
121 if (logger.isDebugEnabled()) {
\r
122 logger.debug("Poker timers set to " + timer1 + " and " + timer2);
\r
129 * Return the last provisioning string built.
\r
131 * @return the last provisioning string built.
\r
133 public String getProvisioningString() {
\r
138 * The method to run at the predefined interval (once per second). This method checks to see if either of the two
\r
139 * timers has expired, and if so, will rebuild the provisioning string, and poke all the nodes and other PODs. The
\r
140 * timers are then reset to 0.
\r
143 public void run() {
\r
146 long now = System.currentTimeMillis();
\r
147 boolean fire = false;
\r
148 synchronized (lock) {
\r
149 if (now > timer1 || now > timer2) {
\r
150 timer1 = timer2 = 0;
\r
158 } catch (Exception e) {
\r
159 logger.warn("PROV0020: Caught exception in Poker: " + e);
\r
163 private void pokeNodes() {
\r
164 // Rebuild the prov string
\r
165 provString = buildProvisioningString();
\r
166 // Only the active POD should poke nodes, etc.
\r
167 boolean active = SynchronizerTask.getSynchronizer().isActive();
\r
169 // Poke all the DR nodes
\r
170 for (String n : BaseServlet.getNodes()) {
\r
173 // Poke the pod that is not us
\r
174 for (String n : BaseServlet.getPods()) {
\r
175 if (n.length() > 0 && !n.equals(thisPod)) {
\r
182 private void pokeNode(final String nodename) {
\r
183 logger.debug("PROV0012 Poking node " + nodename + " ...");
\r
184 String nodeUrl = String.format(POKE_URL_TEMPLATE, nodename + ":" + DB.getHttpPort());
\r
185 Runnable r = () -> {
\r
187 URL url = new URL(nodeUrl);
\r
188 HttpURLConnection conn = (HttpURLConnection) url.openConnection();
\r
189 conn.setConnectTimeout(60000); //Fixes for Itrack DATARTR-3, poke timeout
\r
191 conn.getContentLength(); // Force the GET through
\r
193 } catch (MalformedURLException e) {
\r
195 "PROV0013 MalformedURLException Error poking node at " + nodeUrl + " : " + e
\r
197 } catch (IOException e) {
\r
198 logger.warn("PROV0013 IOException Error poking node at " + nodeUrl + " : " + e
\r
205 private String buildProvisioningString() {
\r
206 StringBuilder sb = new StringBuilder("{\n");
\r
208 // Append Feeds to the string
\r
210 sb.append("\"feeds\": [");
\r
211 for (Feed f : Feed.getAllFeeds()) {
\r
213 sb.append(f.asJSONObject().toString());
\r
216 sb.append("\n],\n");
\r
218 //Append groups to the string - Rally:US708115 - 1610
\r
220 sb.append("\"groups\": [");
\r
221 for (Group s : Group.getAllgroups()) {
\r
223 sb.append(s.asJSONObject().toString());
\r
226 sb.append("\n],\n");
\r
228 // Append Subscriptions to the string
\r
230 sb.append("\"subscriptions\": [");
\r
231 for (Subscription s : Subscription.getAllSubscriptions()) {
\r
234 sb.append(s.asJSONObject().toString());
\r
238 sb.append("\n],\n");
\r
240 // Append Parameters to the string
\r
242 sb.append("\"parameters\": {");
\r
243 Map<String, String> props = Parameters.getParameters();
\r
244 Set<String> ivals = new HashSet<String>();
\r
245 String intv = props.get("_INT_VALUES");
\r
246 if (intv != null) {
\r
247 ivals.addAll(Arrays.asList(intv.split("\\|")));
\r
249 for (String key : new TreeSet<String>(props.keySet())) {
\r
250 String v = props.get(key);
\r
252 sb.append(" \"").append(key).append("\": ");
\r
253 if (ivals.contains(key)) {
\r
256 } else if (key.endsWith("S")) {
\r
257 // Split and append array of strings
\r
258 String[] pp = v.split("\\|");
\r
261 for (String t : pp) {
\r
262 sb.append(p2).append("\"").append(quote(t)).append("\"");
\r
267 sb.append("\"").append(quote(v)).append("\"");
\r
271 sb.append("\n},\n");
\r
273 // Append Routes to the string
\r
275 sb.append("\"ingress\": [");
\r
276 for (IngressRoute in : IngressRoute.getAllIngressRoutes()) {
\r
278 sb.append(in.asJSONObject().toString());
\r
281 sb.append("\n],\n");
\r
284 sb.append("\"egress\": {");
\r
285 for (EgressRoute eg : EgressRoute.getAllEgressRoutes()) {
\r
287 String t = eg.asJSONObject().toString();
\r
288 t = t.substring(1, t.length() - 1);
\r
292 sb.append("\n},\n");
\r
295 sb.append("\"routing\": [");
\r
296 for (NetworkRoute ne : NetworkRoute.getAllNetworkRoutes()) {
\r
298 sb.append(ne.asJSONObject().toString());
\r
304 // Convert to string and verify it is valid JSON
\r
305 String tempProvString = sb.toString();
\r
307 new JSONObject(new JSONTokener(tempProvString));
\r
308 } catch (JSONException e) {
\r
309 logger.warn("PROV0016: Possible invalid prov string: " + e);
\r
311 return tempProvString;
\r
314 private String quote(String s) {
\r
315 StringBuilder sb = new StringBuilder();
\r
316 for (char ch : s.toCharArray()) {
\r
317 if (ch == '\\' || ch == '"') {
\r
322 return sb.toString();
\r