1 /*******************************************************************************
2 * ============LICENSE_START==================================================
4 * * ===========================================================================
5 * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
6 * * ===========================================================================
7 * * Licensed under the Apache License, Version 2.0 (the "License");
8 * * you may not use this file except in compliance with the License.
9 * * You may obtain a copy of the License at
11 * * http://www.apache.org/licenses/LICENSE-2.0
13 * * Unless required by applicable law or agreed to in writing, software
14 * * distributed under the License is distributed on an "AS IS" BASIS,
15 * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * * See the License for the specific language governing permissions and
17 * * limitations under the License.
18 * * ============LICENSE_END====================================================
20 * * ECOMP is a trademark and service mark of AT&T Intellectual Property.
22 ******************************************************************************/
25 package org.onap.dmaap.datarouter.node;
27 import com.att.eelf.configuration.EELFLogger;
28 import com.att.eelf.configuration.EELFManager;
29 import java.io.BufferedReader;
30 import java.io.FileOutputStream;
31 import java.io.FileReader;
32 import java.io.IOException;
33 import java.io.OutputStream;
34 import java.util.Hashtable;
35 import java.util.Timer;
38 * Track redirections of subscriptions
40 public class RedirManager {
41 private static EELFLogger eelfLogger = EELFManager.getInstance().getLogger(RedirManager.class);
42 private Hashtable<String, String> sid2primary = new Hashtable<String, String>();
43 private Hashtable<String, String> sid2secondary = new Hashtable<String, String>();
44 private String redirfile;
45 RateLimitedOperation op;
48 * Create a mechanism for maintaining subscription redirections.
50 * @param redirfile The file to store the redirection information.
51 * @param mininterval The minimum number of milliseconds between writes to the redirection
53 * @param timer The timer thread used to run delayed file writes.
55 public RedirManager(String redirfile, long mininterval, Timer timer) {
56 this.redirfile = redirfile;
57 op = new RateLimitedOperation(mininterval, timer) {
60 StringBuffer sb = new StringBuffer();
61 for (String s : sid2primary.keySet()) {
62 sb.append(s).append(' ').append(sid2primary.get(s)).append(' ')
63 .append(sid2secondary.get(s)).append('\n');
65 try (OutputStream os = new FileOutputStream(RedirManager.this.redirfile)) {
66 os.write(sb.toString().getBytes());
68 } catch (Exception e) {
69 eelfLogger.error("Exception", e);
75 try (BufferedReader br = new BufferedReader(new FileReader(redirfile))) {
76 while ((s = br.readLine()) != null) {
78 String[] sx = s.split(" ");
79 if (s.startsWith("#") || sx.length != 3) {
82 sid2primary.put(sx[0], sx[1]);
83 sid2secondary.put(sx[0], sx[2]);
86 } catch (Exception e) {
87 eelfLogger.error("Missing file is normal", e);
92 * Set up redirection. If a request is to be sent to subscription ID sid, and that is
93 * configured to go to URL primary, instead, go to secondary.
95 * @param sid The subscription ID to be redirected
96 * @param primary The URL associated with that subscription ID
97 * @param secondary The replacement URL to use instead
99 public synchronized void redirect(String sid, String primary, String secondary) {
100 sid2primary.put(sid, primary);
101 sid2secondary.put(sid, secondary);
106 * Cancel redirection. If a request is to be sent to subscription ID sid, send it to its
109 * @param sid The subscription ID to remove from the table.
111 public synchronized void forget(String sid) {
112 sid2primary.remove(sid);
113 sid2secondary.remove(sid);
118 * Look up where to send a subscription. If the primary has changed or there is no redirection,
119 * use the primary. Otherwise, redirect to the secondary URL.
121 * @param sid The subscription ID to look up.
122 * @param primary The configured primary URL.
123 * @return The destination URL to really use.
125 public synchronized String lookup(String sid, String primary) {
126 String oprim = sid2primary.get(sid);
127 if (primary.equals(oprim)) {
128 return (sid2secondary.get(sid));
129 } else if (oprim != null) {
136 * Is a subscription redirected?
138 public synchronized boolean isRedirected(String sid) {
139 return (sid != null && sid2secondary.get(sid) != null);