在Spring Data JPA存储库中使用@Primary

Question

如果有人需要在Spring Data仓库中使用@Primary：

看起来Spring Data JPA忽略了存储库上的@Primary注释。

作为解决方法，我创建了一个BeanFactoryPostProcessor，它检查给定的存储库是否具有@Primary注释，并将该bean设置为主要bean。

代码如下：

@Component
public class SpringDataPrimaryPostProcessor implements BeanFactoryPostProcessor {
    public static final String REPOSITORY_INTERFACE_PROPERTY = "repositoryInterface";


    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
        makeRepositoriesPrimary(getRepositoryBeans(beanFactory));
    }

    protected List<BeanDefinition> getRepositoryBeans(ConfigurableListableBeanFactory beanFactory) {
        List<BeanDefinition> springDataRepositoryDefinitions = Lists.newArrayList();
        for (String beanName : beanFactory.getBeanDefinitionNames()) {
            BeanDefinition beanDefinition = beanFactory.getBeanDefinition(beanName);

            String beanClassName = beanDefinition.getBeanClassName();
            try {
                Class<?> beanClass = Class.forName(beanClassName);
                if (isSpringDataJpaRepository(beanClass)) {
                    springDataRepositoryDefinitions.add(beanDefinition);
                }
            } catch (ClassNotFoundException e) {
                throw new ApplicationContextException(String.format("Error when trying to create instance of %s", beanClassName), e);
            }
        }

        return springDataRepositoryDefinitions;
    }

  protected void makeRepositoriesPrimary(List<BeanDefinition> repositoryBeans) {
    for (BeanDefinition repositoryBeanDefinition : repositoryBeans) {
        String repositoryInterface = (String) repositoryBeanDefinition.getPropertyValues().get(REPOSITORY_INTERFACE_PROPERTY);
            if (isPrimary(repositoryInterface)) {
                log.debug("Making site repository bean primary, class: {}", repositoryInterface);
                repositoryBeanDefinition.setPrimary(true);
            }
    }
}

protected boolean isSpringDataJpaRepository(Class<?> beanClass) {
    return RepositoryFactoryInformation.class.isAssignableFrom(beanClass);
}

private boolean isPrimary(String repositoryInterface) {
    return AnnotationUtils.findAnnotation(getClassSafely(repositoryInterface), Primary.class) != null;
}

    private Class<?> getClassSafely(String repositoryInterface) {
        try {
            return Class.forName(repositoryInterface);
        } catch (ClassNotFoundException e) {
            throw new ApplicationContextException(String.format("Error when trying to create instance of %s", repositoryInterface), e);
        }
    }

- rgrebski

---

你的问题是什么？ - M. Deinum

@Primary 是什么？这个注解在哪个包/软件中？ - Neil Stockton

@Primary是一个Spring注解。详见http://docs.spring.io/spring/docs/current/javadoc-api/org/springframework/context/annotation/Primary.html。 - rgrebski

无法解析方法 isSpringDataJpaRepository()。 - mkczyk

谢谢mk321，我已经更新了代码。 - rgrebski

Answer 1

我尝试将解决方案应用于具有两个Mongo存储库的Spring Boot应用程序。 但是它无法在propertyValues中找到repositoryInterface。 进一步调查发现，有一个属性可以识别存储库接口factoryBeanObjectType。

因此，需要更改方法makeRepositoriesPrimary()中的代码：

String repositoryInterface = (String) repositoryBeanDefinition.getPropertyValues().get(REPOSITORY_INTERFACE_PROPERTY);

To:

String repositoryInterface = (String) repositoryBeanDefinition.getAttribute("factoryBeanObjectType");

按预期工作。

希望这有所帮助。

- Nilesh Padwal