Spring Data Cassandra连接断开无异常抛出，如何检测连接状态？

我之前也遇到过一模一样的问题——Spring Data Cassandra默认会默默重试连接，业务代码调用Repository时完全感知不到连接已经断了。下面是几个我实际验证过的有效方案，你可以根据自己的场景选择：

1. 注册Cassandra Session状态监听器 #

Spring Data Cassandra底层依赖Datastax Java Driver，我们可以直接利用Driver提供的Session.StateListener来监听连接状态变化。这个监听器能在连接断开（onDown）、重新恢复（onUp）时触发回调，非常适合做后台告警或者状态记录。

示例代码：

@Component
public class CassandraConnectionListener implements Session.StateListener {

    private static final Logger log = LoggerFactory.getLogger(CassandraConnectionListener.class);

    @Autowired
    private CassandraTemplate cassandraTemplate;

    @PostConstruct
    public void registerListener() {
        cassandraTemplate.getSession().register(this);
    }

    @Override
    public void onUp(Session session) {
        log.info("Cassandra connection restored successfully");
        // 这里可以加告警恢复、状态更新到监控系统等逻辑
    }

    @Override
    public void onDown(Session session) {
        log.error("Critical: Cassandra connection lost!");
        // 触发邮件/短信告警、标记服务状态等
    }

    // 其他接口方法按需实现，不需要的可以留空
    @Override
    public void onAdd(Session session, Node node) {}
    @Override
    public void onRemove(Session session, Node node) {}
    @Override
    public void onClose(Session session) {}
}

2. 自定义Spring Boot Actuator健康检查端点 #

如果你的项目用了Spring Boot Actuator，自定义一个健康检查器是最方便的运维监控方式。我们可以通过执行一个超轻量的Cassandra查询（比如查询系统时间）来验证连接是否可用，失败时直接标记健康状态为DOWN。

示例代码：

@Component
public class CustomCassandraHealthIndicator implements HealthIndicator {

    @Autowired
    private CassandraTemplate cassandraTemplate;

    @Override
    public Health health() {
        try {
            // 执行最简单的查询验证连通性
            cassandraTemplate.execute("SELECT now() FROM system.local");
            return Health.up().withDetail("cassandra", "connection active").build();
        } catch (Exception e) {
            return Health.down().withDetail("error", e.getMessage()).build();
        }
    }
}

启动项目后，访问/actuator/health就能看到Cassandra的连接状态了（记得在配置里开启Actuator的健康端点权限）。

3. 让Repository方法调用时主动抛出异常 #

如果你希望业务代码调用Repository时，连接断开能直接抛出异常，可以通过配置Driver的重试策略来改变默认的静默重试行为。默认策略会无限重试，我们可以修改为有限次数重试，超过次数后抛出异常。

在application.yml里配置：

spring:
  data:
    cassandra:
      driver:
        config:
          profile:
            default:
              retry-policy:
                # 第一次失败就直接抛出异常，不重试
                class: com.datastax.oss.driver.api.core.retry.FallthroughRetryPolicy
                # 或者用默认重试策略，设置最大重试次数
                # class: com.datastax.oss.driver.api.core.retry.DefaultRetryPolicy
                # max-retries: 3

用FallthroughRetryPolicy的话，连接失败时会直接抛出DriverException；如果用DefaultRetryPolicy，超过设定的重试次数后也会抛出异常，这样业务代码就能捕获到并处理连接断开的情况。

4. AOP切面主动检查连接状态 #

如果需要在特定Repository方法调用前强制检查连接，可以用Spring AOP做一个切面，在方法执行前验证Session的状态。

示例代码：

@Aspect
@Component
public class CassandraConnectionCheckAspect {

    @Autowired
    private CassandraTemplate cassandraTemplate;

    // 匹配你项目中所有Repository的方法
    @Before("execution(* com.yourpackage.repository.*.*(..))")
    public void checkConnectionBeforeRepositoryCall() {
        Session session = cassandraTemplate.getSession();
        if (session.getState().isDown()) {
            throw new IllegalStateException("Cassandra connection is unavailable!");
        }
    }
}

这个方法会在所有Repository方法调用前检查连接状态，不过要注意频繁检查可能会有轻微性能损耗，适合对实时性要求极高的核心业务场景。

内容的提问来源于stack exchange，提问作者venkat g