用 Go 语言写的一个 kv 存储引擎

2022-03-21 09:34:24 +08:00
 roseduan

经历了大概 4 个月的打磨,LotusDB 的第一个 release 版本终于发布了,我看了下,有 200 多次 commit (接近 rosedb 一年多的 commit 次数了)。

项目地址: https://github.com/flower-corp/lotusdb

有了 rosedb 在 bitcask 模型上的实践之后,以及自己在存储这方面的一些经验积累,去年底的时候,在上班路上突然想到的一个 idea ,让我有了做一个新的 kv 存储引擎的想法。

有了想法之后便是验证,因为其实心里还是没谱,我又在 Github 上翻了翻,并没有同类型的实现。后来又找一些大佬沟通了下,证明我的想法是可行的。

这期间还发现了 Usenix Fast 上的一篇关于优化 LSM 的论文,发现论文的内容跟我的 idea 非常类似,这算是又多了一个理论依据,于是便决定开干了。

感兴趣的可以参考下论文,叫做 SLM-DB ,地址: https://www.usenix.org/conference/fast19/presentation/kaiyrakhmet

众所周知,数据存储引擎,目前最主流的两种模型是 B+ 树和 LSM 树,B+ 树在关系型数据库例如 Mysql 中应用比较广泛,而 LSM 的典型代表 rocksdb 也是大多数分布式系统数据落盘的首选。

B+ 树读性能稳定,而 LSM 写吞吐高,LotusDB 在这基础上做了一个巨大的改动,就是完全舍弃掉 LSM 中的 SST 文件,改由 B+ 树来存储索引,而 value 存放则参考了 Wisckey 和 bitcask 模型的设计,存储到单独的 value log 文件中。

LotusDB 是对 LSM 和 B+ 树的优势结合,目前并没有同类型的实现,我们应该是第一个吃螃蟹的人。

LotusDB 的架构图如下:

前台的写入和 LSM 完全一致,先写 wal 再写 memtable 。

而读取则会从 memtable 开始,如果 memtable 找到了,直接返回;没找到的话则从 B+ 树中查询索引,然后根据索引信息到 value log 中获取 value 。

大家可以先了解个大概,后续我会出一个完整的《 LotusDB 设计与实现》系列文章,全面解析 LotusDB 的架构细节以及代码实现,目前已经写了几篇待发布,欢迎关注公众号的后续更新:

再来看看 LotusDB 提供的一些基本接口,目前实现了基础的 Put 、Get 、Delete 接口,并且支持 Column Family (借鉴于 rocksdb ),以及 value log 的自动 GC 回收。

简单的使用方法如下:

package main

import (
	"github.com/flower-corp/lotusdb"
	"io/ioutil"
	"time"
)

// basic operations for LotusDB:
// put
// put with options
// get
// delete
// delete with options
func main() {
	path, _ := ioutil.TempDir("", "lotusdb")
	opts := lotusdb.DefaultOptions(path)
	db, err := lotusdb.Open(opts)
	if err != nil {
		panic(err)
	}
	defer db.Close()

	// 1.----put----
	key1 := []byte("name")
	err = db.Put(key1, []byte("lotusdb"))
	if err != nil {
		// ...
	}

	key2 := []byte("feature")
	// 2.----put with options----
	writeOpts := &lotusdb.WriteOptions{
		Sync:      true,
		ExpiredAt: time.Now().Add(time.Second * 100).Unix(),
	}
	err = db.PutWithOptions(key2, []byte("store data"), writeOpts)
	if err != nil {
		// ...
	}

	// 3.----get----
	val, err := db.Get(key1)
	if err != nil {
		// ...
	}
	if len(val) > 0 {
		// ...
	}

	// 4.----delete----
	err = db.Delete(key1)
	if err != nil {
		// ...
	}

	// 5.----delete with options----
	deleteOpts := &lotusdb.WriteOptions{
		Sync:       false,
		DisableWal: true,
	}
	err = db.DeleteWithOptions([]byte("dummy key"), deleteOpts)
	if err != nil {
		// ...
	}
}

目前自认为 LotusDB 的代码质量比之前的 RoseDB 好多了,单元测试更加完备,注释清晰,代码也更加简洁规范,如果你是 Go 新手,或者准备学习 Go ,也能够把项目当做练习素材,自己对照着来学习。

当然我们的愿景还是打造一个能够在生产环境中实际落地的存储引擎,目前的版本只是一个开始,后续还会有非常多的工作,包括但不限于:

如果大家在使用或者学习的过程中,有任何问题都可以微信联系我:

觉得有帮助的话,欢迎给项目点个 star 支持下

项目地址: https://github.com/flower-corp/lotusdb

3721 次点击
所在节点    分享创造
16 条回复
shisang
2022-03-21 09:49:48 +08:00
lsm + B+ 整合感觉怪怪的,一个 append-only , 一个 replace update ,lsm 就相当于一个 buffer ?
roseduan
2022-03-21 10:21:55 +08:00
@shisang B+ tree 只存储索引信息,其实也有类似的实现,rust 的 sled 比较类似
linglin0924
2022-03-21 10:40:08 +08:00
已 star ,留着日后看数据库学习。
calano
2022-03-21 11:09:52 +08:00
以前研究过 boltDB ,这个也学习下
roseduan
2022-03-21 11:41:21 +08:00
@calano 项目也用到了 boltdb 存索引
roseduan
2022-03-21 11:41:40 +08:00
@linglin0924 感谢支持
dilu
2022-03-21 12:35:50 +08:00
支持 duan 大佬,加油
Sailwww
2022-03-21 12:59:02 +08:00
支持
roseduan
2022-03-21 14:21:15 +08:00
@dilu 感谢
roseduan
2022-03-21 14:21:25 +08:00
@Sailwww 感谢
dongcidaci
2022-03-21 14:21:30 +08:00
支持一下,感觉最近流行造 lsm tree 的轮子啊,前不久才看到一个类似的项目
BeijingBaby
2022-03-21 16:40:58 +08:00
有性能测试数据吗?和其他同类项目对比?
roseduan
2022-03-21 17:01:22 +08:00
@BeijingBaby 暂时还没来得及写
roseduan
2022-03-21 17:01:39 +08:00
@dongcidaci 哈哈,折腾下
A01514035
2022-03-22 14:41:18 +08:00
那几篇设计与实现的设计文章哪里可以看啊,想看看技术原理
roseduan
2022-03-22 16:36:52 +08:00

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/841756

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX