在Node.js中實現(xiàn)視頻收藏功能

更新時間：2024年04月28日 08:27:55 作者：snakeshe1010

在構(gòu)建視頻分享平臺時,視頻的收藏功能是用戶互動的重要組成部分,本文將介紹如何在Node.js應用中實現(xiàn)視頻收藏功能,包括數(shù)據(jù)模型的創(chuàng)建、業(yè)務邏輯的實現(xiàn)以及接口的驗證測試,需要的朋友可以參考下

視頻收藏功能實現(xiàn)：Node.js與MongoDB的實踐

在構(gòu)建視頻分享平臺時，視頻的收藏功能是用戶互動的重要組成部分。本文將介紹如何在Node.js應用中實現(xiàn)視頻收藏功能，包括數(shù)據(jù)模型的創(chuàng)建、業(yè)務邏輯的實現(xiàn)以及接口的驗證測試。

1. 創(chuàng)建數(shù)據(jù)模型

首先，我們需要定義一個用于存儲用戶收藏信息的數(shù)據(jù)模型。我們將使用Mongoose來定義這個模型。

a. 創(chuàng)建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. 實現(xiàn)業(yè)務邏輯

在實現(xiàn)了數(shù)據(jù)模型后，我們需要添加路由和編寫控制器來處理收藏邏輯。

a. 添加路由

在video.js路由文件中，我們添加了一個用于處理收藏請求的路由。

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

這里使用了verifyToken()中間件來驗證用戶身份。

b. 編寫控制器

在控制器中，我們處理了查找視頻、檢查是否已收藏以及創(chuàng)建收藏記錄的邏輯。

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 })
}

注意，這里我們首先檢查視頻是否存在，然后檢查用戶是否已經(jīng)收藏了該視頻。如果用戶已經(jīng)收藏，則返回相應的錯誤信息。否則，我們創(chuàng)建一個新的收藏記錄并保存。