快捷導(dǎo)航

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

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

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

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

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

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

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

a. 創(chuàng)建collectModel

在model/collectModel.js文件中，我們定義了一個(gè)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. 導(dǎo)出模型

接下來(lái)，在model/index.js中導(dǎo)出我們的collectModel。

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

2. 實(shí)現(xiàn)業(yè)務(wù)邏輯

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

a. 添加路由

在video.js路由文件中，我們添加了一個(gè)用于處理收藏請(qǐng)求的路由。

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

這里使用了verifyToken()中間件來(lái)驗(yàn)證用戶身份。

b. 編寫(xiě)控制器

在控制器中，我們處理了查找視頻、檢查是否已收藏以及創(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)收藏，則返回相應(yīng)的錯(cuò)誤信息。否則，我們創(chuàng)建一個(gè)新的收藏記錄并保存。