Scala中实现多键映射同一值的最优方式

这个场景我之前也碰到过！确实手动给每个基础键都加一个带"Extra"的重复键值对太繁琐，尤其是键多的时候。这里有几个更优雅的实现方式，你可以根据需求选：

方案1：查询时用正则预处理键（最轻量化） #

这种方式不需要提前生成所有可能的键，而是在查询前把带"Extra"的键转换成基础键，再去普通Map里查找。好处是节省内存，逻辑也清晰：

// 定义键的标准化函数：移除末尾的" Extra"
def normalizeKey(key: String): String = {
  // 用正则匹配末尾的空格+Extra，确保只替换结尾的部分
  key.replaceAll("\\s+Extra$", "")
}

// 只需要维护基础键的映射
val baseMap = Map(
  "Test 1" -> 1,
  "Test 2" -> 2,
  "Demo" -> 3
)

// 封装查询方法
def getValue(key: String): Option[Int] = {
  baseMap.get(normalizeKey(key))
}

// 测试一下
getValue("Test 1")        // 返回 Some(1)
getValue("Test 1 Extra")  // 返回 Some(1)
getValue("Demo Extra")    // 返回 Some(3)
getValue("Unknown")       // 返回 None

方案2：自定义FlexibleMap类（封装性最好） #

如果你希望这个映射能像普通Scala Map一样直接使用（比如用map("key")语法），可以封装一个自定义Map类，内部自动处理键的匹配逻辑：

import scala.collection.Map

class FlexibleMap[V](private val baseMap: Map[String, V]) extends Map[String, V] {
  // 正则匹配带"Extra"的键，提取基础部分
  private val extraKeyPattern = "(.*?)\\s+Extra$".r

  override def get(key: String): Option[V] = {
    // 先查原键，找不到再尝试匹配带Extra的情况
    baseMap.get(key) orElse {
      key match {
        case extraKeyPattern(baseKey) => baseMap.get(baseKey)
        case _ => None
      }
    }
  }

  // 其他Map方法可以直接委托给baseMap，按需实现
  override def iterator: Iterator[(String, V)] = baseMap.iterator
  override def +[V1 >: V](kv: (String, V1)): Map[String, V1] = new FlexibleMap(baseMap + kv)
  override def -(key: String): Map[String, V] = new FlexibleMap(baseMap - key)
}

// 使用示例
val myMap = new FlexibleMap(Map("Test 1" -> 1, "Test 2" -> 2))
myMap("Test 1")        // 1
myMap("Test 1 Extra")  // 1
myMap.get("Test 3")    // None

方案3：批量生成所有键（适合键数量少的场景） #

如果你的基础键不多，也可以用for推导式一次性生成所有包含"Extra"的键值对，避免手动重复写：

// 先定义基础键值对列表
val basePairs = List(
  "Test 1" -> 1,
  "Test 2" -> 2,
  "Demo" -> 3
)

// 批量生成带Extra的键，合并成完整Map
val fullMap = Map(
  basePairs ++ basePairs.map { case (key, value) => s"$key Extra" -> value }: _*
)

// 现在fullMap包含所有键：
// "Test 1" ->1, "Test 1 Extra"->1, "Test 2"->2, "Test 2 Extra"->2, "Demo"->3, "Demo Extra"->3

选哪个？ #

• 如果键很多，优先方案1或2，避免生成大量重复键占用内存
• 如果需要像普通Map一样无缝使用，选方案2
• 如果键很少，图省事可以用方案3

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