web-dev-qa-db-ja.com

Scalaでタプルに要素を追加または追加する方法

タプルがあり、型の安全性を失うことなく要素を追加したい。これは私が達成したいことです:

val Tuple = ("", 1, 1f) // (String, Int, Float)

val newTuple:(String, Int, Float, Double) = Tuple :+ 1d
28
EECOLOR

このためのコードジェネレーターを数行で書くこともできます。

val tupadd = for (n <- 2 to 20) yield {
  val t = (0 until n).map(i => ('A'+i).toChar).mkString(", ")
  val u = ('A'+n).toChar
  val i = (0 until n).map(i => "x._"+(i+1)).mkString(", ")
  List(
    s"implicit class TupOps$n[$t](val x: ($t)) extends AnyVal {",
    s"  def :+[$u](y: $u) = ($i, y)",
    s"  def +:[$u](y: $u) = (y, $i)",
    "}"
  ).mkString("\n")
}

これらを印刷して、ファイルに貼り付けてください。

scala> implicit class TupOps2[A, B](val x: (A, B)) extends AnyVal {
     |   def :+[C](y: C) = (x._1, x._2, y)
     |   def +:[C](y: C) = (y, x._1, x._2)
     | }
defined class TupOps2

scala> (1,"salmon") :+ true
res15: (Int, String, Boolean) = (1,salmon,true)
14
Rex Kerr

Shapelessが今のところうまくいきます。追加

import shapeless.syntax.std.Tuple._

コードがコンパイルされる前。

14

私は素晴らしい Shapelessライブラリ の助けを借りて解決策を見つけました、そしてそれはHListです

/*
  Define an implicit class to add the methods
  It accepts any Tuple as we have typed it as Product
*/
implicit class TupleOps[A <: Product](t: A) {

  /*
    Declare the method to append

    B - The type of element we want to append
    L - The HList representing the Tuple A
    P - The HList after appending B
    R - The final result
  */
  def :+[B, L <: HList, P <: HList, R <: Product](b: B)(
    /*
      We need some tools to help with the conversion

      hlister - Converts a Tuple into an HList
      prepend - Can prepend one HList before another
      tupler  - Can convert an HList into a Tuple
    */
    implicit hlister: HListerAux[A, L], prepend: PrependAux[L, B :: HNil, P], tupler: TuplerAux[P, R]):R =
      // Let the helpers do their job
      tupler(prepend(hlister(t), b :: HNil))

  /*
    The prepend method is similar to the append method but does not require
    the extra effort to append
  */
  def +:[B, L <: HList, R <: Product](b: B)(
    // Here we use the :: type of shapeless
    implicit hlister: HListerAux[A, L], tupler: TuplerAux[B :: L, R]):R =
      tupler(b :: hlister(t))
}

// usage is like this
("", 1, 1f) :+ 1d  //> res0: (String, Int, Float, Double) = ("",1,1.0,1.0)
1d +: ("", 1, 1f)  //> res1: (Double, String, Int, Float) = (1.0,"",1,1.0)

編集する

暗黙的な変換を処理する必要があるいくつかのケースでは、このソリューションはケースクラスと組み合わせて機能しません。私は次の実装に戻しました(Rex Kerrのコードに基づく)

def char(n: Int) = ('A' + n).toChar
def prop(n: Int) = "t._" + (n + 1)

val result =
  for (n <- 1 to 21) yield {

    val range = (0 until n)
    val tupleTypeParams = range map char mkString ", "
    val tupleProperties = range map prop mkString ", "

    val elementType = char(n)
    val elementProperty = prop(n)

    val first = n == 1
    val tupleType = if (first) s"Tuple1[$tupleTypeParams]" else s"($tupleTypeParams)"
    val tupleInstance = if (first) s"Tuple1($tupleProperties)" else s"($tupleProperties)"

    val resultType = s"($tupleTypeParams, $elementType)"

    s"""|implicit def tupleOps$n[$tupleTypeParams, $elementType] = 
        |  new TupleAppendOps[$tupleType, $elementType, $resultType] {
        |    def append(t: $tupleType, e: $elementType) = ($tupleProperties, e)
        |    def init(t: $resultType) = $tupleInstance
        |    def last(t: $resultType) = $elementProperty
        |  }""".stripMargin
  }

println(result mkString "\n")
9
EECOLOR

マクロパラダイスに基づいた実験的なマクロライブラリを試すことができますscala https://github.com/leonardschneider/macrogen にあるブランチは、まさにあなたの例です。

現在、まだいくつかのバグがありますが、呼び出しをチェーンできません。

2
Leo