条件存储库注入 - Spring 引导

Question

我有两个连接MongoDB和CouchBase的存储库接口：

public interface UserRepositoryMongo extends MongoRepository<User, Long> {
}
public interface UserRepositoryCouch extends  CouchbasePagingAndSortingRepository<User, Long> {
}

有没有办法在条件下将这些存储库@Autowire互换到UserService中？ 条件将在属性文件中： application.properties 。

**Note :**

*These repositories can have custom methods too.*

Answer 1

我们可以使用ConditionalOnProperty或ConditionalOnExpression在两个不同的存储库实现之间切换。

1. 如果我们想用简单的属性存在/不存在或属性值来控制自动装配，那么可以使用ConditionalOnProperty 。

2. 如果需要复杂的评估，那么我们可以使用ConditionalOnExpression 。

ConditionalOnProperty（属性的存在/不存在）

@Qualifier("specificRepo")
@ConditionalOnProperty("mongo.url")
public interface UserRepositoryMongo extends MongoRepository<User, Long>{
}

@Qualifier("specificRepo")   
@ConditionalOnProperty("couch.url")
public interface UserRepositoryCouch extends  CouchbasePagingAndSortingRepository<User, Long>{
}

ConditionalOnProperty（基于值）

@ConditionalOnProperty("repo.url", havingValue="mongo", matchIfMissing = true) //this will be default implementation if no value is matching
public interface UserRepositoryMongo extends MongoRepository<User, Long> {
}

@ConditionalOnProperty("repo.url", havingValue="couch")
public interface UserRepositoryCouch extends  CouchbasePagingAndSortingRepository<User, Long> {
}

条件表达式

@ConditionalOnExpression("#{'${repository.url}'.contains('couch')}")
public interface UserRepositoryCouch extends  CouchbasePagingAndSortingRepository<User, Long> {
}

更新

根据您的要求使用CrudRepository/Repository类型注入。

public class DemoService {

    @Autowired
    @Qualifier("specificRepo")
    private CrudRepository repository;
}

根据创建的 bean， UserRepositoryMongo或UserRepositoryCouch将被自动装配。 确保只实例化一个 bean 以避免歧义错误。

Answer 2

解决这个问题的另一个尝试。

根据该属性，自动装配的 customRepository 将是 SimpleMongoRepository 或 SimpleCouchRepository 的一个实例。

    public interface CustomRepository extends CrudRepository<User, Long> {
        User findByLastName(String lastName); //custom methods
    }

    @Qualifier("specificRepo")
    @ConditionalOnProperty("mongo.url")
    public interface UserRepositoryMongo extends MongoRepository<User, Long>, CustomRepository{
    }

    @Qualifier("specificRepo")   
    @ConditionalOnProperty("couch.url")
    public interface UserRepositoryCouch extends  CouchbasePagingAndSortingRepository<User, Long>, CustomRepository{
    }

    public class UserService {

        @Autowired
        @Qualifier("specificRepo")
        private CustomRepository repository;
    }