源码里没有configure_TiKV源码略读-Config

32831a1f0eaaf4d9d7ad1aaf8c55fee2.png

TiKV 是一个分布式事务型的键值数据库,是TiDB的存储层,提供了满足 ACID 约束的分布式事务接口,并且通过 Raft 协议保证了多副本数据一致性以及高可用。关于TiDB、TiKV的详细介绍可以从官网查阅,这里就不多赘述了。

知乎上已经有一篇高屋建瓴的文章,由TiKV亲爹Ed写的TiKV代码初探,可以从整体了解TiKV的内部功能,不过作为喜欢阅读代码的攻城狮来说,更喜欢来一个庖丁解牛式的分析。所以我们从代码级别初略的分析一下TiKV。

Ed Huang:TiKV 源码初探​zhuanlan.zhihu.com

我们首先选择了Config这部分代码逻辑来分析,一个是相对其他功能模块来说,这部分代码没有太烧脑的算法逻辑,另一个原因是这部分代码是整个TiKV启动后马上运行的部分,是最先碰到的代码逻辑,再有就是可以通过配置代码大体了解TiKV内部那些重要的功能模块们,从而避免繁杂的细节导致窥测一斑的局限。如果想整体了解配置字段,我们可以从官网上查阅完整的配置说明文档。

TiKV Configuration File​docs.pingcap.com
f5c8e935dd93cb1ac0ae2acce78e6214.png

文章中所参考的代码是基于Oct 29, 2020 master分支来进行分析的,可能会和最新的代码有出入,读者需要按照实际情况判别。

25943bd5b90dfbeff8deccc0a48ce8ad.png

从配置流程主干流程上可以看到,系统从cmd/bin/tikv-server开始运行,进入cmd/src/setup获取配置文件参数,之后进入src/config.rs执行各个模块的配置逻辑。

3759fe4a0101dc782aeead9ba7d67441.png
Setup Config Main Flow

src/config.rs内以TiKvConfig struct为起始点,从外部读取配置信息后完成配置初始化工作,以下是TiKvConfig内部的字段。

14b2169a571659aaa87adddcdae77c6f.png

所以我们下面按照代码顺序,依次介绍每个模块的作用和配置检查逻辑所做的事情,流程图内镂空的图例是每个流程图的起始点。

readpool

这是一个独立的用于数据读取的线程池,主要解决单一线程池导致的读写性能阻塞问题,具体的设计细节可以查看Read Pool RFC。

https://github.com/tikv/rfcs/blob/master/text/2017-12-22-read-pool.md​github.com

readpool内的unifed read pool还处于试验阶段,其他两个config都是通过readpool_config宏来定义的,所以他们的逻辑都是一样的,针对并发配置做了检查。

b5b68d7801a24f6f7f43eb40e69ee99c.png

storage

从名字上就能看出这块代码主要负责存储相关的内容,打开项目代码,可以看出不光包含kv数据落盘的逻辑,还包括mvcc,txn一系列相关操作。validate部分比较简单,就是对数据存储的目录进行检查和校验,另外对4.0版本之后的优化配置也进行了检测。

f69609e1f433c3057739a918ce6be9f1.png

src/storage/config.rs, Config struct

49554e3382b867617329be15fc6218c8.png

paths and grpc

这部分配置逻辑没有独立成一个validate方法,我们作为一个整体看一下主要做了哪几件事情:

  1. 设置region拆分检查的大小=6MB
  2. 配置config目录,当空时配置cfg_path为storage.data_dir,默认配置下为"./"
  3. 配置raftdb目录,当空时配置为storage.data_dir"/raft",默认配置下为"./raft"
  4. 配置raft-engine目录,当空时配置为storage.data_dir"/raft-engine",默认配置下为"./raft-engine"
  5. 配置rocksdb目录,默认配置为storage.dat_dir"/db",也就是"./db"。但是这个路径不能和raftdb放在一起,所以会有一个检查,检查有问题会抛出"raft_store.raftdb_path can not same with storage.data_dir/db"的错误。之后会根据kv_db_path, raft_store.raftdb_path, raft_engine.config.dir对目录内的数据库文件进行检查,判断是否存在对应的数据库。

2334a190083331e66bb67c709b7109f9.png

rocksdb

从变量名字上可以知道,这部分的代码逻辑是配置核心数据库的,也就是rocksdb的配置。我们可以看到这里主要是对cf做检测,cf的命名猜测应该是column family的简称,而主要检查内容就是块大小不能大于32MB。另外还有对titan和rocksdb的unordered_write配置检查。titan是pingcap开发的一个用来减少写放大问题的rocksdb插件,titan的理论基础来自于WiscKey。而unorder_write是一个提高rocksdb写性能的配置。这里有几篇文章可以扩展阅读一下。

