001/* 002 * Copyright 2012-2018 the original author or authors. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016 017package org.springframework.boot.actuate.autoconfigure.cassandra; 018 019import java.util.Map; 020 021import com.datastax.driver.core.Cluster; 022 023import org.springframework.boot.actuate.autoconfigure.health.CompositeHealthIndicatorConfiguration; 024import org.springframework.boot.actuate.autoconfigure.health.ConditionalOnEnabledHealthIndicator; 025import org.springframework.boot.actuate.autoconfigure.health.HealthIndicatorAutoConfiguration; 026import org.springframework.boot.actuate.cassandra.CassandraHealthIndicator; 027import org.springframework.boot.actuate.health.HealthIndicator; 028import org.springframework.boot.autoconfigure.AutoConfigureAfter; 029import org.springframework.boot.autoconfigure.AutoConfigureBefore; 030import org.springframework.boot.autoconfigure.EnableAutoConfiguration; 031import org.springframework.boot.autoconfigure.cassandra.CassandraAutoConfiguration; 032import org.springframework.boot.autoconfigure.condition.ConditionalOnBean; 033import org.springframework.boot.autoconfigure.condition.ConditionalOnClass; 034import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; 035import org.springframework.boot.autoconfigure.data.cassandra.CassandraDataAutoConfiguration; 036import org.springframework.context.annotation.Bean; 037import org.springframework.context.annotation.Configuration; 038import org.springframework.data.cassandra.core.CassandraOperations; 039 040/** 041 * {@link EnableAutoConfiguration Auto-configuration} for 042 * {@link CassandraHealthIndicator}. 043 * 044 * @author Julien Dubois 045 * @author Stephane Nicoll 046 * @since 2.1.0 047 */ 048@Configuration 049@ConditionalOnClass({ Cluster.class, CassandraOperations.class }) 050@ConditionalOnBean(CassandraOperations.class) 051@ConditionalOnEnabledHealthIndicator("cassandra") 052@AutoConfigureBefore(HealthIndicatorAutoConfiguration.class) 053@AutoConfigureAfter({ CassandraAutoConfiguration.class, 054 CassandraDataAutoConfiguration.class, 055 CassandraReactiveHealthIndicatorAutoConfiguration.class }) 056public class CassandraHealthIndicatorAutoConfiguration extends 057 CompositeHealthIndicatorConfiguration<CassandraHealthIndicator, CassandraOperations> { 058 059 private final Map<String, CassandraOperations> cassandraOperations; 060 061 public CassandraHealthIndicatorAutoConfiguration( 062 Map<String, CassandraOperations> cassandraOperations) { 063 this.cassandraOperations = cassandraOperations; 064 } 065 066 @Bean 067 @ConditionalOnMissingBean(name = "cassandraHealthIndicator") 068 public HealthIndicator cassandraHealthIndicator() { 069 return createHealthIndicator(this.cassandraOperations); 070 } 071 072}