MongoDB Mongoose 动态引用的 Activity 模型

在本文中，我们将介绍如何使用 MongoDB 和 Mongoose 构建一个具有动态引用的 Activity 模型。Activity 模型是一个常见的数据模型，用于记录用户的活动或事件。

MongoDB 简介

MongoDB 是一个流行的 NoSQL 数据库，它以文档的形式存储数据。与传统的关系型数据库不同，MongoDB 使用集合（Collection）来存储数据，而集合中的文档（Document）则类似于关系型数据库中的行。每个文档都是一个键值对的集合，可以嵌套存储数据。

Mongoose 简介

Mongoose 是一个运行在 Node.js 环境中与 MongoDB 交互的框架。它提供了一种定义数据模型的方式，并且可以通过模型进行 CRUD 操作。使用 Mongoose，我们可以定义模式（Schema），并基于模式创建模型（Model），从而方便地操作 MongoDB 数据库。

动态引用的 Activity 模型

通常，Activity 模型中的每一条记录都会有一个引用字段，用于指向用户或其他对象。在某些情况下，这个引用字段的类型是不确定的，需要在运行时确定。

为了支持动态引用的 Activity 模型，我们可以利用 Mongoose 的强大功能。首先，我们定义一个基础的 Activity 模式，其中包含了一些共同字段和方法：

const mongoose = require('mongoose');

const activitySchema = new mongoose.Schema({
  type: {
    type: String,
    required: true
  },
  createdAt: {
    type: Date,
    default: Date.now
  },
  // 其他共同字段...
});

activitySchema.methods.log = function() {
  console.log(`[ ${this.createdAt}]$ {this.type}`);
};

const Activity = mongoose.model('Activity', activitySchema);

module.exports = Activity;

上述代码创建了一个名为 Activity 的模型，包含了一个必需的 type 字段和一个默认值为当前时间的 createdAt 字段。我们还定义了一个 log 方法，用于在控制台打印出记录的时间和类型。

接下来，我们创建一个函数，它接受一个引用字段的类型作为参数，并返回一个基于 Activity 模型的新模型。这样，我们就可以根据需要创建不同类型的 Activity 模型了：

function createActivityModel(refType) {
  const activitySchema = new mongoose.Schema({
    type: {
      type: String,
      required: true
    },
    ref: {
      type: mongoose.Schema.Types.ObjectId,
      required: true,
      ref: refType // 动态引用字段类型
    },
    createdAt: {
      type: Date,
      default: Date.now
    },
    // 其他共同字段...
  });

  activitySchema.methods.log = function() {
    console.log(`[ ${this.createdAt}]$ {this.type}`);
  };

  return mongoose.model('Activity' + refType, activitySchema);
}

module.exports = createActivityModel;

上述代码通过动态设置 ref 字段的类型，创建了一个基于 Activity 模型的新模型。我们可以通过以下方式使用它：

const createActivityModel = require('./activityModel');

const UserActivity = createActivityModel('User');
const PostActivity = createActivityModel('Post');

const userActivity = new UserActivity({
  type: 'login',
  ref: user._id
});

const postActivity = new PostActivity({
  type: 'create',
  ref: post._id
});

userActivity.log(); // [2021-01-01T00:00:00Z] login
postActivity.log(); // [2021-01-02T00:00:00Z] create

在上面的示例中，我们使用 createActivityModel 函数创建了两个不同类型的 Activity 模型：UserActivity 和 PostActivity。这两个模型都继承了基础的 Activity 模型，并具有不同的引用字段类型。