使用 Spring Data Cassandra 对多个命名空间运行查询

Question

有没有什么方法可以使用 Spring Data 对 Cassandra 中的所有键空间执行查询？

Answer 1

这个答案分为两部分：

1. 使用 Spring Data Cassandra 1.x 时，您需要为要使用的每个键空间设置单独的 CassandraTemplate 实例。
2. 在 Spring Data Cassandra 2.x 中，我们引入了SessionFactory 接口来控制使用哪个Session。我们附带routing SessionFactory support，因此您可以提供多个会话和一个鉴别器（通常是基于ThreadLocal 的东西）来选择合适的Session。

2.0 的一些示例代码如下所示：

class MyRoutingSessionFactory extends AbstractRoutingSessionFactory {

    ThreadLocal<String> lookupKey = ThreadLocal.withInitial(() -> "default-session");

    void setLookupKey(String lookupKey) {
        this.lookupKey.set(lookupKey);
    }

    @Override
    protected Object determineCurrentLookupKey() {
        return lookupKey.get();
    }
}

class MyConfig extends AbstractCassandraConfiguration {

    @Bean
    @Override
    public SessionFactory sessionFactory() {

        MyRoutingSessionFactory factory = new MyRoutingSessionFactory();
        factory.setDefaultTargetSessionFactory(getRequiredSession());

        MapSessionFactoryLookup lookup = new MapSessionFactoryLookup();

        Session myOtherSession = …;

        lookup.addSessionFactory("default-session", getRequiredSession());          
        lookup.addSessionFactory("my-other-session", myOtherSession);

        factory.setSessionFactoryLookup(lookup);

        return factory;
    }

    // …
}