LotusDB 是一个全新的 KV 存储引擎,Github 地址:https://github.com/flower-corp/lotusdb,希望大家多多支持呀,点个 star 或者参与进来!
WAL 是 Write Ahead Log
的简称,通常叫做预写日志,是为了预防内存崩溃,保证数据不丢失的常用手段。WAL 是 LSM 存储模型中重要的组件,在 LotusDB 当中的重要性是一样的。
design-overview.png
试想一下,如果没有 WAL,写入的数据直接到内存的话,由于内存是易失性的,崩溃之后数据无法恢复,如果数据写到一半发生了这种情况,会造成数据不一致甚至丢失,在一个系统底层的存储引擎当中,这通常是不可接受的。
日志结构的数据文件一般是追加写的,WAL 也是一样。在 LotusDB 当中写入 k/v 时,会先将数据封装成一条日志项 LogEntry,并将其追加到 WAL,日志项 LogEntry 的结构体定义如下:
// LogEntry is the data will be appended in log file.
type LogEntry struct {
Key []byte
Value []byte
ExpiredAt int64 // time.Unix
Type EntryType
}
写入前需要将 LogEntry 结构体进行编码,然后再追加到文件中,编码后主要包含 key、value 信息,还有对应的长度 size 信息,以及对整条数据有效性做校验的 crc 值,格式如下:
// EncodeEntry will encode entry into a byte slice.
// The encoded Entry looks like:
// +-------+--------+----------+------------+-----------+-------+---------+
// | crc | type | key size | value size | expiresAt | key | value |
// +-------+--------+----------+------------+-----------+-------+---------+
// |------------------------HEADER----------------------|
// |--------------------------crc check---------------------------|
编码后的数据写到 WAL 时,LotusDB 提供了两种 IO 模式:系统标准 IO 和 mmap,可在打开数据库时通过配置项进行选择,LogFile 结构体定义的 IoSelector 负责实现:
// LogFile is an abstraction of a disk file, entry`s read and write will go through it.
type LogFile struct {
sync.RWMutex
Fid uint32
WriteAt int64
IoSelector ioselector.IOSelector
}
LogFile 是操作文件数据读写的结构体,其中最重要的是 IoSelector,它是一个 interface,负责具体的读写操作,具体的实现有 FileIO 和 MMap。
// IOSelector io selector for fileio and mmap, used by wal and value log right now.
type IOSelector interface {
// Write a slice to log file at offset.
// It returns the number of bytes written and an error, if any.
Write(b []byte, offset int64) (int, error)
// Read a slice from offset.
// It returns the number of bytes read and any error encountered.
Read(b []byte, offset int64) (int, error)
// Sync commits the current contents of the file to stable storage.
// Typically, this means flushing the file system's in-memory copy
// of recently written data to disk.
Sync() error
// Close closes the File, rendering it unusable for I/O.
// It will return an error if it has already been closed.
Close() error
// Delete delete the file.
// Must close it before delete, and will unmap if in MMapSelector.
Delete() error
}
LotusDB 中的一个 WAL 和 一个 memtable 绑定,从 memtable 的结构体定义就能够体现出来:
type memtable struct {
sync.RWMutex
sklIter *arenaskl.Iterator
skl *arenaskl.Skiplist
wal *logfile.LogFile
bytesWritten uint32 // number of bytes written, used for flush wal file.
opts memOptions
}
WAL 文件的大小和跟之相绑定的 memtable 的容量相关,由于 memtable 通常会有一个阈值,写满之后就关闭了,WAL 此时也不会接受新的写入,因此 WAL 文件的容量通常不会无限膨胀。memtable 中的数据被后台线程 flush 到磁盘之后,并且没有其他的错误发生,WAL 就可以被安全的删除了。
在 LotusDB 启动打开 memtable 的时候,会全量加载 WAL 中的数据,逻辑很简单,就是打开 WAL 文件,然后遍历其中的每条数据,将其应用到 memtable 的跳表数据结构当中,通过这样的方式,达到了 WAL 恢复数据的作用,参考代码如下:
for {
if entry, size, err := wal.ReadLogEntry(offset); err == nil {
offset += size
// No need to use atomic updates.
// This function is only be executed in one goroutine at startup.
wal.WriteAt += size
// build memtable...
// ...
} else {
if err == io.EOF || err == logfile.ErrEndOfEntry {
break
}
return nil, err
}
}
WAL 刷盘策略
数据写到 WAL 文件中,实际上并没有完全落到磁盘,由于操作系统的实现,可能只是到了 page cache,需要我们手动调用 Flush 才能够真正将数据持久化。
针对 Flush 的策略,LotusDB 提供了两个配置项:
一是在 WriteOptions
中的 Sync
,如果在数据写入时传递了这个 Options 并且将 Sync 设置为 true,那么写 WAL 完成后会立即 Flush,这种策略能够保证数据不丢,但性能是最差的。
二是 ColumnFamilyOptions
中的 WalBytesFlush
,表示写入累积到配置的字节后进行 Flush,可以理解为每 WalBytesFlush 个字节 Flush 一次。这种情况下,如果系统发生异常,最多丢失 WalBytesFlush 个字节的数据。
如果都不设置,则完全交给操作系统,这是默认的策略,性能是最好的,但是如果系统崩溃,丢失的数据可能是最多的。
LotusDB 也提供了手动进行刷盘的方法,用户可以在适当的时候,调用 Sync 方法进行数据刷盘持久化。
其他和 WAL 相关的配置项:
ColumnFamilyOptions:
WalMMap
是否使用 mmap 进行写入,默认:false
WriteOptions:
DisableWal
是否禁用 WAL,默认:false
LotusDB Github 地址:https://github.com/flower-corp/lotusdb
Copyright© 2013-2020
All Rights Reserved 京ICP备2023019179号-8