NutsDB迭代器使用指南：高效遍历键值数据-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00960/article/details/148577470

NutsDB迭代器使用指南：高效遍历键值数据

nutsdb 项目地址: https://gitcode.com/gh_mirrors/nut/nutsdb

概述

NutsDB作为一款高性能的键值存储数据库，其内部采用字节排序的方式存储键值数据。这种存储结构使得顺序遍历操作非常高效。本文将详细介绍NutsDB提供的多种迭代器使用方法，帮助开发者充分利用这一特性。

前缀扫描（Prefix Scan）

前缀扫描是NutsDB中最常用的遍历方式之一，特别适合处理具有相同前缀的键集合。

基本用法

err := db.View(func(tx *nutsdb.Tx) error {
    prefix := []byte("user_")
    bucket := "user_list"
    // 限制返回100条记录，从第25条开始
    if entries, _, err := tx.PrefixScan(bucket, prefix, 25, 100); err != nil {
        return err
    } else {
        for _, entry := range entries {
            fmt.Println(string(entry.Key), string(entry.Value))
        }
    }
    return nil
})

技术要点

offsetNum参数用于跳过指定数量的记录，实现分页功能
limitNum参数限制返回结果的最大数量，避免内存溢出
返回的entries是按键排序的切片，保证遍历顺序的一致性

前缀搜索扫描（Prefix Search Scan）

前缀搜索扫描在前缀扫描基础上增加了正则表达式匹配功能，适用于更复杂的查询场景。

典型应用

err := db.View(func(tx *nutsdb.Tx) error {
    prefix := []byte("user_")
    reg := "username"
    bucket := "user_list"
    if entries, _, err := tx.PrefixSearchScan(bucket, prefix, reg, 25, 100); err != nil {
        return err
    } else {
        for _, entry := range entries {
            fmt.Println(string(entry.Key), string(entry.Value))
        }
    }
    return nil
})

使用场景

查找特定模式的键，如匹配特定用户名的用户记录
实现复杂的键过滤逻辑
结合分页参数处理大数据集

范围扫描（Range Scan）

范围扫描允许开发者指定键的起始和结束范围，精确控制遍历的数据集。

示例代码

err := db.View(func(tx *nutsdb.Tx) error {
    start := []byte("user_0010001")
    end := []byte("user_0010010")
    bucket := "user_list"
    if entries, err := tx.RangeScan(bucket, start, end); err != nil {
        return err
    } else {
        for _, entry := range entries {
            fmt.Println(string(entry.Key), string(entry.Value))
        }
    }
    return nil
})

最佳实践

适用于已知键范围的精确查询
可以替代部分区间查询功能
起始和结束键都是包含在结果中的（闭区间）

全量获取（Get All）

对于小型数据集，可以直接获取整个桶的所有键值对。

实现方式

err := db.View(func(tx *nutsdb.Tx) error {
    bucket := "user_list"
    entries, err := tx.GetAll(bucket)
    if err != nil {
        return err
    }

    for _, entry := range entries {
        fmt.Println(string(entry.Key),string(entry.Value))
    }

    return nil
})

注意事项

仅适用于数据量小的桶，否则可能导致内存问题
结果已按键排序
没有分页功能，一次性返回所有数据

迭代器（Iterator）

NutsDB提供了更灵活的迭代器接口，支持正向和反向遍历。

正向迭代示例

tx, err := db.Begin(false)
iterator := nutsdb.NewIterator(tx, bucket, nutsdb.IteratorOptions{Reverse: false})
i := 0
for i < 10 {
    ok, err := iterator.SetNext()
    fmt.Println("ok, err", ok, err)
    fmt.Println("Key: ", string(iterator.Entry().Key))
    fmt.Println("Value: ", string(iterator.Entry().Value))
    i++
}
err = tx.Commit()

反向迭代示例

tx, err := db.Begin(false)
iterator := nutsdb.NewIterator(tx, bucket, nutsdb.IteratorOptions{Reverse: true})
i := 0
for i < 10 {
    ok, err := iterator.SetNext()
    fmt.Println("ok, err", ok, err)
    fmt.Println("Key: ", string(iterator.Entry().Key))
    fmt.Println("Value: ", string(iterator.Entry().Value))
    i++
}
err = tx.Commit()