Aerospike Java Client 6.0.0中RecordSet.next()无限挂起问题求助

问题分析 #

从调用栈可见，线程阻塞在ArrayBlockingQueue.take()方法，这是因为RecordSet的消费逻辑一直在等待查询线程推送数据，但6.0.0版本存在内部bug，导致查询线程未发送结束信号，进而引发无限阻塞。且该版本中totalTimeout仅覆盖查询请求的发送阶段，未覆盖后续从队列取数据的等待逻辑，所以设置的超时未生效。

解决方案 #

1. 优先升级客户端版本 #

Aerospike在6.1.0及后续版本中修复了RecordSet的阻塞bug，直接升级到最新稳定版是最彻底的解决办法。

2. 手动添加超时兜底逻辑 #

如果暂时无法升级客户端，可在循环中手动检查超时时间，避免无限阻塞：

Map<Key, Record> map = new HashMap<>();
RecordSet records = null;
final long timeoutMs = 30000; // 30秒超时
long startTime = System.currentTimeMillis();

try {
    QueryPolicy queryPolicy = aerospike.getQueryPolicy();
    queryPolicy.totalTimeout = timeoutMs;
    records = aerospike.getAerospikeClient().query(queryPolicy, stmt);

    while (records != null && records.next()) {
        // 检查是否超过设定超时时间
        if (System.currentTimeMillis() - startTime > timeoutMs) {
            throw new RuntimeException("Query timed out after " + timeoutMs + "ms");
        }
        Record record = records.getRecord();
        Key key = records.getKey();
        map.put(key, record);
    }
} catch (Exception e) {
    // 处理超时或其他异常
    System.err.println("Query failed: " + e.getMessage());
} finally {
    // 务必关闭RecordSet释放资源
    if (records != null) {
        records.close();
    }
}

3. 改用异步查询API #

异步查询不会阻塞当前线程，通过回调处理结果，从根源避免同步阻塞问题：

QueryPolicy queryPolicy = aerospike.getQueryPolicy();
queryPolicy.totalTimeout = 30000;

aerospike.getAerospikeClient().queryAsync(queryPolicy, stmt, new RecordListener() {
    private final Map<Key, Record> resultMap = new HashMap<>();

    @Override
    public void onRecord(Key key, Record record) {
        if (key != null && record != null) {
            resultMap.put(key, record);
        }
    }

    @Override
    public void onSuccess() {
        // 查询完成，处理结果集合
        System.out.println("Query completed, " + resultMap.size() + " records found");
    }

    @Override
    public void onFailure(AerospikeException exception) {
        // 处理查询失败
        System.err.println("Query failed: " + exception.getMessage());
    }
});

4. 检查并完善QueryPolicy参数 #

• 同时设置socketTimeout参数，确保网络层面的超时逻辑生效，与totalTimeout形成双重保障；
• 确认QueryPolicy实例未被复用导致参数被覆盖，建议每次查询创建独立的Policy实例（或确保复用的Policy参数配置正确）。

内容的提问来源于stack exchange，提问作者Sandeep Rastogi