使用 java api 如何查询 tikv集群中key的数量

🪐 TiDB 技术问题
1

使用 tikv-client-java 3.1.0 如何操作tikv集群 v5.0.1 查看集群中key的数量

2

以下是使用 Java 语言和 TiKV API 获取 keys 总数的示例代码：

import org.tikv.common.TiConfiguration;
import org.tikv.common.TiSession;
import org.tikv.raw.RawKVClient;

public class TikvClientExample {
    public static void main(String[] args) {
        // 创建 TiKV 客户端
        TiConfiguration conf = TiConfiguration.createDefault();
        conf.setPdAddrs("127.0.0.1:2379"); // TiKV 集群的 PD 地址
        TiSession session = TiSession.create(conf);
        RawKVClient client = session.createRawClient();

        // 扫描整个 TiKV 集群，并统计 keys 的数量
        byte[] startKey = new byte[0];
        byte[] endKey = new byte[0];
        long limit = 0;
        boolean keyOnly = false;
        long count = 0;
        while (true) {
            // 扫描 TiKV 集群
            Iterable<org.tikv.raw.Coprocessor.KeyValue> pairs = client.scan(startKey, endKey, limit, keyOnly);
            for (org.tikv.raw.Coprocessor.KeyValue pair : pairs) {
                count++;
            }
            // 如果扫描到最后一个 key，则退出循环
            if (pairs.iterator().hasNext()) {
                startKey = pairs.iterator().next().getKey().toByteArray();
            } else {
                break;
            }
        }

        // 输出 keys 的总数
        System.out.printf("Total keys: %d\n", count);

        // 关闭 TiKV 客户端
        client.close();
        session.close();
    }
}

需要注意的是，上述示例代码使用了 TiKV 的 Java 语言客户端 API，需要在项目中添加以下依赖：

<dependency>
    <groupId>org.tikv</groupId>
    <artifactId>tikv-client</artifactId>
    <version>4.0.0</version>
</dependency>

另外，TiKV 的 Java 语言客户端 API 目前还处于开发阶段，可能存在一些限制和不稳定性，建议在生产环境中谨慎使用。

3

tikv-client-java 3.1.0 版本 好像不支持此方法

4

https://github.com/tikv/client-java/blob/v3.1.0/src/test/java/org/tikv/raw/RawKVClientTest.java#L802

我觉得你可以尝试用这个3.1.0版本测试用例里面的方法看看是否可以遍历。
rawkvclient倒是有这个测试用例。txnclient就没有了。如果你用的是txnclient可能还要再调整一下。

5

在我们这里的笔记上找到这段
Response<Pair<byte[], Long>> response = client.scanKeys(
KeyUtils.EMPTY_BYTE_ARRAY,
KeyUtils.MAX_BYTE_ARRAY,
Long.MAX_VALUE
);