如何将字符串处理代码改写为Scala函数式风格实现？

你作为Scala和函数式编程新手，希望将一段带if判断的字符串处理代码改写成更符合Scala风格的实现——需求是：当逗号分隔的第4部分为空时，仅拼接前三列；否则拼接包含值在内的所有列。

原代码 #

// 第一段代码（注意：`case`是Scala关键字，此处应为笔误）
val str = "aa,bb,1668268540040,34.0::aa,bb,1668268540040"
val parts = str.split("::")
for (elem <- parts) {
  val ret = elem.map(c => if (c.value.isEmpty) {
    c.columnFamily + "," + c.qualifier + "," + c.ts
  } else {
    c.columnFamily + "," + c.qualifier + "," + c.ts + "," + c.value
  })
}

// 第二段代码
val str = "aa,bb,1668268540040,34.0::aa,bb,166826434343"
val parts = str.split("::")
for (part <- parts) {
    val elem = part.split(",", 4)
    if (elem.length == 4) {
      val Array(f, q, t, v) = elem
        state.put(f + ":" + q, (v, t.toLong))
    } else {
        val Array(f, q, t) = elem
        state.put(f + ":" + q, ("", t.toLong))
    }
}

改写后的Scala风格代码 #

针对第一段代码：集合拼接+Option处理空值

用List存储基础字段，结合Option过滤空值后拼接，最后用mkString生成结果，完全避开显式if分支：

val str = "aa,bb,1668268540040,34.0::aa,bb,1668268540040"
val parts = str.split("::")

val results = parts.map { elem =>
  elem.map { c =>
    val baseFields = List(c.columnFamily, c.qualifier, c.ts)
    // 仅当value非空时追加到列表
    val allFields = baseFields ++ Option(c.value).filter(_.nonEmpty)
    allFields.mkString(",")
  }
}

也可以用模式匹配实现更直观的分支逻辑：

val results = parts.map { elem =>
  elem.map {
    case c if c.value.isEmpty => List(c.columnFamily, c.qualifier, c.ts).mkString(",")
    case c => List(c.columnFamily, c.qualifier, c.ts, c.value).mkString(",")
  }
}

针对第二段代码：模式匹配替代长度判断

Scala的数组模式匹配可直接匹配不同长度的数组，替代命令式的if-else：

val str = "aa,bb,1668268540040,34.0::aa,bb,166826434343"
val parts = str.split("::")

parts.foreach { part =>
  part.split(",", 4) match {
    case Array(f, q, t, v) => state.put(s"$f:$q", (v, t.toLong))
    case Array(f, q, t) => state.put(s"$f:$q", ("", t.toLong))
    // 可选：处理数组长度不足3的异常情况
    case invalid => println(s"无效格式：$invalid")
  }
}

核心思路 #

1. 模式匹配替代if分支：作为Scala函数式编程的核心特性，模式匹配能更清晰地表达分支逻辑，提升代码可读性。
2. 集合操作替代手动拼接：通过List存储字段，结合++拼接、mkString生成结果，避免繁琐的字符串拼接操作。
3. Option处理空值：利用Option的过滤特性优雅处理空值场景，符合函数式编程避免null的最佳实践。

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