使用 mysql2 连接 TiDB
TiDB 是兼容 MySQL 的数据库,mysql2 是 Ruby 生态中最流行的 MySQL 驱动之一。
在本教程中,你可以学习如何使用 TiDB 和 mysql2 完成以下任务:
- 搭建你的开发环境。
- 使用 mysql2 连接到你的 TiDB 集群。
- 构建并运行你的应用程序。你还可以在 示例代码片段 中找到基本的 CRUD 操作示例。
前置条件
完成本教程,你需要:
如果你还没有 TiDB 集群,可以按如下方式创建:
- (推荐)参考 创建 TiDB Cloud Starter 集群 创建属于你自己的 TiDB Cloud 集群。
- 参考 部署本地测试 TiDB 集群 或 部署生产环境 TiDB 集群 创建本地集群。
运行示例应用连接 TiDB
本节演示如何运行示例应用代码并连接到 TiDB。
步骤 1:克隆示例应用仓库
在终端窗口中运行以下命令,克隆示例代码仓库:
git clone https://github.com/tidb-samples/tidb-ruby-mysql2-quickstart.git
cd tidb-ruby-mysql2-quickstart
步骤 2:安装依赖
运行以下命令,为示例应用安装所需的依赖包(包括 mysql2 和 dotenv):
bundle install
为已有项目安装依赖
对于你已有的项目,运行以下命令安装依赖包:
bundle add mysql2 dotenv
步骤 3:配置连接信息
根据你选择的 TiDB 部署方式,连接到你的 TiDB 集群。
进入 Clusters 页面,点击目标集群名称,进入集群概览页。
点击右上角的 Connect,弹出连接信息对话框。
确保连接对话框中的配置与你的操作环境一致。
- Connection Type 设置为
Public。 - Branch 设置为
main。 - Connect With 设置为
General。 - Operating System 与你运行应用的操作系统一致。
- Connection Type 设置为
如果你还未设置密码,点击 Generate Password 生成一个随机密码。
运行以下命令,复制
.env.example并重命名为.env:cp .env.example .env编辑
.env文件,按如下方式设置环境变量,并将对应的占位符{}替换为连接对话框中的参数:DATABASE_HOST={host} DATABASE_PORT=4000 DATABASE_USER={user} DATABASE_PASSWORD={password} DATABASE_NAME=test DATABASE_ENABLE_SSL=true保存
.env文件。
进入 Clusters 页面,点击目标集群名称,进入集群概览页。
点击右上角的 Connect,弹出连接信息对话框。
在连接对话框中,从 Connection Type 下拉列表选择 Public,然后点击 CA cert 下载 CA 证书。
如果你还未配置 IP 访问列表,点击 Configure IP Access List,或参考 配置 IP 访问列表 进行配置,以便首次连接。
除了 Public 连接类型,TiDB Cloud Dedicated 还支持 Private Endpoint 和 VPC Peering 连接类型。更多信息请参考 连接到你的 TiDB Cloud Dedicated 集群。
运行以下命令,复制
.env.example并重命名为.env:cp .env.example .env编辑
.env文件,按如下方式设置环境变量,并将对应的占位符{}替换为连接对话框中的参数:DATABASE_HOST={host} DATABASE_PORT=4000 DATABASE_USER={user} DATABASE_PASSWORD={password} DATABASE_NAME=test DATABASE_ENABLE_SSL=true DATABASE_SSL_CA={downloaded_ssl_ca_path}保存
.env文件。
运行以下命令,复制
.env.example并重命名为.env:cp .env.example .env编辑
.env文件,按如下方式设置环境变量,并将对应的占位符{}替换为你自己的 TiDB 连接信息:DATABASE_HOST={host} DATABASE_PORT=4000 DATABASE_USER={user} DATABASE_PASSWORD={password} DATABASE_NAME=test如果你在本地运行 TiDB,默认主机地址为
127.0.0.1,密码为空。保存
.env文件。
步骤 4:运行代码并检查结果
运行以下命令执行示例代码:
ruby app.rb
如果连接成功,控制台会输出 TiDB 集群的版本信息,如下所示:
🔌 Connected to TiDB cluster! (TiDB version: 8.0.11-TiDB-v8.5.3)
⏳ Loading sample game data...
✅ Loaded sample game data.
🆕 Created a new player with ID 12.
ℹ️ Got Player 12: Player { id: 12, coins: 100, goods: 100 }
🔢 Added 50 coins and 50 goods to player 12, updated 1 row.
🚮 Deleted 1 player data.
示例代码片段
你可以参考以下示例代码片段,完成你自己的应用开发。
完整示例代码及运行方式请参考 tidb-samples/tidb-ruby-mysql2-quickstart 仓库。
使用连接参数连接 TiDB
以下代码通过环境变量中定义的参数建立 TiDB 连接:
require 'dotenv/load'
require 'mysql2'
Dotenv.load # Load the environment variables from the .env file
options = {
host: ENV['DATABASE_HOST'] || '127.0.0.1',
port: ENV['DATABASE_PORT'] || 4000,
username: ENV['DATABASE_USER'] || 'root',
password: ENV['DATABASE_PASSWORD'] || '',
database: ENV['DATABASE_NAME'] || 'test'
}
options.merge(ssl_mode: :verify_identity) unless ENV['DATABASE_ENABLE_SSL'] == 'false'
options.merge(sslca: ENV['DATABASE_SSL_CA']) if ENV['DATABASE_SSL_CA']
client = Mysql2::Client.new(options)
插入数据
以下查询语句创建一个包含两个字段的玩家,并返回 last_insert_id:
def create_player(client, coins, goods)
result = client.query(
"INSERT INTO players (coins, goods) VALUES (#{coins}, #{goods});"
)
client.last_id
end
更多信息请参考 插入数据。
查询数据
以下查询语句根据 ID 返回指定玩家的记录:
def get_player_by_id(client, id)
result = client.query(
"SELECT id, coins, goods FROM players WHERE id = #{id};"
)
result.first
end
更多信息请参考 查询数据。
更新数据
以下查询语句根据 ID 更新指定玩家的记录:
def update_player(client, player_id, inc_coins, inc_goods)
result = client.query(
"UPDATE players SET coins = coins + #{inc_coins}, goods = goods + #{inc_goods} WHERE id = #{player_id};"
)
client.affected_rows
end
更多信息请参考 更新数据。
删除数据
以下查询语句删除指定玩家的记录:
def delete_player_by_id(client, id)
result = client.query(
"DELETE FROM players WHERE id = #{id};"
)
client.affected_rows
end
更多信息请参考 删除数据。
最佳实践
默认情况下,mysql2 gem 会按特定顺序自动查找本地已存在的 CA 证书文件,直到找到为止。
/etc/ssl/certs/ca-certificates.crt(适用于 Debian、Ubuntu、Gentoo、Arch 或 Slackware)/etc/pki/tls/certs/ca-bundle.crt(适用于 RedHat、Fedora、CentOS、Mageia、Vercel 或 Netlify)/etc/ssl/ca-bundle.pem(适用于 OpenSUSE)/etc/ssl/cert.pem(适用于 macOS 或 Alpine(docker 容器))
虽然你可以手动指定 CA 证书路径,但在多环境部署场景下,不同机器和环境可能存储 CA 证书的位置不同,这样做会带来较大不便。因此,推荐将 sslca 设置为 nil,以便在不同环境下灵活部署和使用。
后续步骤
- 通过 mysql2 的官方文档 学习更多 mysql2 驱动的用法。
- 通过 开发者指南 各章节,学习 TiDB 应用开发最佳实践,例如:插入数据、更新数据、删除数据、查询数据、事务 以及 SQL 性能优化。
- 通过专业的 TiDB 开发者课程,并在通过考试后获得 TiDB 认证。