org.jarbframework.migrations.MigrationsNamespaceHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jarb-migrations Show documentation
Database migration support
There is a newer version: 2.3.0
package org.jarbframework.migrations;

import org.jarbframework.migrations.liquibase.LiquibaseMigrator;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.xml.BeanDefinitionParser;
import org.springframework.beans.factory.xml.NamespaceHandlerSupport;
import org.springframework.beans.factory.xml.ParserContext;
import org.w3c.dom.Element;

public class MigrationsNamespaceHandler extends NamespaceHandlerSupport {

    @Override
    public void init() {
        registerBeanDefinitionParser("migrate", new MigrateBeanDefinitionParser());
    }

    public static class MigrateBeanDefinitionParser implements BeanDefinitionParser {

        @Override
        public BeanDefinition parse(Element element, ParserContext parserContext) {
            final String dataSourceId = element.getAttribute("data-source");
            BeanDefinitionRegistry beanRegistry = parserContext.getRegistry();
            
            // Overwrite our existing data source definition with the migrating data source
            BeanDefinition migratingDataSource = createMigratingDataSource(element, dataSourceId, beanRegistry);
            beanRegistry.registerBeanDefinition(dataSourceId, migratingDataSource);
            return null;
        }

        private BeanDefinition createMigratingDataSource(Element element, String dataSourceId, BeanDefinitionRegistry beanRegistry) {
            BeanDefinitionBuilder migratingDataSourceBuilder = BeanDefinitionBuilder.genericBeanDefinition(MigratingDataSource.class);
            migratingDataSourceBuilder.addConstructorArgValue(beanRegistry.getBeanDefinition(dataSourceId));
            addMigratorArgument(element, migratingDataSourceBuilder);
            migratingDataSourceBuilder.addPropertyValue("username", element.getAttribute("username"));
            migratingDataSourceBuilder.addPropertyValue("password", element.getAttribute("password"));
            return migratingDataSourceBuilder.getBeanDefinition();
        }

        private void addMigratorArgument(Element element, BeanDefinitionBuilder migratingDataSourceBuilder) {
            if (element.hasAttribute("migrator")) {
                migratingDataSourceBuilder.addConstructorArgReference(element.getAttribute("migrator"));
            } else {
                migratingDataSourceBuilder.addConstructorArgValue(createLiquibaseMigrator(element));
            }
        }

        private BeanDefinition createLiquibaseMigrator(Element element) {
            BeanDefinitionBuilder liquibaseMigratorBuilder = BeanDefinitionBuilder.genericBeanDefinition(LiquibaseMigrator.class);
            liquibaseMigratorBuilder.addPropertyValue("changeLogPath", element.getAttribute("path"));
            return liquibaseMigratorBuilder.getBeanDefinition();
        }

    }

}