如何在Sequelize.js中实现用户表自关联的多对多好友关系

嗨，我完全懂你的困惑——跨表的多对多写多了，突然碰到自关联的场景，确实会有点懵。不过Sequelize对这种自关联多对多的支持很完善，核心就是给不同方向的关联起别名，再明确指定外键就行，这样就不会出现重复定义的问题啦！

下面一步步给你拆解实现方法：

1. 先定义模型 #

首先分别定义User和Friendship模型：

const { DataTypes, Sequelize } = require('sequelize');
const sequelize = new Sequelize('your-db-url');

// User模型
const User = sequelize.define('User', {
  id: {
    type: DataTypes.INTEGER,
    primaryKey: true,
    autoIncrement: true
  },
  username: {
    type: DataTypes.STRING,
    allowNull: false,
    unique: true
  },
  password: {
    type: DataTypes.STRING,
    allowNull: false
  },
  email: {
    type: DataTypes.STRING,
    allowNull: false,
    unique: true
  }
});

// Friendship中间表模型
const Friendship = sequelize.define('Friendship', {
  id: {
    type: DataTypes.INTEGER,
    primaryKey: true,
    autoIncrement: true
  },
  accepted: {
    type: DataTypes.BOOLEAN,
    defaultValue: false, // 默认是未通过的好友请求
    allowNull: false
  }
});

2. 建立自关联多对多关系 #

关键来了：给两个方向的关联设置不同的别名，并指定对应的foreignKey和otherKey，让Sequelize能区分开两个关联逻辑：

// 关联1：当前用户发起的好友请求（userId1是当前用户ID）
User.belongsToMany(User, {
  as: 'RequestedFriends', // 别名：我发起请求的用户
  through: Friendship,
  foreignKey: 'userId1', // 中间表中指向当前用户的外键
  otherKey: 'userId2'    // 中间表中指向对方用户的外键
});

// 关联2：当前用户收到的好友请求（userId2是当前用户ID）
User.belongsToMany(User, {
  as: 'ReceivedFriendRequests', // 别名：向我发起请求的用户
  through: Friendship,
  foreignKey: 'userId2', // 中间表中指向当前用户的外键
  otherKey: 'userId1'    // 中间表中指向对方用户的外键
});

如果需要直接查询已通过的好友，还可以额外加一个带过滤条件的关联：

// 关联3：当前用户已通过的双向好友
User.belongsToMany(User, {
  as: 'Friends',
  through: Friendship,
  foreignKey: 'userId1',
  otherKey: 'userId2',
  where: { accepted: true }, // 只筛选已通过的请求
  constraints: false // 避免自动生成可能冲突的约束，按需开启
});

3. 如何使用这些关联查询 #

现在你就可以通过别名来查询不同场景的好友关系了：

• 查询用户发起的所有好友请求（包括未通过的）：

const user = await User.findByPk(1, {
  include: [{
    model: User,
    as: 'RequestedFriends',
    through: { attributes: ['accepted'] } // 带上中间表的请求状态
  }]
});

• 查询用户收到的所有好友请求：

const user = await User.findByPk(1, {
  include: [{
    model: User,
    as: 'ReceivedFriendRequests',
    through: { attributes: ['accepted'] }
  }]
});

• 查询用户已通过的好友：

const user = await User.findByPk(1, {
  include: [{
    model: User,
    as: 'Friends'
  }]
});

关键注意点 #

• 好友关系的双向性：因为我们通过别名区分了发起和接收方，所以不需要在中间表中创建两条记录（比如userId1=1&userId2=2和userId1=2&userId2=1），只需要一条记录，通过不同的关联别名就能从两个方向查询到。
• 业务逻辑处理：当用户同意好友请求时，只需要把对应Friendship记录的accepted字段设为true即可。
• 外键约束：Sequelize会自动给Friendship表的userId1和userId2添加外键关联到User表的id，确保数据一致性。

这样就完美解决了自关联多对多的重复定义问题，而且能灵活处理好友请求的不同状态~

内容的提问来源于stack exchange，提问作者Onyx