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.liquibase; 018 019import liquibase.integration.spring.SpringLiquibase; 020 021import org.springframework.beans.BeansException; 022import org.springframework.beans.factory.config.BeanPostProcessor; 023import org.springframework.boot.actuate.autoconfigure.endpoint.condition.ConditionalOnEnabledEndpoint; 024import org.springframework.boot.actuate.liquibase.LiquibaseEndpoint; 025import org.springframework.boot.autoconfigure.AutoConfigureAfter; 026import org.springframework.boot.autoconfigure.EnableAutoConfiguration; 027import org.springframework.boot.autoconfigure.condition.ConditionalOnBean; 028import org.springframework.boot.autoconfigure.condition.ConditionalOnClass; 029import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; 030import org.springframework.boot.autoconfigure.liquibase.DataSourceClosingSpringLiquibase; 031import org.springframework.boot.autoconfigure.liquibase.LiquibaseAutoConfiguration; 032import org.springframework.context.ApplicationContext; 033import org.springframework.context.annotation.Bean; 034import org.springframework.context.annotation.Configuration; 035 036/** 037 * {@link EnableAutoConfiguration Auto-configuration} for {@link LiquibaseEndpoint}. 038 * 039 * @author Phillip Webb 040 * @since 2.0.0 041 */ 042@Configuration 043@ConditionalOnClass(SpringLiquibase.class) 044@AutoConfigureAfter(LiquibaseAutoConfiguration.class) 045public class LiquibaseEndpointAutoConfiguration { 046 047 @Bean 048 @ConditionalOnBean(SpringLiquibase.class) 049 @ConditionalOnMissingBean 050 @ConditionalOnEnabledEndpoint 051 public LiquibaseEndpoint liquibaseEndpoint(ApplicationContext context) { 052 return new LiquibaseEndpoint(context); 053 } 054 055 @Bean 056 @ConditionalOnBean(SpringLiquibase.class) 057 @ConditionalOnEnabledEndpoint(endpoint = LiquibaseEndpoint.class) 058 public static BeanPostProcessor preventDataSourceCloseBeanPostProcessor() { 059 return new BeanPostProcessor() { 060 061 @Override 062 public Object postProcessBeforeInitialization(Object bean, String beanName) 063 throws BeansException { 064 if (bean instanceof DataSourceClosingSpringLiquibase) { 065 ((DataSourceClosingSpringLiquibase) bean) 066 .setCloseDataSourceOnceMigrated(false); 067 } 068 return bean; 069 } 070 071 @Override 072 public Object postProcessAfterInitialization(Object bean, String beanName) 073 throws BeansException { 074 return bean; 075 } 076 077 }; 078 } 079 080}