如何在JavaScript中按自定义大小写规则排序Java风格包名数组

优化方案 #

你的现有实现可以跑通，但完全不需要构建树形结构，直接通过自定义Array.sort比较函数就能实现需求，代码更简洁、性能更好。

核心思路

1. 优先判断排序开关：不需要排序时直接返回原数组副本，避免原地修改原数组
2. 逐段对比包名：将两个待比较的包名按.拆成数组，从最高层级到最低层级逐段对比
3. 大小写优先级判断：同层级段首字符大小写不同时，按配置决定优先级
4. 同优先级按字母序排序：大小写优先级一致时，用原生localeCompare做字母序对比
5. 前缀完全一致时短包名优先：比如foo.bar排在foo.bar.baz前，符合常规包排序逻辑

完整实现代码

// 可直接替换你现有逻辑的排序函数
function sortPackageNames(packageNames, options) {
  // 不需要排序直接返回原顺序副本
  if (!shouldOrganizeImports(options)) {
    return [...packageNames]
  }
  const isUppercaseFirst = shouldOrganizeImportsUppercaseFirst(options)
  // 拷贝数组后排序，避免修改原数组
  return [...packageNames].sort((a, b) => {
    const partsA = a.split('.')
    const partsB = b.split('.')
    const minLength = Math.min(partsA.length, partsB.length)
    // 逐段对比
    for (let i = 0; i < minLength; i++) {
      const partA = partsA[i]
      const partB = partsB[i]
      if (partA === partB) continue
      // 判断当前段首字符大小写
      const isUpperA = partA[0] === partA[0].toUpperCase()
      const isUpperB = partB[0] === partB[0].toUpperCase()
      // 大小写不同时按配置返回优先级
      if (isUpperA !== isUpperB) {
        return isUppercaseFirst 
          ? (isUpperA ? -1 : 1) 
          : (isUpperA ? 1 : -1)
      }
      // 大小写相同时按字母序对比
      const compareResult = partA.localeCompare(partB)
      if (compareResult !== 0) return compareResult
    }
    // 前缀完全一致时短包名排在前面
    return partsA.length - partsB.length
  })
}

使用示例

const importStrings = [
  "foo.bar.Zonk",
  "lorem.ipsum.acme.Rocket",
  "foo.bar.BazFoo",
  "lorem.ipsum.Blah",
  "foo.bar.baz.Gnarf",
  "lorem.ipsum.acme.Amboss",
];
// 大写优先排序
const sortedUppercaseFirst = sortPackageNames(importStrings, {
  organizeImports: true,
  uppercaseFirst: true
})
// 小写优先排序
const sortedLowercaseFirst = sortPackageNames(importStrings, {
  organizeImports: true,
  uppercaseFirst: false
})
// 保留原顺序
const originOrder = sortPackageNames(importStrings, {
  organizeImports: false
})

方案优势

• 代码量仅为原实现的1/3，无冗余的树构建、遍历、拼接逻辑，可读性和可维护性更高
• 不需要额外存储树形结构，内存占用更低，包数量大时性能优势更明显
• 完全兼容你现有逻辑中的shouldOrganizeImports、shouldOrganizeImportsUppercaseFirst配置判断，替换成本极低

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