org.bitsofinfo.hazelcast.discovery.consul.HttpHealthCheckBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hazelcast-consul-discovery-spi Show documentation

hazelcast-consul-discovery-spi - Consul based discovery strategy SPI for Hazelcast enabled applications

The newest version!

package org.bitsofinfo.hazelcast.discovery.consul;

import java.util.Map;

import com.hazelcast.logging.ILogger;
import com.hazelcast.logging.Logger;
import com.hazelcast.cluster.Address;
import com.orbitz.consul.model.agent.Registration;
import com.orbitz.consul.model.agent.Registration.RegCheck;

/**
 * Implementation of the health check builder for HTTP health check
 * 
 * @author bmudda
 *
 */
public class HttpHealthCheckBuilder implements HealthCheckBuilder {
	
	// variables for http template support
	private static final String HTTP_TEMPLATE_MYPORT = "#MYPORT";
	private static final String HTTP_TEMPLATE_MYIP = "#MYIP";

	public static final String CONFIG_PROP_HEALTH_CHECK_HTTP = "healthCheckHttp";
	public static final String CONFIG_PROP_HEALTH_CHECK_HTTP_INTERVAL_SECONDS = "healthCheckHttpIntervalSeconds";
	
	private static final ILogger logger = Logger.getLogger(HttpHealthCheckBuilder.class);
	
	@Override
	public RegCheck buildRegistrationCheck( Map registratorConfig, Address localAddress) {
		
		RegCheck regCheck = null;
		
		try{
			/**
			 * Deal with health check http
			 */
			String healthCheckHttp = (String)registratorConfig.get(CONFIG_PROP_HEALTH_CHECK_HTTP);
			
			if (healthCheckHttp != null && !healthCheckHttp.trim().isEmpty()) {
				
				healthCheckHttp = healthCheckHttp.replaceAll(HTTP_TEMPLATE_MYIP, localAddress.getInetAddress().getHostAddress())
						  						 .replaceAll(HTTP_TEMPLATE_MYPORT, String.valueOf(localAddress.getPort()));
				
				Long healthCheckHttpIntervalSeconds = Long.valueOf((Integer)registratorConfig.get(CONFIG_PROP_HEALTH_CHECK_HTTP_INTERVAL_SECONDS));
				regCheck = Registration.RegCheck.http(healthCheckHttp, healthCheckHttpIntervalSeconds);
			}
		
		}catch(Exception e){
			logger.severe("Unexpected error occured trying to build HTTP health check : " + e.getMessage(), e);

		}
		
		return regCheck;
	}

}