1 files changed, 114 insertions, 0 deletions
diff --git a/ecomp-portal-BE-common/src/main/java/org/onap/portalapp/portal/service/ConsulHealthServiceImpl.java b/ecomp-portal-BE-common/src/main/java/org/onap/portalapp/portal/service/ConsulHealthServiceImpl.java
new file mode 100644
index 00000000..ebb43767
--- /dev/null
+++ b/ecomp-portal-BE-common/src/main/java/org/onap/portalapp/portal/service/ConsulHealthServiceImpl.java
@@ -0,0 +1,114 @@
+/*-
+ * ============LICENSE_START==========================================
+ * ONAP Portal
+ * ===================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ===================================================================
+ *
+ * Unless otherwise specified, all software contained herein is licensed
+ * under the Apache License, Version 2.0 (the "License");
+ * you may not use this software 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.
+ *
+ * Unless otherwise specified, all documentation contained herein is licensed
+ * under the Creative Commons License, Attribution 4.0 Intl. (the "License");
+ * you may not use this documentation except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *             https://creativecommons.org/licenses/by/4.0/
+ *
+ * Unless required by applicable law or agreed to in writing, documentation
+ * 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.
+ *
+ * ============LICENSE_END============================================
+ *
+ * ECOMP is a trademark and service mark of AT&T Intellectual Property.
+ */
+package org.onap.portalapp.portal.service;
+
+import java.util.List;
+
+import org.onap.portalapp.portal.utils.EcompPortalUtils;
+import org.onap.portalsdk.core.logging.logic.EELFLoggerDelegate;
+import org.springframework.stereotype.Component;
+
+import com.ecwid.consul.ConsulException;
+import com.orbitz.consul.Consul;
+import com.orbitz.consul.HealthClient;
+import com.orbitz.consul.model.health.ServiceHealth;
+
+@Component
+public class ConsulHealthServiceImpl implements ConsulHealthService {
+
+	private EELFLoggerDelegate logger = EELFLoggerDelegate.getLogger(ConsulHealthServiceImpl.class);
+
+	@Override
+	public String getServiceLocation(String service, String fallbackPortOnLocalHost) {
+
+		List<ServiceHealth> nodes = null;
+
+		try {
+			Consul consul = Consul.builder().build();
+			HealthClient healthClient = consul.healthClient();
+			nodes = healthClient.getHealthyServiceInstances(service).getResponse();
+		} catch (Exception e) {
+			String localFallbackServiceLocation = EcompPortalUtils.localOrDockerHost() + ":" + fallbackPortOnLocalHost;
+			logger.debug(EELFLoggerDelegate.debugLogger,
+					" problem getting nodes for service {1}. Defaulting to {2}. Exception: {3}", service,
+					localFallbackServiceLocation, e.getMessage());
+			logger.error(EELFLoggerDelegate.errorLogger,
+					" problem getting nodes for service {1}. Defaulting to {2}. Exception: {3}", service,
+					localFallbackServiceLocation, e);
+			return localFallbackServiceLocation;
+		}
+
+		if (nodes == null || nodes.size() == 0) {
+			logger.debug(EELFLoggerDelegate.debugLogger, "No healthy node found in the consul cluster running service " + service
+					+ ". Defaulting to localhost");
+			return EcompPortalUtils.localOrDockerHost() + ":" + fallbackPortOnLocalHost;
+		} else {
+			String locationFromConsul;
+			ServiceHealth node = nodes.get(0);
+			locationFromConsul = node.getNode().getNode() + ":" + node.getService().getPort();
+			logger.debug(EELFLoggerDelegate.debugLogger,
+					"Found healthy service location using consul - returning location " + locationFromConsul);
+
+			// if locationFromConsul is null for some reason (very unlikely at
+			// this point), default to localhost
+			if (null == locationFromConsul || "".equals(locationFromConsul)) {
+				logger.debug(EELFLoggerDelegate.debugLogger,
+						"Couldn't get location from consul for service " + service + ". Defaulting to localhost");
+				return "localhost:" + fallbackPortOnLocalHost;
+			} else {
+				logger.debug(EELFLoggerDelegate.debugLogger, "Found service location from consul for service " + service
+						+ ". Location is " + locationFromConsul);
+				return locationFromConsul;
+			}
+		}
+	}
+
+	@Override
+	public List<ServiceHealth> getAllHealthyNodes(String service) throws ConsulException {
+		Consul consul = Consul.builder().build();
+		HealthClient healthClient = consul.healthClient();
+		return healthClient.getHealthyServiceInstances(service).getResponse();
+	}
+
+	@Override
+	public List<ServiceHealth> getAllNodes(String service) {
+		Consul consul = Consul.builder().build();
+		HealthClient healthClient = consul.healthClient();
+		return healthClient.getAllServiceInstances(service).getResponse();
+	}
+}