Spring Boot集成AWS Keyspaces时如何配置一致性级别

问题根因 #

AWS Keyspaces 所有写入操作强制要求使用LOCAL_QUORUM一致性级别，Spring Data Cassandra默认读写一致性级别为ONE/LOCAL_ONE，与服务端强制要求不匹配，就会抛出你看到的InvalidQueryException。网上流传的大部分旧版本配置方案针对3.x版本DataStax驱动，在Spring Boot 2.7+、Spring Data Cassandra 3.x+默认集成的4.x版DataStax驱动下不生效。

可行配置方案 #

1. 全局代码配置（适配你当前手动构造CqlSession的代码结构） #

直接修改现有AppConfig类，在构造CqlSession时通过编程式配置指定全局一致性级别，同时注册CassandraTemplate避免配置被Spring默认逻辑覆盖：

import com.datastax.oss.driver.api.core.CqlSession;
import com.datastax.oss.driver.api.core.config.DriverConfigLoader;
import com.datastax.oss.driver.api.core.config.DefaultDriverOption;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.data.cassandra.core.CassandraTemplate;
import org.springframework.data.cassandra.repository.config.EnableCassandraRepositories;

@Configuration
@EnableCassandraRepositories(basePackages = "com.demo")
public class AppConfig {

    private final static String KEYSPACE = "demo";

    @Primary
    @Bean
    public CqlSession cqlSession() {
        DriverConfigLoader configLoader = DriverConfigLoader.programmaticBuilder()
                // 普通读写请求一致性设为LOCAL_QUORUM，满足Keyspaces要求
                .withString(DefaultDriverOption.REQUEST_CONSISTENCY, "LOCAL_QUORUM")
                // 轻量事务、条件写入场景用LOCAL_SERIAL
                .withString(DefaultDriverOption.REQUEST_SERIAL_CONSISTENCY, "LOCAL_SERIAL")
                .build();

        return CqlSession.builder()
                .withKeyspace(KEYSPACE)
                .withConfigLoader(configLoader)
                .build();
    }

    @Bean
    public CassandraTemplate cassandraTemplate(CqlSession cqlSession) {
        return new CassandraTemplate(cqlSession);
    }
}

2. 配置文件全局配置 #

如果移除手动构造CqlSession的逻辑，使用Spring Boot默认的Cassandra自动装配，可以直接在application.yml中添加配置实现全局生效：

spring:
  data:
    cassandra:
      keyspace-name: demo
      # 替换为你的Keyspaces所属地域，例如cn-northwest-1、us-east-1
      local-datacenter: your-keyspaces-region
      request:
        consistency: LOCAL_QUORUM
        serial-consistency: LOCAL_SERIAL

3. 细粒度单操作配置 #

如果只需要给特定接口指定一致性级别，不需要修改全局配置，可以直接在Repository接口的方法上添加注解指定：

import org.springframework.data.cassandra.repository.Consistency;
import org.springframework.data.cassandra.repository.CassandraRepository;
import com.datastax.oss.driver.api.core.ConsistencyLevel;
import java.util.List;

public interface UserRepository extends CassandraRepository<User, Integer> {
    @Consistency(ConsistencyLevel.LOCAL_QUORUM)
    <S extends User> S save(S entity);

    @Consistency(ConsistencyLevel.LOCAL_QUORUM)
    List<User> findAll();
}

额外注意事项 #

AWS Keyspaces不支持ONE、LOCAL_ONE级别的写入操作，读操作使用LOCAL_QUORUM也能避免读取到不一致的数据。你当前createUser方法中通过findAll()拉取全表数据判断手机号重复的逻辑存在严重性能问题，数据量超过千条后就会触发全表扫描超时，建议给phone字段创建二级索引，或者调整表主键结构用手机号作为查询条件直接查单条数据判断重复，禁止全表扫描。

内容的提问来源于stack exchange，提问作者Michael Torres Ospina