libraries/sql-liquibase/src/main/java/org/apache/zest/library/sql/liquibase/LiquibaseService.java

/*
 * Copyright (c) 2011, Rickard Öberg. All Rights Reserved.
 * Copyright (c) 2012, Paul Merlin. All Rights Reserved.
 *
 * 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.apache.zest.library.sql.liquibase;

import java.net.ConnectException;
import java.sql.Connection;
import java.sql.SQLException;
import javax.sql.DataSource;
import liquibase.Liquibase;
import liquibase.database.DatabaseConnection;
import liquibase.database.jvm.JdbcConnection;
import liquibase.resource.ClassLoaderResourceAccessor;
import org.apache.zest.api.activation.ActivatorAdapter;
import org.apache.zest.api.activation.Activators;
import org.apache.zest.api.configuration.Configuration;
import org.apache.zest.api.injection.scope.Service;
import org.apache.zest.api.injection.scope.This;
import org.apache.zest.api.mixin.Mixins;
import org.apache.zest.api.service.ServiceComposite;
import org.apache.zest.api.service.ServiceImporterException;
import org.apache.zest.api.service.ServiceReference;
import org.apache.zest.library.sql.common.SQLUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Wrapper service for Liquibase.
 */
@Mixins( LiquibaseService.Mixin.class )
@Activators( LiquibaseService.Activator.class )
public interface LiquibaseService
        extends ServiceComposite
{

    void activateLiquibase()
            throws Exception;

    public static class Activator
            extends ActivatorAdapter<ServiceReference<LiquibaseService>>
    {

        @Override
        public void afterActivation( ServiceReference<LiquibaseService> activated )
                throws Exception
        {
            activated.get().activateLiquibase();
        }

    }

    public static abstract class Mixin
            implements LiquibaseService
    {

        private static final Logger LOGGER = LoggerFactory.getLogger( "org.qi4j.library.sql" );

        @This
        Configuration<LiquibaseConfiguration> config;

        @Service
        ServiceReference<DataSource> dataSource;

        @Override
        public void activateLiquibase()
                throws Exception
        {
            config.refresh();
            boolean enabled = config.get().enabled().get();
            if ( !enabled ) {
                return;
            }

            Connection connection = null;
            try {

                connection = dataSource.get().getConnection();
                DatabaseConnection dc = new JdbcConnection( connection );
                Liquibase liquibase = new Liquibase( config.get().changeLog().get(), new ClassLoaderResourceAccessor(), dc );
                liquibase.update( config.get().contexts().get() );

            } catch ( SQLException e ) {

                Throwable ex = e;
                while ( ex.getCause() != null ) {
                    ex = ex.getCause();
                }

                if ( ex instanceof ConnectException ) {
                    LOGGER.warn( "Could not connect to database; Liquibase should be disabled" );
                    return;
                }

                LOGGER.error( "Liquibase could not perform database migration", e );

            } catch ( ServiceImporterException ex ) {

                LOGGER.warn( "DataSource is not available - database refactoring skipped" );

            } finally {

                SQLUtil.rollbackQuietly( connection );
                SQLUtil.closeQuietly( connection );

            }
        }

    }

}