/tidb-lite

Using tidb-lite to create a TiDB server with mocktikv mode in your application or unit test.

Primary LanguageGoApache License 2.0Apache-2.0

tidb-lite test

English README

tidb-lite 是一个库,可以方便地使用该库在 golang 应用或者单元测试中创建 mocktikv 模式的 TiDB。

提供的接口

  • func NewTiDBServer(options *Options) (*TiDBServer, error)

    创建一个 TiDB Server,使用 options 来设置 TiDB 存储数据的路径和服务的端口号。

  • func GetTiDBServer() (*TiDBServer, error)

    获取已经创建的 TiDB Server。

  • func (t *TiDBServer) CreateConn() (*sql.DB, error)

    获取一个 TiDB 的链接。

  • func (t *TiDBServer) Close()

    关闭 TiDB 服务。

  • func (t *TiDBServer) CloseGracefully()

    优雅地关闭 TiDB 服务。

使用示例

可以通过 example 路径下的示例代码了解 tidb-lite 的使用方法。

example.go 中定义了一个函数 GetRowCount 计算指定表中符合条件的数据的行数。 在 example_test.go 中使用以下代码创建 TiDB Server 并获取数据库链接:

tidbServer, err := tidblite.NewTiDBServer(tidblite.NewOptions(c.MkDir()))
c.Assert(err, IsNil)
defer tidbServer.Close()

dbConn, err := tidbServer.CreateConn()
c.Assert(err, IsNil)

然后就可以使用链接 dbConn 生成测试数据,对函数 GetRowCount 进行测试。

注意

tidb-lite 只允许同一时刻运行一个 TiDB 实例,需要保证已经创建的 TiDB 实例 close 后再创建新的实例。