/*
* Copyright (c) MuleSoft, Inc. All rights reserved. http://www.mulesoft.com
* The software in this package is published under the terms of the CPAL v1.0
* license, a copy of which has been included with this distribution in the
* LICENSE.txt file.
*/
package org.mule.runtime.config.spring.processors;
import org.mule.runtime.core.api.lifecycle.LifecycleState;
import org.mule.runtime.core.api.lifecycle.LifecycleStateAware;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
/**
* A {@link BeanPostProcessor} which sets a given {@link #state} into instances of {@link LifecycleStateAware}
*
* @since 3.7.0
*/
public final class LifecycleStatePostProcessor implements BeanPostProcessor {
private final LifecycleState state;
public LifecycleStatePostProcessor(LifecycleState state) {
this.state = state;
}
@Override
public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
if (bean instanceof LifecycleStateAware) {
((LifecycleStateAware) bean).setLifecycleState(state);
}
return bean;
}
@Override
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
return bean;
}
}