Mongoose Upsert

简介

在使用 MongoDB 数据库时，我们经常需要进行数据的插入和更新操作。Mongoose 是一个优秀的 MongoDB 对象模型工具，在 Node.js 环境下操作 MongoDB 数据库变得更加方便。在 Mongoose 中，upsert 是一种常用的操作方法，它可以在插入数据时如果已存在就进行更新，如果不存在就插入新数据。

本文将详细介绍 Mongoose 中 upsert 的用法，并给出示例代码和运行结果。

连接 MongoDB

在使用 Mongoose 进行操作之前，首先需要连接 MongoDB 数据库。

首先，我们需要安装 Mongoose 模块。在终端中输入以下命令：

npm install mongoose

接下来，在你的代码中引入 Mongoose 模块，并使用 mongoose.connect() 方法连接到 MongoDB 数据库。示例代码如下：

const mongoose = require('mongoose');

mongoose.connect('mongodb://localhost/test', {
  useNewUrlParser: true,
  useUnifiedTopology: true,
})
  .then(() => console.log('Connected to MongoDB'))
  .catch((err) => console.error('Failed to connect to MongoDB:', err));

定义模型和 Schema

在插入和更新数据之前，我们需要定义模型和对应的 Schema。

模型是 Mongoose 中用于操作数据库的对象。在模型中，我们可以定义操作数据库的方法，如插入、更新和查询等。

Schema 定义了集合中的字段和字段类型。我们可以在 Schema 中定义要操作集合的字段名和字段类型，并设置一些验证规则。

下面是一个简单的示例，定义了一个 User 模型，其中包含了 name 和 age 字段。

const mongoose = require('mongoose');

// 定义 Schema
const userSchema = new mongoose.Schema({
  name: {
    type: String,
    required: true, // 必填字段
  },
  age: {
    type: Number,
    required: true,
    min: 18, // 最小值为 18
  },
});

// 定义模型
const User = mongoose.model('User', userSchema);

使用 upsert 插入或更新数据

使用 upsert 方法可以在插入数据时如果已存在就进行更新，如果不存在就插入新数据。

Model.updateOne(条件, 更新内容, { upsert: true })

• Model：指定要操作的模型。
• 条件：使用 Mongoose 的查询语法定义操作的条件。
• 更新内容：通过 Mongoose 的更新操作符定义要更新的字段和对应的值。
• { upsert: true }：指定开启 upsert 操作。

注意，在使用 upsert 方法时，必须使用 Model.updateOne()，而不是 Model.updateMany() 或 Model.update()。

下面是一个示例代码，实现了在用户表中插入或更新一条数据的功能：

User.updateOne(
  { name: 'Bob' },
  { age: 25 },
  { upsert: true },
)
  .then((result) => console.log('Upsert successful:', result))
  .catch((err) => console.error('Upsert failed:', err));

运行以上代码后，如果数据库中已存在名为 “Bob” 的用户，将会更新该用户的年龄为 25；如果数据库中不存在名为 “Bob” 的用户，则会插入一条新数据，名字为 “Bob”，年龄为 25。

运行结果

在执行 upsert 操作后，updateOne() 方法返回一个 Promise 对象。通过该 Promise 对象的回调函数可以获取到 upsert 操作的结果。

当 upsert 操作成功时，Promise 对象的回调函数将会被调用，并且传入的参数 result 中包含了执行 upsert 操作的信息。

当 upsert 操作失败时，Promise 对象的回调函数将会被调用，并且通过 err 参数获取到错误信息。

下面是一个示例代码的运行结果：

Upsert successful: { n: 1, nModified: 0, upserted: [ { index: 0, _id: 609fa0f32d9a5f3084e03e53 } ], ...

可以看到，upsert 操作成功后，n 字段为 1，表示有 1 条数据被更新或插入，nModified 字段为 0，表示没有数据被修改。

总结

在本文中，我们详细介绍了 Mongoose 中 upsert 的用法，并给出了示例代码和运行结果。使用 upsert 方法可以在插入数据时如果已存在就进行更新，如果不存在就插入新数据。通过该方法，我们可以更方便地操作 MongoDB 数据库。