compile 'org.springframework.retry:spring-retry:1.1.2.RELEASE'

    compile 'org.springframework.boot:spring-boot-actuator:1.4.5.RELEASE'

/*

 * Copyright 2013-2015 the original author or authors.

 *

 * 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.springframework.cloud.consul;

import org.aspectj.lang.annotation.Aspect;

import org.springframework.boot.actuate.autoconfigure.ConditionalOnEnabledHealthIndicator;

import org.springframework.boot.actuate.condition.ConditionalOnEnabledEndpoint;

import org.springframework.boot.actuate.endpoint.Endpoint;

import org.springframework.boot.autoconfigure.aop.AopAutoConfiguration;

import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;

import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;

import org.springframework.boot.context.properties.EnableConfigurationProperties;

import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import org.springframework.context.annotation.Import;

import org.springframework.retry.annotation.EnableRetry;

import org.springframework.retry.annotation.Retryable;

import org.springframework.retry.interceptor.RetryInterceptorBuilder;

import org.springframework.retry.interceptor.RetryOperationsInterceptor;

import com.ecwid.consul.v1.ConsulClient;

/**

 * @author Spencer Gibb

 */

@Configuration

@EnableConfigurationProperties

@ConditionalOnConsulEnabled

public class ConsulAutoConfiguration {

    @Bean

    @ConditionalOnMissingBean

    public ConsulProperties consulProperties() {

        return new ConsulProperties();

    }

    @Bean

    @ConditionalOnMissingBean

    public ConsulClient consulClient(ConsulProperties consulProperties) {

        return new ConsulClient(consulProperties.getHost(), consulProperties.getPort());

    }

    @Configuration

    @ConditionalOnClass(Endpoint.class)

    protected static class ConsulHealthConfig {

        @Bean

        @ConditionalOnMissingBean

        @ConditionalOnEnabledEndpoint("consul")

        public ConsulEndpoint consulEndpoint(ConsulClient consulClient) {

            return new ConsulEndpoint(consulClient);

        }

        @Bean

        @ConditionalOnMissingBean

        @ConditionalOnEnabledHealthIndicator("consul")

        public PosConsulHealthIndicator consulHealthIndicator(ConsulClient consulClient) {

            return new PosConsulHealthIndicator(consulClient);

        }

    }

    @ConditionalOnClass({ Retryable.class, Aspect.class, AopAutoConfiguration.class })

    @Configuration

    @EnableRetry(proxyTargetClass = true)

    @Import(AopAutoConfiguration.class)

    @EnableConfigurationProperties(RetryProperties.class)

    protected static class RetryConfiguration {

        @Bean(name = "consulRetryInterceptor")

        @ConditionalOnMissingBean(name = "consulRetryInterceptor")

        public RetryOperationsInterceptor consulRetryInterceptor(

                RetryProperties properties) {

            return RetryInterceptorBuilder

                    .stateless()

                    .backOffOptions(properties.getInitialInterval(),

                            properties.getMultiplier(), properties.getMaxInterval())

                    .maxAttempts(properties.getMaxAttempts()).build();

        }

    }

}

/*

 * Copyright 2013-2015 the original author or authors.

 *

 * 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.springframework.cloud.consul;

import java.util.List;

import java.util.Map;

import org.springframework.boot.actuate.health.AbstractHealthIndicator;

import org.springframework.boot.actuate.health.Health;

import com.ecwid.consul.v1.ConsulClient;

import com.ecwid.consul.v1.QueryParams;

import com.ecwid.consul.v1.Response;

import com.ecwid.consul.v1.agent.model.Self;

import com.ecwid.consul.v1.agent.model.Self.Config;

import com.dxz.IoSessionHelper;

import com.dxz.utils.SpringUtil;

/**

 * @author Spencer Gibb

 */

public class PosConsulHealthIndicator extends AbstractHealthIndicator {

    private ConsulClient consul;

    private IoSessionHelper ioSessionHelper;

    public PosConsulHealthIndicator(ConsulClient consul) {

        this.consul = consul;

    }

    private IoSessionHelper getIoSessionHelper() {

        if(null == ioSessionHelper) {

            ioSessionHelper = SpringUtil.getBean("ioSessionHelper", IoSessionHelper.class);

        }

        return ioSessionHelper;

    }

    private boolean ioSessionCheck() {

        return getIoSessionHelper().ioSessionCheck();

    }

    @Override

    protected void doHealthCheck(Health.Builder builder) throws Exception {

        try {

            Response<Self> self = consul.getAgentSelf();

            Config config = self.getValue().getConfig();

            Response<Map<String, List<String>>> services = consul

                    .getCatalogServices(QueryParams.DEFAULT);

            builder

            .withDetail("services", services.getValue())

            .withDetail("advertiseAddress", config.getAdvertiseAddress())

            .withDetail("datacenter", config.getDatacenter())

            .withDetail("domain", config.getDomain())

            .withDetail("nodeName", config.getNodeName())

            .withDetail("bindAddress", config.getBindAddress())

            .withDetail("clientAddress", config.getClientAddress());

            if(ioSessionCheck()) {

                builder.up();

            } else {

                builder.outOfService()

                .withDetail("description", "ioSession not available");

            }

        }

        catch (Exception e) {

            builder.down(e);

        }

    }

}