在Node.js中实现视频收藏功能

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)

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

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

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