Spring Cassandra:在运行时在另一个键空间中创建模式

时间:2019-01-24 10:28:44

标签: spring spring-boot cassandra spring-data spring-data-cassandra

我有一个带有spring-cassandra的Spring Boot 2项目,我在其中使用@RefreshScope在运行时创建键空间,并且正在执行查询以使用该键空间,但是架构不是从带有@Table批注的实体创建的。

启动应用程序时,如果不存在则创建一个新的键空间,并创建架构。之后,我更改app.properties中的键空间,并将POST请求发送到执行器/刷新。创建并使用了新的键空间,但未在该键空间中创建架构。

这是我的Cassandra配置:

@Configuration
@EnableCassandraRepositories
public class CassandraConfig extends AbstractCassandraConfiguration {

  @Value("${cassandra.port}")
  private int cassandraPort;
  @Value("${cassandra.keyspace}")
  private String KEYSPACE;
  @Value("${cassandra.contactpoints}")
  private String NODES;

  @Bean
  @Override
  public CassandraCqlClusterFactoryBean cluster() {
    CassandraCqlClusterFactoryBean bean = new CassandraCqlClusterFactoryBean();
    bean.setKeyspaceCreations(getKeyspaceCreations());
    bean.setContactPoints(NODES);
    bean.setPort(cassandraPort);
    return bean;
  }

  @Override
  public SchemaAction getSchemaAction() {
    return SchemaAction.CREATE_IF_NOT_EXISTS;
  }

  @Override
  protected String getKeyspaceName() {
    return KEYSPACE;
  }

  @Override
  protected String getContactPoints() {
    return NODES;
  }

  @Override
  protected int getPort() {
    return cassandraPort;
  }

  @Override
  public String[] getEntityBasePackages() {
    return new String[] {"package.entities"};
  }

  @Override
  protected List<CreateKeyspaceSpecification> getKeyspaceCreations() {
    List<CreateKeyspaceSpecification> createKeyspaceSpecifications = new ArrayList<>();
    createKeyspaceSpecifications.add(getKeySpaceSpecification());
    return createKeyspaceSpecifications;
  }

  // Below method creates "keyspace" if it doesn't exist.
  private CreateKeyspaceSpecification getKeySpaceSpecification() {
    DataCenterReplication dcr = DataCenterReplication.of("datacenter1", 3L);
    return CreateKeyspaceSpecification.createKeyspace(KEYSPACE)
        .ifNotExists()
        .withNetworkReplication(dcr);
  }
}

这是RefreshScope类:

@Component
class MySessionRefresh extends AbstractCassandraConfiguration{

  @Autowired
  Environment env;

  @Autowired
  Session session;

  @EventListener
  @Order(Ordered.HIGHEST_PRECEDENCE)
  public void handle(RefreshScopeRefreshedEvent event) {

    String keyspace = env.getProperty("cassandra.keyspace");
    session.execute("CREATE KEYSPACE IF NOT EXISTS " + keyspace + " WITH REPLICATION = { 'class' : 'NetworkTopologyStrategy', 'datacenter1' : 3 };");
    session.execute("USE " + keyspace + " ;");

  }

在运行时,需要使用@Table为新的键空间从实体自动创建架构的其他操作吗?

1 个答案:

答案 0 :(得分:1)

Spring Data Cassandra中的架构创建与会话工厂(CassandraSessionFactoryBean)绑定,键空间创建与Cluster实例(CassandraClusterFactoryBean)绑定。您需要在两个工厂上都放@RefreshScope

请记住,刷新会中断您的Cassandra操作。您还应注意,架构创建不适用于生产。在设计时考虑了集成测试和快速的应用程序开发。