com.playtika.testcontainer.common.spring.AbstractDependsOnPostProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of testcontainers-common Show documentation

The newest version!

package com.playtika.testcontainer.common.spring;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanFactoryUtils;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public abstract class AbstractDependsOnPostProcessor implements BeanFactoryPostProcessor {

    final Class beansOfType;
    final String[] dependsOn;

    protected AbstractDependsOnPostProcessor(Class beansOfType, String ... dependsOn) {
        this.beansOfType = beansOfType;
        this.dependsOn = dependsOn;
    }

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
        List beanNamesForType = asList(getBeanNamesForType(beanFactory));
        beanNamesForType.forEach(
                datastoreClientBeanName -> {
                    setupDependsOn(beanFactory, datastoreClientBeanName);
                }
        );
    }

    abstract protected List getDefaultDependsOn();

    private void setupDependsOn(ConfigurableListableBeanFactory beanFactory, String dataSourceBeanName) {
        BeanDefinition dataStoreClientBeanDefinition = beanFactory.getBeanDefinition(dataSourceBeanName);
        List dependsOnExisting = asList(dataStoreClientBeanDefinition.getDependsOn());
        dependsOnExisting.addAll(asList(dependsOn));
        dependsOnExisting.addAll(getDefaultDependsOn());
        dataStoreClientBeanDefinition.setDependsOn(dependsOnExisting.toArray(new String[]{}));
    }

    private String[] getBeanNamesForType(ConfigurableListableBeanFactory beanFactory) {
        boolean includeNonSingletons = true;
        boolean allowEagerInit = true;
        return BeanFactoryUtils.beanNamesForTypeIncludingAncestors(beanFactory, beansOfType, includeNonSingletons, allowEagerInit);
    }

    private static List asList(String[] array) {
        return (array == null ? new ArrayList<>() : new ArrayList<>(Arrays.asList(array)));
    }
}