如何在Mongoose pre save钩子中生成用户名前缀关键词优化WebSocket查询

循环逻辑补全方案 #

你只需要将循环中占位的*???*替换为user.username.slice(0, i + 1)即可实现前缀截取需求。如果需要支持大小写不敏感的搜索，建议统一将用户名转为小写后再生成前缀，补全后的完整代码如下：

userSchema.pre(`save`, async function (next) {
  const user = this

  if (user.isModified(`username`)) {
    const parts = []
    // 统一转为小写，实现大小写不敏感搜索，不需要可删除此行直接用user.username
    const formattedUsername = user.username.toLowerCase()
    for (let i = 0; i < formattedUsername.length; i++) {
      // slice(0, i+1) 表示截取从第0位开始，长度为i+1的子串
      parts.push(formattedUsername.slice(0, i + 1))
    }
    user.keywords = parts
  }

  next()
})

注：当前逻辑没有异步操作，函数前的async关键字可以按需删除，不影响运行。

索引设置正确性说明 #

你当前的keywords字段索引配置是完全正确的：

• MongoDB天然支持数组类型字段的索引，设置index: true后会自动为数组中的每一个字符串元素建立独立索引，完全适配你前缀模糊搜索的需求，查询时可以直接命中索引性能很高。
• 如果你需要实现大小写不敏感的搜索，可以给索引增加collation配置进一步优化：

keywords: {
  // for live search of username
  type: [String],
  index: true,
  // 忽略大小写和重音差异，适配用户名搜索场景
  collation: { locale: 'en', strength: 2 }
}

配套查询示例 #

后续需要查询前缀匹配的用户时，直接使用如下语句即可命中索引，返回所有符合条件的用户：

// 示例：搜索用户名前缀为john的用户
User.find({ keywords: 'john' })

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