18 使用 Sequelize 操作数据库

在上一篇教程中，我们介绍了如何连接和操作 MongoDB 数据库。这一篇，我们将聚焦于使用 Sequelize 操作关系型数据库，例如 MySQL、PostgreSQL、SQLite 等。Sequelize 是一个基于 Promise 的 Node.js ORM（对象关系映射）库，它简化了与数据库间的交互，使得我们可以通过 JavaScript 代码方便地进行数据操作。

1. 安装 Sequelize 和数据库驱动

首先，我们需要安装 Sequelize 和相应的数据库驱动。以 MySQL 为例，运行以下命令安装：

npm install sequelize mysql2

如果你使用其他数据库，例如 PostgreSQL，则可以安装 pg 和 pg-hstore：

npm install sequelize pg pg-hstore

2. 初始化 Sequelize 实例

我们可以通过以下方式初始化 Sequelize 实例：

const { Sequelize } = require('sequelize');

// 创建 Sequelize 实例，连接数据库
const sequelize = new Sequelize('数据库名称', '用户名', '密码', {
  host: 'localhost',
  dialect: 'mysql' // 或者 'postgres', 'sqlite', 'mariadb' 等
});

在上述代码中，请将 '数据库名称'、'用户名' 和 '密码' 替换为你自己的数据库信息。

3. 测试数据库连接

在使用 Sequelize 操作数据库之前，我们应该确认数据库连接是否成功。可以使用以下代码进行测试：

async function testConnection() {
  try {
    await sequelize.authenticate();
    console.log('数据库连接成功');
  } catch (error) {
    console.error('无法连接到数据库:', error);
  }
}

testConnection();

4. 定义模型

在 Sequelize 中，我们可以通过定义模型来描述数据库表。接下来，我们创建一个用户模型 User，它将对应于数据库中的 users 表。

const { DataTypes } = require('sequelize');

// 定义用户模型
const User = sequelize.define('User', {
  name: {
    type: DataTypes.STRING,
    allowNull: false,
  },
  email: {
    type: DataTypes.STRING,
    allowNull: false,
    unique: true,
  },
  password: {
    type: DataTypes.STRING,
    allowNull: false,
  },
}, {
  tableName: 'users', // 实际表名
  timestamps: true, // 自动创建时间戳字段 createdAt 和 updatedAt
});

5. 同步模型

一旦定义好模型，可以使用 sync 方法创建对应的数据库表：

async function syncDatabase() {
  await User.sync();
  console.log('数据库表已同步');
}

syncDatabase();

6. 数据库操作示例

现在我们可以进行基本的数据库操作，例如 CRUD（创建、读取、更新、删除）。以下是一些示例。

6.1 创建记录

你可以使用 create 方法创建一条用户记录：

async function createUser() {
  const newUser = await User.create({
    name: '张三',
    email: 'zhangsan@example.com',
    password: 'password123',
  });
  console.log('用户创建成功:', newUser.toJSON());
}

createUser();

6.2 读取记录

使用 findAll 方法可以读取所有用户记录：

async function getAllUsers() {
  const users = await User.findAll();
  console.log('所有用户:', users.map(user => user.toJSON()));
}

getAllUsers();

6.3 更新记录

更新用户记录非常简单，只需通过 update 方法：

async function updateUser(id) {
  const [updated] = await User.update({ name: '李四' }, {
    where: { id: id },
  });
  if (updated) {
    console.log(`用户ID为${id}的用户已更新`);
  } else {
    console.log(`用户ID为${id}的用户未找到`);
  }
}

updateUser(1); // 假设更新ID为1的用户

6.4 删除记录

删除用户记录使用 destroy 方法：

async function deleteUser(id) {
  const deleted = await User.destroy({
    where: { id: id },
  });
  if (deleted) {
    console.log(`用户ID为${id}的用户已删除`);
  } else {
    console.log(`用户ID为${id}的用户未找到`);
  }
}

deleteUser(1); // 假设删除ID为1的用户

7. 处理查询结果

在处理查询结果时，我们可以使用各种查询选项来满足需求。例如，使用 where 子句进行条件查询：

async function getUserByEmail(email) {
  const user = await User.findOne({ where: { email: email } });
  if (user) {
    console.log('找到用户:', user.toJSON());
  } else {
    console.log('用户未找到');
  }
}

getUserByEmail('zhangsan@example.com');

结语

在这一篇中，我们介绍了如何使用 Sequelize 操作数据库，包括连接数据库、定义模型、执行基本的 CRUD 操作等。这些操作为后续的功能开发提供了坚实的基础。

在下一篇教程中，我们将讨论部署与优化，并探讨如何选择合适的部署平台，让你的 Node.js 应用能够在生产环境中高效运行。请继续关注！