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.OutputStream;
33 import java.util.HashMap;
35 import java.util.Timer;
38 * Track redirections of subscriptions.
42 private static EELFLogger eelfLogger = EELFManager.getInstance().getLogger(RedirManager.class);
43 private RateLimitedOperation op;
44 private HashMap<String, String> sid2primary = new HashMap<>();
45 private HashMap<String, String> sid2secondary = new HashMap<>();
46 private String redirfile;
49 * Create a mechanism for maintaining subscription redirections.
51 * @param redirfile The file to store the redirection information.
52 * @param mininterval The minimum number of milliseconds between writes to the redirection information file.
53 * @param timer The timer thread used to run delayed file writes.
55 RedirManager(String redirfile, long mininterval, Timer timer) {
56 this.redirfile = redirfile;
57 op = new RateLimitedOperation(mininterval, timer) {
60 StringBuilder sb = new StringBuilder();
61 for (Map.Entry<String, String> entry : sid2primary.entrySet()) {
62 String key = entry.getKey();
63 String value = entry.getValue();
64 sb.append(key).append(' ').append(value).append(' ')
65 .append(sid2secondary.get(key)).append('\n');
67 try (OutputStream os = new FileOutputStream(RedirManager.this.redirfile)) {
68 os.write(sb.toString().getBytes());
70 } catch (Exception e) {
71 eelfLogger.error("Exception", e);
77 try (BufferedReader br = new BufferedReader(new FileReader(redirfile))) {
78 while ((line = br.readLine()) != null) {
79 addSubRedirInfo(line);
82 } catch (Exception e) {
83 eelfLogger.debug("Missing file is normal", e);
88 * Set up redirection. If a request is to be sent to subscription ID sid, and that is configured to go to URL
89 * primary, instead, go to secondary.
91 * @param sid The subscription ID to be redirected
92 * @param primary The URL associated with that subscription ID
93 * @param secondary The replacement URL to use instead
95 synchronized void redirect(String sid, String primary, String secondary) {
96 sid2primary.put(sid, primary);
97 sid2secondary.put(sid, secondary);
102 * Cancel redirection. If a request is to be sent to subscription ID sid, send it to its primary URL.
104 * @param sid The subscription ID to remove from the table.
106 synchronized void forget(String sid) {
107 sid2primary.remove(sid);
108 sid2secondary.remove(sid);
113 * Look up where to send a subscription. If the primary has changed or there is no redirection, use the primary.
114 * Otherwise, redirect to the secondary URL.
116 * @param sid The subscription ID to look up.
117 * @param primary The configured primary URL.
118 * @return The destination URL to really use.
120 synchronized String lookup(String sid, String primary) {
121 String oprim = sid2primary.get(sid);
122 if (primary.equals(oprim)) {
123 return (sid2secondary.get(sid));
124 } else if (oprim != null) {
131 * Is a subscription redirected.
133 synchronized boolean isRedirected(String sid) {
134 return (sid != null && sid2secondary.get(sid) != null);
137 private void addSubRedirInfo(String subRedirInfo) {
138 subRedirInfo = subRedirInfo.trim();
139 String[] sx = subRedirInfo.split(" ");
140 if (subRedirInfo.startsWith("#") || sx.length != 3) {
143 sid2primary.put(sx[0], sx[1]);
144 sid2secondary.put(sx[0], sx[2]);