RocksDB Config​tikv.org
7e4b5835759cac3350a11ac594b31cea.png
How TiKV reads and writes​tikv.org
7e4b5835759cac3350a11ac594b31cea.png
Tune TiKV Performance​docs.pingcap.com
f5c8e935dd93cb1ac0ae2acce78e6214.png
https://tikv.org/docs/3.0/tasks/configure/titan/​tikv.org https://www.usenix.org/system/files/conference/fast16/fast16-papers-lu.pdf​www.usenix.org Higher write throughput with `unordered_write` feature​rocksdb.org
92021b0fbed0a2e0565fcba8a29f9d76.png

5f22543b6aa8182b6d242e0295e9b597.png

src/config.rs, DbConfig struct

35ec121220cea9b3e3e7b2be908a6fb2.png

raftdb

每一个TiKV都包含两个rocksdb实例,一个是用来存储真实数据的我们称之为kv rocksdb,就是我们上面介绍的rocksdb变量对应的配置;另一个用来存储raft log,我们称之为raft rocksdb,也就是这个raftdb变量对应的配置,用来存放multi-raft log的数据。可以参考下面的文章进行配置和调优工作。

Raftstore Config​tikv.org
7e4b5835759cac3350a11ac594b31cea.png
Tune TiKV Performance​docs.pingcap.com
f5c8e935dd93cb1ac0ae2acce78e6214.png

73a4a6688768e0906d9df88b625038e7.png

src/config.rs, RaftDbConfig struct

8fd6c078dfa724a6f736470b21304dcb.png

raft_engine

raft engine应用了另一个tikv的git库,所对应的validate逻辑也不复杂,就检验了purge阈值。但从raft engine的readme看,这是一个存放multi-raft log的数据引擎,用来解决raftdb出现的性能问题,最终会替换raftdb。从文中声称的性能测试结果来看,确实有非常大的提升。

https://github.com/tikv/raft-engine​github.com

关于multi-raft的设计和实现可以参考这篇文章。

TiKV 源码解析系列 - multi-raft 设计与实现 | PingCAP​pingcap.com
f5c8e935dd93cb1ac0ae2acce78e6214.png

9d64d9c501a1e5158d87077deb57175a.png

server

从流程图上也可以看出,服务器配置相对来说是所有配置里面算逻辑复杂的,但大体上分成几个部分:对ip地址/端口的检查,snapshot收发量检查,被调用的递归深度和时长的检查,grpc的配置检查。

02339c103df98f244d832feec02f223a.png

src/server/config.rs, Config struct

062d715e29dbf5b1a98c3987576729c1.png

raft_store

这个是所有配置逻辑里面最复杂的部分了,不过主要还是围绕raft算法进行配置,包括:Leader选举(Leader election)、日志同步(Log replication)、安全性(Safety)、日志压缩(Log compaction)、成员变更(Membership change)等。到此我们其实看到有三处地方与raft配置有关系的代码,设置raft存储目录,raftdb配置,raftstore配置,是不是可以把这些代码放在一起来维护?

以下是对于raft store逻辑的扩展阅读:

https://tikv.org/docs/dev/tasks/configure/raftstore/​tikv.org
7e4b5835759cac3350a11ac594b31cea.png
Raftstore Config​tikv.org
7e4b5835759cac3350a11ac594b31cea.png

关于raft算法这里就不详细讲解了,需要深入了解的同学可以参考这篇知乎。

祥光:Raft算法详解​zhuanlan.zhihu.com
93d889ce75b97fdfe8e8a6a6648d0580.png

当然pingcap在raft paper的基础上做了很多优化措施,具体细节可以参考这篇文章。

TiKV 功能介绍 - Raft 的优化 | PingCAP​pingcap.com
f5c8e935dd93cb1ac0ae2acce78e6214.png

5c10b1492536338fcd9d4985a12c961a.png

component/raftstore/src/store/config.rs, Config struct

let split_size = ReadableSize::mb(coprocessor::config::SPLIT_SIZE_MB) // SPLIT_SIZE_MB=96

08b4d59b1f3301ca3ee0c8f0f6687361.png

pd

pd是placement driver的缩写,用来管理整个tikv集群,是整个集群的中央控制器,负责整个集群的调度工作。tikv内是pd client的逻辑,所以对配置的检查逻辑相对比较简单。

相关的介绍可以先看看这篇文章,后续的文章也会从代码级别来分析pd。

TiKV 源码解析系列 - multi-raft 设计与实现 | PingCAP​pingcap.com
f5c8e935dd93cb1ac0ae2acce78e6214.png

e8c6d94c2b2e06864030ad17cc63cebc.png

