视频收藏功能实现：Node.js与MongoDB的实践

在构建视频分享平台时，视频的收藏功能是用户互动的重要组成部分。本文将介绍如何在Node.js应用中实现视频收藏功能，包括数据模型的创建、业务逻辑的实现以及接口的验证测试。

1. 创建数据模型

首先，我们需要定义一个用于存储用户收藏信息的数据模型。我们将使用Mongoose来定义这个模型。

a. 创建collectModel

在model/collectModel.js文件中，我们定义了一个collectSchema，它包含了用户和视频的引用。

const mongoose = require('mongoose')
const baseModel = require('./baseModel')

const collectSchema = new mongoose.Schema({
  user: {
    type: mongoose.ObjectId,
    required: true,
    ref: "User"
  },
  video: {
    type: mongoose.ObjectId,
    required: true,
    ref: "Video"
  },
  ...baseModel
})

module.exports = mongoose.model('CollectModel', collectSchema)

b. 导出模型

接下来，在model/index.js中导出我们的collectModel。

module.exports = {
  collectModel: require('./collectModel')
}

2. 实现业务逻辑

在实现了数据模型后，我们需要添加路由和编写控制器来处理收藏逻辑。

a. 添加路由

在video.js路由文件中，我们添加了一个用于处理收藏请求的路由。

router
  .get('/collect/:videoId', verifyToken(), videoController.collect)

这里使用了verifyToken()中间件来验证用户身份。

b. 编写控制器

在控制器中，我们处理了查找视频、检查是否已收藏以及创建收藏记录的逻辑。

exports.collect = async (req, res) => {
  const videoId = req.params.videoId
  const userId = req.user.userinfo._id
  const video = await Video.findById(videoId)
  if (!video) {
    return res.status(404).json({ err: '视频不存在' })
  }
  let doc = await collectModel.findOne({
    user: userId,
    video: videoId
  })
  if (doc) {
    return res.status(401).json({ err: '视频已被收藏' })
  }
  const myCollect = await new collectModel({
    user: userId,
    video: videoId
  }).save()
  res.status(200).json({ myCollect })
}

注意，这里我们首先检查视频是否存在，然后检查用户是否已经收藏了该视频。如果用户已经收藏，则返回相应的错误信息。否则，我们创建一个新的收藏记录并保存。

3. 验证测试

最后，我们使用Postman进行接口验证，并在数据库中检查数据。

a. Postman接口验证

使用Postman，我们可以发送GET请求到/collect/:videoId来测试收藏功能。

• 收藏成功：如果操作成功，将返回新创建的收藏记录。

• 重复收藏：如果用户尝试收藏一个已经收藏的视频，将返回错误信息。

b. 数据库数据验证

通过检查MongoDB数据库，我们可以验证收藏记录是否正确创建。

结语

视频收藏功能是提升用户参与度和平台互动的关键。通过上述步骤，我们成功地在Node.js应用中实现了这一功能。这不仅为用户提供了更好的体验，也为平台的推荐算法提供了重要的数据支持。