summaryrefslogtreecommitdiffstats
path: root/ecomp-sdk/epsdk-core/src/main/java/org/openecomp/portalsdk/core/onboarding/ueb/Publisher.java
blob: 8dbdda2bc6f2913ca21f86cdede69b59404f6516 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*-
 * ================================================================================
 * ECOMP Portal SDK
 * ================================================================================
 * Copyright (C) 2017 AT&T Intellectual Property
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ================================================================================
 */
package org.openecomp.portalsdk.core.onboarding.ueb;

import java.io.IOException;
import java.net.MalformedURLException;
import java.security.GeneralSecurityException;
import java.util.LinkedList;
import java.util.concurrent.TimeUnit;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.openecomp.portalsdk.core.onboarding.util.PortalApiConstants;

import com.att.nsa.cambria.client.CambriaBatchingPublisher;
import com.att.nsa.cambria.client.CambriaClientBuilders;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.ObjectWriter;

/**
 * Provides a publisher that sends messages to a UEB topic.
 * 
 * Utilizes AT&T's UEB/Cambria subscriber/publisher messaging service. 
 */
public class Publisher {

	private final Log logger = LogFactory.getLog(getClass());

	protected final LinkedList<String> urlList = Helper.uebUrlList();

	private final String topicName;
	private final String publisherKey;
	private final String publisherSecret;

	/**
	 * Accepts coordinates needed to publish to a UEB topic.
	 * 
	 * @param publisherKey
	 *            UEB key used to publish to the topic
	 * @param publisherSecret
	 *            UEB secret used to publish to the topic
	 * @param topicName
	 *            UEB topic name
	 */
	public Publisher(String publisherKey, String publisherSecret, String topicName) {
		this.publisherKey = publisherKey;
		this.publisherSecret = publisherSecret;
		this.topicName = topicName;
		logger.info("Publisher instantiated for topic " + topicName);
	}

	/**
	 * Creates a publisher, subscribes to the topic, sends the specified message
	 * to the topic, then closes the publisher. This ensures that the single
	 * message goes immediately. UEB is designed for high throughput and tries
	 * to batch up multiple messages in each send, but this method wants the
	 * single message to go immediately.
	 * 
	 * @param uebMsg
	 *            Message object to send as the payload.
	 * @throws UebException
	 *             If anything goes wrong, including JSON serialization of the
	 *             specified message object.
	 */
	public void send(UebMsg uebMsg) throws UebException {
		String msg = null;

		CambriaBatchingPublisher pub;
		try {
			pub = new CambriaClientBuilders.PublisherBuilder()
					.authenticatedBy(publisherKey, publisherSecret).usingHosts(urlList).onTopic(topicName).build();
		} catch (MalformedURLException e1) {
			logger.error("pub.build Exception ", e1);
			throw new UebException(PortalApiConstants.ECOMP_UEB_UNKNOWN_PUBLISH_ERROR, e1, topicName, null, msg);
		} catch (GeneralSecurityException e1) {
			logger.error("pub.build Exception ", e1);
			throw new UebException(PortalApiConstants.ECOMP_UEB_UNKNOWN_PUBLISH_ERROR, e1, topicName, null, msg);
		}

		try {
			ObjectWriter mapper = new ObjectMapper().writer().withDefaultPrettyPrinter();
			msg = mapper.writeValueAsString(uebMsg);
		} catch (JsonProcessingException e) {
			throw new UebException(PortalApiConstants.ECOMP_UEB_INVALID_MSG, topicName, null, null);
		}

		try {
			logger.debug("Publishing to " + topicName + " msg: " + msg);
			int NumSent = pub.send(PortalApiConstants.ECOMP_GENERAL_UEB_PARTITION, msg);
			if (NumSent == 0) {
				throw new UebException(PortalApiConstants.ECOMP_UEB_UNKNOWN_PUBLISH_ERROR, topicName, null, msg);
			}
		} catch (IOException ex) {
			logger.error("Failed to publish", ex);
			throw new UebException(PortalApiConstants.ECOMP_UEB_UNKNOWN_PUBLISH_ERROR, ex, topicName, null, msg);
		}

		try {
			// close the publisher to make sure everything's sent before exiting
			pub.close(5, TimeUnit.SECONDS);
		} catch (Exception ex) {
			logger.error("pub.close Exception ", ex);
			throw new UebException(PortalApiConstants.ECOMP_UEB_UNKNOWN_PUBLISH_ERROR, ex, topicName, null, msg);
		}

	}
}