以更实用的风格更改 if-else 结构？

Question

是否有可能以更实用的 Scala 风格更改此 if-else 结构？

def getMIMEType(document: String): String = {

  if (document.endsWith(".pdf")) {
    return "application/pdf"
  } else if (document.endsWith(".dxf")) {
    return "application/dxf"
  } else if (document.endsWith(".jpg")) {
    return "image/jpeg"
  } else return "application/octet-stream"
}

我尝试使用模式匹配，但它不起作用。所以我很好奇一个好的解决方案。

Answer 1

MIME 类型映射

我同意@glowcoder。我发现至少在这种特殊情况下，最好使用 MIME 类型映射。这是一个 Scala 版本：

val MimeTypesMapping = Map(
  ".pdf" -> "application/pdf",
  ".dxf" -> "application/dxf",
  ".jpg" -> "image/jpeg"
)

def extension(fileName: String) = fileName substring (fileName lastIndexOf ".")

def getMIMEType(document: String) =
  MimeTypesMapping get extension(document) getOrElse "application/octet-stream"

此外，正如 Rex Kerr 所指出的，您可以直接将默认值添加到映射定义中：

val MimeTypesMapping = Map(
  ".pdf" -> "application/pdf",
  ".dxf" -> "application/dxf",
  ".jpg" -> "image/jpeg"
) withDefaultValue "application/octet-stream" 

然后像这样使用它： MimeTypesMapping(extension(document))

模式匹配

如果您不这样做不像第一个解决方案，那么您可以使用模式匹配来代替：

def getMIMEType(document: String) = extension(document) match {
  case ".pdf" => "application/pdf"
  case ".dxf" => "application/dxf"
  case ".jpg" => "image/jpeg"
  case _      => "application/octet-stream"
}

此解决方案的优点是您可以在条件中进行更复杂的逻辑，例如：

def getMIMEType(document: String) = extension(document) match {
  // ...
  case ".jpg" | ".jpeg" => "image/jpeg"
  // ...
}

使用自定义提取器进行模式匹配

还有另一种使用模式匹配的方法。您可以为文件扩展名编写自己的提取器，然后在 match 中使用它：

object Extension {
  def unapply(fileName: String): Option[String] = {
    val idx = fileName lastIndexOf "."

    if (idx != -1) Some(fileName substring idx) else None
  }
}

def getMIMEType(document: String) = document match {
  case Extension(".pdf") => "application/pdf"
  case Extension(".dxf") => "application/dxf"
  case Extension(".jpg") => "image/jpeg"
  case _                 => "application/octet-stream"
}

Answer 2

我不知道 scala 的内置结构到底是什么，但听起来你想要一张地图。就像，Java 使用

val map = scala.collection.mutable.HashMap[String, String]()
map.put(".pdf","application/pdf");
map.put(".dxf","application/dxf");
map.put(".jpg","image/jpeg");

Then 在你的函数中使用

def ext = getExtension(document); // assume this exists, eh?
if(map.containsKey(ext)) return map.get(ext);
return "application/octet-stream";

Update：看起来 Scala 使用了一个非常相似的映射：

http://www.scala-lang.org/api/rc/scala/collection/mutable/Map.html

Answer 3

除了我上面的评论之外，我还想对其进行优化，使其实际上成为 scala。

    def getMIMEType(document: String): String = {

        if (document endsWith ".pdf") {
             return "application/pdf"
        } 
        if (document endsWith ".dxf") {
             return "application/dxf"
        } 
        if (document endsWith ".jpg") {
             return "image/jpeg"
        } 
             "application/octet-stream"
    }

我在这里看到代码不能同时满足两个条件中的任何一个。所以我建议使用上面的代码。另外，如果对于最后一个返回语句，即“application/octet-stream”，您可以使用适当的“if条件”，如果您找出最后一行的“if”，则可以删除所有返回，否则上面的代码很好。

返回类型也无需提及。

    def getMIMEType(document: String): String = {} 

可以写为，

    def getMIMEType(document: String) = {} 

检查它是否适合您。