将MongoTemplate传递到自定义存储库实现

Question

项目配置为使用多个MongoTemplate

Mongo Ref传递为

@EnableMongoRepositories(basePackages={"com.mypackage.one"}, mongoTemplateRef="mongoTemplateOne")

@EnableMongoRepositories(basePackages={"com.mypackage.two"}, mongoTemplateRef="mongoTemplateTwo")

问题：我需要访问MongoTemplate，它是类似的标准存储库。

例如，如果MyRepo正在将MyRepoCustom接口扩展为

@Repository
interface MyRepo extends MongoRepository<MyEntity, String>, MyRepoCustom{}

MyRepoCustomImpl

@Service
    public class MyRepoCustomImpl implements MyRepoCustom{
        @Autowired
        @Qualifier("mongoTemplateOne")
        MongoTemplate mongoTmpl;

        @Override
        MyEntity myCustomNeedFunc(String arg){
            // MyImplemenation goes here
        }

}

@EnableMongoRepositories(basePackages={"com.mypackage.one"}, mongoTemplateRef="mongoTemplateThree")

问题：相反，难道没有任何方法可以让要使用的MongoTemplate根据它扩展到的Repo自动注入或解析吗？

Answer 1

MongoTemplate不由MongoRepository接口公开。它们可能会公开MongoTemplate@bean的名称，这可以为您的问题提供一个解决方案。然而，鉴于他们不这样做的事实，我将在下面提供一个可能适合您需要的例子。

首先，MongoTemplateRef引用要使用的@bean的名称，它不指定MongoTemplate的名称。

您需要提供每个MongoTemplate@bean，然后在@enableMongoRepositories注释中引用它。

由于您使用的是spring-boot，因此可以利用MongoDataAutoConfiguration类。请看一下它在这里做了什么https://github.com/spring-projects/spring-boot/blob/master/spring-boot-autociguration/src/main/java/org/springframework/boot/autociguration/data/mongo/mongodataautociguration.java。

最简单的例子是这样的。

包：com.xyz.repo（此实现依赖于MongoDataAutoConfiguration提供的配置）

@Configuration
@EnableMongoRepositories(basePackages={"com.xyz.repo"}) //mongoTemplateRef defaults to mongoTemplate
public class XyzRepoConfiguration {

}

public abstract class BaseRepo {
    @Autowired
    MongoTemplate mongoTemplate;
}

@Service
public class MyRepoCustomImpl extends BaseRepo implements MyRepoCustom {    
    @Override
    MyEntity myCustomNeedFunc(String arg){
        // access to this.mongoTemplate is present
    }
}

包装：com.abc.repo

@Configuration
@EnableMongoRepositories(basePackages={"com.abc.repo"}, mongoTemplateRef=AbcRepConfiguration.TEMPLATE_NAME)
public class AbcRepoConfiguration {
    public static final String TEMPLATE_NAME = "mongoTemplateTwo";

    @Bean(name="mongoPropertiesTwo")
    @ConfigurationProperties(prefix="spring.data.mongodb2")
    public MongoProperties mongoProperties() {
        return new MongoProperties();
    }

    @Bean(name="mongoDbFactoryTwo")
    public SimpleMongoDbFactory mongoDbFactory(MongoClient mongo, @Qualifier("mongoPropertiesTwo") MongoProperties mongoProperties) throws Exception {
        String database = this.mongoProperties.getMongoClientDatabase();
        return new SimpleMongoDbFactory(mongo, database);
    }

    @Bean(name=AbcRepoConfiguration.TEMPLATE_NAME)
    public MongoTemplate mongoTemplate(@Qualifier("mongoDbFactoryTwo") MongoDbFactory mongoDbFactory, MongoConverter converter) throws UnknownHostException {
        return new MongoTemplate(mongoDbFactory, converter);
    }
}

public abstract class BaseRepo {
    @Autowired
    @Qualifier(AbcRepoConfiguration.TEMPLATE_NAME)
    MongoTemplate mongoTemplate;
}

@Service
public class MyRepoCustomImpl extends BaseRepo implements MyRepoCustom {    
    @Override
    MyEntity myCustomNeedFunc(String arg){
        // access to this.mongoTemplate is present
    }
}

com.xyz.repo将依赖于application.properties中的spring.data.mongoDB属性。repo将依赖于application.properties中的spring.data.mongoDB2属性

我以前没有使用过abcrepoConfiguration.template_name方法，但它是在我的IDE中编译的。

如果你需要任何澄清，请告诉我。