component/pd_client/src/config.rs, Config struct

d676d4d476b509ad3bd7350db2e5ff1e.png

coprocessor

类似于Hbase,tikv提供了一个协处理器框架来支持分布式计算。主要的功能是,每个节点在接收到分布式请求处理之后把自己负责的数据先做一次处理。这里不仅能提高整体运算效率,还能有效减少网络开销。需要具体了解coprocessor,可以参考这篇文章。

TiKV 源码解析系列文章(十四)Coprocessor 概览 | PingCAP​pingcap.com
f5c8e935dd93cb1ac0ae2acce78e6214.png

2501d21ace8094eacc9d311945ca0da9.png

components/raftstore/src/coprocessor/config.rs, Config struct

let split_size = ReadableSize::mb(96);

e2f63e3a64cabee549ac15ea8c19f010.png

security

security一看就知道是负责安全相关的逻辑,validate代码里主要检查了证书,密钥等一系列配置。security的config struct和其他模块不太一样,居然放在lib.rs里面,是不是可以考虑移出来,跟其他模块保持一致呢?

40b995be766a93f9df61b83226b93cb1.png

components/security/src/lib.rs, SecurityConfig struct

8c14e28ab818b295e18903df6e2b52e9.png

import

这里的import老实说没有能从代码引用关系找到对应代码文件,当然在浏览全项目文件之后找到了一个有近似逻辑的文件,所以这部分之后还需要跟tikv团队确认一下。

import的主要功能是磁盘加载rocksdb的文件,在代码中会看到sst这样的缩写,sst其实是rocksdb生成的文件格式名称,但由于不同的应用场景,这种文件格式其实有好几种结构:block-based table, plain table, cuckoo table和index block。

block-based table是sst文件的默认格式结构,这种格式下默认block大小为4kb,所有存储在文件内的key都是被排序过的,所以利用二叉搜索算法能够快速搜索到对应的键值。具体block-based table详解和其他格式结构可以参考一下文章:

SST File FormatsLinks
Rocksdb BlockBasedTable Formathttps://github.com/facebook/rocksdb/wiki/Rocksdb-BlockBasedTable-Format
PlainTable Formathttps://github.com/facebook/rocksdb/wiki/PlainTable-Format
CuckooTable Formathttps://github.com/facebook/rocksdb/wiki/CuckooTable-Format
Index Block Formathttps://github.com/facebook/rocksdb/wiki/Index-Block-Format

validate检查的内容也很简单,就是threads和window两个变量。

4217a7a460b4708588927b0aa17acc65.png

components/sst_importer/src/config.rs, Config struct

da57f5b3032290b229d7bae6f71f8ce5.png

backup

这部应该是关于数据备份相关的配置代码,里面其实就一个字段num_threads,默认最多75%cpu来做备份的工作。

dc6c85a2850ac06095edb17334b3417b.png

pessimistic_txn

从名字上看,这是一个悲观锁事务的配置验证。悲观锁和乐观锁的概念是大学数据库课程的基本概念,这里可以看一下tikv的解释说明,这篇文章也指出tikv主要使用乐观锁,所以这里的配置检查只是保证在使用悲观锁时的锁定时长不能等于0毫秒。

https://tikv.org/deep-dive/distributed-transaction/locking/​tikv.org

c785fdc93ba0404e8d4fecf95fbd2e5d.png

src/server/lock_manager/config.rs, Config struct

1cac022d638392b6466c37e8a6e4c051.png

gc

这部分代码应该是检查垃圾回收配置的逻辑,关于垃圾回收的官方文档目前只有在2.1版本里能找到,不过在用户文档里有相关的内容。

TiDB 垃圾回收 (GC)​docs.pingcap.com
f5c8e935dd93cb1ac0ae2acce78e6214.png
垃圾回收 (GC) - GC 机制简介 - 《TiDB v4.0 用户文档》 - 书栈网 · BookStack​www.bookstack.cn
2253e725211d3f36a45fade93c94b141.png

gc的任务也比较清楚,就是清理不再需要的旧数据。这里检查的batch_keys会在src/server/gc_worker/gc_worker.rs内被用来设定批量扫描的范围,所以这个值肯定不能等于0。

// Scans at most `GcConfig.batch_keys` keys.

2e751c7a14236d0e83837fb07595e7bf.png

src/server/gc_worker/config.rs, GcConfig struct

a7a1382326bfad5a99ee48ad8d850adf.png

封面上都是我们的团队成员,上个月刚刚在集团内部获得技术创新大奖,恭喜他们~~!

最后贴上完整的config流程图,图有点大,只能看个大概 !-P

e511be43e540b701d06babf9cbab7bf8.png
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值