scala - 如何使用 Scala 泛型做到这一点

Question

目前我有几种非常相似的方法，我想将它们合并为一种方法。这里有2个方法

  def toInt(attrType: String, attrValue: String): Int = {
    attrType match {
      case "N" => attrValue.toInt
      case _ => -1
    }
  }

  def toString(attrType: String, attrValue: String): String = {
    attrType match {
      case "S" => attrValue
      case _ => ""
    }
  }

我在想有一种更简单的方法可以在 Scala 中使用泛型来做到这一点？

Answer 1

您可以执行以下操作：

trait Converter[T] {
  def convert(attrType: String, attrValue: String): T
}

object ConverterTest {

  implicit object IntConverter extends Converter[Int] {
    def convert(attrType: String, attrValue: String): Int = {
      attrType match {
        case "N" => attrValue.toInt
        case _ => -1
      }
    }
  }

  implicit object StringConverter extends Converter[String] {
    def convert(attrType: String, attrValue: String): String = {
      attrType match {
        case "S" => attrValue
        case _ => ""
      }
    }
  }

  def to[T: Converter](attrType: String, attrValue: String): T = {
    implicitly[Converter[T]].convert(attrType, attrValue)
  }

  def main(args: Array[String]) {
    println(to[String]("S", "B"))
    println(to[String]("N", "B"))

    println(to[Int]("S", "23"))
    println(to[Int]("N", "23"))
  }
}

它的代码更多，我无法让类型推断起作用，因此它的用途可能有限。

但它是一个单一的方法加上一堆可以在呼叫现场控制的转换器，因此您可以获得一些额外的灵活性。

值得付出努力吗？取决于实际用例。