如何为 Instrumentation 扩展 cassandra SessionManager

Question

我想检测通过 Cassandra 中的 datastax java 驱动程序执行的查询。是扩展 com.datastax.driver.core.SessionManager 的方式吗？此外，如果扩展 SessionManager 我必须通过修改集群源在 com.datastax.driver.core.Cluster 中实例化它吗？ 这是最好的方法还是有其他可能的方法来实现这一目标？

Answer 1

我只是通过定时调用 session.execute() 来收集指标：

private Session session;
private AtomicLongMap<String> executeCount;
private AtomicLongMap<String> executeTime;

....

@Override
public ResultSet execute(BoundStatement statement, Object... bindVariables) {
    try {
        long startTime = System.currentTimeMillis();
        ResultSet resultSet = session.execute(statement);
        long endTime = System.currentTimeMillis();
        executeCount.incrementAndGet(command);
        executeTime.addAndGet(command, endTime - startTime);
        return resultSet;
    } catch (IllegalArgumentException e) {
        LOG.error("IllegalArgumentException executing:" + command + ":" + Arrays.toString(bindVariables));
        throw e;
    }
}