funkygao/cp-ddd-framework

package io.github.dddplus.mybatis;

import org.apache.ibatis.reflection.factory.DefaultObjectFactory;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

import java.util.List;

/**
 * 为repo里获取的(对象，关联对象)赋予Spring autoware功能.
 *
 * <p>这样，HasMany/HasOne 在 infrastructure 的实现类就可以自动注入其他对象，例如DAO</p>
 * <p>需要配置mybatis-config.xml</p>
 * <pre>
 * {@code
 *
 * <objectFactory type="io.github.dddplus.mybatis.AutowireObjectFactory" />
 * }
 * </pre>
 */
@Component
public class AutowiredObjectFactory extends DefaultObjectFactory implements ApplicationContextAware {
    private ApplicationContext context;
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.context = applicationContext;
    }

    @Override
    public <T> T create(Class<T> type) {
        T object = super.create(type);
        context.getAutowireCapableBeanFactory().autowireBean(object);
        return object;
    }

    @Override
    public <T> T create(Class<T> type, List<Class<?>> constructorArgTypes, List<Object> constructorArgs) {
        T object = super.create(type, constructorArgTypes, constructorArgs);
        context.getAutowireCapableBeanFactory().autowireBean(object);
        return object;
    }
}