Go 用 BadgerDB 实现本地 KV 存储方法

BadgerDB 是 Go 语言中高性能、纯 Go 编写的本地 KV 存储，但其生产级使用存在多个极易被忽视的关键陷阱：必须显式指定 Dir 和 ValueDir（哪怕路径相同）、写操作务必通过 `NewTransaction(true)` 开启写事务并正确 Commit、读取值时必须调用 `ValueCopy(nil)` 获取安全副本而非直接使用 `Value()` 返回的内部切片、且程序退出前必须且仅能调用一次 `Close()`，否则轻则数据丢失、读取为空，重则引发 panic 或文件损坏；避开这四大坑——尤其是最常被遗漏的 ValueDir 设置和 ValueCopy 调用，才能真正让 BadgerDB 稳健可靠地服务于本地存储场景。

BadgerDB 能直接用作生产级本地 KV 存储，但默认配置下容易因误关数据库或未设值类型导致数据丢失、读取为空或 panic。

打开数据库时必须指定 Dir 和 ValueDir

BadgerDB 区分键索引（Dir）和值存储（ValueDir），两者路径可以相同，但不能省略。若只传一个路径，或传空字符串，badger.Open() 会静默创建临时目录，进程退出后数据即消失。

• 正确写法：

  opt := badger.DefaultOptions("/tmp/badger")
  opt.ValueDir = "/tmp/badger"

• 错误写法：badger.Open(badger.DefaultOptions("")) 或 badger.Open(badger.DefaultOptions("/tmp/badger"))（没显式设 ValueDir）
• 若想复用同一目录，务必显式赋值，避免依赖旧版行为

写入前必须用 db.NewTransaction(true) 开启写事务

BadgerDB 不提供直接 Put 接口，所有写操作必须包裹在写事务中。漏掉 true 参数（即传 false）会得到只读事务，调用 Set 时 panic 报错 "cannot perform write on read-only transaction"。

• 写事务： txn := db.NewTransaction(true) → txn.Set([]byte("k"), []byte("v")) → txn.Commit()
• 读事务： txn := db.NewTransaction(false) → 只能用 Get，不能 Set
• 忘记 Commit() 不会报错，但数据不会落盘；建议用 defer txn.Discard() 配合 if err != nil 提前返回

读取值需手动调用 Item.ValueCopy(nil)

BadgerDB 的 Item.Value() 返回的是内部内存切片，生命周期绑定于事务。事务结束后再访问该切片会读到脏数据或 panic。必须用 ValueCopy 拷贝一份独立副本。

• 安全读法：

  item, err := txn.Get([]byte("k"))
  if err == nil {
    val, err := item.ValueCopy(nil)
    if err == nil {
      fmt.Printf("%s", val) // val 是独立字节切片
    }
  }

• 危险写法：item.Value() 直接用在 txn.Commit() 之后 → 数据可能已被覆写
• ValueCopy(nil) 内部会分配新内存，无需预估长度；传非 nil 切片可复用底层数组，但要注意容量是否足够

关闭数据库前必须调用 db.Close()，且不能重复 Close

BadgerDB 使用 mmap 和后台 goroutine 管理文件，不调用 Close() 就退出进程，会导致 value log 文件损坏、下次启动失败（报错 "manifest has unsupported version" 或 "checksum mismatch"）。重复 Close() 会 panic。

• 务必在程序退出前（如 main 结束、signal.Notify 捕获 SIGINT 后）调用一次 db.Close()
• 不要在 defer 中对全局 *badger.DB 多次 defer Close()，容易重复调用
• 如果用 panic 流程控制，需确保 Close() 在 recover 前已执行

最常被跳过的其实是 ValueDir 显式设置和 ValueCopy —— 这两个点不踩坑，BadgerDB 的本地 KV 就算稳住了。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Go 用 BadgerDB 实现本地 KV 存储方法》文章吧，也可关注golang学习网公众号了解相关技术文章。