How to add schema method in mongoose?

在 Mongoose 中添加 schema 方法是一个非常有用的功能，它允许您在模型层面上封装和重用业务逻辑。以下是如何添加自定义方法到 Mongoose schema 的步骤：

步骤 1: 定义 Schema

首先，您需要定义一个 Mongoose Schema。例如，我们定义一个简单的 User Schema：

javascript
const mongoose = require('mongoose');
const Schema = mongoose.Schema;

let UserSchema = new Schema({
    username: { type: String, required: true },
    email: { type: String, required: true },
    password: { type: String, required: true }
});

步骤 2: 添加实例方法

一旦您有了一个 Schema，您可以添加实例方法。这些方法将能被任何该模型的实例调用。

javascript
UserSchema.methods.encryptPassword = function(password) {
    // 这里是加密逻辑，例如使用 bcrypt 加密密码
    return bcrypt.hashSync(password, bcrypt.genSaltSync(8), null);
};

在这个例子中，encryptPassword 是一个实例方法，可以在任何 User 模型的实例上调用来加密密码。

步骤 3: 编译模型

定义完 Schema 和方法后，您需要编译这个模型。

javascript
const User = mongoose.model('User', UserSchema);

步骤 4: 使用方法

现在，您可以创建一个 User 实例并使用定义的方法。

javascript
let newUser = new User({
    username: 'testUser',
    email: 'test@example.com',
    password: 'mypassword'
});

// 使用实例方法
newUser.password = newUser.encryptPassword(newUser.password);

这样，您就能在创建用户的同时加密密码。这是一个将业务逻辑封装在模型方法中的好例子。

总结

通过给 Mongoose Schema 添加方法，您可以很方便地在模型层面封装业务逻辑，使代码更加模块化和重用。在开发大型应用时，这种做法可以提高开发效率和代码质量。