/*-
 * ============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============================================
 *
 * 
 */
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();
	}
}