PingKai Logo下载

使用 Sequelize 连接到 TiDB

TiDB 是一个兼容 MySQL 的数据库。Sequelize 是当前流行的 Node.js ORM 框架之一。

本文档将展示如何使用 TiDB 和 Sequelize 来构造一个简单的 CRUD 应用程序。

  • 配置你的环境。
  • 使用 Sequelize 连接到 TiDB。
  • 构建并运行你的应用程序。你也可以参考示例代码片段,完成基本的 CRUD 操作。

前置需求

为了能够顺利完成本教程,你需要提前:

  • 在你的机器上安装 Node.js 18.x 或以上版本。
  • 在你的机器上安装 Git
  • 准备一个 TiDB 集群。

如果你还没有 TiDB 集群,可以按照以下方式创建:

运行代码并连接到 TiDB

本小节演示如何运行示例应用程序的代码,并连接到 TiDB。

第 1 步:克隆示例代码仓库到本地

运行以下命令,将示例代码仓库克隆到本地:

git clone git@github.com:tidb-samples/tidb-nodejs-sequelize-quickstart.git
cd tidb-nodejs-sequelize-quickstart

第 2 步:安装依赖

运行以下命令,安装示例代码所需要的依赖(包括 sequelize):

npm install

第 3 步:配置连接信息

根据不同的 TiDB 部署方式,使用不同的方法连接到 TiDB。

  1. 在 TiDB Cloud 的 My TiDB 页面中,选择你的 TiDB Cloud Starter 或 Essential 实例,进入实例的 Overview 页面。

  2. 点击右上角的 Connect 按钮,将会弹出连接对话框。

  3. 确认对话框中的选项配置和你的运行环境一致。

    • Connection TypePublic
    • Branch 选择 main
    • Connect With 选择 General
    • Operating System 为运行示例代码所在的操作系统。
  4. 如果你还没有设置密码,点击 Generate Password 按钮生成一个随机的密码。

  5. 运行以下命令,将 .env.example 复制并重命名为 .env

    cp .env.example .env
  6. 编辑 .env 文件,按照如下格式设置连接信息,将占位符 {} 替换为从连接对话框中复制的参数值:

    TIDB_HOST='{host}'
    TIDB_PORT='4000'
    TIDB_USER='{user}'
    TIDB_PASSWORD='{password}'
    TIDB_DB_NAME='test'
    TIDB_ENABLE_SSL='true'
  7. 保存 .env 文件。

第 4 步:运行代码并查看结果

运行以下命令,执行示例代码:

npm start

示例代码片段

你可参考以下关键代码片段,完成自己的应用开发。

完整代码及其运行方式,见代码仓库 tidb-samples/tidb-nodejs-sequelize-quickstart

连接到 TiDB

下面的代码使用环境变量中定义的连接选项来建立与 TiDB 集群的连接。

// src/lib/tidb.ts
import { Sequelize } from 'sequelize';

export function initSequelize() {
  return new Sequelize({
    dialect: 'mysql',
    host: process.env.TIDB_HOST || 'localhost',     // TiDB host, for example: {gateway-region}.aws.tidbcloud.com
    port: Number(process.env.TIDB_PORT) || 4000,    // TiDB port, default: 4000
    username: process.env.TIDB_USER || 'root',      // TiDB user, for example: {prefix}.root
    password: process.env.TIDB_PASSWORD || 'root',  // TiDB password
    database: process.env.TIDB_DB_NAME || 'test',   // TiDB database name, default: test
    dialectOptions: {
      ssl:
        process.env?.TIDB_ENABLE_SSL === 'true'     // (Optional) Enable SSL
          ? {
              minVersion: 'TLSv1.2',
              rejectUnauthorized: true,
              ca: process.env.TIDB_CA_PATH          // (Optional) Path to the custom CA certificate
                ? readFileSync(process.env.TIDB_CA_PATH)
                : undefined,
            }
          : null,
    },
}

export async function getSequelize() {
  if (!sequelize) {
    sequelize = initSequelize();
    try {
      await sequelize.authenticate();
      logger.info('Connection has been established successfully.');
    } catch (error) {
      logger.error('Unable to connect to the database:');
      logger.error(error);
      throw error;
    }
  }
  return sequelize;
}

插入数据

下面的查询会创建一条单独的 Players 记录,并返回一个 Players 对象:

logger.info('Creating a new player...');
const newPlayer = await playersModel.create({
  id: 6,
  coins: 600,
  goods: 600,
});
logger.info('Created a new player.');
logger.info(newPlayer.toJSON());

更多信息参考插入数据

查询数据

下面的查询会返回一条 Players 记录,其金币数量大于 300

logger.info('Reading all players with coins > 300...');
const allPlayersWithCoinsGreaterThan300 = await playersModel.findAll({
  where: {
    coins: {
      [Op.gt]: 300,
    },
  },
});
logger.info('Read all players with coins > 300.');
logger.info(allPlayersWithCoinsGreaterThan300.map((p) => p.toJSON()));

更多信息参考查询数据

更新数据

下面的查询会将 ID 为 6Player 的金币数量和物品数量设置为 700,这个记录是在插入数据部分创建的:

logger.info('Updating the new player...');
await newPlayer.update({ coins: 700, goods: 700 });
logger.info('Updated the new player.');
logger.info(newPlayer.toJSON());

更多信息参考更新数据

删除数据

下面的查询会删除在插入数据部分创建的 Player 记录,其 ID 为 6

logger.info('Deleting the new player...');
await newPlayer.destroy();
const deletedNewPlayer = await playersModel.findByPk(6);
logger.info('Deleted the new player.');
logger.info(deletedNewPlayer?.toJSON());

更多信息参考删除数据

下一步

需要帮助?