如何在Node.js应用中实现关注列表和粉丝列表

土豆大侠 • 2024年4月14日下午9:04 • 前端开发 • 阅读 194

如何在Node.js应用中实现关注列表和粉丝列表

在社交网络或者任何需要用户交互的应用中，实现关注和被关注的功能是非常常见的需求。本文将通过一个简单的例子，展示如何在Node.js环境下，使用Express框架和Mongoose库来实现用户的关注列表和粉丝列表。

1. 创建路由

首先，我们需要为获取用户的关注列表和粉丝列表创建相应的路由。

获取关注列表

我们创建一个GET路由，通过该路由可以查询某个用户关注了哪些人：

// 在user.js文件中定义路由
router.get('/getSubscribe/:userId', userController.getSubscribe);

这里的/getSubscribe/:userId路由将接受一个用户ID作为参数，用来查找这个用户关注的所有人。

2. 编写Controller

接下来，我们需要在controller中实现具体的逻辑来处理路由请求。

关注列表的实现

在userController中，我们添加getSubscribe方法来处理获取关注列表的逻辑：

const lodash = require('lodash');
const Subscribe = require('../models/subscribe');

exports.getSubscribe = async (req, res) => {
  let subscribeList = await Subscribe.find({
    user: req.params.userId
  }).populate('channel');
  
  subscribeList = subscribeList.map(item => {
    return lodash.pick(item.channel, [
      '_id', 'username', 'image', 'subscribeCount', 'cover', 'channelDes'
    ]);
  });
  
  res.status(200).json(subscribeList);
};

在这段代码中，我们使用Subscribe.find()来查找所有该用户关注的记录，并使用.populate('channel')来填充关联的用户信息。然后我们使用lodash的pick函数从每个关注的用户中挑选出需要的信息返回。

3. Postman测试

在完成路由和控制器的编写后，我们使用Postman进行接口测试，确保我们的功能正常。

4. 实现粉丝列表

粉丝列表的实现逻辑与关注列表类似，但是在查询时我们将关注点放在了被关注者（即当前用户）。

路由定义

// 在user.js文件中添加路由
router.get('/getChannel', verifyToken(), userController.getChannel);

这里我们使用verifyToken()来确保只有验证过的用户才可以查询他们的粉丝列表。

Controller实现

exports.getChannel = async (req, res) => {
  let channelList = await Subscribe.find({
    channel: req.user.userinfo._id
  }).populate('user');
  
  channelList = channelList.map(item => {
    return lodash.pick(item.user, [
      '_id', 'username', 'image', 'subscribeCount', 'cover', 'channelDes'
    ]);
  });
  
  res.status(200).json(channelList);
};